Function makeDepthColorModifier

src/modifiers/depthColor.ts:17–36

(
  splatToView: SplatTransformer,
  minDepth: DynoVal<"float">,
  maxDepth: DynoVal<"float">,
  reverse: DynoVal<"bool">,
)

} from "../dyno";

export function makeDepthColorModifier(
  splatToView: SplatTransformer,
  minDepth: DynoVal<"float">,
  maxDepth: DynoVal<"float">,
  reverse: DynoVal<"bool">,
) {
  return dynoBlock({ gsplat: Gsplat }, { gsplat: Gsplat }, ({ gsplat }) => {
    if (!gsplat) {
      throw new Error("No gsplat input");
    }
    let { center } = splitGsplat(gsplat).outputs;
    center = splatToView.apply(center);
    const { z } = split(center).outputs;
    let depth = normalizedDepth(neg(z), minDepth, maxDepth);
    depth = select(reverse, sub(dynoConst("float", 1), depth), depth);

    gsplat = combineGsplat({ gsplat, r: depth, g: depth, b: depth });
    return { gsplat };
  });
}

export function setDepthColor(
  splats: SplatMesh,