Adding all beam pipe materials to both material scan scripts.

key4hep · Sep 24, 2024 · e1aa6b2 · e1aa6b2
2 parents 699790c + 0881aa1
commit e1aa6b2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/utils/material_plots.py b/utils/material_plots.py
@@ -34,6 +34,7 @@ def main():
             material = material.replace("66D","")
             material = material.replace("Vtx","")
 
+            # Ignore materials used in beam pipe as vertex material budget does not work without beam pipe. You can add more materials to ignore here
             if material in ["Air","Tungsten","Copper","beam","LiquidNDecane", "AlBeMet162", "Gold"]:
                 continue
             if material not in histDict.keys():

diff --git a/utils/material_plots_2D.py b/utils/material_plots_2D.py
@@ -36,8 +36,9 @@ def main():
 
         entry_x0, entry_lambda, entry_depth = 0.0, 0.0, 0.0
         for i in range(nMat):
-            if entry.material.at(i) == "Air": continue
-
+            # Ignore materials used in beam pipe as vertex material budget does not work without beam pipe. You can add more materials to ignore here
+            if entry.material.at(i) in ["Air","Tungsten","Copper","beam","LiquidNDecane", "AlBeMet162", "Gold"]:
+                continue
             entry_x0        += entry.nX0.at(i)*100.0
             entry_lambda    += entry.nLambda.at(i)
             entry_depth     += entry.matDepth.at(i)