Skip to content
Merged
Changes from all commits
Commits
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
74 changes: 52 additions & 22 deletions PWGLF/Tasks/GlobalEventProperties/flattenictyPikp.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -229,12 +229,18 @@ struct FlattenictyPikp {
Configurable<bool> cfgFillV0Hist{"cfgFillV0Hist", false, "fill V0 histograms"};
Configurable<bool> cfgFillChrgType{"cfgFillChrgType", false, "fill histograms per charge types"};
Configurable<bool> cfgFillChrgTypeV0s{"cfgFillChrgTypeV0s", false, "fill V0s histograms per charge types"};
Configurable<std::vector<float>> paramsFuncMIPpos{"paramsFuncMIPpos", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPneg{"paramsFuncMIPneg", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPall{"paramsFuncMIPall", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUpos{"paramsFuncPlateaUpos", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUneg{"paramsFuncPlateaUneg", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUall{"paramsFuncPlateaUall", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPposEtaP{"paramsFuncMIPposEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPnegEtaP{"paramsFuncMIPnegEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPallEtaP{"paramsFuncMIPallEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPposEtaN{"paramsFuncMIPposEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPnegEtaN{"paramsFuncMIPnegEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncMIPallEtaN{"paramsFuncMIPallEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUposEtaP{"paramsFuncPlateaUposEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUnegEtaP{"paramsFuncPlateaUnegEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUallEtaP{"paramsFuncPlateaUallEtaP", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUposEtaN{"paramsFuncPlateaUposEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUnegEtaN{"paramsFuncPlateaUnegEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::vector<float>> paramsFuncPlateaUallEtaN{"paramsFuncPlateaUallEtaN", std::vector<float>{-1.f}, "parameters of pol2"};
Configurable<std::string> cfgGainEqCcdbPath{"cfgGainEqCcdbPath", "Users/g/gbencedi/flattenicity/GainEq", "CCDB path for gain equalization constants"};
Configurable<std::string> cfgVtxEqCcdbPath{"cfgVtxEqCcdbPath", "Users/g/gbencedi/flattenicity/ZvtxEq", "CCDB path for z-vertex equalization constants"};
Configurable<std::string> cfgDeDxCalibCcdbPath{"cfgDeDxCalibCcdbPath", "Users/g/gbencedi/flattenicity/dEdxCalib", "CCDB path for dEdx calibration"};
Expand Down Expand Up @@ -403,25 +409,38 @@ struct FlattenictyPikp {

void init(InitContext&)
{
auto vecParamsMIPpos = (std::vector<float>)paramsFuncMIPpos;
auto vecParamsMIPneg = (std::vector<float>)paramsFuncMIPneg;
auto vecParamsMIPall = (std::vector<float>)paramsFuncMIPall;
auto vecParamsPLApos = (std::vector<float>)paramsFuncPlateaUpos;
auto vecParamsPLAneg = (std::vector<float>)paramsFuncPlateaUneg;
auto vecParamsPLAall = (std::vector<float>)paramsFuncPlateaUall;
auto vecParamsMIPposEtaP = (std::vector<float>)paramsFuncMIPposEtaP;
auto vecParamsMIPposEtaN = (std::vector<float>)paramsFuncMIPposEtaN;
auto vecParamsMIPnegEtaP = (std::vector<float>)paramsFuncMIPnegEtaP;
auto vecParamsMIPnegEtaN = (std::vector<float>)paramsFuncMIPnegEtaN;
auto vecParamsMIPallEtaP = (std::vector<float>)paramsFuncMIPallEtaP;
auto vecParamsMIPallEtaN = (std::vector<float>)paramsFuncMIPallEtaN;

auto vecParamsPLAposEtaP = (std::vector<float>)paramsFuncPlateaUposEtaP;
auto vecParamsPLAposEtaN = (std::vector<float>)paramsFuncPlateaUposEtaN;
auto vecParamsPLAnegEtaP = (std::vector<float>)paramsFuncPlateaUnegEtaP;
auto vecParamsPLAnegEtaN = (std::vector<float>)paramsFuncPlateaUnegEtaN;
auto vecParamsPLAallEtaP = (std::vector<float>)paramsFuncPlateaUallEtaP;
auto vecParamsPLAallEtaN = (std::vector<float>)paramsFuncPlateaUallEtaN;

auto addVec = [&](std::vector<std::vector<float>>& targetVec, const std::string& name, bool isMIP) {
if (isMIP) {
targetVec.emplace_back(vecParamsMIPpos);
targetVec.emplace_back(vecParamsMIPneg);
targetVec.emplace_back(vecParamsMIPall);
targetVec.emplace_back(vecParamsMIPposEtaP);
targetVec.emplace_back(vecParamsMIPnegEtaP);
targetVec.emplace_back(vecParamsMIPallEtaP);
targetVec.emplace_back(vecParamsMIPposEtaN);
targetVec.emplace_back(vecParamsMIPnegEtaN);
targetVec.emplace_back(vecParamsMIPallEtaN);
if (!vecParamsMIP.size()) {
LOG(info) << "size of " << name << "is zero.";
}
} else {
targetVec.emplace_back(vecParamsPLApos);
targetVec.emplace_back(vecParamsPLAneg);
targetVec.emplace_back(vecParamsPLAall);
targetVec.emplace_back(vecParamsPLAposEtaP);
targetVec.emplace_back(vecParamsPLAnegEtaP);
targetVec.emplace_back(vecParamsPLAallEtaP);
targetVec.emplace_back(vecParamsPLAposEtaN);
targetVec.emplace_back(vecParamsPLAnegEtaN);
targetVec.emplace_back(vecParamsPLAallEtaN);
if (!vecParamsPLA.size()) {
LOG(info) << "size of " << name << "is zero.";
}
Expand Down Expand Up @@ -1144,12 +1163,24 @@ struct FlattenictyPikp {
float valCalib = -1.;
if constexpr (isChrg) {
if (track.sign() * track.tpcInnerParam() > Cnull) {
valCalib = fCalib.at(0)->Eval(track.eta());
if (track.eta() >= 0) {
valCalib = fCalib.at(0)->Eval(track.eta());
} else {
valCalib = fCalib.at(3)->Eval(track.eta());
}
} else {
valCalib = fCalib.at(1)->Eval(track.eta());
if (track.eta() >= 0) {
valCalib = fCalib.at(1)->Eval(track.eta());
} else {
valCalib = fCalib.at(4)->Eval(track.eta());
}
}
} else {
valCalib = fCalib.at(2)->Eval(track.eta());
if (track.eta() >= 0) {
valCalib = fCalib.at(2)->Eval(track.eta());
} else {
valCalib = fCalib.at(5)->Eval(track.eta());
}
}
return valCalib;
}
Expand Down Expand Up @@ -2233,7 +2264,6 @@ struct FlattenictyPikp {

// using Particles = soa::Filtered<aod::McParticles>;
// expressions::Filter primaries = (aod::mcparticle::flags & (uint8_t)o2::aod::mcparticle::enums::PhysicalPrimary) == (uint8_t)o2::aod::mcparticle::enums::PhysicalPrimary;

void processMCclosure(Colls::iterator const& collision,
MyPIDTracks const& tracks,
MyLabeledTracks const& mcTrackLabels,
Expand Down
Loading