Multitrack seeding improvement and fix to KF track refit

Control/CalypsoExample/Reconstruction/scripts/faser_reco.py

@@ -29,6 +29,8 @@ parser.add_argument("-r", "--reco", default="",
                     help="Specify reco tag (to append to output filename)")
 parser.add_argument("-n", "--nevents", type=int, default=-1,
                     help="Specify number of events to process (default: all)")
+parser.add_argument("--skip", type=int, default=0,
+                    help="Specify number of events to skip (default: none)")
 parser.add_argument("-v", "--verbose", action='store_true', 
                     help="Turn on DEBUG output")
 parser.add_argument("--isMC", action='store_true',
@@ -37,6 +39,8 @@ parser.add_argument("--MC_calibTag", default="",
                     help="Specify tag used to reconstruct MC calo energy: (WAVE-Calibration-01-LG-nofilt, WAVE-Calibration-01-LG, WAVE-Calibration-01-HG-nofilt, or WAVE-Calibration-01-HG) ")
 parser.add_argument("--testBeam", action='store_true',
                     help="Set geometry for 2021 test beam")
+parser.add_argument("--isOverlay", action='store_true', 
+                    help="Set overlaid data input")
 
 args = parser.parse_args()
 
@@ -81,6 +85,8 @@ else:
 print(f"Starting reconstruction of {filepath.name} with type {runtype}")
 if args.nevents > 0:
     print(f"Reconstructing {args.nevents} events by command-line option")
+if args.skip > 0:
+    print(f"Skipping {args.skip} events by command-line option")
 
 # Start reconstruction
 
@@ -103,6 +109,8 @@ else:
 ConfigFlags.Input.ProjectName = "data20"
 ConfigFlags.GeoModel.Align.Dynamic    = False
 
+ConfigFlags.Exec.SkipEvents = args.skip
+
 # Flags for later
 useCKF = True
 useCal = False
@@ -173,7 +181,7 @@ acc.merge(PoolWriteCfg(ConfigFlags))
 #
 # Set up RAW data access
 
-if args.isMC:
+if args.isMC or args.isOverlay:
     from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
     acc.merge(PoolReadCfg(ConfigFlags))
 else:    
@@ -185,21 +193,23 @@ else:
 from FaserGeoModel.FaserGeoModelConfig import FaserGeometryCfg
 acc.merge(FaserGeometryCfg(ConfigFlags))
 
-if useLHC:
+if useLHC and not args.isOverlay:
     from LHCDataAlgs.LHCDataAlgConfig import LHCDataAlgCfg
     acc.merge(LHCDataAlgCfg(ConfigFlags))
 
 # Set up algorithms
-from WaveRecAlgs.WaveRecAlgsConfig import WaveformReconstructionCfg    
-acc.merge(WaveformReconstructionCfg(ConfigFlags))
+if not args.isOverlay:
+    from WaveRecAlgs.WaveRecAlgsConfig import WaveformReconstructionCfg    
+    acc.merge(WaveformReconstructionCfg(ConfigFlags))
 
-# Calorimeter Energy reconstruction
-if useCal:
-    from CaloRecAlgs.CaloRecAlgsConfig import CalorimeterReconstructionCfg
-    acc.merge(CalorimeterReconstructionCfg(ConfigFlags, MC_calibTag=args.MC_calibTag))
+    # Calorimeter Energy reconstruction
+    if useCal:
+        from CaloRecAlgs.CaloRecAlgsConfig import CalorimeterReconstructionCfg
+        acc.merge(CalorimeterReconstructionCfg(ConfigFlags, MC_calibTag=args.MC_calibTag))
 
 # Tracker clusters
 from TrackerPrepRawDataFormation.TrackerPrepRawDataFormationConfig import FaserSCT_ClusterizationCfg
+# acc.merge(FaserSCT_ClusterizationCfg(ConfigFlags, DataObjectName="Pos_SCT_RDOs"))
 acc.merge(FaserSCT_ClusterizationCfg(ConfigFlags, DataObjectName="SCT_RDOs"))
 
 #
@@ -226,7 +236,8 @@ if useCKF:
     #
     # Kalman Filter for tracking
     from FaserActsKalmanFilter.CKF2Config import CKF2Cfg
-    acc.merge(CKF2Cfg(ConfigFlags, noDiagnostics=True))
+    if not args.isOverlay:
+        acc.merge(CKF2Cfg(ConfigFlags, noDiagnostics=True))
 
     # Add tracking collection with no IFT 
     acc.merge(CKF2Cfg(ConfigFlags, maskedLayers=[0, 1, 2], name="CKF_woIFT", 
@@ -246,10 +257,10 @@ itemList = [ "xAOD::EventInfo#*"
              , "TrackCollection#*"
 ]
 #
-if useLHC:
+if useLHC and not args.isOverlay:
     itemList.extend( ["xAOD::FaserLHCData#*", "xAOD::FaserLHCDataAux#*"] )
 
-if args.isMC:
+if args.isMC and not args.isOverlay:
     # Make xAOD versions of truth
     from Reconstruction.xAODTruthCnvAlgConfig import xAODTruthCnvAlgCfg
     acc.merge(xAODTruthCnvAlgCfg(ConfigFlags))
@@ -266,13 +277,14 @@ tagBuilder = CompFactory.EventInfoTagBuilder()
 tagBuilder.PropagateInput=False
 acc.addEventAlgo(tagBuilder)
 
-# Waveform reconstruction output
-from WaveRecAlgs.WaveRecAlgsConfig import WaveformReconstructionOutputCfg    
-acc.merge(WaveformReconstructionOutputCfg(ConfigFlags))
+if not args.isOverlay:
+    # Waveform reconstruction output
+    from WaveRecAlgs.WaveRecAlgsConfig import WaveformReconstructionOutputCfg    
+    acc.merge(WaveformReconstructionOutputCfg(ConfigFlags))
 
-# Calorimeter reconstruction output
-from CaloRecAlgs.CaloRecAlgsConfig import CalorimeterReconstructionOutputCfg
-acc.merge(CalorimeterReconstructionOutputCfg(ConfigFlags))
+    # Calorimeter reconstruction output
+    from CaloRecAlgs.CaloRecAlgsConfig import CalorimeterReconstructionOutputCfg
+    acc.merge(CalorimeterReconstructionOutputCfg(ConfigFlags))
 
 # Check what we have
 print( "Writing out xAOD objects:" )