Skip to content

EventFiltering: Add the option for using TPC splines PID calibration #1996

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
fgrosa merged 2 commits into from
Feb 15, 2023
Merged

EventFiltering: Add the option for using TPC splines PID calibration #1996

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
4e00e4b
Add the option for using TPC splines PID calibration
zhangbiao-phy Feb 14, 2023
873ec8b
Change to proper name
zhangbiao-phy Feb 14, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 29 additions & 20 deletions EventFiltering/PWGHF/HFFilter.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -113,8 +113,12 @@ struct HfFilter { // Main struct for HF triggers
Configurable<std::string> ccdbPathGrpMag{"ccdbPathGrpMag", "GLO/Config/GRPMagField", "CCDB path of the GRPMagField object"};
Configurable<string> ccdbPathTPC{"ccdbPathTPC", "Users/i/iarsene/Calib/TPCpostCalib", "base path to the ccdb object"};

// TPC PID post calibrations
Configurable<bool> computeTPCPostCalib{"computeTPCPostCalib", false, "If true, compute TPC post-calibrated n-sigmas"};
// TPC PID calibrations
Configurable<int> setTPCCalib{"setTPCCalib", 0, "0 is not use re-calibrations, 1 is compute TPC post-calibrated n-sigmas, 2 is using TPC Spline"};
Configurable<std::string> ccdbBBProton{"ccdbBBProton", "Users/l/lserksny/PIDProton", "Path to the CCDB ocject for proton BB param"};
Configurable<std::string> ccdbBBAntiProton{"ccdbBBAntiProton", "Users/l/lserksny/PIDAntiProton", "Path to the CCDB ocject for antiproton BB param"};
Configurable<std::string> ccdbBBPion{"ccdbBBPion", "Users/l/lserksny/PIDPion", "Path to the CCDB ocject for Pion BB param"};
Configurable<std::string> ccdbBBAntiPion{"ccdbBBAntiPion", "Users/l/lserksny/PIDAntiPion", "Path to the CCDB ocject for antiPion BB param"};

// parameter for Optimisation Tree
Configurable<bool> applyOptimisation{"applyOptimisation", false, "Flag to enable or disable optimisation"};
Expand All @@ -140,12 +144,11 @@ struct HfFilter { // Main struct for HF triggers
std::shared_ptr<TH1> hGammaSelected, hGammaEtaBefore, hGammaEtaAfter;
std::shared_ptr<TH2> hGammaArmPodBefore, hGammaArmPodAfter;

// Histogram of TPC postcalibration map for pion and proton
TH3F* hMapPionMean = nullptr;
TH3F* hMapPionSigma = nullptr;
TH3F* hMapProtonMean = nullptr;
TH3F* hMapProtonSigma = nullptr;

// Histograms of TPC calibration for pion and proton
std::array<TH3F*, 2> hMapPion = {nullptr, nullptr};
std::array<TH3F*, 2> hMapProton = {nullptr, nullptr};
std::array<std::vector<double>, 2> hBBProton{};
std::array<std::vector<double>, 2> hBBPion{};
// ONNX
std::array<std::shared_ptr<Ort::Experimental::Session>, kNCharmParticles> sessionML = {nullptr, nullptr, nullptr, nullptr, nullptr};
std::array<std::vector<std::vector<int64_t>>, kNCharmParticles> inputShapesML{};
Expand Down Expand Up @@ -275,20 +278,26 @@ struct HfFilter { // Main struct for HF triggers
o2::base::Propagator::initFieldFromGRP(grpo);

// needed for TPC PID postcalibrations
if (computeTPCPostCalib) {
if (setTPCCalib == 1) {
auto calibList = ccdb->getForTimeStamp<TList>(ccdbPathTPC.value, bc.timestamp());
if (!calibList) {
LOG(fatal) << "Can not find the TPC Post Calibration object!";
}

hMapPionMean = (TH3F*)calibList->FindObject("mean_map_pion");
hMapPionSigma = (TH3F*)calibList->FindObject("sigma_map_pion");
hMapProtonMean = (TH3F*)calibList->FindObject("mean_map_proton");
hMapProtonSigma = (TH3F*)calibList->FindObject("sigma_map_proton");
hMapPion[0] = (TH3F*)calibList->FindObject("mean_map_pion");
hMapPion[1] = (TH3F*)calibList->FindObject("sigma_map_pion");
hMapProton[0] = (TH3F*)calibList->FindObject("mean_map_proton");
hMapProton[1] = (TH3F*)calibList->FindObject("sigma_map_proton");

if (!hMapPionMean || !hMapPionSigma || !hMapProtonMean || !hMapProtonSigma) {
if (!hMapPion[0] || !hMapPion[1] || !hMapProton[0] || !hMapProton[1]) {
LOG(fatal) << "Can not find histograms!";
}
} else if (setTPCCalib > 1) {

hBBProton[0] = setValuesBB(ccdbApi, bc, ccdbBBProton);
hBBProton[1] = setValuesBB(ccdbApi, bc, ccdbBBAntiProton);
hBBPion[0] = setValuesBB(ccdbApi, bc, ccdbBBPion);
hBBPion[1] = setValuesBB(ccdbApi, bc, ccdbBBAntiPion);
}

currentRun = bc.runNumber();
Expand All @@ -311,7 +320,7 @@ struct HfFilter { // Main struct for HF triggers
auto trackPos = cand2Prong.prong0_as<BigTracksPID>(); // positive daughter
auto trackNeg = cand2Prong.prong1_as<BigTracksPID>(); // negative daughter

auto preselD0 = isDzeroPreselected(trackPos, trackNeg, nsigmaTPCPionKaonDzero, nsigmaTOFPionKaonDzero, computeTPCPostCalib, hMapPionMean, hMapPionSigma);
auto preselD0 = isDzeroPreselected(trackPos, trackNeg, nsigmaTPCPionKaonDzero, nsigmaTOFPionKaonDzero, setTPCCalib, hMapPion, hBBPion);
if (!preselD0) {
continue;
}
Expand Down Expand Up @@ -465,7 +474,7 @@ struct HfFilter { // Main struct for HF triggers

// 2-prong femto
if (!keepEvent[kFemto2P] && isCharmTagged && track.collisionId() == thisCollId) {
bool isProton = isSelectedProton4Femto(track, trackParThird, femtoMinProtonPt, femtoMaxNsigmaProton, femtoProtonOnlyTOF, computeTPCPostCalib, hMapProtonMean, hMapProtonSigma, activateQA, hProtonTPCPID, hProtonTOFPID);
bool isProton = isSelectedProton4Femto(track, trackParThird, femtoMinProtonPt, femtoMaxNsigmaProton, femtoProtonOnlyTOF, setTPCCalib, hMapProton, hBBProton, activateQA, hProtonTPCPID, hProtonTOFPID);
if (isProton) {
float relativeMomentum = computeRelativeMomentum(pVecThird, pVec2Prong, massD0);
if (applyOptimisation) {
Expand Down Expand Up @@ -543,13 +552,13 @@ struct HfFilter { // Main struct for HF triggers
}

if (is3Prong[0]) { // D+ preselections
is3Prong[0] = isDplusPreselected(trackSecond, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, computeTPCPostCalib, hMapPionMean, hMapPionSigma);
is3Prong[0] = isDplusPreselected(trackSecond, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, setTPCCalib, hMapPion, hBBPion);
}
if (is3Prong[1]) { // Ds preselections
is3Prong[1] = isDsPreselected(pVecFirst, pVecThird, pVecSecond, trackSecond, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, computeTPCPostCalib, hMapPionMean, hMapPionSigma);
is3Prong[1] = isDsPreselected(pVecFirst, pVecThird, pVecSecond, trackSecond, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, setTPCCalib, hMapPion, hBBPion);
}
if (is3Prong[2] || is3Prong[3]) { // charm baryon preselections
auto presel = isCharmBaryonPreselected(trackFirst, trackThird, trackSecond, nsigmaTPCProtonLc, nsigmaTOFProtonLc, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, computeTPCPostCalib, hMapProtonMean, hMapProtonSigma, hMapPionMean, hMapPionSigma);
auto presel = isCharmBaryonPreselected(trackFirst, trackThird, trackSecond, nsigmaTPCProtonLc, nsigmaTOFProtonLc, nsigmaTPCKaon3Prong, nsigmaTOFKaon3Prong, setTPCCalib, hMapProton, hBBProton, hMapPion, hBBPion);
if (is3Prong[2]) {
is3Prong[2] = presel;
}
Expand Down Expand Up @@ -696,7 +705,7 @@ struct HfFilter { // Main struct for HF triggers
} // end beauty selection

// 3-prong femto
bool isProton = isSelectedProton4Femto(track, trackParFourth, femtoMinProtonPt, femtoMaxNsigmaProton, femtoProtonOnlyTOF, computeTPCPostCalib, hMapProtonMean, hMapProtonSigma, activateQA, hProtonTPCPID, hProtonTOFPID);
bool isProton = isSelectedProton4Femto(track, trackParFourth, femtoMinProtonPt, femtoMaxNsigmaProton, femtoProtonOnlyTOF, setTPCCalib, hMapProton, hBBProton, activateQA, hProtonTPCPID, hProtonTOFPID);
if (isProton && track.collisionId() == thisCollId) {
for (int iHypo{0}; iHypo < kNCharmParticles - 1 && !keepEvent[kFemto3P]; ++iHypo) {
if (isCharmTagged[iHypo]) {
Expand Down
Loading