Add changes in calibration parameters

Author: gbencedi

PWGLF/Tasks/GlobalEventProperties/flattenictyPikp.cxx

@@ -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"};
@@ -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.";
         }
@@ -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;
   }
@@ -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,