@@ -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