GeneratorDBLabel with Initial State

key4hep · Dec 4, 2024 · 35a9cf5 · 35a9cf5
1 parent a6f5d2a
commit 35a9cf5
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/python/Process.py b/python/Process.py
@@ -49,15 +49,24 @@ def process_info(self):
             self._proclabel += f"{self._finfo[p].name} "
             self._parts.append(self._finfo[p])
         # generate the label for the generatorDB
+        # first the initial state
+        initialstate = [abs(self.initial[0]), abs(self.initial[1])]
+        #sort ascending
+        initialstate.sort()
+        # add to label
+        for pdg in initialstate:
+            self.generatorDBLabel += f"_{str(abs(pdg))}"
+        # remove leading "_"
+        self.generatorDBLabel = self.generatorDBLabel[
+            (self.generatorDBLabel.index("_") + 1) :
+        ]
+        # now the final state
         finalstate = [abs(part) for part in self.final]
         # sort ascending
         finalstate.sort()
+        # add to label
         for pdg in finalstate:
             self.generatorDBLabel += f"_{str(abs(pdg))}"
-        # remove leading _
-        self.generatorDBLabel = self.generatorDBLabel[
-            (self.generatorDBLabel.index("_") + 1) :
-        ]
 
     def set_particle_data(self, pdata):
         if pdata is None or self._init: