Skip to content

Commit 4417c4e

Browse files
authored
[PWGEM/Dilepton] remove collinear from treeCreatorElectronMLDDA.cxx (#12522)
1 parent 2bea623 commit 4417c4e

File tree

1 file changed

+19
-22
lines changed

1 file changed

+19
-22
lines changed

PWGEM/Dilepton/TableProducer/treeCreatorElectronMLDDA.cxx

Lines changed: 19 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -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"};
@@ -749,7 +748,7 @@ struct TreeCreatorElectronMLDDA {
749748
Partition<MyTracks> negTracks = o2::aod::track::signed1Pt < 0.f && ncheckbit(aod::track::v001::detectorMap, (uint8_t)o2::aod::track::ITS) == true;
750749
std::vector<uint64_t> stored_trackIds;
751750

752-
void processPID(filteredMyCollisions const& collisions, aod::BCsWithTimestamps const&, filteredV0s const& v0s, filteredCascades const& cascades, MyTracks const& tracks, aod::V0s const&)
751+
void processPID(filteredMyCollisions const& collisions, aod::BCsWithTimestamps const&, filteredV0s const& v0s, filteredCascades const& cascades, MyTracks const& tracks)
753752
{
754753
stored_trackIds.reserve(tracks.size());
755754
for (const auto& collision : collisions) {
@@ -769,7 +768,7 @@ struct TreeCreatorElectronMLDDA {
769768

770769
auto v0s_coll = v0s.sliceBy(perCollision_v0, collision.globalIndex());
771770
for (const auto& v0 : v0s_coll) {
772-
auto o2v0 = v0.template v0_as<aod::V0s>();
771+
// auto o2v0 = v0.template v0_as<aod::V0s>();
773772
auto pos = v0.template posTrack_as<MyTracks>();
774773
auto neg = v0.template negTrack_as<MyTracks>();
775774
// LOGF(info, "v0.globalIndex() = %d, v0.collisionId() = %d, v0.posTrackId() = %d, v0.negTrackId() = %d", v0.globalIndex(), v0.collisionId(), v0.posTrackId(), v0.negTrackId());
@@ -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

Comments
 (0)