Add support for new modules to tracking validation timing plots

cms-sw · Oct 13, 2016 · 37a21ea · 37a21ea
1 parent 028e6ac
commit 37a21ea
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/Validation/RecoTrack/python/plotting/trackingPlots.py b/Validation/RecoTrack/python/plotting/trackingPlots.py
@@ -1029,7 +1029,9 @@ def _set(param, name, modules):
                 setattr(self, name, modules)
 
         _set(clusterMasking, "_clusterMasking", [self._name+"Clusters"])
-        _set(seeding, "_seeding", [self._name+"SeedingLayers", self._name+"Seeds"])
+        # it's fine to include e.g. quadruplets here also for pair
+        # steps, as non-existing modules are just ignored
+        _set(seeding, "_seeding", [self._name+"SeedingLayers", self._name+"TrackingRegions", self._name+"HitDoublets", self._name+"HitTriplets", self._name+"HitQuadruplets", self._name+"Seeds"])
         _set(building, "_building", [self._name+"TrackCandidates"])
         _set(fit, "_fit", [self._name+"Tracks"])
         _set(selection, "_selection", [self._name])
@@ -1071,6 +1073,10 @@ def modules(self):
 _iterations = [
     Iteration("initialStepPreSplitting", clusterMasking=[],
               seeding=["initialStepSeedLayersPreSplitting",
+                       "initialStepTrackingRegionsPreSplitting",
+                       "initialStepHitDoubletsPreSplitting",
+                       "initialStepHitTripletsPreSplitting",
+                       "initialStepHitQuadrupletsPreSplitting",
                        "initialStepSeedsPreSplitting"],
               building=["initialStepTrackCandidatesPreSplitting"],
               fit=["initialStepTracksPreSplitting"],
@@ -1108,6 +1114,12 @@ def modules(self):
     Iteration("mixedTripletStep",
               seeding=["mixedTripletStepSeedLayersA",
                        "mixedTripletStepSeedLayersB",
+                       "mixedTripletStepTrackingRegionsA",
+                       "mixedTripletStepTrackingRegionsB",
+                       "mixedTripletStepHitDoubletsA",
+                       "mixedTripletStepHitDoubletsB",
+                       "mixedTripletStepHitTripletsA",
+                       "mixedTripletStepHitTripletsB",
                        "mixedTripletStepSeedsA",
                        "mixedTripletStepSeedsB",
                        "mixedTripletStepSeeds"],
@@ -1121,6 +1133,11 @@ def modules(self):
     Iteration("tobTecStep",
               seeding=["tobTecStepSeedLayersTripl",
                        "tobTecStepSeedLayersPair",
+                       "tobTecStepTrackingRegionsTripl",
+                       "tobTecStepTrackingRegionsPair",
+                       "tobTecStepHitDoubletsTripl",
+                       "tobTecStepHitDoubletsPair",
+                       "tobTecStepHitTripletsTripl",
                        "tobTecStepSeedsTripl",
                        "tobTecStepSeedsPair",
                        "tobTecStepSeeds"],
@@ -1162,6 +1179,9 @@ def modules(self):
               building=["convTrackCandidates"],
               fit=["convStepTracks"],
               selection=["convStepSelector"]),
+    Iteration("Other", clusterMasking=[], seeding=[], building=[], fit=[], selection=[],
+              other=["trackerClusterCheckPreSplitting",
+                     "trackerClusterCheck"]),
 ]
 
 def _iterModuleMap(includeConvStep=True, onlyConvStep=False):