File indexing completed on 2025-12-16 09:22:12
0001
0002
0003 import tempfile
0004 from pathlib import Path
0005 import shutil
0006
0007 import acts
0008 from acts.examples.simulation import (
0009 addParticleGun,
0010 MomentumConfig,
0011 EtaConfig,
0012 PhiConfig,
0013 ParticleConfig,
0014 addFatras,
0015 addDigitization,
0016 ParticleSelectorConfig,
0017 addDigiParticleSelection,
0018 )
0019
0020 from acts.examples.reconstruction import (
0021 addSeeding,
0022 TrackSmearingSigmas,
0023 SeedFinderConfigArg,
0024 SeedFinderOptionsArg,
0025 SeedingAlgorithm,
0026 TruthEstimatedSeedingAlgorithmConfigArg,
0027 CkfConfig,
0028 addCKFTracks,
0029 TrackSelectorConfig,
0030 )
0031
0032 from physmon_common import makeSetup
0033
0034 u = acts.UnitConstants
0035
0036 setup = makeSetup()
0037
0038
0039 def run_ckf_tracking(label, seeding):
0040 with tempfile.TemporaryDirectory() as temp:
0041 s = acts.examples.Sequencer(
0042 events=10000,
0043 numThreads=-1,
0044 logLevel=acts.logging.INFO,
0045 )
0046
0047 tp = Path(temp)
0048
0049 for d in setup.decorators:
0050 s.addContextDecorator(d)
0051
0052 rnd = acts.examples.RandomNumbers(seed=42)
0053
0054 addParticleGun(
0055 s,
0056 MomentumConfig(1.0 * u.GeV, 10.0 * u.GeV, transverse=True),
0057 EtaConfig(-3.0, 3.0, uniform=True),
0058 PhiConfig(0.0, 360.0 * u.degree),
0059 ParticleConfig(1, acts.PdgParticle.eMuon, randomizeCharge=True),
0060 vtxGen=acts.examples.GaussianVertexGenerator(
0061 mean=acts.Vector4(0, 0, 0, 0),
0062 stddev=acts.Vector4(
0063 0.0125 * u.mm, 0.0125 * u.mm, 55.5 * u.mm, 1.0 * u.ns
0064 ),
0065 ),
0066 multiplicity=1,
0067 rnd=rnd,
0068 )
0069
0070 addFatras(
0071 s,
0072 setup.trackingGeometry,
0073 setup.field,
0074 enableInteractions=True,
0075 rnd=rnd,
0076 )
0077
0078 addDigitization(
0079 s,
0080 setup.trackingGeometry,
0081 setup.field,
0082 digiConfigFile=setup.digiConfig,
0083 rnd=rnd,
0084 )
0085
0086 addDigiParticleSelection(
0087 s,
0088 ParticleSelectorConfig(
0089 pt=(0.9 * u.GeV, None),
0090 measurements=(9, None),
0091 removeNeutral=True,
0092 ),
0093 )
0094
0095 addSeeding(
0096 s,
0097 setup.trackingGeometry,
0098 setup.field,
0099 TrackSmearingSigmas(
0100
0101 loc0=0,
0102 loc0PtA=0,
0103 loc0PtB=0,
0104 loc1=0,
0105 loc1PtA=0,
0106 loc1PtB=0,
0107 time=0,
0108 phi=0,
0109 theta=0,
0110 ptRel=0,
0111 ),
0112 SeedFinderConfigArg(
0113 r=(33 * u.mm, 200 * u.mm),
0114 deltaR=(1 * u.mm, 300 * u.mm),
0115 collisionRegion=(-250 * u.mm, 250 * u.mm),
0116 z=(-2000 * u.mm, 2000 * u.mm),
0117 maxSeedsPerSpM=1,
0118 sigmaScattering=5,
0119 radLengthPerSeed=0.1,
0120 minPt=0.5 * u.GeV,
0121 impactMax=3 * u.mm,
0122 ),
0123 SeedFinderOptionsArg(bFieldInZ=2 * u.T),
0124 TruthEstimatedSeedingAlgorithmConfigArg(deltaR=(10.0 * u.mm, None)),
0125 seedingAlgorithm=seeding,
0126 initialSigmas=[
0127 1 * u.mm,
0128 1 * u.mm,
0129 1 * u.degree,
0130 1 * u.degree,
0131 0 * u.e / u.GeV,
0132 1 * u.ns,
0133 ],
0134 initialSigmaQoverPt=0.1 * u.e / u.GeV,
0135 initialSigmaPtRel=0.1,
0136 initialVarInflation=[1.0] * 6,
0137 geoSelectionConfigFile=setup.geoSel,
0138 rnd=rnd,
0139 outputDirRoot=tp,
0140 )
0141
0142 addCKFTracks(
0143 s,
0144 setup.trackingGeometry,
0145 setup.field,
0146 TrackSelectorConfig(
0147 pt=(0.9 * u.GeV, None),
0148 loc0=(-4.0 * u.mm, 4.0 * u.mm),
0149 nMeasurementsMin=6,
0150 maxHoles=2,
0151 maxOutliers=2,
0152 ),
0153 CkfConfig(
0154 chi2CutOffMeasurement=15.0,
0155 chi2CutOffOutlier=25.0,
0156 numMeasurementsCutOff=10,
0157 seedDeduplication=(
0158 True if seeding != SeedingAlgorithm.TruthSmeared else False
0159 ),
0160 stayOnSeed=True if seeding != SeedingAlgorithm.TruthSmeared else False,
0161 ),
0162 outputDirRoot=tp,
0163 )
0164
0165 s.run()
0166
0167 for file in (
0168 ["performance_seeding.root"]
0169 if seeding != SeedingAlgorithm.TruthSmeared
0170 else []
0171 ) + [
0172 "tracksummary_ckf.root",
0173 "performance_finding_ckf.root",
0174 "performance_fitting_ckf.root",
0175 ]:
0176 perf_file = tp / file
0177 assert perf_file.exists(), f"Performance file not found {perf_file}"
0178 (setup.outdir / label).mkdir(parents=True, exist_ok=True)
0179 shutil.copy(perf_file, setup.outdir / label / file)
0180
0181
0182 for label, seeding in [
0183 ("truth_smeared", SeedingAlgorithm.TruthSmeared),
0184 ("truth_estimated", SeedingAlgorithm.TruthEstimated),
0185 ("seeded", SeedingAlgorithm.GridTriplet),
0186 ("orthogonal", SeedingAlgorithm.OrthogonalTriplet),
0187 ]:
0188 run_ckf_tracking(label, seeding)
