From ec9cfa1fe131e75a130d9f20ad266e7352739be2 Mon Sep 17 00:00:00 2001 From: cetin Date: Wed, 17 Jan 2024 16:00:01 +0100 Subject: [PATCH] trlosses --- moore_options/get_ghost_data.py | 14 ++++++++++---- moore_options/get_resolution_and_eff_data.py | 2 +- moore_options/get_tracking_losses.py | 2 +- 3 files changed, 12 insertions(+), 6 deletions(-) diff --git a/moore_options/get_ghost_data.py b/moore_options/get_ghost_data.py index 28c18d8..f0f22e1 100644 --- a/moore_options/get_ghost_data.py +++ b/moore_options/get_ghost_data.py @@ -33,7 +33,7 @@ import glob options.evt_max = -1 -decay = "D" # D, B +decay = "BJpsi" # D, B options.ntuple_file = f"data/ghost_data_{decay}_default_phi_eta.root" options.input_type = "ROOT" @@ -105,7 +105,9 @@ def run_tracking_debug(): MCParticles=mc_unpackers()["MCParticles"], SciFiHitLinks=links_to_hits, SciFiHits=make_PrStoreSciFiHits_hits(), - TrackInfo=make_data_with_FetchDataFromFile("/Event/MC/TrackInfo"), + TrackInfo=make_data_with_FetchDataFromFile( + "/Event/MC/TrackInfo", "LHCb::MCProperty" + ), ), **loose_forward_params, ) @@ -120,7 +122,9 @@ def run_tracking_debug(): MCParticles=mc_unpackers()["MCParticles"], SciFiHitLinks=links_to_hits, SciFiHits=make_PrStoreSciFiHits_hits(), - TrackInfo=make_data_with_FetchDataFromFile("/Event/MC/TrackInfo"), + TrackInfo=make_data_with_FetchDataFromFile( + "/Event/MC/TrackInfo", "LHCb::MCProperty" + ), ), **loose_forward_params, ) @@ -141,7 +145,9 @@ def run_tracking_debug(): SeedTracks=seed_tracks["v1"], VeloTrackLinks=links_to_velo_tracks, SeedTrackLinks=links_to_seed_tracks, - TrackInfo=make_data_with_FetchDataFromFile("/Event/MC/TrackInfo"), + TrackInfo=make_data_with_FetchDataFromFile( + "/Event/MC/TrackInfo", "LHCb::MCProperty" + ), MCParticles=mc_unpackers()["MCParticles"], ), AddUTHitsToolName=get_global_ut_hits_tool(enable=True), diff --git a/moore_options/get_resolution_and_eff_data.py b/moore_options/get_resolution_and_eff_data.py index 82f1d88..75d8e8d 100644 --- a/moore_options/get_resolution_and_eff_data.py +++ b/moore_options/get_resolution_and_eff_data.py @@ -40,7 +40,7 @@ decay = "test" options.evt_max = -1 -options.ntuple_file = f"data/resolutions_and_effs_{decay}_default_weights.root" +options.ntuple_file = f"data/resolutions_and_effs_{decay}_default.root" options.input_type = "ROOT" diff --git a/moore_options/get_tracking_losses.py b/moore_options/get_tracking_losses.py index 68bba1d..27253cf 100644 --- a/moore_options/get_tracking_losses.py +++ b/moore_options/get_tracking_losses.py @@ -38,7 +38,7 @@ tested by mc_matching_example.py """ -decay = "BJpsi" +decay = "B" if decay == "B": options.input_files = glob.glob("/auto/data/guenther/Bd_Kstee/*.xdigi")