fix pdg toggle for mcparticle tree view

key4hep · Jul 1, 2024 · 0dc5988 · 0dc5988
1 parent 16288eb
commit 0dc5988
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 15 deletions.
diff --git a/js/filter/mcparticle.js b/js/filter/mcparticle.js
@@ -8,6 +8,7 @@ import {
   start,
   getWidthFilterContent,
 } from "../menu/filter/filter.js";
+import { drawAll } from "../draw.js";
 
 const filter = document.getElementById("filter");
 const filters = document.getElementById("filters");

diff --git a/js/menu/show-pdg.js b/js/menu/show-pdg.js
@@ -1,29 +1,22 @@
 import { Toggle } from "./toggle.js";
-import { selectedObjectTypes } from "../main.js";
 
 export class PdgToggle extends Toggle {
   constructor(id) {
     super(id);
   }
 
   toggle(currentObjects, redraw) {
-    const validObjects = selectedObjectTypes.types;
-
+    const collection =
+      currentObjects.datatypes["edm4hep::MCParticle"].collection;
     if (this.isSliderActive) {
-      for (const objectType of validObjects) {
-        const collection = currentObjects[objectType].collection;
-        if (collection[0].PDG === undefined) return;
-        for (const object of collection) {
-          object.updateTexImg(`${object.PDG}`);
-        }
+      if (collection[0].PDG === undefined) return;
+      for (const object of collection) {
+        object.updateTexImg(`${object.PDG}`);
       }
     } else {
-      for (const objectType of validObjects) {
-        const collection = currentObjects[objectType].collection;
-        if (collection[0].PDG === undefined) return;
-        for (const object of collection) {
-          object.updateTexImg(`${object.name}`);
-        }
+      if (collection[0].PDG === undefined) return;
+      for (const object of collection) {
+        object.updateTexImg(`${object.name}`);
       }
     }