diff --git a/DaVinciTests/python/DaVinciTests/options_lhcbintegration_track_refitting.py b/DaVinciTests/python/DaVinciTests/options_lhcbintegration_track_refitting.py index d3d88f0e41c81dbb119ff8f7f130f78e99ce4666..52f647365ec4be895b7ba1a36340ac365a0f44ec 100644 --- a/DaVinciTests/python/DaVinciTests/options_lhcbintegration_track_refitting.py +++ b/DaVinciTests/python/DaVinciTests/options_lhcbintegration_track_refitting.py @@ -28,7 +28,7 @@ from RecoConf.standard_particles import make_long_pions from PyConf.reading import get_pvs from PyConf.Algorithms import FunctionalChargedProtoParticleMaker -from PyConf.Algorithms import TrackMonitor +from PyConf.Algorithms import TrackCorrelationsMonitor, TrackMonitor from PyConf.Algorithms import SelectTracksForParticles from PyConf.Algorithms import LHCbIDOverlapRelationTable from PyConf.Algorithms import FunctionalParticleMaker @@ -36,6 +36,7 @@ from PyConf.Algorithms import FunctionalParticleMaker from RecoConf import track_refitting from RecoConf.calorimeter_reconstruction import make_digits from RecoConf.decoders import default_VeloCluster_source +from RecoConf.hlt2_tracking import make_PrStoreSciFiHits_hits import Functors as F @@ -100,15 +101,27 @@ def main(options: Options): muons = make_muons(partial(make_protos, tracks_from_raw)) # original new_muons = make_muons(partial(make_protos, tracks_from_dst)) + scifihits = make_PrStoreSciFiHits_hits() + # the refitted tracks + corr_plots_refi = TrackCorrelationsMonitor( + TracksInContainer=tracks_from_dst, + SciFiHits=scifihits, + name="RefittedTrackCorrelationsMonitor", + ) resolution_plots_refi = TrackMonitor( TracksInContainer=tracks_from_dst, name="RefittedTrackMonitor" ) # the original tracks - resolution_plots_orig = TrackMonitor( + corr_plots_orig = TrackMonitor( TracksInContainer=tracks_from_particles, name="TrackMonitor" ) + resolution_plots_orig = TrackCorrelationsMonitor( + TracksInContainer=tracks_from_particles, + SciFiHits=scifihits, + name="TrackCorrelationsMonitor", + ) pvs = get_pvs() refitted_pvs = track_refitting.refit_pvs(input_pvs=pvs).OutputVertices @@ -227,6 +240,8 @@ def main(options: Options): my_filter, tracks_from_dst, relation_table_match_by_lhcbid, + corr_plots_refi, + corr_plots_orig, resolution_plots_refi, resolution_plots_orig, my_tuple,