Skip to content

Please consider the following formatting changes to #17535 #28

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

Closed
alibuild wants to merge 2 commits into from
Closed

Please consider the following formatting changes to #17535 #28

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1789b38
[PWGDQ] fix linter errors
aferrero2707 Aug 21, 2026
99a671a
Please consider the following formatting changes
alibuild Aug 21, 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
142 changes: 76 additions & 66 deletions PWGDQ/Tasks/mftMchMatcher.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@
//
/// \file mftMchMatcher.cxx
/// \brief MFT-MCH matching tool for data preparation
/// \author Andrea Ferrero <andrea.ferrero@cern.ch>
/// \author Marice Coquet <maurice.louis.coquet@cern.ch>
/// \author Nivendra Ramasubramanian <niv.ram@cern.ch>
/// \author Yoshini Bailung <yoshini.bailung@cern.ch>

#include "PWGDQ/Core/VarManager.h"

Expand Down Expand Up @@ -218,43 +222,47 @@ DECLARE_SOA_TABLE(FwdMatchMLCandidates, "AOD", "FWDMLCAND",
} // namespace o2::aod

struct mftMchMatcher {
static constexpr int GlobalTrackTypeMax = 2;
static constexpr int ThetaAbsBoundaryDeg = 3;
static constexpr double SlopeResolutionZ = 535.;

Produces<o2::aod::FwdMatchMLCandidates> fwdMatchMLCandidates;
//// Variables for selecting muon tracks
Configurable<float> fPMchLow{"cfgPMchLow", 0.0f, ""};
Configurable<float> fPtMchLow{"cfgPtMchLow", 0.7f, ""};
Configurable<float> fEtaMchLow{"cfgEtaMchLow", -4.0f, ""};
Configurable<float> fEtaMchUp{"cfgEtaMchUp", -2.5f, ""};
Configurable<float> fRabsLow{"cfgRabsLow", 17.6f, ""};
Configurable<float> fRabsUp{"cfgRabsUp", 89.5f, ""};
Configurable<float> fSigmaPdcaUp{"cfgPdcaUp", 6.f, ""};
Configurable<float> fTrackChi2MchUp{"cfgTrackChi2MchUp", 5.f, ""};
Configurable<float> fMatchingChi2MchMidUp{"cfgMatchingChi2MchMidUp", 999.f, ""};
Configurable<float> cfgPMchLow{"cfgPMchLow", 0.0f, ""};
Configurable<float> cfgPtMchLow{"cfgPtMchLow", 0.7f, ""};
// Configurable<float> cfgEtaMchLow{"cfgEtaMchLow", -4.0f, ""};
// Configurable<float> cfgEtaMchUp{"cfgEtaMchUp", -2.5f, ""};
Configurable<float> cfgRabsLow{"cfgRabsLow", 17.6f, ""};
Configurable<float> cfgRabsUp{"cfgRabsUp", 89.5f, ""};
Configurable<float> cfgPdcaUp{"cfgPdcaUp", 6.f, ""};
Configurable<float> cfgTrackChi2MchUp{"cfgTrackChi2MchUp", 5.f, ""};
// Configurable<float> cfgMatchingChi2MchMidUp{"cfgMatchingChi2MchMidUp", 999.f, ""};

//// Variables for selecting mft tracks
Configurable<float> fEtaMFTLow{"cfgEtaMFTlow", -3.6f, ""};
Configurable<float> fEtaMFTUp{"cfgEtaMFTup", -2.5f, ""};
Configurable<float> fTrackChi2MFTUp{"cfgTrackChi2MFTUp", 10.f, ""};
Configurable<float> fPtMFTLow{"cfgPtMFTLow", 0.1f, ""};
Configurable<float> cfgEtaMFTlow{"cfgEtaMFTlow", -3.6f, ""};
Configurable<float> cfgEtaMFTup{"cfgEtaMFTup", -2.5f, ""};
// Configurable<float> cfgTrackChi2MFTUp{"cfgTrackChi2MFTUp", 10.f, ""};
// Configurable<float> cfgPtMFTLow{"cfgPtMFTLow", 0.1f, ""};

//// Variables for matching configuration
Configurable<int> fMaxCandidates{"cfgMaxCandidates", 0, ""};

Configurable<bool> fKeepBestMatch{"cfgKeepBestMatch", false, "Keep only the best match global muons in the skimming"};
Configurable<float> fzMatching{"cfgzMatching", -77.5f, "Plane for MFT-MCH matching"};

Configurable<float> fSamplingFraction{"cfgSamplingFraction", 1.f, "Fraction of randomly selected events to be processed"};
Configurable<float> fSamplingFractionTrueLeadingMatches{"cfgSamplingFractionTrueLeadingMatches", 1.f, "Fraction of randomly selected leading true matches to be processed"};
Configurable<float> fSamplingFractionWrongLeadingMatches{"cfgSamplingFractionWrongLeadingMatches", 1.f, "Fraction of randomly selected leading wrong matches to be processed"};
Configurable<float> fSamplingFractionDecayLeadingMatches{"cfgSamplingFractionDecayLeadingMatches", 1.f, "Fraction of randomly selected leading decay matches to be processed"};
Configurable<float> fSamplingFractionFakeLeadingMatches{"cfgSamplingFractionFakeLeadingMatches", 1.f, "Fraction of randomly selected leading fake matches to be processed"};
Configurable<float> fSamplingFractionTrueNonLeadingMatches{"cfgSamplingFractionTrueNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading true matches to be processed"};
Configurable<float> fSamplingFractionWrongNonLeadingMatches{"cfgSamplingFractionWrongNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading wrong matches to be processed"};
Configurable<float> fSamplingFractionDecayNonLeadingMatches{"cfgSamplingFractionDecayNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading decay matches to be processed"};
Configurable<float> fSamplingFractionFakeNonLeadingMatches{"cfgSamplingFractionFakeNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading fake matches to be processed"};
Configurable<int> fSamplingBcOddness{"cfgSamplingBcOddness", -1, "Select only events with even (0) or odd (1) global BCs"};
// Configurable<int> cfgMaxCandidates{"cfgMaxCandidates", 0, ""};

Configurable<bool> cfgKeepBestMatch{"cfgKeepBestMatch", false, "Keep only the best match global muons in the skimming"};
Configurable<float> cfgzMatching{"cfgzMatching", -77.5f, "Plane for MFT-MCH matching"};

Configurable<float> cfgSamplingFraction{"cfgSamplingFraction", 1.f, "Fraction of randomly selected events to be processed"};
Configurable<float> cfgSamplingFractionTrueLeadingMatches{"cfgSamplingFractionTrueLeadingMatches", 1.f, "Fraction of randomly selected leading true matches to be processed"};
Configurable<float> cfgSamplingFractionWrongLeadingMatches{"cfgSamplingFractionWrongLeadingMatches", 1.f, "Fraction of randomly selected leading wrong matches to be processed"};
Configurable<float> cfgSamplingFractionDecayLeadingMatches{"cfgSamplingFractionDecayLeadingMatches", 1.f, "Fraction of randomly selected leading decay matches to be processed"};
Configurable<float> cfgSamplingFractionFakeLeadingMatches{"cfgSamplingFractionFakeLeadingMatches", 1.f, "Fraction of randomly selected leading fake matches to be processed"};
Configurable<float> cfgSamplingFractionTrueNonLeadingMatches{"cfgSamplingFractionTrueNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading true matches to be processed"};
Configurable<float> cfgSamplingFractionWrongNonLeadingMatches{"cfgSamplingFractionWrongNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading wrong matches to be processed"};
Configurable<float> cfgSamplingFractionDecayNonLeadingMatches{"cfgSamplingFractionDecayNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading decay matches to be processed"};
Configurable<float> cfgSamplingFractionFakeNonLeadingMatches{"cfgSamplingFractionFakeNonLeadingMatches", 1.f, "Fraction of randomly selected non-leading fake matches to be processed"};
Configurable<int> cfgSamplingBcOddness{"cfgSamplingBcOddness", -1, "Select only events with even (0) or odd (1) global BCs"};

//// Variables for ccdb
Configurable<std::string> ccdburl{"ccdb-url", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
Configurable<std::string> ccdbUrl{"ccdbUrl", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
Configurable<std::string> grpPath{"grpPath", "GLO/GRP/GRP", "Path of the grp file"};
Configurable<std::string> grpmagPath{"grpmagPath", "GLO/Config/GRPMagField", "CCDB path of the GRPMagField object"};
Configurable<std::string> geoPath{"geoPath", "GLO/Config/GeometryAligned", "Path of the geometry file"};
Expand Down Expand Up @@ -314,7 +322,9 @@ struct mftMchMatcher {
static const double relPRes = 0.0004;
static const double slopeRes = 0.0005;

double thetaAbs = TMath::ATan(mchTrack.rAtAbsorberEnd() / 505.) * TMath::RadToDeg();
constexpr double AbsorberEndZ = 505.;
constexpr double RadToDeg = 180. / o2::constants::math::PI;
double thetaAbs = std::atan(mchTrack.rAtAbsorberEnd() / AbsorberEndZ) * RadToDeg;

// propagate muon track to vertex
auto trackConv = VarManager::FwdToTrackPar(mchTrack, mchTrack);
Expand All @@ -329,17 +339,17 @@ struct mftMchMatcher {
double p = mchTrackAtVertex.p();

double pDCA = mchTrack.pDca();
double sigmaPDCA = (thetaAbs < 3) ? sigmaPDCA23 : sigmaPDCA310;
double sigmaPDCA = (thetaAbs < ThetaAbsBoundaryDeg) ? sigmaPDCA23 : sigmaPDCA310;
double nrp = nSigmaPDCA * relPRes * p;
double pResEffect = sigmaPDCA / (1. - nrp / (1. + nrp));
double slopeResEffect = 535. * slopeRes * p;
double sigmaPDCAWithRes = TMath::Sqrt(pResEffect * pResEffect + slopeResEffect * slopeResEffect);
double slopeResEffect = SlopeResolutionZ * slopeRes * p;
double sigmaPDCAWithRes = std::sqrt(pResEffect * pResEffect + slopeResEffect * slopeResEffect);

return (pDCA <= nSigmaPDCA * sigmaPDCAWithRes);
}

template <class T, class C>
bool IsGoodMuon(const T& mchTrack, const C& collision,
bool isGoodMuon(const T& mchTrack, const C& collision,
double chi2Cut,
double pCut,
double pTCut,
Expand Down Expand Up @@ -383,7 +393,7 @@ struct mftMchMatcher {
}

template <class T>
bool IsGoodMFT(const T& mftTrack,
bool isGoodMFT(const T& mftTrack,
double chi2Cut,
double pTCut,
std::array<double, 2> etaCut)
Expand Down Expand Up @@ -427,18 +437,18 @@ struct mftMchMatcher {
void init(o2::framework::InitContext&)
{
// Load geometry
ccdbManager->setURL(ccdburl);
ccdbManager->setURL(ccdbUrl);
ccdbManager->setCaching(true);
ccdbManager->setLocalObjectValidityChecking();
fCCDBApi.init(ccdburl);
fCCDBApi.init(ccdbUrl);
mRunNumber = 0;

if (!o2::base::GeometryManager::isGeometryLoaded()) {
LOGF(info, "Load geometry from CCDB");
ccdbManager->get<TGeoManager>(geoPath);
}

if (fSamplingFraction < 1.0) {
if (cfgSamplingFraction < 1.0) {
std::random_device rd;
mGenerator = std::mt19937(rd());
}
Expand Down Expand Up @@ -476,7 +486,7 @@ struct mftMchMatcher {
fBestMatch.clear();
std::unordered_map<int, std::pair<float, int>> mCandidates;
for (const auto& muon : muons) {
if (static_cast<int>(muon.trackType()) < 2) {
if (static_cast<int>(muon.trackType()) < GlobalTrackTypeMax) {
auto muonID = muon.matchMCHTrackId();
auto chi2 = muon.chi2MatchMCHMFT();
if (mCandidates.find(muonID) == mCandidates.end()) {
Expand All @@ -501,7 +511,7 @@ struct mftMchMatcher {
// outer loop on muon tracks
for (const auto& muonTrack : muonTracks) {
// only consider MCH standalone or MCH-MID matches
if (static_cast<int>(muonTrack.trackType()) <= 2) {
if (static_cast<int>(muonTrack.trackType()) <= GlobalTrackTypeMax) {
continue;
}

Expand Down Expand Up @@ -585,7 +595,7 @@ struct mftMchMatcher {
{
MuonMatchType result{kMatchTypeUndefined};

if (static_cast<int>(muonTrack.trackType()) > 2) {
if (static_cast<int>(muonTrack.trackType()) > GlobalTrackTypeMax) {
return result;
}

Expand Down Expand Up @@ -656,9 +666,9 @@ struct mftMchMatcher {

registry.get<TH1>(HIST("acceptedEvents"))->Fill(0);
// reject a randomly selected fraction of events
if (fSamplingFraction < 1.0) {
if (cfgSamplingFraction < 1.0) {
double rnd = mDistribution(mGenerator);
if (rnd > fSamplingFraction) {
if (rnd > cfgSamplingFraction) {
return;
}
}
Expand Down Expand Up @@ -688,41 +698,41 @@ struct mftMchMatcher {

bool isBestMatch = fBestMatch.find(muon.globalIndex()) != fBestMatch.end();

if (fKeepBestMatch && !isBestMatch) {
if (cfgKeepBestMatch && !isBestMatch) {
continue;
}

const auto& collision = collisions.rawIteratorAt(muon.collisionId());
auto bc_coll = collision.template bc_as<TBCS>();
auto bcColl = collision.template bc_as<TBCS>();

auto muontrack = muon.template matchMCHTrack_as<TMUONS>();
auto mfttrack = muon.template matchMFTTrack_as<TMFTS>();
auto const& mfttrackcov = mftCovs.rawIteratorAt(mftCovIndexes[mfttrack.globalIndex()]);

auto muonTime = muontrack.trackTime() + bc_coll.globalBC() * o2::constants::lhc::LHCBunchSpacingNS;
auto mftTime = mfttrack.trackTime() + bc_coll.globalBC() * o2::constants::lhc::LHCBunchSpacingNS;
auto muonTime = muontrack.trackTime() + bcColl.globalBC() * o2::constants::lhc::LHCBunchSpacingNS;
auto mftTime = mfttrack.trackTime() + bcColl.globalBC() * o2::constants::lhc::LHCBunchSpacingNS;

o2::track::TrackParCovFwd mftprop = VarManager::FwdToTrackPar(mfttrack, mfttrackcov);
o2::dataformats::GlobalFwdTrack muonprop = trackToGlobalFwd(muontrack, muontrack);
if (fzMatching.value < 0.) {
mftprop = VarManager::PropagateFwd(mfttrack, mfttrackcov, fzMatching.value);
if (cfgzMatching.value < 0.) {
mftprop = VarManager::PropagateFwd(mfttrack, mfttrackcov, cfgzMatching.value);
muonprop = VarManager::PropagateMuon(muontrack, collision, VarManager::kToMatching);
}
auto muonpropCov = muonprop.getCovariances();
auto mftpropCov = mftprop.getCovariances();

if (!IsGoodMuon(muontrack, collision, fTrackChi2MchUp, fPMchLow, fPtMchLow, {fEtaMFTLow, fEtaMFTUp}, {fRabsLow, fRabsUp}, fSigmaPdcaUp)) {
if (!isGoodMuon(muontrack, collision, cfgTrackChi2MchUp, cfgPMchLow, cfgPtMchLow, {cfgEtaMFTlow, cfgEtaMFTup}, {cfgRabsLow, cfgRabsUp}, cfgPdcaUp)) {
continue;
}

// at this level we consider all the matching candidates, regardless of the MFT tracks quality
// MFT track quality cuts should be applied only after having selected the best candidate
// if (!IsGoodMFT(mfttrack, fTrackChi2MFTUp, fPtMFTLow, {fEtaMFTLow, fEtaMFTUp})){
// if (!isGoodMFT(mfttrack, cfgTrackChi2MFTUp, cfgPtMFTLow, {cfgEtaMFTlow, cfgEtaMFTup})){
// continue;
//}

bool IsAmbig = (muon.compatibleCollIds().size() != 1);
int MFTMult = collision.mftNtracks();
bool isAmbig = (muon.compatibleCollIds().size() != 1);
int mftMult = collision.mftNtracks();
int matchAttempts = 0;
auto matchAttemptsIt = matchAttemptsMap.find(muontrack.globalIndex());
if (matchAttemptsIt == matchAttemptsMap.end()) {
Expand Down Expand Up @@ -750,35 +760,35 @@ struct mftMchMatcher {
registry.get<TH1>(HIST("matchType"))->Fill(static_cast<int>(matchType));

// skipp odd/even BCs if requested
if (fSamplingBcOddness.value >= 0 && (static_cast<int>((bc_coll.globalBC() % 2)) != fSamplingBcOddness.value)) {
if (cfgSamplingBcOddness.value >= 0 && (static_cast<int>((bcColl.globalBC() % 2)) != cfgSamplingBcOddness.value)) {
continue;
}

float matchTypeSamplingFraction = 1.0;
switch (matchType) {
case kMatchTypeTrueLeading:
matchTypeSamplingFraction = fSamplingFractionTrueLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionTrueLeadingMatches;
break;
case kMatchTypeTrueNonLeading:
matchTypeSamplingFraction = fSamplingFractionTrueNonLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionTrueNonLeadingMatches;
break;
case kMatchTypeWrongLeading:
matchTypeSamplingFraction = fSamplingFractionWrongLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionWrongLeadingMatches;
break;
case kMatchTypeWrongNonLeading:
matchTypeSamplingFraction = fSamplingFractionWrongNonLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionWrongNonLeadingMatches;
break;
case kMatchTypeDecayLeading:
matchTypeSamplingFraction = fSamplingFractionDecayLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionDecayLeadingMatches;
break;
case kMatchTypeDecayNonLeading:
matchTypeSamplingFraction = fSamplingFractionDecayNonLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionDecayNonLeadingMatches;
break;
case kMatchTypeFakeLeading:
matchTypeSamplingFraction = fSamplingFractionFakeLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionFakeLeadingMatches;
break;
case kMatchTypeFakeNonLeading:
matchTypeSamplingFraction = fSamplingFractionFakeNonLeadingMatches;
matchTypeSamplingFraction = cfgSamplingFractionFakeNonLeadingMatches;
break;
default:
break;
Expand Down Expand Up @@ -848,8 +858,8 @@ struct mftMchMatcher {
muon.chi2MatchMCHMFT(),
muon.fwdDcaX(),
muon.fwdDcaY(),
IsAmbig,
MFTMult,
isAmbig,
mftMult,
matchAttempts,
mcMaskMuon,
mcMaskMft,
Expand All @@ -869,7 +879,7 @@ struct mftMchMatcher {
if (bcs.size() > 0) {
auto bc = bcs.begin();
initCCDB(bc);
VarManager::SetMatchingPlane(fzMatching.value);
VarManager::SetMatchingPlane(cfgzMatching.value);
}

fillTable<true>(collisions, bcs, muonTracks, mftTracks, mftCovs);
Expand All @@ -886,7 +896,7 @@ struct mftMchMatcher {
if (bcs.size() > 0) {
auto bc = bcs.begin();
initCCDB(bc);
VarManager::SetMatchingPlane(fzMatching.value);
VarManager::SetMatchingPlane(cfgzMatching.value);
}

fillTable<false>(collisions, bcs, muonTracks, mftTracks, mftCovs);
Expand Down
Loading