Skip to content

Commit 67c38d1

Browse files
committed
[PWGDQ] added processing of realigned MCH forward tracks
The new code allows to process tracks produced by the MCH realignment task, as well as the planned global muon rematching task (see PR O2Physics#16678).
1 parent 847085a commit 67c38d1

1 file changed

Lines changed: 29 additions & 0 deletions

File tree

PWGDQ/Tasks/qaMatching.cxx

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
#include "Common/DataModel/EventSelection.h"
2323
#include "Common/DataModel/Multiplicity.h"
2424
#include "Common/DataModel/TrackSelectionTables.h"
25+
#include "Common/DataModel/FwdTrackReAlignTables.h"
2526
#include "Tools/ML/MlResponse.h"
2627

2728
#include <CCDB/BasicCCDBManager.h>
@@ -178,6 +179,7 @@ DECLARE_SOA_INDEX_COLUMN_FULL_CUSTOM(Candidate, candidate, int32_t, o2::aod::QaM
178179

179180
using MyEvents = soa::Join<aod::Collisions, aod::EvSels, aod::FT0Mults, aod::MFTMults, aod::PVMults, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs>;
180181
using MyMuons = soa::Join<aod::FwdTracks, aod::FwdTracksCov>;
182+
using MyMuonsReAlign = soa::Join<aod::FwdTracksReAlign, aod::FwdTrksCovReAlign>;
181183
using MyMuonsMC = soa::Join<aod::FwdTracks, aod::FwdTracksCov, aod::McFwdTrackLabels, aod::FwdTracksDCA, aod::FwdTrkCompColls>;
182184
using MyMFTs = aod::MFTTracks;
183185
using MyMFTCovariances = aod::MFTTracksCov;
@@ -3233,6 +3235,33 @@ struct QaMatching {
32333235
}
32343236

32353237
PROCESS_SWITCH(QaMatching, processQA, "processQA", false);
3238+
3239+
void processQAReAlign(MyEvents const& collisions,
3240+
aod::BCsWithTimestamps const& bcs,
3241+
MyMuonsReAlign const& muonTracks,
3242+
MyMFTs const& mftTracks,
3243+
MyMFTCovariances const& mftCovs)
3244+
{
3245+
auto bc = bcs.begin();
3246+
initCcdb(bc);
3247+
3248+
for (const auto& muon : muonTracks) {
3249+
registry.get<TH1>(HIST("nTracksPerType"))->Fill(static_cast<int>(muon.trackType()));
3250+
}
3251+
3252+
mftTrackCovs.clear();
3253+
for (const auto& mftTrackCov : mftCovs) {
3254+
mftTrackCovs[mftTrackCov.matchMFTTrackId()] = mftTrackCov.globalIndex();
3255+
}
3256+
3257+
fillCollisions<false>(collisions, bcs, muonTracks, mftTracks, mftCovs, fCollisionInfos);
3258+
3259+
for (auto const& [collisionIndex, collisionInfo] : fCollisionInfos) {
3260+
processCollision<false>(collisionInfo, collisions, bcs, muonTracks, mftTracks, mftCovs);
3261+
}
3262+
}
3263+
3264+
PROCESS_SWITCH(QaMatching, processQAReAlign, "processQAReAlign", false);
32363265
};
32373266

32383267
WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)

0 commit comments

Comments
 (0)