Skip to content

Extend V0 tracking QC #1465

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
Nov 14, 2022
Merged

Extend V0 tracking QC #1465

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
3611ba9
Extend V0 tracking QC
njacazio Nov 2, 2022
d08be25
Fix
njacazio Nov 14, 2022
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
69 changes: 65 additions & 4 deletions DPG/Tasks/AOTTrack/qaK0sTrackingEfficiency.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,18 +25,24 @@
using namespace o2;
using namespace o2::framework;

using PIDTracks = soa::Join<aod::Tracks, aod::TracksExtra, aod::pidTPCFullPi>;
using PIDTracks = soa::Join<aod::Tracks, aod::TracksExtra, aod::pidTPCFullPi, aod::pidTPCFullPr, aod::pidTOFFullPi>;
using PIDTracksIU = soa::Join<aod::TracksIU, aod::TracksExtra, aod::pidTPCFullPi, aod::pidTPCFullPr, aod::pidTOFFullPi>;
using SelectedCollisions = soa::Join<aod::Collisions, aod::EvSels>;

struct qaK0sTrackingEfficiency {
ConfigurableAxis nSigmaBins{"nSigmaBins", {1000, -100.f, 100.f}, "Binning for the nSigma histograms"};

HistogramRegistry registry{"K0sTrackingEfficiency"};
#define fillHistogram(name, ...) registry.fill(HIST(name), __VA_ARGS__)

void init(InitContext const&)
{
const AxisSpec RAxis{500, 0.f, 50.f, "#it{R} (cm)"};
const AxisSpec pTAxis{200, 0.f, 10.f, "#it{p}_{T} (GeV/#it{c})"};
const AxisSpec mAxis{200, 0.4f, 0.6f, "#it{m} (GeV/#it{c}^{2})"};
const AxisSpec etaAxis{200, -1.f, 1.0f, "#it{#eta}"};
const AxisSpec phiAxis{200, -1.f, 1.0f, "#it{#phi}"};
const AxisSpec nsigmaAxis{200, -1.f, 1.0f, "#it{#phi}"};
const AxisSpec statusAxis{2, -0.5f, 1.5f, ""};
const AxisSpec hitMapAxis{128, -0.5f, 127.5f, ""};

Expand All @@ -53,6 +59,24 @@ struct qaK0sTrackingEfficiency {
registry.add("Test/h_posITSStatus", "h_posITSStatus", {HistType::kTH1I, {statusAxis}});
registry.add("Test/h_negITSHitMap", "h_negITSHitMap", {HistType::kTH1I, {hitMapAxis}});
registry.add("Test/h_posITSHitMap", "h_posITSHitMap", {HistType::kTH1I, {hitMapAxis}});
registry.add("VsPt/h_mass", "h_mass", HistType::kTH2F, {pTAxis, mAxis});
registry.add("VsPt/h_tofnsigma", "h_tofnsigma", HistType::kTH2F, {pTAxis, nSigmaBins});
registry.add("VsPt/h_tpcnsigma", "h_tpcnsigma", HistType::kTH2F, {pTAxis, nSigmaBins});
registry.add("VsEta/h_mass", "h_mass", HistType::kTH2F, {etaAxis, mAxis});
registry.add("VsPhi/h_mass", "h_mass", HistType::kTH2F, {phiAxis, mAxis});
registry.add("VsRadius/h_mass", "h_mass", HistType::kTH2F, {RAxis, mAxis});

if (!doprocessIU) {
return;
}
// IU info
registry.add("VsRadiusPosIU/h_mass", "h_mass", HistType::kTH2F, {RAxis, mAxis});
registry.add("VsRadiusPosIU/h_radius", "IU Radius of the negative track", HistType::kTH2F, {RAxis, RAxis});
registry.add("VsRadiusPosIU/h_phi", "Phi", HistType::kTH2F, {RAxis, phiAxis});

registry.add("VsRadiusNegIU/h_mass", "h_mass", HistType::kTH2F, {RAxis, mAxis});
registry.add("VsRadiusNegIU/h_radius", "IU Radius of the positive track", HistType::kTH2F, {RAxis, RAxis});
registry.add("VsRadiusNegIU/h_phi", "Phi", HistType::kTH2F, {RAxis, phiAxis});
}

// Selection criteria
Expand All @@ -62,7 +86,7 @@ struct qaK0sTrackingEfficiency {
Configurable<bool> eventSelection{"eventSelection", true, "event selection"};

template <typename T1, typename T2, typename C>
bool acceptV0(T1 v0, T2 ntrack, T2 ptrack, C collision)
bool acceptV0(const T1& v0, const T2& ntrack, const T2& ptrack, const C& collision)
{
// Apply selections on V0
if (v0.v0cosPA(collision.posX(), collision.posY(), collision.posZ()) < v0cospa)
Expand All @@ -89,13 +113,19 @@ struct qaK0sTrackingEfficiency {

for (auto& v0 : fullV0s) {

auto reconegtrack = v0.negTrack_as<PIDTracks>();
auto recopostrack = v0.posTrack_as<PIDTracks>();
const auto& recopostrack = v0.posTrack_as<PIDTracks>();
const auto& reconegtrack = v0.negTrack_as<PIDTracks>();

if (acceptV0(v0, reconegtrack, recopostrack, collision)) {
registry.fill(HIST("Test/h_R"), v0.v0radius());
registry.fill(HIST("Test/h_pT"), v0.pt());
registry.fill(HIST("Test/h_mass"), v0.mK0Short());
registry.fill(HIST("VsPt/h_mass"), v0.pt(), v0.mK0Short());
registry.fill(HIST("VsPt/h_tpcnsigma"), v0.pt(), recopostrack.tpcNSigmaPi());
registry.fill(HIST("VsPt/h_tofnsigma"), v0.pt(), recopostrack.tofNSigmaPi());
registry.fill(HIST("VsEta/h_mass"), v0.eta(), v0.mK0Short());
registry.fill(HIST("VsPhi/h_mass"), v0.phi(), v0.mK0Short());
registry.fill(HIST("VsRadius/h_mass"), v0.v0radius(), v0.mK0Short());

registry.fill(HIST("Test/h_negITSStatus"), reconegtrack.hasITS());
registry.fill(HIST("Test/h_posITSStatus"), recopostrack.hasITS());
Expand All @@ -108,6 +138,37 @@ struct qaK0sTrackingEfficiency {
}
}
}

void processIU(SelectedCollisions::iterator const& collision,
aod::V0Datas const& fullV0s,
PIDTracksIU const&)
// TODO: add centrality
{
if (eventSelection && !collision.sel8()) {
return;
}

float posRadius = 0.f;
float negRadius = 0.f;
for (auto& v0 : fullV0s) {
const auto& posTrackIU = v0.posTrack_as<PIDTracksIU>();
const auto& negTrackIU = v0.negTrack_as<PIDTracksIU>();
if (!acceptV0(v0, negTrackIU, posTrackIU, collision)) {
continue;
}
posRadius = sqrt(posTrackIU.x() * posTrackIU.x() + posTrackIU.y() * posTrackIU.y());
negRadius = sqrt(negTrackIU.x() * negTrackIU.x() + negTrackIU.y() * negTrackIU.y());

fillHistogram("VsRadiusPosIU/h_mass", posRadius, v0.mK0Short());
fillHistogram("VsRadiusPosIU/h_radius", posRadius, negRadius);
fillHistogram("VsRadiusPosIU/h_phi", posRadius, posTrackIU.phi());

fillHistogram("VsRadiusNegIU/h_mass", negRadius, v0.mK0Short());
fillHistogram("VsRadiusNegIU/h_radius", negRadius, posRadius);
fillHistogram("VsRadiusNegIU/h_phi", negRadius, posTrackIU.phi());
}
}
PROCESS_SWITCH(qaK0sTrackingEfficiency, processIU, "Run also on IU", true);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
11 changes: 9 additions & 2 deletions PWGLF/TableProducer/cascadebuilder.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -356,8 +356,15 @@ struct cascadeBuilder {
// Act on copies for minimization
auto tV0Copy = o2::track::TrackParCov(tV0);
auto bTrackCopy = o2::track::TrackParCov(bTrack);

int nCand2 = fitterCasc.process(tV0Copy, bTrackCopy);
int nCand2 = 0;
try {
nCand2 = fitterCasc.process(tV0Copy, bTrackCopy);
} catch (...) {
LOG(error) << "Exception caught in fitterCasc.process";
};
if (nCand2 == 0) {
continue;
}
double finalXv0 = fitterCasc.getTrack(0).getX();
double finalXbach = fitterCasc.getTrack(1).getX();

Expand Down