Skip to content

[Common] improvements to the muon QA task #18024

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

Draft
aferrero2707 wants to merge 1 commit into
base: master
Choose a base branch
from
+64 −4
Draft

[Common] improvements to the muon QA task #18024

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
68 changes: 64 additions & 4 deletions Common/Tasks/qaMuon.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@
/// \author Paul Veen <paul.veen@cern.ch>
/// \author Chi Zhang <chi.zhang@cern.ch>

#include "Common/CCDB/EventSelectionParams.h"
#include "Common/CCDB/RCTSelectionFlags.h"
#include "Common/DataModel/EventSelection.h"

#include <CCDB/BasicCCDBManager.h>
Expand Down Expand Up @@ -83,6 +85,7 @@ using namespace o2::framework;
using namespace o2::framework::expressions;

using MyEvents = soa::Join<aod::Collisions, aod::EvSels>;
using MyBCs = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels>;
using MyMuonsWithCov = soa::Join<aod::FwdTracks, aod::FwdTracksCov>;
using MyMFTs = aod::MFTTracks;

Expand Down Expand Up @@ -185,6 +188,16 @@ struct muonQa {
Configurable<bool> fEnableSingleMuonDiMuonCorrelations{"cfgEnableMuonDiMuonCorrelations", false, "Enable muon-dimuon QA checks"};
} configQAs;

//// Variables for selecting events
struct : ConfigurableGroup {
Configurable<bool> fEvSelRequireGoodRCT{"cfgEvSelRequireGoodRCT", false, "Require good detector flags in Run Condition Table"};
Configurable<bool> fEvSelRequireFrameBordersExclusion{"cfgEvSelRequireFrameBordersExclusion", false, "Exclude events in the TF and ITS ROF borders"};
Configurable<bool> fEvSelRequireNoCollInTimeRange{"cfgEvSelRequireNoCollInTimeRange", false, "Exclude events with multiple collisions in time range (standard)"};
Configurable<bool> fEvSelRequireNoSameBunch{"cfgEvSelRequireNoSameBunch", false, "Exclude events with multiple collisions in the same bunch crossing"};
Configurable<float> fEvSelVtxZMin{"cfgEvSelVtxZMin", -10.0f, "Minimum accepted value for vertex z position"};
Configurable<float> fEvSelVtxZMax{"cfgEvSelVtxZMax", 10.0f, "Maximum accepted value for vertex z position"};
} configEvSel;

//// Variables for selecting muon tracks
struct : ConfigurableGroup {
Configurable<float> fPMchLow{"cfgPMchLow", 0.0f, ""};
Expand Down Expand Up @@ -214,6 +227,9 @@ struct muonQa {
Configurable<float> fDimuonDCAMassLow{"cfgDimuonDCAMassLow", 2.8f, ""};
Configurable<float> fDimuonDCAMassHigh{"cfgDimuonDCAMassHigh", 3.4f, ""};

// Magnetic field position bias
Configurable<float> fFieldOriginBiasZ{"cfgFieldOriginBiasZ", 0.0f, "Bias applied to the magnetic field z position"};

//// Variables for alignment corrections
Configurable<bool> fEnableMFTAlignmentCorrections{"cfgEnableMFTAlignmentCorrections", false, ""};

Expand Down Expand Up @@ -273,6 +289,8 @@ struct muonQa {
o2::field::MagneticField* fieldB = nullptr;
double Bz; // Bz for MFT

o2::aod::rctsel::RCTFlagsChecker rctChecker{"CBT_muon_glo", false, false, true};

geo::TransformationCreator transformation;
std::map<int, math_utils::Transform3D> transformRef; // reference geometry w.r.t track data
std::map<int, math_utils::Transform3D> transformNew; // new geometry
Expand Down Expand Up @@ -2345,7 +2363,8 @@ struct muonQa {
fgValues = {};
}

void initCCDB(aod::BCsWithTimestamps const& bcs)
template <typename BC>
void initCCDB(BC const& bcs)
{
// Update CCDB informations
if (bcs.size() > 0 && fCurrentRun != bcs.begin().runNumber()) {
Expand Down Expand Up @@ -2407,6 +2426,9 @@ struct muonQa {
ccdb->setCaching(true);
ccdb->setLocalObjectValidityChecking();

// configure magnetic field position bias
o2::conf::ConfigurableParam::setValue("FieldOriginBias.z", std::to_string(fFieldOriginBiasZ.value));

// Configuration for track fitter
const auto& trackerParam = TrackerParam::Instance();
trackFitter.setBendingVertexDispersion(trackerParam.bendingVertexDispersion);
Expand Down Expand Up @@ -2685,6 +2707,38 @@ struct muonQa {
{
for (auto const& collision : collisions) {

const auto& bc = bcs.rawIteratorAt(collision.bcId());

if (configEvSel.fEvSelRequireFrameBordersExclusion) {
// remove TF/ROF borders and ambiguous collisions
if (!bc.selection_bit(o2::aod::evsel::kNoTimeFrameBorder) ||
!bc.selection_bit(o2::aod::evsel::kNoITSROFrameBorder)) {
continue;
}
}

// remove events with multiple collisions in one bunch crossing
if (configEvSel.fEvSelRequireNoCollInTimeRange &&
!bc.selection_bit(o2::aod::evsel::kNoCollInTimeRangeStandard)) {
continue;
}

// remove events with multiple collisions in one bunch crossing
if (configEvSel.fEvSelRequireNoSameBunch &&
!bc.selection_bit(o2::aod::evsel::kNoSameBunchPileup)) {
continue;
}

// remove events with bad RCT flags
if (configEvSel.fEvSelRequireGoodRCT && !rctChecker(collision)) {
continue;
}

if (collision.posZ() < configEvSel.fEvSelVtxZMin ||
collision.posZ() > configEvSel.fEvSelVtxZMax) {
continue;
}

uint64_t collisionIndex = collision.globalIndex();
auto muonsThisCollision = muons.sliceBy(fwdtracksPerCollision, collisionIndex);
auto mftsThisCollision = mfts.sliceBy(mftPerCollision, collisionIndex);
Expand All @@ -2695,7 +2749,7 @@ struct muonQa {

auto& fgValuesColl = collisionSel[collisionIndex];
FillCollision(collision, fgValuesColl);
fgValuesColl.bc = bcs.rawIteratorAt(collision.bcId()).globalBC();
fgValuesColl.bc = bc.globalBC();
fgValuesColl.multMFT = mftsThisCollision.size();
}
}
Expand All @@ -2708,7 +2762,7 @@ struct muonQa {

//// Get collision information if associated
VarColl fgValuesColl;
if (muon.has_collision()) {
if (muon.has_collision() && collisions.contains(muon.collisionId())) {
fgValuesColl = collisions.at(muon.collisionId());
} else {
continue;
Expand Down Expand Up @@ -2897,8 +2951,14 @@ struct muonQa {

for (const auto& [muon1, muon2] : muonPairs) {
auto collisionIndex1 = muon1.first;
if (!collisions.contains(collisionIndex1)) {
continue;
}
auto const& collision1 = collisions.at(collisionIndex1);
auto collisionIndex2 = muon2.first;
if (!collisions.contains(collisionIndex2)) {
continue;
}
auto const& collision2 = collisions.at(collisionIndex2);

auto mchIndex1 = muon1.second;
Expand Down Expand Up @@ -3840,7 +3900,7 @@ struct muonQa {
}
}

void processMuonQa(MyEvents const& collisions, aod::BCsWithTimestamps const& bcs, MyMuonsWithCov const& muontracks, MyMFTs const& mfttracks, aod::FwdTrkCls const& muonclusters)
void processMuonQa(MyEvents const& collisions, MyBCs const& bcs, MyMuonsWithCov const& muontracks, MyMFTs const& mfttracks, aod::FwdTrkCls const& muonclusters)
{
std::map<uint64_t, VarColl> collisionSel;
std::map<uint64_t, std::vector<uint64_t>> matchingCandidates;
Expand Down
Loading