@@ -207,7 +207,6 @@ struct TreeCreatorElectronMLDDA {
207207 Configurable<bool > cfg_includeITSsa{" cfg_includeITSsa" , false , " Flag to include ITSsa tracks" };
208208 Configurable<float > cfg_max_pt_itssa{" cfg_max_pt_itssa" , 0.15 , " mix pt for ITSsa track" };
209209 Configurable<float > cfg_min_qt_strangeness{" cfg_min_qt_strangeness" , 0.015 , " min qt for Lambda and K0S" };
210- Configurable<bool > cfg_require_collinearV0{" cfg_require_collinearV0" , false , " require collinear V0 for photon conversions" };
211210
212211 Configurable<float > cfg_min_TPCNsigmaEl{" cfg_min_TPCNsigmaEl" , -5 , " min n sigma e in TPC" };
213212 Configurable<float > cfg_max_TPCNsigmaEl{" cfg_max_TPCNsigmaEl" , +5 , " max n sigma e in TPC" };
@@ -833,27 +832,25 @@ struct TreeCreatorElectronMLDDA {
833832 }
834833 } // end of stangeness
835834
836- if (!v0cuts.cfg_require_collinearV0 || o2v0.isCollinearV0 ()) {
837- if (isElectronTight (pos) && isElectron (neg)) {
838- registry.fill (HIST (" V0/hMassGamma" ), v0.mGamma ());
839- registry.fill (HIST (" V0/hMassGamma_Rxy" ), v0.v0radius (), v0.mGamma ());
840- if (v0cuts.cfg_min_mass_photon < v0.mGamma () && v0.mGamma () < v0cuts.cfg_max_mass_photon ) {
841- registry.fill (HIST (" V0/hXY_Gamma" ), v0.x (), v0.y ());
842- fillTrackTable (collision, neg, static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron ));
843- registry.fill (HIST (" V0/hTPCdEdx_P_El" ), neg.tpcInnerParam (), neg.tpcSignal ());
844- registry.fill (HIST (" V0/hTOFbeta_P_El" ), neg.tpcInnerParam (), neg.beta ());
845- }
835+ if (isElectronTight (pos) && isElectron (neg)) {
836+ registry.fill (HIST (" V0/hMassGamma" ), v0.mGamma ());
837+ registry.fill (HIST (" V0/hMassGamma_Rxy" ), v0.v0radius (), v0.mGamma ());
838+ if (v0cuts.cfg_min_mass_photon < v0.mGamma () && v0.mGamma () < v0cuts.cfg_max_mass_photon ) {
839+ registry.fill (HIST (" V0/hXY_Gamma" ), v0.x (), v0.y ());
840+ fillTrackTable (collision, neg, static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron ));
841+ registry.fill (HIST (" V0/hTPCdEdx_P_El" ), neg.tpcInnerParam (), neg.tpcSignal ());
842+ registry.fill (HIST (" V0/hTOFbeta_P_El" ), neg.tpcInnerParam (), neg.beta ());
846843 }
844+ }
847845
848- if (isElectron (pos) && isElectronTight (neg)) {
849- registry.fill (HIST (" V0/hMassGamma" ), v0.mGamma ());
850- registry.fill (HIST (" V0/hMassGamma_Rxy" ), v0.v0radius (), v0.mGamma ());
851- if (v0cuts.cfg_min_mass_photon < v0.mGamma () && v0.mGamma () < v0cuts.cfg_max_mass_photon ) {
852- registry.fill (HIST (" V0/hXY_Gamma" ), v0.x (), v0.y ());
853- fillTrackTable (collision, pos, static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron ));
854- registry.fill (HIST (" V0/hTPCdEdx_P_El" ), pos.tpcInnerParam (), pos.tpcSignal ());
855- registry.fill (HIST (" V0/hTOFbeta_P_El" ), pos.tpcInnerParam (), pos.beta ());
856- }
846+ if (isElectron (pos) && isElectronTight (neg)) {
847+ registry.fill (HIST (" V0/hMassGamma" ), v0.mGamma ());
848+ registry.fill (HIST (" V0/hMassGamma_Rxy" ), v0.v0radius (), v0.mGamma ());
849+ if (v0cuts.cfg_min_mass_photon < v0.mGamma () && v0.mGamma () < v0cuts.cfg_max_mass_photon ) {
850+ registry.fill (HIST (" V0/hXY_Gamma" ), v0.x (), v0.y ());
851+ fillTrackTable (collision, pos, static_cast <uint8_t >(o2::aod::pwgem::dilepton::ml::PID_Label::kElectron ));
852+ registry.fill (HIST (" V0/hTPCdEdx_P_El" ), pos.tpcInnerParam (), pos.tpcSignal ());
853+ registry.fill (HIST (" V0/hTOFbeta_P_El" ), pos.tpcInnerParam (), pos.beta ());
857854 }
858855 }
859856
0 commit comments