|
22 | 22 | #include "Common/DataModel/EventSelection.h" |
23 | 23 | #include "Common/DataModel/Multiplicity.h" |
24 | 24 | #include "Common/DataModel/TrackSelectionTables.h" |
| 25 | +#include "Common/DataModel/FwdTrackReAlignTables.h" |
25 | 26 | #include "Tools/ML/MlResponse.h" |
26 | 27 |
|
27 | 28 | #include <CCDB/BasicCCDBManager.h> |
@@ -178,6 +179,7 @@ DECLARE_SOA_INDEX_COLUMN_FULL_CUSTOM(Candidate, candidate, int32_t, o2::aod::QaM |
178 | 179 |
|
179 | 180 | using MyEvents = soa::Join<aod::Collisions, aod::EvSels, aod::FT0Mults, aod::MFTMults, aod::PVMults, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs>; |
180 | 181 | using MyMuons = soa::Join<aod::FwdTracks, aod::FwdTracksCov>; |
| 182 | +using MyMuonsReAlign = soa::Join<aod::FwdTracksReAlign, aod::FwdTrksCovReAlign>; |
181 | 183 | using MyMuonsMC = soa::Join<aod::FwdTracks, aod::FwdTracksCov, aod::McFwdTrackLabels, aod::FwdTracksDCA, aod::FwdTrkCompColls>; |
182 | 184 | using MyMFTs = aod::MFTTracks; |
183 | 185 | using MyMFTCovariances = aod::MFTTracksCov; |
@@ -3233,6 +3235,33 @@ struct QaMatching { |
3233 | 3235 | } |
3234 | 3236 |
|
3235 | 3237 | 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); |
3236 | 3265 | }; |
3237 | 3266 |
|
3238 | 3267 | WorkflowSpec defineDataProcessing(ConfigContext const& cfgc) |
|
0 commit comments