CLDReco: add automatic switching of ConformalTracking maxDistance for…

… different options o1: with larger vertex radius other options with smaller vertex radius
key4hep · Dec 12, 2023 · 8810abc · 8810abc
1 parent 4cee350
commit 8810abc
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/CLDConfig/CLDReconstruction.py b/CLDConfig/CLDReconstruction.py
@@ -307,6 +307,13 @@
                                  "UseTruthInPrefit": ["false"]
                                  }
 
+if any(small_vtx in detPath for detPath in geoservice.detectors for small_vtx in ["_o2_", "_o3_", "_o4_"]):
+    CT_MAX_DIST = "0.05;"  # semi-colon is important!
+elif any("_o1_" in detPath for detPath in geoservice.detectors):
+    CT_MAX_DIST = "0.03;"  # semi-colon is important!
+else:
+    raise RuntimeError("Unknown detector model to chose CT_MAX_DISTANCE")
+
 MyConformalTracking = MarlinProcessorWrapper("MyConformalTracking")
 MyConformalTracking.OutputLevel = WARNING
 MyConformalTracking.ProcessorType = "ConformalTrackingV2"
@@ -325,27 +332,27 @@
                                   [
                                       "[VXDBarrel]",
                                       "@Collections", ":", "VXDTrackerHits",
-                                      "@Parameters", ":", "MaxCellAngle", ":", "0.01;", "MaxCellAngleRZ", ":", "0.01;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", "0.05;", "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
+                                      "@Parameters", ":", "MaxCellAngle", ":", "0.01;", "MaxCellAngleRZ", ":", "0.01;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", CT_MAX_DIST, "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
                                       "@Flags", ":", "HighPTFit,", "VertexToTracker",
                                       "@Functions", ":", "CombineCollections,", "BuildNewTracks",
                                       "[VXDEncap]",
                                       "@Collections", ":", "VXDEndcapTrackerHits",
-                                      "@Parameters", ":", "MaxCellAngle", ":", "0.01;", "MaxCellAngleRZ", ":", "0.01;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", "0.05;", "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
+                                      "@Parameters", ":", "MaxCellAngle", ":", "0.01;", "MaxCellAngleRZ", ":", "0.01;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", CT_MAX_DIST, "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
                                       "@Flags", ":", "HighPTFit,", "VertexToTracker",
                                       "@Functions", ":", "CombineCollections,", "ExtendTracks",
                                       "[LowerCellAngle1]",
                                       "@Collections", ":", "VXDTrackerHits,", "VXDEndcapTrackerHits",
-                                      "@Parameters", ":", "MaxCellAngle", ":", "0.05;", "MaxCellAngleRZ", ":", "0.05;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", "0.05;", "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
+                                      "@Parameters", ":", "MaxCellAngle", ":", "0.05;", "MaxCellAngleRZ", ":", "0.05;", "Chi2Cut", ":", "100;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", CT_MAX_DIST, "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
                                       "@Flags", ":", "HighPTFit,", "VertexToTracker,", "RadialSearch",
                                       "@Functions", ":", "CombineCollections,", "BuildNewTracks",
                                       "[LowerCellAngle2]",
                                       "@Collections", ":",
-                                      "@Parameters", ":", "MaxCellAngle", ":", "0.1;", "MaxCellAngleRZ", ":", "0.1;", "Chi2Cut", ":", "2000;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", "0.05;", "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
+                                      "@Parameters", ":", "MaxCellAngle", ":", "0.1;", "MaxCellAngleRZ", ":", "0.1;", "Chi2Cut", ":", "2000;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", CT_MAX_DIST, "SlopeZRange:", "10.0;", "HighPTCut:", "10.0;",
                                       "@Flags", ":", "HighPTFit,", "VertexToTracker,", "RadialSearch",
                                       "@Functions", ":", "BuildNewTracks,", "SortTracks",
                                       "[Tracker]",
                                       "@Collections", ":", "ITrackerHits,", "OTrackerHits,", "ITrackerEndcapHits,", "OTrackerEndcapHits",
-                                      "@Parameters", ":", "MaxCellAngle", ":", "0.1;", "MaxCellAngleRZ", ":", "0.1;", "Chi2Cut", ":", "2000;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", "0.05;", "SlopeZRange:", "10.0;", "HighPTCut:", "1.0;",
+                                      "@Parameters", ":", "MaxCellAngle", ":", "0.1;", "MaxCellAngleRZ", ":", "0.1;", "Chi2Cut", ":", "2000;", "MinClustersOnTrack", ":", "4;", "MaxDistance", ":", CT_MAX_DIST, "SlopeZRange:", "10.0;", "HighPTCut:", "1.0;",
                                       "@Flags", ":", "HighPTFit,", "VertexToTracker,", "RadialSearch",
                                       "@Functions", ":", "CombineCollections,", "ExtendTracks",
                                       "[Displaced]",