Skip to content

Commit 7de67ca

Browse files
Suppress the Bayesian PID for the time being (#1347)
1 parent 0933cec commit 7de67ca

5 files changed

Lines changed: 49 additions & 2 deletions

File tree

PWGCF/TwoParticleCorrelations/Core/PIDSelectionFilterAndAnalysis.cxx

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,14 @@ using namespace o2::framework::expressions;
2525
using namespace o2::analysis::PWGCF;
2626
using namespace boost;
2727

28+
#ifdef INCORPORATEBAYESIANPID
2829
PIDSelectionConfigurable::PIDSelectionConfigurable(std::vector<std::string> pidtpcel, std::vector<std::string> pidtpcmu, std::vector<std::string> pidtpcpi, std::vector<std::string> pidtpcka, std::vector<std::string> pidtpcpr,
2930
std::vector<std::string> pidtofel, std::vector<std::string> pidtofmu, std::vector<std::string> pidtofpi, std::vector<std::string> pidtofka, std::vector<std::string> pidtofpr,
3031
std::vector<std::string> pidbayel, std::vector<std::string> pidbaymu, std::vector<std::string> pidbaypi, std::vector<std::string> pidbayka, std::vector<std::string> pidbaypr)
32+
#else
33+
PIDSelectionConfigurable::PIDSelectionConfigurable(std::vector<std::string> pidtpcel, std::vector<std::string> pidtpcmu, std::vector<std::string> pidtpcpi, std::vector<std::string> pidtpcka, std::vector<std::string> pidtpcpr,
34+
std::vector<std::string> pidtofel, std::vector<std::string> pidtofmu, std::vector<std::string> pidtofpi, std::vector<std::string> pidtofka, std::vector<std::string> pidtofpr)
35+
#endif
3136
: mPidTpcSel_el{},
3237
mPidTpcSel_mu{},
3338
mPidTpcSel_pi{},
@@ -84,11 +89,13 @@ PIDSelectionConfigurable::PIDSelectionConfigurable(std::vector<std::string> pidt
8489
mPidTofSel_pi = storeCutString(pidtofpi, "tofpi");
8590
mPidTofSel_ka = storeCutString(pidtofka, "tofka");
8691
mPidTofSel_pr = storeCutString(pidtofpr, "tofpr");
92+
#ifdef INCORPORATEBAYESIANPID
8793
mPidBaySel_el = storeCutString(pidbayel, "bayel");
8894
mPidBaySel_mu = storeCutString(pidbaymu, "baymu");
8995
mPidBaySel_pi = storeCutString(pidbaypi, "baypi");
9096
mPidBaySel_ka = storeCutString(pidbayka, "bayka");
9197
mPidBaySel_pr = storeCutString(pidbaypr, "baypr");
98+
#endif
9299
}
93100

94101
ClassImp(PIDSelectionFilterAndAnalysis);

PWGCF/TwoParticleCorrelations/Core/PIDSelectionFilterAndAnalysis.h

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020
#include "SkimmingConfigurableCuts.h"
2121
#include "SelectionFilterAndAnalysis.h"
2222

23+
#undef INCORPORATEBAYESIANPID
24+
2325
namespace o2
2426
{
2527
namespace analysis
@@ -35,6 +37,7 @@ class PIDSelectionConfigurable
3537
friend class PIDSelectionFilterAndAnalysis;
3638

3739
public:
40+
#ifdef INCORPORATEBAYESIANPID
3841
PIDSelectionConfigurable(std::string pidtpcel = "", std::string pidtpcmu = "", std::string pidtpcpi = "", std::string pidtpcka = "", std::string pidtpcpr = "",
3942
std::string pidtofel = "", std::string pidtofmu = "", std::string pidtofpi = "", std::string pidtofka = "", std::string pidtofpr = "",
4043
std::string pidbayel = "", std::string pidbaymu = "", std::string pidbaypi = "", std::string pidbayka = "", std::string pidbaypr = "")
@@ -44,6 +47,24 @@ class PIDSelectionConfigurable
4447
PIDSelectionConfigurable(std::vector<std::string> pidtpcel, std::vector<std::string> pidtpcmu, std::vector<std::string> pidtpcpi, std::vector<std::string> pidtpcka, std::vector<std::string> pidtpcpr,
4548
std::vector<std::string> pidtofel, std::vector<std::string> pidtofmu, std::vector<std::string> pidtofpi, std::vector<std::string> pidtofka, std::vector<std::string> pidtofpr,
4649
std::vector<std::string> pidbayel, std::vector<std::string> pidbaymu, std::vector<std::string> pidbaypi, std::vector<std::string> pidbayka, std::vector<std::string> pidbaypr);
50+
#else
51+
PIDSelectionConfigurable(std::string pidtpcel = "", std::string pidtpcmu = "", std::string pidtpcpi = "", std::string pidtpcka = "", std::string pidtpcpr = "",
52+
std::string pidtofel = "", std::string pidtofmu = "", std::string pidtofpi = "", std::string pidtofka = "", std::string pidtofpr = "")
53+
: mPidTpcSel_el{pidtpcel},
54+
mPidTpcSel_mu{pidtpcmu},
55+
mPidTpcSel_pi{pidtpcpi},
56+
mPidTpcSel_ka{pidtpcka},
57+
mPidTpcSel_pr{pidtpcpr},
58+
mPidTofSel_el{pidtofel},
59+
mPidTofSel_mu{pidtofmu},
60+
mPidTofSel_pi{pidtofpi},
61+
mPidTofSel_ka{pidtofka},
62+
mPidTofSel_pr{pidtofpr}
63+
{
64+
}
65+
PIDSelectionConfigurable(std::vector<std::string> pidtpcel, std::vector<std::string> pidtpcmu, std::vector<std::string> pidtpcpi, std::vector<std::string> pidtpcka, std::vector<std::string> pidtpcpr,
66+
std::vector<std::string> pidtofel, std::vector<std::string> pidtofmu, std::vector<std::string> pidtofpi, std::vector<std::string> pidtofka, std::vector<std::string> pidtofpr);
67+
#endif
4768

4869
private:
4970
std::string mPidTpcSel_el = "";
@@ -152,11 +173,13 @@ inline uint64_t PIDSelectionFilterAndAnalysis::Filter(TrackToFilter const& track
152173
filterBrickValue(mCloseNsigmasTOF[kPion], track.tofNSigmaPi());
153174
filterBrickValue(mCloseNsigmasTOF[kKaon], track.tofNSigmaKa());
154175
filterBrickValue(mCloseNsigmasTOF[kProton], track.tofNSigmaPr());
176+
#ifdef INCORPORATEBAYESIANPID
155177
filterBrickValue(mBayesProbability[kElectron], track.bayesEl());
156178
filterBrickValue(mBayesProbability[kMuon], track.bayesMu());
157179
filterBrickValue(mBayesProbability[kPion], track.bayesPi());
158180
filterBrickValue(mBayesProbability[kKaon], track.bayesKa());
159181
filterBrickValue(mBayesProbability[kProton], track.bayesPr());
182+
#endif
160183

161184
mSelectedMask = selectedMask;
162185
return mSelectedMask;

PWGCF/TwoParticleCorrelations/TableProducer/skimmingconf.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ struct : o2::framework::ConfigurableGroup {
1818
o2::framework::Configurable<std::vector<std::string>> bfield{"evtflt_bfield", {"positive-yes", "negative-yes"}, "B filed polarity cut: both 'yes' default, anything else alternative"};
1919
o2::framework::Configurable<std::vector<std::string>> zvtxsel{"evtflt_zvtx", {"rg{-7.0,7.0}-yes", "rg{-10.0,10.0}-no", "rg{-3.0,3.0}-no"}, "Z vertex cut: first, default value, next, alternatives"};
2020
o2::framework::Configurable<std::vector<std::string>> centmultsel{"evtflt_centmult", {"mrg{V0M,0,5,10,20,30,40,50,60,70,80}-yes", "mrg{CL1,0,5,10,20,30,40,50,60,70,80}-no"}, "Centrality/Multiplicity cut: first, default, next, alternatives"};
21-
o2::framework::Configurable<std::vector<std::string>> pileuprej{"evtflt_pileuprej", {"fnrg{V0M_TPCOUT=-0.5+3.7*x-0.14*x*x, 0.5-3.7*x+0.14*x*x}-yes", "fnrg{V0M_TRKLETS=-0.5+3.7*x-0.14*x*x, 0.5-3.7*x+0.14*x*x}-no"}, "Advanced pile-up rejection cut: first, default, next, alternatives"};
21+
o2::framework::Configurable<std::vector<std::string>> pileuprej{"evtflt_pileuprej", {"fnrg{V0M_TPCOUT=-0.5+3.7*x-0.14*x*x,0.5-3.7*x+0.14*x*x}-yes", "fnrg{V0M_TRKLETS=-0.5+3.7*x-0.14*x*x,0.5-3.7*x+0.14*x*x}-no"}, "Advanced pile-up rejection cut: first, default, next, alternatives"};
2222
} eventfilter;
2323

2424
struct : o2::framework::ConfigurableGroup {
@@ -50,13 +50,15 @@ struct : o2::framework::ConfigurableGroup {
5050
o2::framework::Configurable<std::vector<std::string>> tpcka{"pidflt_tof_ka", {"rg{-3.0,3.0}-yes", "rg{-2.0,2.0}-no", "rg{-3.0,5.0}-no"}, "nsigmas to the tof kaon line, below/above: first, default value, next, alternatives"};
5151
o2::framework::Configurable<std::vector<std::string>> tpcpr{"pidflt_tof_pr", {"rg{-3.0,3.0}-yes", "rg{-2.0,2.0}-no", "rg{-3.0,5.0}-no"}, "nsigmas to the tof proton line, below/above: first, default value, next, alternatives"};
5252
} pidtoffilter;
53+
#ifdef INCORPORATEBAYESIANPID
5354
struct : ConfigurableGroup {
5455
o2::framework::Configurable<std::vector<std::string>> bayel{"pidflt_bayes_el", {"th{80}-yes", "th{70}-no", "th{90}-no"}, "Bayesian probability for electron (%%): first, default value, next, alternatives"};
5556
o2::framework::Configurable<std::vector<std::string>> baymu{"pidflt_bayes_mu", {"th{80}-yes", "th{70}-no", "th{90}-no"}, "Bayesian probability for muon (%%): first, default value, next, alternatives"};
5657
o2::framework::Configurable<std::vector<std::string>> baypi{"pidflt_bayes_pi", {"th{80}-yes", "th{70}-no", "th{90}-no"}, "Bayesian probability for pion (%%): first, default value, next, alternatives"};
5758
o2::framework::Configurable<std::vector<std::string>> bayka{"pidflt_bayes_ka", {"th{80}-yes", "th{70}-no", "th{90}-no"}, "Bayesian probability for kaon (%%): first, default value, next, alternatives"};
5859
o2::framework::Configurable<std::vector<std::string>> baypr{"pidflt_bayes_pr", {"th{80}-yes", "th{70}-no", "th{90}-no"}, "Bayesian probability for proton (%%): first, default value, next, alternatives"};
5960
} pidbayesfilter;
61+
#endif
6062
} pidfilter;
6163

6264
#endif // O2_ANALYSIS_CFSKIMMINGCONF_H

PWGCF/TwoParticleCorrelations/TableProducer/twoParticleCorrelationsFiltering.cxx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
#include "Framework/AnalysisTask.h"
1313
#include "PWGCF/TwoParticleCorrelations/Core/EventSelectionFilterAndAnalysis.h"
1414
#include "PWGCF/TwoParticleCorrelations/Core/TrackSelectionFilterAndAnalysis.h"
15+
#include "PWGCF/TwoParticleCorrelations/Core/PIDSelectionFilterAndAnalysis.h"
1516
#include "PWGCF/TwoParticleCorrelations/DataModel/TwoParticleCorrelationsSkimmed.h"
1617
#include "PWGCF/TwoParticleCorrelations/DataModel/TwoParticleCorrelationsFiltered.h"
1718
#include "Framework/runDataProcessing.h"
@@ -83,9 +84,14 @@ struct TwoParticleCorrelationsFilter {
8384
PWGCF::TrackSelectionConfigurable trksel(trackfilter.ttype, trackfilter.nclstpc, trackfilter.nxrtpc, trackfilter.nclsits, trackfilter.chi2clustpc,
8485
trackfilter.chi2clusits, trackfilter.xrofctpc, trackfilter.dcaxy, trackfilter.dcaz, trackfilter.ptrange, trackfilter.etarange);
8586
fTrackFilter = new PWGCF::TrackSelectionFilterAndAnalysis(trksel, PWGCF::SelectionFilterAndAnalysis::kAnalysis);
87+
#ifdef INCORPORATEBAYESIANPID
8688
PWGCF::PIDSelectionConfigurable pidsel(pidfilter.pidtpcfilter.tpcel, pidfilter.pidtpcfilter.tpcmu, pidfilter.pidtpcfilter.tpcpi, pidfilter.pidtpcfilter.tpcka, pidfilter.pidtpcfilter.tpcpr,
8789
pidfilter.pidtoffilter.tpcel, pidfilter.pidtoffilter.tpcmu, pidfilter.pidtoffilter.tpcpi, pidfilter.pidtoffilter.tpcka, pidfilter.pidtoffilter.tpcpr,
8890
pidfilter.pidbayesfilter.bayel, pidfilter.pidbayesfilter.baymu, pidfilter.pidbayesfilter.baypi, pidfilter.pidbayesfilter.bayka, pidfilter.pidbayesfilter.baypr);
91+
#else
92+
PWGCF::PIDSelectionConfigurable pidsel(pidfilter.pidtpcfilter.tpcel, pidfilter.pidtpcfilter.tpcmu, pidfilter.pidtpcfilter.tpcpi, pidfilter.pidtpcfilter.tpcka, pidfilter.pidtpcfilter.tpcpr,
93+
pidfilter.pidtoffilter.tpcel, pidfilter.pidtoffilter.tpcmu, pidfilter.pidtoffilter.tpcpi, pidfilter.pidtoffilter.tpcka, pidfilter.pidtoffilter.tpcpr);
94+
#endif
8995
fPIDFilter = new PWGCF::PIDSelectionFilterAndAnalysis(pidsel, PWGCF::SelectionFilterAndAnalysis::kAnalysis);
9096

9197
nReportedTracks = 0;

PWGCF/TwoParticleCorrelations/TableProducer/twoParticleCorrelationsSkimming.cxx

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,14 @@ namespace o2::analysis::cfskim
3434
{
3535
#define LOGTRACKCOLLISIONS info
3636
#define LOGTRACKTRACKS info
37-
37+
#ifdef INCORPORATEBAYESIANPID
3838
using pidTables = soa::Join<aod::pidTPCEl, aod::pidTPCMu, aod::pidTPCPi, aod::pidTPCKa, aod::pidTPCPr,
3939
aod::pidTOFEl, aod::pidTOFMu, aod::pidTOFPi, aod::pidTOFKa, aod::pidTOFPr,
4040
aod::pidBayesEl, aod::pidBayesMu, aod::pidBayesPi, aod::pidBayesKa, aod::pidBayesPr>;
41+
#else
42+
using pidTables = soa::Join<aod::pidTPCEl, aod::pidTPCMu, aod::pidTPCPi, aod::pidTPCKa, aod::pidTPCPr,
43+
aod::pidTOFEl, aod::pidTOFMu, aod::pidTOFPi, aod::pidTOFKa, aod::pidTOFPr>;
44+
#endif
4145

4246
PWGCF::TrackSelectionFilterAndAnalysis* fTrackFilter = nullptr;
4347
PWGCF::EventSelectionFilterAndAnalysis* fEventFilter = nullptr;
@@ -136,9 +140,14 @@ struct TwoParticleCorrelationsSkimming {
136140
PWGCF::TrackSelectionConfigurable trksel(trackfilter.ttype, trackfilter.nclstpc, trackfilter.nxrtpc, trackfilter.nclsits, trackfilter.chi2clustpc,
137141
trackfilter.chi2clusits, trackfilter.xrofctpc, trackfilter.dcaxy, trackfilter.dcaz, trackfilter.ptrange, trackfilter.etarange);
138142
fTrackFilter = new PWGCF::TrackSelectionFilterAndAnalysis(trksel, PWGCF::SelectionFilterAndAnalysis::kFilter);
143+
#ifdef INCORPORATEBAYESIANPID
139144
PWGCF::PIDSelectionConfigurable pidsel(pidfilter.pidtpcfilter.tpcel, pidfilter.pidtpcfilter.tpcmu, pidfilter.pidtpcfilter.tpcpi, pidfilter.pidtpcfilter.tpcka, pidfilter.pidtpcfilter.tpcpr,
140145
pidfilter.pidtoffilter.tpcel, pidfilter.pidtoffilter.tpcmu, pidfilter.pidtoffilter.tpcpi, pidfilter.pidtoffilter.tpcka, pidfilter.pidtoffilter.tpcpr,
141146
pidfilter.pidbayesfilter.bayel, pidfilter.pidbayesfilter.baymu, pidfilter.pidbayesfilter.baypi, pidfilter.pidbayesfilter.bayka, pidfilter.pidbayesfilter.baypr);
147+
#else
148+
PWGCF::PIDSelectionConfigurable pidsel(pidfilter.pidtpcfilter.tpcel, pidfilter.pidtpcfilter.tpcmu, pidfilter.pidtpcfilter.tpcpi, pidfilter.pidtpcfilter.tpcka, pidfilter.pidtpcfilter.tpcpr,
149+
pidfilter.pidtoffilter.tpcel, pidfilter.pidtoffilter.tpcmu, pidfilter.pidtoffilter.tpcpi, pidfilter.pidtoffilter.tpcka, pidfilter.pidtoffilter.tpcpr);
150+
#endif
142151
fPIDFilter = new PWGCF::PIDSelectionFilterAndAnalysis(pidsel, PWGCF::SelectionFilterAndAnalysis::kFilter);
143152
nReportedTracks = 0;
144153

0 commit comments

Comments
 (0)