Skip to content

PWGEM/Dilepton: merge dielectron and dimuon into dilepton task #7016

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
dsekihat merged 1 commit into from
Jul 27, 2024
Merged

PWGEM/Dilepton: merge dielectron and dimuon into dilepton task #7016

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
718 changes: 466 additions & 252 deletions PWGEM/Dilepton/Tasks/dielectronQC.cxx → PWGEM/Dilepton/Core/Dilepton.h
View file
Open in desktop

Large diffs are not rendered by default.

694 changes: 453 additions & 241 deletions PWGEM/Dilepton/Tasks/dielectronQCMC.cxx → PWGEM/Dilepton/Core/DileptonMC.h
View file
Open in desktop

Large diffs are not rendered by default.

56 changes: 55 additions & 1 deletion PWGEM/Dilepton/Core/PhotonHBT.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -197,6 +197,13 @@ struct PhotonHBT {
Configurable<bool> enableOptimizations{"enableOptimizations", false, "Enables the ONNX extended model-optimization: sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_EXTENDED)"};
} dielectroncuts;

struct : ConfigurableGroup {
std::string prefix = "ggpaircut_group";
Configurable<bool> applydR{"applydR", false, "apply deta-dphi cut to avoid track splitting/merging"};
Configurable<float> cfgMinDeltaEta{"cfgMinDeltaEta", 0.f, "min. delta-eta between 2 photons"};
Configurable<float> cfgMinDeltaPhi{"cfgMinDeltaPhi", 0.f, "min. delta-phi between 2 photons"};
} ggpaircuts;

~PhotonHBT()
{
delete emh1;
Expand Down Expand Up @@ -306,13 +313,14 @@ struct PhotonHBT {
const AxisSpec axis_kt{ConfKtBins, "k_{T} (GeV/c)"};

const AxisSpec axis_qinv{30, 0.0, +0.3, "q_{inv} (GeV/c)"};
const AxisSpec axis_qabs_lcms{30, 0.0, +0.3, "|q|^{LCMS} (GeV/c)"};
const AxisSpec axis_qabs_lcms{30, 0.0, +0.3, "|#bf{q}|^{LCMS} (GeV/c)"};
const AxisSpec axis_qout{60, -0.3, +0.3, "q_{out} (GeV/c)"}; // qout does not change between LAB and LCMS frame
const AxisSpec axis_qside{60, -0.3, +0.3, "q_{side} (GeV/c)"}; // qside does not change between LAB and LCMS frame
const AxisSpec axis_qlong{60, -0.3, +0.3, "q_{long} (GeV/c)"};

fRegistry.add("Pair/same/hs_1d", "diphoton correlation 1D", kTHnSparseD, {axis_kt, axis_qinv, axis_qabs_lcms}, true);
fRegistry.add("Pair/same/hs_3d", "diphoton correlation 3D LCMS", kTHnSparseD, {axis_kt, axis_qout, axis_qside, axis_qlong}, true);
fRegistry.add("Pair/same/hDeltaEtaDeltaPhi", "diphoton distance in #eta-#varphi plane;#Delta#varphi (rad);#Delta#eta", kTH2D, {{200, -0.1, +0.1}, {200, -0.1, 0.1}}, false);
fRegistry.addClone("Pair/same/", "Pair/mix/");
}

Expand Down Expand Up @@ -497,6 +505,10 @@ struct PhotonHBT {
// float qabs_lcms_tmp = q12_lcms.P();
// LOGF(info, "qabs_lcms = %f, qabs_lcms_tmp = %f", qabs_lcms, qabs_lcms_tmp);

float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
fRegistry.fill(HIST("Pair/") + HIST(event_pair_types[ev_id]) + HIST("hDeltaEtaDeltaPhi"), dphi, deta);
fRegistry.fill(HIST("Pair/") + HIST(event_pair_types[ev_id]) + HIST("hs_1d"), kt, qinv, qabs_lcms);
fRegistry.fill(HIST("Pair/") + HIST(event_pair_types[ev_id]) + HIST("hs_3d"), kt, qout_lcms, qside_lcms, qlong_lcms);
}
Expand Down Expand Up @@ -576,6 +588,12 @@ struct PhotonHBT {

ROOT::Math::PtEtaPhiMVector v1(g1.pt(), g1.eta(), g1.phi(), 0.);
ROOT::Math::PtEtaPhiMVector v2(g2.pt(), g2.eta(), g2.phi(), 0.);
float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<0>(collision, v1, v2);
ndiphoton++;

Expand Down Expand Up @@ -651,6 +669,12 @@ struct PhotonHBT {

std::pair pair_tmp = std::make_pair(std::make_pair(pos1.trackId(), ele1.trackId()), std::make_pair(pos2.trackId(), ele2.trackId()));
if (std::find(used_pairs_per_collision.begin(), used_pairs_per_collision.end(), pair_tmp) == used_pairs_per_collision.end()) {
float deta = v1_ee.Eta() - v2_ee.Eta();
float dphi = v1_ee.Phi() - v2_ee.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<0>(collision, v1_ee, v2_ee);
used_pairs_per_collision.emplace_back(std::make_pair(pair_tmp.first, pair_tmp.second));
used_pairs_per_collision.emplace_back(std::make_pair(pair_tmp.second, pair_tmp.first));
Expand Down Expand Up @@ -716,6 +740,12 @@ struct PhotonHBT {
ROOT::Math::PtEtaPhiMVector v_pos2(pos2.pt(), pos2.eta(), pos2.phi(), o2::constants::physics::MassElectron);
ROOT::Math::PtEtaPhiMVector v_ele2(ele2.pt(), ele2.eta(), ele2.phi(), o2::constants::physics::MassElectron);
ROOT::Math::PtEtaPhiMVector v2_ee = v_pos2 + v_ele2;
float deta = v1_gamma.Eta() - v2_ee.Eta();
float dphi = v1_gamma.Phi() - v2_ee.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<0>(collision, v1_gamma, v2_ee);
ndiphoton++;
std::pair<int, int> pair_tmp_id1 = std::make_pair(ndf, g1.globalIndex());
Expand Down Expand Up @@ -762,6 +792,12 @@ struct PhotonHBT {
for (auto& g2 : photons1_from_event_pool) {
ROOT::Math::PtEtaPhiMVector v1(g1.pt(), g1.eta(), g1.phi(), 0.);
ROOT::Math::PtEtaPhiMVector v2(g2.pt(), g2.eta(), g2.phi(), 0.);
float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<1>(collision, v1, v2);
}
}
Expand All @@ -782,6 +818,12 @@ struct PhotonHBT {
for (auto& g2 : photons1_from_event_pool) {
ROOT::Math::PtEtaPhiMVector v1(g1.pt(), g1.eta(), g1.phi(), g1.mass());
ROOT::Math::PtEtaPhiMVector v2(g2.pt(), g2.eta(), g2.phi(), g2.mass());
float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<1>(collision, v1, v2);
}
}
Expand All @@ -802,6 +844,12 @@ struct PhotonHBT {
for (auto& g2 : photons2_from_event_pool) { // dielectron
ROOT::Math::PtEtaPhiMVector v1(g1.pt(), g1.eta(), g1.phi(), 0.0); // keep v1 for PCM
ROOT::Math::PtEtaPhiMVector v2(g2.pt(), g2.eta(), g2.phi(), g2.mass());
float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<1>(collision, v1, v2);
}
}
Expand All @@ -822,6 +870,12 @@ struct PhotonHBT {
for (auto& g2 : photons1_from_event_pool) { // PCM
ROOT::Math::PtEtaPhiMVector v1(g2.pt(), g2.eta(), g2.phi(), 0.0); // keep v1 for PCM
ROOT::Math::PtEtaPhiMVector v2(g1.pt(), g1.eta(), g1.phi(), g1.mass());
float deta = v1.Eta() - v2.Eta();
float dphi = v1.Phi() - v2.Phi();
o2::math_utils::bringToPMPi(dphi);
if (ggpaircuts.applydR && std::pow(deta / ggpaircuts.cfgMinDeltaEta, 2) + std::pow(dphi / ggpaircuts.cfgMinDeltaPhi, 2) < 1.f) {
continue;
}
fillPairHistogram<1>(collision, v1, v2);
}
}
Expand Down
Loading