feat(event-display): ability to scale CaloClusters

HSF · Apr 5, 2021 · ff666e2 · ff666e2
1 parent 6f14d29
commit ff666e2
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 7 deletions.
diff --git a/packages/phoenix-event-display/src/loaders/phoenix-loader.ts b/packages/phoenix-event-display/src/loaders/phoenix-loader.ts
@@ -165,13 +165,14 @@ export class PhoenixLoader implements EventDataLoader {
         typeFolder: any,
         typeFolderPM: PhoenixMenuNode
       ) => {
+        const scaleJets = (value: number) => {
+          this.graphicsLibrary.getSceneManager().scaleJets(value);
+        };
         if (typeFolder) {
           const sizeMenu = typeFolder
             .add({ jetsScale: 100 }, 'jetsScale', 1, 200)
             .name('Jets Size (%)');
-          sizeMenu.onChange((value: number) => {
-            this.graphicsLibrary.getSceneManager().scaleJets(value);
-          });
+          sizeMenu.onChange(scaleJets);
         }
         // Phoenix menu
         if (typeFolderPM) {
@@ -181,9 +182,7 @@ export class PhoenixLoader implements EventDataLoader {
             min: 1,
             max: 200,
             allowCustomValue: true,
-            onChange: (value: number) => {
-              this.graphicsLibrary.getSceneManager().scaleJets(value);
-            },
+            onChange: scaleJets,
           });
         }
       };
@@ -210,11 +209,40 @@ export class PhoenixLoader implements EventDataLoader {
         new Cut('energy', 0, 10000),
       ];
 
+      const addCaloClusterOptions = (
+        typeFolder: any,
+        typeFolderPM: PhoenixMenuNode
+      ) => {
+        const scaleCaloClusters = (value: number) => {
+          this.graphicsLibrary
+            .getSceneManager()
+            .scaleChildrenObjects('CaloClusters', value / 100);
+        };
+        if (typeFolder) {
+          const sizeMenu = typeFolder
+            .add({ caloClustersScale: 100 }, 'caloClustersScale', 1, 400)
+            .name('CaloClusters Size (%)');
+          sizeMenu.onChange(scaleCaloClusters);
+        }
+        // Phoenix menu
+        if (typeFolderPM) {
+          typeFolderPM.addConfig('slider', {
+            label: 'CaloClusters Size (%)',
+            value: 100,
+            min: 1,
+            max: 400,
+            allowCustomValue: true,
+            onChange: scaleCaloClusters,
+          });
+        }
+      };
+
       this.addObjectType(
         eventData.CaloClusters,
         PhoenixObjects.getCluster,
         'CaloClusters',
-        cuts
+        cuts,
+        addCaloClusterOptions
       );
     }
 

diff --git a/packages/phoenix-event-display/src/three/scene-manager.ts b/packages/phoenix-event-display/src/three/scene-manager.ts
@@ -463,6 +463,21 @@ export class SceneManager {
     });
   }
 
+  /**
+   * Scale lowest level objects in a group.
+   * @param groupName Name of the group to scale objects of.
+   * @param value Value of the scale. Default is 1.
+   */
+  public scaleChildrenObjects(groupName: string, value: number) {
+    const object = this.scene.getObjectByName(groupName);
+
+    object.traverse((objectChild: Object3D) => {
+      if (objectChild.children.length === 0) {
+        objectChild.scale.setScalar(value);
+      }
+    });
+  }
+
   /**
    * Add label to the three.js object.
    * @param label Label to add to the event object.