Skip to content

[DPG] adding_mc_table_hmpid #16875

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
njacazio merged 2 commits into from
Jun 30, 2026
+82 −38
Merged

[DPG] adding_mc_table_hmpid #16875

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
140c25e
adding_mc_table_hmpid
rliotino99 Jun 30, 2026
5906985
fixing_format
rliotino99 Jun 30, 2026
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
93 changes: 60 additions & 33 deletions DPG/Tasks/AOTTrack/PID/HMPID/hmpidTableProducer.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,13 +24,8 @@
#include <DetectorsBase/MatLayerCylSet.h>
#include <DetectorsBase/Propagator.h>
#include <Framework/AnalysisDataModel.h>
#include <Framework/AnalysisHelpers.h>
#include <Framework/AnalysisTask.h>
#include <Framework/Configurable.h>
#include <Framework/HistogramRegistry.h>
#include <Framework/HistogramSpec.h>
#include <Framework/InitContext.h>
#include <Framework/OutputObjHeader.h>
#include <Framework/runDataProcessing.h>

#include <TGeoManager.h>
Expand All @@ -51,6 +46,7 @@ using namespace o2::constants::physics;
struct HmpidTableProducer {

Produces<aod::HmpidAnalysis> hmpidAnalysis;
Produces<aod::HmpidAnalysisMC> hmpidAnalysisMC;

HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};

Expand Down Expand Up @@ -80,6 +76,16 @@ struct HmpidTableProducer {
aod::pidTOFFullPi, aod::pidTOFFullKa,
aod::pidTOFFullPr, aod::pidTOFFullDe>;

using TrackCandidatesMC = soa::Join<aod::Tracks, aod::TracksExtra,
aod::TracksDCA, aod::TrackSelection,
aod::pidTPCFullPi, aod::pidTPCFullKa,
aod::pidTPCFullPr, aod::pidTPCFullDe,
aod::pidTOFFullPi, aod::pidTOFFullKa,
aod::pidTOFFullPr, aod::pidTOFFullDe,
aod::McTrackLabels>;

std::unordered_set<uint32_t> mCollisionsWithHmpid;

void init(o2::framework::InitContext&)
{
ccdb->setURL(ccdbConfig.ccdbUrl);
Expand Down Expand Up @@ -294,42 +300,37 @@ struct HmpidTableProducer {
}
PROCESS_SWITCH(HmpidTableProducer, processEvent, "Process event level", true);

void processHmpid(
template <bool isMC, typename TTrackTable>
void runHmpidAnalysis(
aod::HMPIDs const& hmpids,
TrackCandidates const&,
TTrackTable const&,
CollisionCandidates const&,
aod::BCsWithTimestamps const&)
{
static std::unordered_set<uint32_t> collisionsWithHmpid;

for (auto const& t : hmpids) {

const auto& globalTrack = t.track_as<TrackCandidates>();
const auto& globalTrack = t.template track_as<TTrackTable>();

if (!globalTrack.has_collision())
continue;

const auto& col = globalTrack.collision_as<CollisionCandidates>();
initCCDB(col.bc_as<aod::BCsWithTimestamps>());
const auto& col = globalTrack.template collision_as<CollisionCandidates>();
initCCDB(col.template bc_as<aod::BCsWithTimestamps>());
uint32_t collId = col.globalIndex();

// Track quality selection
if ((requireITS && !globalTrack.hasITS()) ||
(requireTPC && !globalTrack.hasTPC()) ||
(requireTOF && !globalTrack.hasTOF())) {
(requireTOF && !globalTrack.hasTOF()))
continue;
}

if (collisionsWithHmpid.insert(collId).second) {
if (mCollisionsWithHmpid.insert(collId).second)
histos.fill(HIST("eventsHmpid"), 0.5);
}

// clusSize diagnostics
histos.fill(HIST("hClusSize"), t.hmpidClusSize());
bool isCorrupt = (t.hmpidClusSize() <= 0);
if (isCorrupt) {
if (isCorrupt)
histos.fill(HIST("hClusSizeCorrupt"), t.hmpidClusSize());
}

// --- M2: clusSize encoding ---
int chamberM2 = t.hmpidClusSize() / 1000000;
Expand All @@ -354,7 +355,7 @@ struct HmpidTableProducer {
int16_t charge = globalTrack.sign();

auto prop = o2::base::Propagator::Instance();
double bz = static_cast<double>(prop->getNominalBz()); // positive sign
double bz = static_cast<double>(prop->getNominalBz());

int chamberM1 = getHmpidChamber(x, p, bz, charge);

Expand All @@ -374,35 +375,30 @@ struct HmpidTableProducer {

// Legend:
// bin 0 = clusSize > 0, chamber found (M2 ok)
// bin 1 = clusSize > 0, chamber not found (non dovrebbe mai accadere)
// bin 1 = clusSize > 0, chamber not found
// bin 2 = clusSize <= 0, M1 recovery (corrupt, M1 ok)
// bin 3 = clusSize <= 0, M1 fails (corrupt, skipped)

if (!isCorrupt && chamberM3 >= 0) {
if (!isCorrupt && chamberM3 >= 0)
histos.fill(HIST("hChamberAssignment"), 0.);
} else if (!isCorrupt && chamberM3 < 0) {
else if (!isCorrupt && chamberM3 < 0)
histos.fill(HIST("hChamberAssignment"), 1.);
} else if (isCorrupt && chamberM3 >= 0) {
else if (isCorrupt && chamberM3 >= 0)
histos.fill(HIST("hChamberAssignment"), 2.);
} else {
else
histos.fill(HIST("hChamberAssignment"), 3.);
}

histos.fill(HIST("hChamberM3"), chamberM3);
histos.fill(HIST("hChamberM3vsM2"), chamberM2, chamberM3);

// Skip track if chamber undetermined
if (chamberM3 < 0) {
if (chamberM3 < 0)
continue;
}

// Fill photon charges
float hmpidPhotsCharge2[o2::aod::kDimPhotonsCharge];
for (int i = 0; i < o2::aod::kDimPhotonsCharge; i++) {
for (int i = 0; i < o2::aod::kDimPhotonsCharge; i++)
hmpidPhotsCharge2[i] = t.hmpidPhotsCharge()[i];
}

// Fill output table
// fill hmpid table
hmpidAnalysis(
t.hmpidSignal(), t.hmpidMom(),
globalTrack.p(), t.hmpidXTrack(), t.hmpidYTrack(),
Expand All @@ -422,9 +418,40 @@ struct HmpidTableProducer {
globalTrack.tpcNSigmaDe(), globalTrack.tofNSigmaDe(),
col.centFV0A());

// fill hmpid table for mc tracks if running on MC
if constexpr (isMC) {
if (globalTrack.has_mcParticle()) {
const auto& mc = globalTrack.mcParticle();
hmpidAnalysisMC(mc.pdgCode(), mc.vx(), mc.vy(), mc.vz(), mc.isPhysicalPrimary(), mc.getProcess());
} else {
hmpidAnalysisMC(-1, 0.f, 0.f, 0.f, false, -100);
}
}

} // end HMPID loop
}

// process real data
void processHmpid(aod::HMPIDs const& hmpids,
TrackCandidates const& tracks,
CollisionCandidates const& cols,
aod::BCsWithTimestamps const& bcs)
{
// isMC False
runHmpidAnalysis<false>(hmpids, tracks, cols, bcs);
}
PROCESS_SWITCH(HmpidTableProducer, processHmpid, "Process HMPID entries", true);

// process MC
void processHmpidMC(aod::HMPIDs const& hmpids,
TrackCandidatesMC const& tracks,
CollisionCandidates const& cols,
aod::BCsWithTimestamps const& bcs,
aod::McParticles const&)
{
runHmpidAnalysis<true>(hmpids, tracks, cols, bcs);
}
PROCESS_SWITCH(HmpidTableProducer, processHmpidMC, "Process HMPID MC entries", true);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfg)
Expand Down
27 changes: 22 additions & 5 deletions DPG/Tasks/AOTTrack/PID/HMPID/tableHMPID.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,22 +34,18 @@ DECLARE_SOA_COLUMN(ChargeMip, chargeMip, float);
DECLARE_SOA_COLUMN(ClusterSize, clusterSize, float);
DECLARE_SOA_COLUMN(Chamber, chamber, float);
DECLARE_SOA_COLUMN(PhotonsCharge, photonsCharge, float[kDimPhotonsCharge]);

DECLARE_SOA_COLUMN(EtaTrack, etaTrack, float);
DECLARE_SOA_COLUMN(PhiTrack, phiTrack, float);
DECLARE_SOA_COLUMN(Px, px, float);
DECLARE_SOA_COLUMN(Py, py, float);
DECLARE_SOA_COLUMN(Pz, pz, float);

DECLARE_SOA_COLUMN(ItsNCluster, itsNCluster, float);
DECLARE_SOA_COLUMN(TpcNCluster, tpcNCluster, float);
DECLARE_SOA_COLUMN(TpcNClsCrossedRows, tpcNClsCrossedRows, float);
DECLARE_SOA_COLUMN(TpcChi2, tpcChi2, float);
DECLARE_SOA_COLUMN(ItsChi2, itsChi2, float);

DECLARE_SOA_COLUMN(DcaXY, dcaXY, float);
DECLARE_SOA_COLUMN(DcaZ, dcaZ, float);

DECLARE_SOA_COLUMN(TpcNSigmaPi, tpcNSigmaPi, float);
DECLARE_SOA_COLUMN(TofNSigmaPi, tofNSigmaPi, float);
DECLARE_SOA_COLUMN(TpcNSigmaKa, tpcNSigmaKa, float);
Expand All @@ -59,7 +55,6 @@ DECLARE_SOA_COLUMN(TofNSigmaPr, tofNSigmaPr, float);
DECLARE_SOA_COLUMN(TpcNSigmaDe, tpcNSigmaDe, float);
DECLARE_SOA_COLUMN(TofNSigmaDe, tofNSigmaDe, float);
DECLARE_SOA_COLUMN(Centrality, centrality, float);

} // namespace variables_table

DECLARE_SOA_TABLE(HmpidAnalysis, "AOD", "HMPIDANALYSIS",
Expand Down Expand Up @@ -97,6 +92,28 @@ DECLARE_SOA_TABLE(HmpidAnalysis, "AOD", "HMPIDANALYSIS",
variables_table::TofNSigmaDe,
variables_table::Centrality);

// -----------------------------------------------------------------------
// MC truth table
// -----------------------------------------------------------------------
namespace hmpid_mc
{
DECLARE_SOA_COLUMN(PdgCode, pdgCode, int);
DECLARE_SOA_COLUMN(McVx, mcVx, float);
DECLARE_SOA_COLUMN(McVy, mcVy, float);
DECLARE_SOA_COLUMN(McVz, mcVz, float);
DECLARE_SOA_COLUMN(IsPhysPrimary, isPhysPrimary, bool);
DECLARE_SOA_COLUMN(ProcessCode, processCode, int);

} // namespace hmpid_mc

DECLARE_SOA_TABLE(HmpidAnalysisMC, "AOD", "HMPIDANALYSISMC",
hmpid_mc::PdgCode,
hmpid_mc::McVx,
hmpid_mc::McVy,
hmpid_mc::McVz,
hmpid_mc::IsPhysPrimary,
hmpid_mc::ProcessCode);

} // namespace o2::aod

#endif // DPG_TASKS_AOTTRACK_PID_HMPID_TABLEHMPID_H_
Loading