From e5c7c4e221a673f81a8f6d0dcdc8b82465921a32 Mon Sep 17 00:00:00 2001 From: Liam O'Sullivan Date: Thu, 11 Jul 2024 09:15:06 -0500 Subject: [PATCH] made Kalman track visible in 2D eve display --- scripts/Reco/draw_spill_3D_projections.py | 42 +++++++++++------------ 1 file changed, 21 insertions(+), 21 deletions(-) diff --git a/scripts/Reco/draw_spill_3D_projections.py b/scripts/Reco/draw_spill_3D_projections.py index 9515ea28..35c30ddc 100644 --- a/scripts/Reco/draw_spill_3D_projections.py +++ b/scripts/Reco/draw_spill_3D_projections.py @@ -327,13 +327,13 @@ def draw_spill(out_dir, name, input_filename, spill_number, time_slice, readout_ kal_true_x[node] = KalmanTruePos[i*600 + node*3 + 0]/1000.0 # from mm to m kal_true_y[node] = KalmanTruePos[i*600 + node*3 + 1]/1000.0 - x_z.plot(kal_z[1:], kal_x[1:], ls='-', lw=2, color=green_cbf) - z_y.plot(kal_z[1:], kal_y[1:], ls='-', lw=2, color=green_cbf) - x_y.plot(kal_x[1:], kal_y[1:], ls='-', lw=2, color=green_cbf) + x_z.plot(kal_z[1:], kal_x[1:], ls='-', lw=2, color=magenta_cbf) + z_y.plot(kal_z[1:], kal_y[1:], ls='-', lw=2, color=magenta_cbf) + x_y.plot(kal_x[1:], kal_y[1:], ls='-', lw=2, color=magenta_cbf) - x_z.plot(kal_z[1:], kal_true_x[1:], ls='--', lw=2, color=magenta_cbf) - z_y.plot(kal_z[1:], kal_true_y[1:], ls='--', lw=2, color=magenta_cbf) - x_y.plot(kal_true_x[1:], kal_true_y[1:], ls='--', lw=2, color=magenta_cbf) + x_z.plot(kal_z[1:], kal_true_x[1:], ls='--', lw=2, color=green_cbf) + z_y.plot(kal_z[1:], kal_true_y[1:], ls='--', lw=2, color=green_cbf) + x_y.plot(kal_true_x[1:], kal_true_y[1:], ls='--', lw=2, color=green_cbf) ### Track start #print(StartPos) @@ -343,9 +343,9 @@ def draw_spill(out_dir, name, input_filename, spill_number, time_slice, readout_ #print("Start", StartPos[i*3 + 0], StartPos[i*3 + 1], StartPos[i*3 + 2]) if not StartPos[i*3 + 1] == 0.0: - x_z.fill_between(*hit_size(StartPos[i*3 + 2], StartPos[i*3 + 0], 'xz', StartPos[i*3 + 2]), color = magenta_cbf) - z_y.fill_between(*hit_size(StartPos[i*3 + 2], StartPos[i*3 + 1], 'zy', StartPos[i*3 + 2]), color = magenta_cbf) - x_y.fill_between(*hit_size(StartPos[i*3 + 0], StartPos[i*3 + 1], 'xy', StartPos[i*3 + 2]), color = magenta_cbf, alpha = 0.5, linewidth = 0.5) + x_z.fill_between(*hit_size(StartPos[i*3 + 2], StartPos[i*3 + 0], 'xz', StartPos[i*3 + 2]), color = black_cbf) + z_y.fill_between(*hit_size(StartPos[i*3 + 2], StartPos[i*3 + 1], 'zy', StartPos[i*3 + 2]), color = black_cbf) + x_y.fill_between(*hit_size(StartPos[i*3 + 0], StartPos[i*3 + 1], 'xy', StartPos[i*3 + 2]), color = black_cbf, alpha = 0.5, linewidth = 0.5) ### Track end @@ -356,9 +356,9 @@ def draw_spill(out_dir, name, input_filename, spill_number, time_slice, readout_ #print("End", EndPos[i*3 + 0], EndPos[i*3 + 1], EndPos[i*3 + 2]) if not EndPos[i*3 + 1] == 0.0: - x_z.fill_between(*hit_size(EndPos[i*3 + 2], EndPos[i*3 + 0], 'xz', EndPos[i*3 + 2]), color = magenta_cbf) - z_y.fill_between(*hit_size(EndPos[i*3 + 2], EndPos[i*3 + 1], 'zy', EndPos[i*3 + 2]), color = magenta_cbf) - x_y.fill_between(*hit_size(EndPos[i*3 + 0], EndPos[i*3 + 1], 'xy', EndPos[i*3 + 2]), color = magenta_cbf, alpha = 0.5, linewidth = 0.5) + x_z.fill_between(*hit_size(EndPos[i*3 + 2], EndPos[i*3 + 0], 'xz', EndPos[i*3 + 2]), color = black_cbf) + z_y.fill_between(*hit_size(EndPos[i*3 + 2], EndPos[i*3 + 1], 'zy', EndPos[i*3 + 2]), color = black_cbf) + x_y.fill_between(*hit_size(EndPos[i*3 + 0], EndPos[i*3 + 1], 'xy', EndPos[i*3 + 2]), color = black_cbf, alpha = 0.5, linewidth = 0.5) ### Track direction #print(Direction) @@ -369,18 +369,18 @@ def draw_spill(out_dir, name, input_filename, spill_number, time_slice, readout_ #print("Direction", StartPos[i*3 + 1], StartPos[i*3 + 2]) if not StartPos[i*3 + 1] == 0.0 or EndPos[i*3 + 1] == 0.0: - x_z.plot([StartPos[i*3 + 2] / 1000.0, EndPos[i*3 + 2] / 1000.0], [StartPos[i*3 + 0] / 1000.0, EndPos[i*3 + 0] / 1000.0], color = magenta_cbf, linewidth = 1.5, linestyle = '--') - z_y.plot([StartPos[i*3 + 2] / 1000.0, EndPos[i*3 + 2] / 1000.0], [StartPos[i*3 + 1] / 1000.0, EndPos[i*3 + 1] / 1000.0], color = magenta_cbf, linewidth = 1.5, linestyle = '--') - x_y.plot([StartPos[i*3 + 0] / 1000.0, EndPos[i*3 + 0] / 1000.0], [StartPos[i*3 + 1] / 1000.0, EndPos[i*3 + 1] / 1000.0], color = magenta_cbf, linewidth = 1.5, linestyle = '--') + x_z.plot([StartPos[i*3 + 2] / 1000.0, EndPos[i*3 + 2] / 1000.0], [StartPos[i*3 + 0] / 1000.0, EndPos[i*3 + 0] / 1000.0], color = black_cbf, linewidth = 1.5, linestyle = '--') + z_y.plot([StartPos[i*3 + 2] / 1000.0, EndPos[i*3 + 2] / 1000.0], [StartPos[i*3 + 1] / 1000.0, EndPos[i*3 + 1] / 1000.0], color = black_cbf, linewidth = 1.5, linestyle = '--') + x_y.plot([StartPos[i*3 + 0] / 1000.0, EndPos[i*3 + 0] / 1000.0], [StartPos[i*3 + 1] / 1000.0, EndPos[i*3 + 1] / 1000.0], color = black_cbf, linewidth = 1.5, linestyle = '--') - x_z.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 0] / 1000.0, c = green_cbf, marker = '2', alpha = 0.5) - x_z.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 0] / 1000.0, c = green_cbf, marker = '1', alpha = 0.5) + x_z.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 0] / 1000.0, c = magenta_cbf, marker = '2', alpha = 0.5) + x_z.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 0] / 1000.0, c = magenta_cbf, marker = '1', alpha = 0.5) - z_y.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 1] / 1000.0, c = green_cbf, marker = '2', alpha = 0.5) - z_y.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 1] / 1000.0, c = green_cbf, marker = '1', alpha = 0.5) + z_y.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 1] / 1000.0, c = magenta_cbf, marker = '2', alpha = 0.5) + z_y.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 2] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 1] / 1000.0, c = magenta_cbf, marker = '1', alpha = 0.5) - x_y.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 0] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 1] / 1000.0, c = green_cbf, marker = '2', alpha = 0.5) - x_y.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 0] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 1] / 1000.0, c = green_cbf, marker = '1', alpha = 0.5) + x_y.scatter(RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 0] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackStart[i*4 + 1] / 1000.0, c = magenta_cbf, marker = '2', alpha = 0.5) + x_y.scatter(RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 0] / 1000.0, RecoTrackPrimaryParticleTruePositionTrackEnd[i*4 + 1] / 1000.0, c = magenta_cbf, marker = '1', alpha = 0.5) output_filename = os.path.join(out_dir, f"{name}_{current_spill_number:03d}")