Skip to content

[PWGEM/Dilepton] add muons in struct CheckSmearing #12314

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 30, 2025
Merged

[PWGEM/Dilepton] add muons in struct CheckSmearing #12314

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
127 changes: 78 additions & 49 deletions PWGEM/Dilepton/Tasks/smearing.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,21 +13,21 @@
// Analysis task to produce smeared pt, eta, phi for electrons/muons in dilepton analysis
// Please write to: daiki.sekihata@cern.ch

#include <array>
#include <string>
#include <chrono>
#include "PWGEM/Dilepton/DataModel/dileptonTables.h"
#include "PWGEM/Dilepton/Utils/MomentumSmearer.h"

#include "CCDB/BasicCCDBManager.h"
#include "Framework/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/ASoA.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/AnalysisTask.h"
#include "Framework/DataTypes.h"
#include "Framework/HistogramRegistry.h"
// #include "PWGDQ/DataModel/ReducedInfoTables.h" // remove this later, because 2 data tables (covariant matrix) in this header confilict against EM tables.
#include "PWGEM/Dilepton/Utils/MomentumSmearer.h"
#include "PWGEM/Dilepton/DataModel/dileptonTables.h"
#include "Framework/runDataProcessing.h"

#include <array>
#include <chrono>
#include <string>

using namespace o2;
using namespace o2::framework;
Expand Down Expand Up @@ -194,7 +194,7 @@
template <o2::aod::pwgem::dilepton::smearing::EMAnaType type, typename TTracksMC, typename TCollisions, typename TMCCollisions>
void applySmearing(TTracksMC const& tracksMC, TCollisions const& collisions, TMCCollisions const&)
{
for (auto& mctrack : tracksMC) {

Check failure on line 197 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[const-ref-in-for-loop] 

Use constant references for non-modified iterators in range-based for loops.
float ptgen = mctrack.pt();
float etagen = mctrack.eta();
float phigen = mctrack.phi();
Expand All @@ -218,7 +218,7 @@
}

int pdgCode = mctrack.pdgCode();
if (std::abs(pdgCode) == 11) {

Check failure on line 221 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
int ch = -1;
if (pdgCode < 0) {
ch = 1;
Expand All @@ -232,7 +232,7 @@
// fill the table
smearedelectron(ptsmeared, etasmeared, phismeared, efficiency, dca);
smearedmuon(ptgen, etagen, phigen, 1.f, 0.f, ptgen, etagen, phigen, 1.f, 0.f);
} else if (std::abs(pdgCode) == 13) {

Check failure on line 235 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
int ch = -1;
if (pdgCode < 0) {
ch = 1;
Expand Down Expand Up @@ -276,12 +276,12 @@
void processDummyCocktail(aod::McParticles const& tracksMC)
{
// don't apply smearing
for (auto& mctrack : tracksMC) {

Check failure on line 279 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[const-ref-in-for-loop] 

Use constant references for non-modified iterators in range-based for loops.
int pdgCode = mctrack.pdgCode();
if (std::abs(pdgCode) == 11) {

Check failure on line 281 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
smearedelectron(mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0);
smearedmuon(mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0, mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0);
} else if (std::abs(pdgCode) == 13) {

Check failure on line 284 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
smearedelectron(mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0);
smearedmuon(mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0, mctrack.pt(), mctrack.eta(), mctrack.phi(), 1.0, 0.0);
} else {
Expand All @@ -292,20 +292,16 @@
}

void processDummyMCanalysisEM(aod::EMMCParticles const&) {}
// void processDummyMCanalysisDQ(ReducedMCTracks const&) {}

PROCESS_SWITCH(ApplySmearing, processMCanalysisEM, "Run for MC analysis which uses skimmed EM data format", false);
// PROCESS_SWITCH(ApplySmearing, processMCanalysisDQ, "Run for MC analysis which uses skimmed DQ data format", false);
PROCESS_SWITCH(ApplySmearing, processCocktail, "Run for cocktail analysis", false);
PROCESS_SWITCH(ApplySmearing, processDummyMCanalysisEM, "Dummy process function", false);
// PROCESS_SWITCH(ApplySmearing, processDummyMCanalysisDQ, "Dummy process function", false);
PROCESS_SWITCH(ApplySmearing, processDummyCocktail, "Dummy process function", true);
};

struct CheckSmearing {
using EMMCParticlesWithSmearing = soa::Join<aod::EMMCParticles, aod::SmearedElectrons>; // this is only for electrons
// using MyReducedTracks = soa::Join<ReducedMCTracks, aod::SmearedElectrons>; // this is only for electrons
using MyCocktailTracks = soa::Join<aod::McParticles, aod::SmearedElectrons>; // this is only for electrons
using EMMCParticlesWithSmearing = soa::Join<aod::EMMCParticles, aod::SmearedElectrons, aod::SmearedMuons>;
using MyCocktailTracks = soa::Join<aod::McParticles, aod::SmearedElectrons, aod::SmearedMuons>;

// Run for electrons or muons
Configurable<int> fPdgCode{"cfgPdgCode", 11, "Set the type of particle to be checked"};
Expand All @@ -322,9 +318,9 @@

void init(o2::framework::InitContext&)
{
registry.add<TH2>("hCorrelation_Pt", "pT correlation;p_{T,l}^{gen} (GeV/c);p_{T,l}^{smeared} (GeV/c)", {HistType::kTH2F, {{1000, 0.0f, 10.0f}, {1000, 0.0f, 10.0f}}});
registry.add<TH2>("hCorrelation_Eta", "eta correlation;#eta_{l}^{gen};#eta_{l}^{smeared}", {HistType::kTH2F, {{200, -1.0f, +1.0f}, {200, -1.0f, +1.0f}}});
registry.add<TH2>("hCorrelation_Phi", "phi correlation;#varphi_{l}^{gen} (rad.);#varphi_{l}^{smeared} (rad.)", {HistType::kTH2F, {{100, 0.0f, TMath::TwoPi()}, {100, 0.0f, TMath::TwoPi()}}});
registry.add<TH2>("Electron/hCorrelation_Pt", "pT correlation;p_{T,l}^{gen} (GeV/c);p_{T,l}^{smeared} (GeV/c)", {HistType::kTH2F, {{1000, 0.0f, 10.0f}, {1000, 0.0f, 10.0f}}});
registry.add<TH2>("Electron/hCorrelation_Eta", "eta correlation;#eta_{l}^{gen};#eta_{l}^{smeared}", {HistType::kTH2F, {{200, -1.0f, +1.0f}, {200, -1.0f, +1.0f}}});
registry.add<TH2>("Electron/hCorrelation_Phi", "phi correlation;#varphi_{l}^{gen} (rad.);#varphi_{l}^{smeared} (rad.)", {HistType::kTH2F, {{100, 0.0f, TMath::TwoPi()}, {100, 0.0f, TMath::TwoPi()}}});

Check failure on line 323 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[external-pi] 

Use the PI constant (and its multiples and fractions) defined in o2::constants::math.

// Binning for resolution
AxisSpec axisPtRes{ptResBins, "#it{p}^{gen}_{T,l} (GeV/#it{c})"};
Expand All @@ -333,22 +329,25 @@
AxisSpec axisDeltaphiRes{deltaphiResBins, "#varphi^{gen} - #varphi^{rec} (rad.)"};

if (!fConfigUsePtVecRes) {
registry.add<TH2>("PtGen_DeltaPtOverPtGen", "", HistType::kTH2D, {axisPtRes, axisDeltaptRes}, true);
registry.add<TH2>("PtGen_DeltaEta", "", HistType::kTH2D, {axisPtRes, axisDeltaetaRes}, true);
registry.add<TH2>("PtGen_DeltaPhi_Neg", "", HistType::kTH2D, {axisPtRes, axisDeltaphiRes}, true);
registry.add<TH2>("PtGen_DeltaPhi_Pos", "", HistType::kTH2D, {axisPtRes, axisDeltaphiRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPtOverPtGen", "", HistType::kTH2D, {axisPtRes, axisDeltaptRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaEta", "", HistType::kTH2D, {axisPtRes, axisDeltaetaRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPhi_Neg", "", HistType::kTH2D, {axisPtRes, axisDeltaphiRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPhi_Pos", "", HistType::kTH2D, {axisPtRes, axisDeltaphiRes}, true);
} else {
registry.add<TH2>("PtGen_DeltaPtOverPtGen", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaptRes}, true);
registry.add<TH2>("PtGen_DeltaEta", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaetaRes}, true);
registry.add<TH2>("PtGen_DeltaPhi_Neg", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaphiRes}, true);
registry.add<TH2>("PtGen_DeltaPhi_Pos", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaphiRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPtOverPtGen", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaptRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaEta", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaetaRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPhi_Neg", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaphiRes}, true);
registry.add<TH2>("Electron/PtGen_DeltaPhi_Pos", "", HistType::kTH2D, {{ptResBinsVec, "#it{p}^{gen}_{T,l} (GeV/#it{c})"}, axisDeltaphiRes}, true);
}

registry.addClone("Electron/", "GlobalMuon/");
registry.addClone("Electron/", "StandaloneMuon/");
}

template <o2::aod::pwgem::dilepton::smearing::EMAnaType type, typename TTracksMC, typename TMCCollisions>
void Check(TTracksMC const& tracksMC, TMCCollisions const&)
{
for (auto& mctrack : tracksMC) {

Check failure on line 350 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[const-ref-in-for-loop] 

Use constant references for non-modified iterators in range-based for loops.
if (std::abs(mctrack.pdgCode()) != fPdgCode) {
continue;
}
Expand All @@ -360,21 +359,59 @@
}
}

float deltaptoverpt = -1000.;
if (mctrack.pt() > 0.)
deltaptoverpt = (mctrack.pt() - mctrack.ptSmeared()) / mctrack.pt();
float deltaeta = mctrack.eta() - mctrack.etaSmeared();
float deltaphi = mctrack.phi() - mctrack.phiSmeared();
registry.fill(HIST("PtGen_DeltaPtOverPtGen"), mctrack.pt(), deltaptoverpt);
registry.fill(HIST("PtGen_DeltaEta"), mctrack.pt(), deltaeta);
if (mctrack.pdgCode() < 0) {
registry.fill(HIST("PtGen_DeltaPhi_Neg"), mctrack.pt(), deltaphi);
} else {
registry.fill(HIST("PtGen_DeltaPhi_Pos"), mctrack.pt(), deltaphi);
if (std::abs(mctrack.pdgCode()) == 11) { // for electrons

Check failure on line 362 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
float deltaptoverpt = -1000.f;
if (mctrack.pt() > 0.f) {
deltaptoverpt = (mctrack.pt() - mctrack.ptSmeared()) / mctrack.pt();
}
float deltaeta = mctrack.eta() - mctrack.etaSmeared();
float deltaphi = mctrack.phi() - mctrack.phiSmeared();
registry.fill(HIST("Electron/PtGen_DeltaPtOverPtGen"), mctrack.pt(), deltaptoverpt);
registry.fill(HIST("Electron/PtGen_DeltaEta"), mctrack.pt(), deltaeta);
if (mctrack.pdgCode() < 0) { // e+
registry.fill(HIST("Electron/PtGen_DeltaPhi_Pos"), mctrack.pt(), deltaphi);
} else { // e-
registry.fill(HIST("Electron/PtGen_DeltaPhi_Neg"), mctrack.pt(), deltaphi);
}
registry.fill(HIST("Electron/hCorrelation_Pt"), mctrack.pt(), mctrack.ptSmeared());
registry.fill(HIST("Electron/hCorrelation_Eta"), mctrack.eta(), mctrack.etaSmeared());
registry.fill(HIST("Electron/hCorrelation_Phi"), mctrack.phi(), mctrack.phiSmeared());
} else if (std::abs(mctrack.pdgCode()) == 13) { // for muons

Check failure on line 379 in PWGEM/Dilepton/Tasks/smearing.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[pdg/explicit-code] 

Avoid hard-coded PDG codes. Use named values from PDG_t or o2::constants::physics::Pdg instead.
float deltaptoverpt = -1000.f;
// for standalone muons
if (mctrack.pt() > 0.f) {
deltaptoverpt = (mctrack.pt() - mctrack.ptSmeared_sa_muon()) / mctrack.pt();
}
float deltaeta = mctrack.eta() - mctrack.etaSmeared_sa_muon();
float deltaphi = mctrack.phi() - mctrack.phiSmeared_sa_muon();
registry.fill(HIST("StandaloneMuon/PtGen_DeltaPtOverPtGen"), mctrack.pt(), deltaptoverpt);
registry.fill(HIST("StandaloneMuon/PtGen_DeltaEta"), mctrack.pt(), deltaeta);
if (mctrack.pdgCode() < 0) { // mu+
registry.fill(HIST("StandaloneMuon/PtGen_DeltaPhi_Pos"), mctrack.pt(), deltaphi);
} else { // mu-
registry.fill(HIST("StandaloneMuon/PtGen_DeltaPhi_Neg"), mctrack.pt(), deltaphi);
}
registry.fill(HIST("StandaloneMuon/hCorrelation_Pt"), mctrack.pt(), mctrack.ptSmeared_sa_muon());
registry.fill(HIST("StandaloneMuon/hCorrelation_Eta"), mctrack.eta(), mctrack.etaSmeared_sa_muon());
registry.fill(HIST("StandaloneMuon/hCorrelation_Phi"), mctrack.phi(), mctrack.phiSmeared_sa_muon());

// for global muons
if (mctrack.pt() > 0.f) {
deltaptoverpt = (mctrack.pt() - mctrack.ptSmeared_gl_muon()) / mctrack.pt();
}
deltaeta = mctrack.eta() - mctrack.etaSmeared_gl_muon();
deltaphi = mctrack.phi() - mctrack.phiSmeared_gl_muon();
registry.fill(HIST("GlobalMuon/PtGen_DeltaPtOverPtGen"), mctrack.pt(), deltaptoverpt);
registry.fill(HIST("GlobalMuon/PtGen_DeltaEta"), mctrack.pt(), deltaeta);
if (mctrack.pdgCode() < 0) { // mu+
registry.fill(HIST("GlobalMuon/PtGen_DeltaPhi_Pos"), mctrack.pt(), deltaphi);
} else { // mu-
registry.fill(HIST("GlobalMuon/PtGen_DeltaPhi_Neg"), mctrack.pt(), deltaphi);
}
registry.fill(HIST("GlobalMuon/hCorrelation_Pt"), mctrack.pt(), mctrack.ptSmeared_gl_muon());
registry.fill(HIST("GlobalMuon/hCorrelation_Eta"), mctrack.eta(), mctrack.etaSmeared_gl_muon());
registry.fill(HIST("GlobalMuon/hCorrelation_Phi"), mctrack.phi(), mctrack.phiSmeared_gl_muon());
}
registry.fill(HIST("hCorrelation_Pt"), mctrack.pt(), mctrack.ptSmeared());
registry.fill(HIST("hCorrelation_Eta"), mctrack.eta(), mctrack.etaSmeared());
registry.fill(HIST("hCorrelation_Phi"), mctrack.phi(), mctrack.phiSmeared());
} // end of mctrack loop
}

Expand All @@ -383,25 +420,17 @@
Check<o2::aod::pwgem::dilepton::smearing::EMAnaType::kEfficiency>(tracksMC, mccollisions);
}

// void processCheckMCanalysisDQ(MyReducedTracks const& tracksMC)
// {
// Check(tracksMC);
// }

void processCheckCocktail(MyCocktailTracks const& tracksMC)
{
Check<o2::aod::pwgem::dilepton::smearing::EMAnaType::kCocktail>(tracksMC, nullptr);
}

void processDummyMCanalysisEM(aod::EMMCParticles const&) {}
// void processDummyMCanalysisDQ(ReducedMCTracks const&) {}
void processDummyCocktail(aod::McParticles const&) {}

PROCESS_SWITCH(CheckSmearing, processCheckMCanalysisEM, "Run for MC analysis", false);
// PROCESS_SWITCH(CheckSmearing, processCheckMCanalysisDQ, "Run for MC analysis", false);
PROCESS_SWITCH(CheckSmearing, processCheckCocktail, "Run for cocktail analysis", false);
PROCESS_SWITCH(CheckSmearing, processDummyMCanalysisEM, "Dummy process function", false);
// PROCESS_SWITCH(CheckSmearing, processDummyMCanalysisDQ, "Dummy process function", false);
PROCESS_SWITCH(CheckSmearing, processDummyCocktail, "Dummy process function", true);
};
WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
19 changes: 12 additions & 7 deletions PWGEM/Dilepton/Utils/MomentumSmearer.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,22 +15,27 @@
#ifndef PWGEM_DILEPTON_UTILS_MOMENTUMSMEARER_H_
#define PWGEM_DILEPTON_UTILS_MOMENTUMSMEARER_H_

#include <vector>
#include "CCDB/BasicCCDBManager.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/AnalysisTask.h"
#include "Framework/Logger.h"
#include "Framework/runDataProcessing.h"

#include <TFile.h>
#include <TGrid.h>
#include <TH1.h>
#include <TH2.h>
#include <TH3.h>
#include <THnSparse.h>
#include <TString.h>
#include <TGrid.h>
#include <TFile.h>
#include <TKey.h>
#include <TString.h>

#include "CCDB/BasicCCDBManager.h"
#include "Framework/Logger.h"
#include <vector>

using namespace o2::framework;
using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;
using namespace o2::soa;

class MomentumSmearer
{
Expand Down
Loading