[WIP] Prototype TruthInfo graph for truth history navigation and association studies

Configuration/ProcessModifiers/python/enableTruth_cff.py

@@ -0,0 +1,7 @@
+# Original author: Felice Pantaleo (CERN) <felice.pantaleo@cern.ch>
+# Part of the MC-truth-graph prototype - under heavy development, not yet open
+# to external contributions (see PhysicsTools/TruthInfo/README.md).
+
+import FWCore.ParameterSet.Config as cms
+
+enableTruth = cms.Modifier()

Configuration/PyReleaseValidation/python/upgradeWorkflowComponents.py

@@ -1006,6 +1006,45 @@ def condition(self, fragment, stepList, key, hasHarvest):
 upgradeWFs['ticlv5_TrackLinkingGNN'].step3 = {'--procModifiers': 'ticlv5_TrackLinkingGNN'}
 upgradeWFs['ticlv5_TrackLinkingGNN'].step4 = {'--procModifiers': 'ticlv5_TrackLinkingGNN'}
 
+
+
+class UpgradeWorkflow_enableTruth(UpgradeWorkflow):
+    def setup_(self, step, stepName, stepDict, k, properties):
+        # enableTruth runs the truth-graph producers in RecoGlobal (step3) and,
+        # in GenSim (step1), keeps the full ancestor branch of every stored
+        # SimTrack (g4SimHits PersistencyEmin -> 0 via the modifier) so the
+        # truth graph stays connected to the generator.
+        if 'GenSim' in step or 'RecoGlobal' in step:
+            stepDict[stepName][k] = deepcopy(stepDict[step][k])
+
+            if '--procModifiers' in stepDict[stepName][k]:
+                stepDict[stepName][k]['--procModifiers'] += ',enableTruth'
+            else:
+                stepDict[stepName][k]['--procModifiers'] = 'enableTruth'
+
+    def condition(self, fragment, stepList, key, hasHarvest):
+        return 'Run4' in key
+
+
+upgradeWFs['enableTruth'] = UpgradeWorkflow_enableTruth(
+    steps = [
+        'GenSim',
+        'GenSimHLBeamSpot',
+        'GenSimHLBeamSpot14',
+        'GenSimHLBeamSpotCloseBy',
+        'RecoGlobal',
+    ],
+    PU = [
+        'GenSim',
+        'GenSimHLBeamSpot',
+        'GenSimHLBeamSpot14',
+        'GenSimHLBeamSpotCloseBy',
+        'RecoGlobal',
+    ],
+    suffix = '_enableTruth',
+    offset = 0.88,
+)
+
 # L3 Tracker Muon Outside-In reconstruction first
 class UpgradeWorkflow_phase2L3MuonsOIFirst(UpgradeWorkflow):
     def setup_(self, step, stepName, stepDict, k, properties):

PhysicsTools/TruthInfo/BuildFile.xml

@@ -0,0 +1,12 @@
+<use name="FWCore/Framework"/>
+<use name="FWCore/MessageLogger"/>
+<use name="FWCore/Utilities"/>
+<use name="DataFormats/Common"/>
+<use name="DataFormats/Math"/>
+<use name="FWCore/ParameterSet"/>
+<use name="SimDataFormats/EncodedEventId"/>
+<use name="heppdt"/>
+
+<export>
+  <lib name="1"/>
+</export>