add validity guard to fitcentroid test

rat-pac · Dec 9, 2024 · 7cd3061 · 7cd3061
1 parent 6d26a33
commit 7cd3061
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/full/fitcentroid/fitcentroid.C b/test/full/fitcentroid/fitcentroid.C
@@ -12,12 +12,12 @@ void fitcentroid(std::string event_filename, std::string out_filename) {
       new TProfile("hReconVsEnergy",
                    "Electrons at center (error bars are RMS);Kinetic energy (MeV);Reconstructed X (mm)", 5, 3, 15, "s");
 
-  T->Draw("ds.ev.fitResults.fit_position.fX>>hCenterRes", "ds.mc.particle.ke < 0.0205 && ds.mc.particle.pos.fX < 0.05",
-          "goff");
+  T->Draw("ds.ev.fitResults.fit_position.fX>>hCenterRes",
+          "ds.mc.particle.ke < 0.0205 && ds.mc.particle.pos.fX < 0.05 && ds.ev.fitResults.valid_position", "goff");
   T->Draw("ds.ev.fitResults.fit_position.fX-ds.mc.particle.pos.fX:(ds.mc.particle.pos.Mag()/437)>>hReconVsRadius",
-          "ds.mc.particle.ke < 0.0205 && ds.mc.particle.pos.fX > 0.05", "goff prof");
+          "ds.mc.particle.ke < 0.0205 && ds.mc.particle.pos.fX > 0.05 && ds.ev.fitResults.valid_position", "goff prof");
   T->Draw("ds.ev.fitResults.fit_position.fX:ds.mc.particle.ke>>hReconVsEnergy",
-          "ds.mc.particle.ke > 0.0205 && ds.mc.particle.pos.fX < 0.05", "goff prof");
+          "ds.mc.particle.ke > 0.0205 && ds.mc.particle.pos.fX < 0.05 && ds.ev.fitResults.valid_position", "goff prof");
 
   hCenterRes->Fit("gaus");
   hReconVsRadius->SetMinimum(-2000);