Skip to content

Commit 9118502

Browse files
authored
Add files via upload
1 parent e89f40a commit 9118502

File tree

2 files changed

+25
-34
lines changed

2 files changed

+25
-34
lines changed

PWGDQ/Core/HistogramsLibrary.cxx

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -984,21 +984,9 @@ void o2::aod::dqhistograms::DefineHistograms(HistogramManager* hm, const char* h
984984
for (int i = 0; i < 26; i++) {
985985
coschiBins[i] = -1.0 + 2.0 * TMath::Power(0.04 * i, 2.0);
986986
}
987-
988987
hm->AddHistogram(histClass, "Coschi", "", false, 25, coschiBins, VarManager::kMCCosChi, 0, nullptr, -1, 0, nullptr, -1, "", "", "", -1, VarManager::kMCWeight);
989-
990-
// for bkg
991-
hm->AddHistogram(histClass, "Coschi_randomPhi_trans", "", false, 25, coschiBins, VarManager::kMCCosChi_randomPhi_trans, 0, nullptr, -1, 0, nullptr, -1, "", "", "", -1, VarManager::kMCWeight_randomPhi_trans);
992-
hm->AddHistogram(histClass, "Coschi_randomPhi_toward", "", false, 25, coschiBins, VarManager::kMCCosChi_randomPhi_toward, 0, nullptr, -1, 0, nullptr, -1, "", "", "", -1, VarManager::kMCWeight_randomPhi_toward);
993-
hm->AddHistogram(histClass, "Coschi_randomPhi_away", "", false, 25, coschiBins, VarManager::kMCCosChi_randomPhi_away, 0, nullptr, -1, 0, nullptr, -1, "", "", "", -1, VarManager::kMCWeight_randomPhi_away);
994-
}
995-
if (!groupStr.CompareTo("energy-correlator-unfolding")) {
996-
double coschiBins[26];
997-
for (int i = 0; i < 26; i++) {
998-
coschiBins[i] = -1.0 + 0.08 * i;
999-
}
1000-
hm->AddHistogram(histClass, "Coschi_unfolding", "", false, 25, coschiBins, VarManager::kMCCosChi_rec, 25, coschiBins, VarManager::kMCCosChi_gen);
1001988
}
989+
1002990
if (!groupStr.CompareTo("polarization-pseudoproper-gen")) {
1003991
int varspTHE[3] = {VarManager::kMCPt, VarManager::kMCCosThetaHE, VarManager::kMCVertexingTauxyProjected};
1004992
int varspTCS[3] = {VarManager::kMCPt, VarManager::kMCCosThetaCS, VarManager::kMCVertexingTauxyProjected};
@@ -1947,6 +1935,7 @@ void o2::aod::dqhistograms::DefineHistograms(HistogramManager* hm, const char* h
19471935
coschiBins[i] = -1.0 + 2.0 * TMath::Power(0.04 * i, 2.0);
19481936
}
19491937

1938+
hm->AddHistogram(histClass, "Coschi_unfolding", "", false, 25, coschiBins, VarManager::kMCCosChi_rec, 25, coschiBins, VarManager::kMCCosChi_gen);
19501939
hm->AddHistogram(histClass, "Coschi", "", false, 25, coschiBins, VarManager::kCosChi, 0, nullptr, -1, 0, nullptr, -1, "", "", "", -1, VarManager::kECWeight);
19511940
hm->AddHistogram(histClass, "DeltaEta_DeltaPhi_weight", "", false, 20, -2.0, 2.0, VarManager::kDeltaEta, 50, -2.0, 6.0, VarManager::kDeltaPhi, 0, 0, 0, -1, "", "", "", -1, VarManager::kPtDau);
19521941
}

PWGDQ/Core/VarManager.h

Lines changed: 23 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -3015,28 +3015,30 @@ void VarManager::FillEnergyCorrelatorsMC(T const& track, T1 const& t1, float* va
30153015
template <int pairType, typename T1, typename T2, typename T, typename T3>
30163016
void VarManager::FillEnergyCorrelatorsMCUnfolding(T1 const& dilepton, T2 const& hadron, T const& track, T3 const& t1, float* values)
30173017
{
3018-
// energy correlators
3019-
float MassHadron;
3020-
if constexpr (pairType == kJpsiHadronMass) {
3021-
MassHadron = TMath::Sqrt(t1.e() * t1.e() - t1.p() * t1.p());
3022-
}
3023-
if constexpr (pairType == kJpsiPionMass) {
3024-
MassHadron = o2::constants::physics::MassPionCharged;
3018+
if (fgUsedVars[kMCCosChi_gen] || fgUsedVars[kMCWeight_gen] || fgUsedVars[kMCdeltaeta_gen] || fgUsedVars[kMCCosChi_rec] || fgUsedVars[kMCWeight_rec] || fgUsedVars[kMCdeltaeta_rec]) {
3019+
// energy correlators
3020+
float MassHadron;
3021+
if constexpr (pairType == kJpsiHadronMass) {
3022+
MassHadron = TMath::Sqrt(t1.e() * t1.e() - t1.p() * t1.p());
3023+
}
3024+
if constexpr (pairType == kJpsiPionMass) {
3025+
MassHadron = o2::constants::physics::MassPionCharged;
3026+
}
3027+
ROOT::Math::PtEtaPhiMVector v1_gen(track.pt(), track.eta(), track.phi(), o2::constants::physics::MassJPsi);
3028+
ROOT::Math::PtEtaPhiMVector v2_gen(t1.pt(), t1.eta(), t1.phi(), MassHadron);
3029+
float E_boost_gen = LorentzTransformJpsihadroncosChi("weight_boost", v1_gen, v2_gen);
3030+
float CosChi_gen = LorentzTransformJpsihadroncosChi("coschi", v1_gen, v2_gen);
3031+
values[kMCCosChi_gen] = CosChi_gen;
3032+
values[kMCWeight_gen] = E_boost_gen / o2::constants::physics::MassJPsi;
3033+
values[kMCdeltaeta_gen] = track.eta() - t1.eta();
3034+
3035+
ROOT::Math::PtEtaPhiMVector v1_rec(dilepton.pt(), dilepton.eta(), dilepton.phi(), dilepton.mass());
3036+
ROOT::Math::PtEtaPhiMVector v2_rec(hadron.pt(), hadron.eta(), hadron.phi(), o2::constants::physics::MassPionCharged);
3037+
values[kMCCosChi_rec] = LorentzTransformJpsihadroncosChi("coschi", v1_rec, v2_rec);
3038+
float E_boost_rec = LorentzTransformJpsihadroncosChi("weight_boost", v1_rec, v2_rec);
3039+
values[kMCWeight_rec] = E_boost_rec / v1_rec.M();
3040+
values[kMCdeltaeta_rec] = dilepton.eta() - hadron.eta();
30253041
}
3026-
ROOT::Math::PtEtaPhiMVector v1_gen(track.pt(), track.eta(), track.phi(), o2::constants::physics::MassJPsi);
3027-
ROOT::Math::PtEtaPhiMVector v2_gen(t1.pt(), t1.eta(), t1.phi(), MassHadron);
3028-
float E_boost_gen = LorentzTransformJpsihadroncosChi("weight_boost", v1_gen, v2_gen);
3029-
float CosChi_gen = LorentzTransformJpsihadroncosChi("coschi", v1_gen, v2_gen);
3030-
values[kMCCosChi_gen] = CosChi_gen;
3031-
values[kMCWeight_gen] = E_boost_gen / o2::constants::physics::MassJPsi;
3032-
values[kMCdeltaeta_gen] = track.eta() - t1.eta();
3033-
3034-
ROOT::Math::PtEtaPhiMVector v1_rec(dilepton.pt(), dilepton.eta(), dilepton.phi(), dilepton.mass());
3035-
ROOT::Math::PtEtaPhiMVector v2_rec(hadron.pt(), hadron.eta(), hadron.phi(), o2::constants::physics::MassPionCharged);
3036-
values[kMCCosChi_rec] = LorentzTransformJpsihadroncosChi("coschi", v1_rec, v2_rec);
3037-
float E_boost_rec = LorentzTransformJpsihadroncosChi("weight_boost", v1_rec, v2_rec);
3038-
values[kMCWeight_rec] = E_boost_rec / v1_rec.M();
3039-
values[kMCdeltaeta_rec] = dilepton.eta() - hadron.eta();
30403042
}
30413043

30423044
template <uint32_t fillMap, typename T1, typename T2, typename C>

0 commit comments

Comments
 (0)