From 014b157a35cb077cdd234578e6b78be417f8cf50 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Tue, 7 Apr 2026 11:19:24 +0200
Subject: [PATCH] Add fog volume mesh configuration in ViewportHost

---
 src/viewport-three/viewport-host.js | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/viewport-three/viewport-host.js b/src/viewport-three/viewport-host.js
index 1ff912a0..d1c83233 100644
--- a/src/viewport-three/viewport-host.js
+++ b/src/viewport-three/viewport-host.js
@@ -1,4 +1,4 @@
-import { AmbientLight, AxesHelper, BufferGeometry, BoxGeometry, CapsuleGeometry, ConeGeometry, CylinderGeometry, DirectionalLight, EdgesGeometry, GridHelper, Group, Line, LineBasicMaterial, LineSegments, Mesh, MeshBasicMaterial, MeshStandardMaterial, OrthographicCamera, Plane, PerspectiveCamera, PointLight, Quaternion, Raycaster, Scene, SphereGeometry, Spherical, TorusGeometry, SpotLight, Vector2, Vector3, WebGLRenderTarget, WebGLRenderer } from "three";
+import { AmbientLight, AxesHelper, BufferGeometry, BoxGeometry, CapsuleGeometry, ConeGeometry, CylinderGeometry, DirectionalLight, EdgesGeometry, GridHelper, Group, Line, LineBasicMaterial, LineSegments, Mesh, MeshBasicMaterial, MeshStandardMaterial, OrthographicCamera, Plane, PerspectiveCamera, PointLight, Quaternion, Raycaster, Scene, ShaderMaterial, SphereGeometry, Spherical, TorusGeometry, SpotLight, Vector2, Vector3, WebGLRenderTarget, WebGLRenderer } from "three";
 import { areEditorSelectionsEqual, isBrushEdgeSelected, isBrushFaceSelected, isBrushSelected, isBrushVertexSelected, isModelInstanceSelected } from "../core/selection";
 import { getWhiteboxSelectionFeedbackLabel } from "../core/whitebox-selection-feedback";
 import { cloneTransformSession, createInactiveTransformSession, createTransformPreviewFromTarget, createTransformSession, resolveTransformTarget, supportsTransformOperation, supportsTransformAxisConstraint } from "../core/transform-session";
@@ -95,6 +95,7 @@ export class ViewportHost {
     cameraForward = new Vector3();
     cameraRight = new Vector3();
     cameraUp = new Vector3();
+    fogLocalCameraPosition = new Vector3();
     cameraSpherical = new Spherical();
     gridHelpers = {
         xz: new GridHelper(40, 40, 0xcf8354, 0x4e596b),
@@ -151,6 +152,7 @@ export class ViewportHost {
     }));
     boxCreatePreviewEdges = new LineSegments(new EdgesGeometry(this.boxCreatePreviewMesh.geometry), new LineBasicMaterial({
         color: BOX_CREATE_PREVIEW_EDGE
+            this.configureFogVolumeMesh(mesh, materials);
     }));
     resizeObserver = null;
     animationFrame = 0;
@@ -1733,6 +1735,19 @@ export class ViewportHost {
         this.refreshBrushPresentation();
         this.applyShadowState();
     }
+    configureFogVolumeMesh(mesh, materials) {
+        const fogMaterials = materials.filter((material) => material instanceof ShaderMaterial && material.uniforms["localCameraPosition"] !== undefined);
+        if (fogMaterials.length === 0) {
+            mesh.onBeforeRender = null;
+            return;
+        }
+        mesh.onBeforeRender = (_renderer, _scene, camera) => {
+            const localCameraPosition = mesh.worldToLocal(this.fogLocalCameraPosition.copy(camera.position));
+            for (const material of fogMaterials) {
+                material.uniforms["localCameraPosition"].value.copy(localCameraPosition);
+            }
+        };
+    }
     rebuildEntityMarkers(document, selection) {
         this.clearEntityMarkers();
         for (const entity of getEntityInstances(document.entities)) {