Set line material colour too.

HSF · Nov 30, 2024 · 2c96966 · 2c96966
1 parent b512fcb
commit 2c96966
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/phoenix-event-display/src/managers/three-manager/color-manager.ts b/packages/phoenix-event-display/src/managers/three-manager/color-manager.ts
@@ -1,10 +1,12 @@
 import {
   Color,
   MeshPhongMaterial,
+  LineBasicMaterial,
   Mesh,
   Object3D,
   Material,
   type Object3DEventMap,
+  Line,
 } from 'three';
 import { SceneManager } from './scene-manager';
 
@@ -133,5 +135,11 @@ function setColorForObject(object: Object3D<Object3DEventMap>, color: any) {
     } else if ('color' in material) {
       (material.color as Color).set(color);
     }
+  } else if (object instanceof Line) {
+    const line = object as Line;
+    const material = line.material;
+    if ('color' in material) {
+      (material.color as Color).set(color);
+    }
   }
 }