Skip to content

Commit a1bd55b

Browse files
chore: expand TrackFinderPerformanceWriter usage boolean throughout python (#5515)
Refactor current TruthMatching for particle to track matching & update TrackFinderPerformance writer accordingly
1 parent 0dda19a commit a1bd55b

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

Python/Examples/python/reconstruction.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1631,6 +1631,7 @@ def addCKFTracks(
16311631
writeTrackStates: bool = False,
16321632
writePerformance: bool = True,
16331633
writeCovMat=False,
1634+
writeMatchingDetails: bool = False,
16341635
logLevel: Optional[acts.logging.Level] = None,
16351636
prefix: str = "",
16361637
) -> None:
@@ -1770,6 +1771,7 @@ def addCKFTracks(
17701771
writeSummary=writeTrackSummary,
17711772
writeStates=writeTrackStates,
17721773
writeFitterPerformance=writePerformance,
1774+
writeMatchingDetails=writeMatchingDetails,
17731775
writeFinderPerformance=writePerformance,
17741776
writeCovMat=writeCovMat,
17751777
logLevel=logLevel,
@@ -1851,6 +1853,7 @@ def addTrackWriters(
18511853
writeFinderNTuple: bool = False,
18521854
logLevel: Optional[acts.logging.Level] = None,
18531855
writeCovMat=False,
1856+
writeMatchingDetails: bool = False,
18541857
):
18551858
customLogLevel = acts.examples.defaultLogging(s, logLevel)
18561859

@@ -1906,6 +1909,7 @@ def addTrackWriters(
19061909
inputParticleTrackMatching="particle_track_matching",
19071910
inputParticleMeasurementsMap="particle_measurements_map",
19081911
filePath=str(outputDirRoot / f"performance_finding_{name}.root"),
1912+
writeMatchingDetails=writeMatchingDetails,
19091913
)
19101914
s.addWriter(trackFinderPerfWriter)
19111915

@@ -1987,6 +1991,7 @@ def addGnn(
19871991
inputClusters: str = "",
19881992
outputDirRoot: Optional[Union[Path, str]] = None,
19891993
device=None,
1994+
writeMatchingDetails: bool = False,
19901995
logLevel: Optional[acts.logging.Level] = None,
19911996
) -> acts.examples.Sequencer:
19921997
"""
@@ -2075,6 +2080,7 @@ def addGnn(
20752080
writeFinderPerformance=True,
20762081
writeSummary=False,
20772082
writeFinderNTuple=True,
2083+
writeMatchingDetails=writeMatchingDetails,
20782084
)
20792085

20802086
return s
@@ -2093,6 +2099,7 @@ def addAmbiguityResolution(
20932099
writeTrackStates: bool = False,
20942100
writePerformance: bool = True,
20952101
writeCovMat=False,
2102+
writeMatchingDetails: bool = False,
20962103
logLevel: Optional[acts.logging.Level] = None,
20972104
prefix: str = "",
20982105
) -> None:
@@ -2142,6 +2149,7 @@ def addAmbiguityResolution(
21422149
writeSummary=writeTrackSummary,
21432150
writeStates=writeTrackStates,
21442151
writeFitterPerformance=writePerformance,
2152+
writeMatchingDetails=writeMatchingDetails,
21452153
writeFinderPerformance=writePerformance,
21462154
writeCovMat=writeCovMat,
21472155
logLevel=logLevel,
@@ -2164,6 +2172,7 @@ def addScoreBasedAmbiguityResolution(
21642172
writeTrackStates: bool = False,
21652173
writePerformance: bool = True,
21662174
writeCovMat=False,
2175+
writeMatchingDetails: bool = False,
21672176
logLevel: Optional[acts.logging.Level] = None,
21682177
) -> None:
21692178
from acts.examples import ScoreBasedAmbiguityResolutionAlgorithm
@@ -2213,6 +2222,7 @@ def addScoreBasedAmbiguityResolution(
22132222
writeSummary=writeTrackSummary,
22142223
writeStates=writeTrackStates,
22152224
writeFitterPerformance=writePerformance,
2225+
writeMatchingDetails=writeMatchingDetails,
22162226
writeFinderPerformance=writePerformance,
22172227
writeCovMat=writeCovMat,
22182228
logLevel=logLevel,
@@ -2235,6 +2245,7 @@ def addAmbiguityResolutionML(
22352245
writeTrackStates: bool = False,
22362246
writePerformance: bool = True,
22372247
writeCovMat=False,
2248+
writeMatchingDetails: bool = False,
22382249
logLevel: Optional[acts.logging.Level] = None,
22392250
) -> None:
22402251
from acts.examples.onnx import AmbiguityResolutionMLAlgorithm
@@ -2293,6 +2304,7 @@ def addAmbiguityResolutionML(
22932304
writeSummary=writeTrackSummary,
22942305
writeStates=writeTrackStates,
22952306
writeFitterPerformance=writePerformance,
2307+
writeMatchingDetails=writeMatchingDetails,
22962308
writeFinderPerformance=writePerformance,
22972309
writeCovMat=writeCovMat,
22982310
logLevel=logLevel,

0 commit comments

Comments
 (0)