[PWGHF] Centrality changes in Lc correlator, Correlation task #11724
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -27,6 +27,7 @@ | |
|
|
||
| #include "Common/CCDB/EventSelectionParams.h" | ||
| #include "Common/Core/RecoDecay.h" | ||
| #include "Common/DataModel/Centrality.h" | ||
| #include "Common/DataModel/EventSelection.h" | ||
| #include "Common/DataModel/Multiplicity.h" | ||
| #include "Common/DataModel/PIDResponseTOF.h" | ||
|
|
@@ -65,7 +66,6 @@ using namespace o2::constants::math; | |
| using namespace o2::framework; | ||
| using namespace o2::framework::expressions; | ||
| using namespace o2::analysis::hf_correlations; | ||
| /// | ||
| /// Returns deltaPhi values in range [-pi/2., 3.*pi/2.], typically used for correlation studies | ||
| /// | ||
|
|
@@ -88,11 +88,13 @@ struct HfCorrelatorLcHadronsSelection { | |
| Configurable<int> selectionFlagLc{"selectionFlagLc", 1, "Selection Flag for Lc"}; | ||
| Configurable<float> yCandMax{"yCandMax", 0.8, "max. cand. rapidity"}; | ||
| Configurable<float> ptCandMin{"ptCandMin", 1., "min. cand. pT"}; | ||
| Configurable<float> centMin{"centMin", 0., "Minimum Centrality"}; | ||
| Configurable<float> centMax{"centMax", 100., "Maximum Centrality"}; | ||
|
|
||
| HfHelper hfHelper; | ||
| SliceCache cache; | ||
|
|
||
| using SelCollisions = soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Ms>; | ||
| using CandidatesLcData = soa::Filtered<soa::Join<aod::HfCand3Prong, aod::HfSelLc>>; | ||
| using CandidatesLcMcRec = soa::Filtered<soa::Join<aod::HfCand3Prong, aod::HfSelLc, aod::HfCand3ProngMcRec>>; | ||
| using CandidatesLcMcGen = soa::Join<aod::McParticles, aod::HfCand3ProngMcGen>; | ||
|
|
@@ -108,6 +110,7 @@ struct HfCorrelatorLcHadronsSelection { | |
| bool isLcFound = true; | ||
| bool isSel8 = true; | ||
| bool isNosameBunchPileUp = true; | ||
| bool isCentInRange = false; | ||
| if (doSelLcCollision) { | ||
| for (const auto& candidate : candidates) { | ||
| if (std::abs(hfHelper.yLc(candidate)) > yCandMax || candidate.pt() < ptCandMin) { | ||
|
|
@@ -118,13 +121,18 @@ struct HfCorrelatorLcHadronsSelection { | |
| break; | ||
| } | ||
| } | ||
|
|
||
| float cent = collision.centFT0M(); | ||
|
|
||
| if (useSel8) { | ||
| isSel8 = collision.sel8(); | ||
| } | ||
| if (selNoSameBunchPileUpColl) { | ||
| isNosameBunchPileUp = static_cast<bool>(collision.selection_bit(o2::aod::evsel::kNoSameBunchPileup)); | ||
| } | ||
|
|
||
| isCentInRange = (cent >= centMin && cent < centMax); | ||
| isSelColl = isLcFound && isSel8 && isNosameBunchPileUp && isCentInRange; | ||
| lcSel(isSelColl); | ||
| } | ||
| PROCESS_SWITCH(HfCorrelatorLcHadronsSelection, processLcSelectionData, "Process Lc Collision Selection Data", true); | ||
|
|
@@ -226,6 +234,7 @@ struct HfCorrelatorLcHadrons { | |
| Configurable<bool> isRecTrkPhyPrimary{"isRecTrkPhyPrimary", true, "Calculate the efficiency of reconstructed primary physical tracks"}; | ||
| Configurable<bool> calEffLcEvent{"calEffLcEvent", true, "Calculate the efficiency of Lc candidate"}; | ||
| Configurable<float> eventFractionToAnalyze{"eventFractionToAnalyze", -1, "Fraction of events to analyze (use only for ME offline on very large samples)"}; | ||
| Configurable<bool> isMultiplicityDependent{"isMultiplicityDependent", false, "Flag for multiplicity dependent analyses"}; | ||
|
|
||
| HfHelper hfHelper; | ||
| SliceCache cache; | ||
|
|
@@ -236,7 +245,7 @@ struct HfCorrelatorLcHadrons { | |
| TRandom3* rnd = new TRandom3(0); | ||
|
|
||
| // Event Mixing for the Data Mode | ||
| using SelCollisionsWithLc = soa::Filtered<soa::Join<aod::Collisions, aod::Mults, aod::EvSels, aod::LcSelection, aod::CentFT0Ms>>; | ||
| using SelCollisionsWithLcMc = soa::Filtered<soa::Join<aod::McCollisions, aod::LcSelection, aod::MultsExtraMC>>; // collisionFilter applied | ||
| using CandidatesLcData = soa::Filtered<soa::Join<aod::HfCand3Prong, aod::HfSelLc, aod::HfMlLcToPKPi>>; | ||
| // Event Mixing for the MCRec Mode | ||
|
|
@@ -267,6 +276,7 @@ struct HfCorrelatorLcHadrons { | |
| ConfigurableAxis binsPoolBin{"binsPoolBin", {9, 0., 9.}, "PoolBin"}; | ||
| ConfigurableAxis binsMultFT0M{"binsMultFT0M", {600, 0., 6000.}, "Multiplicity as FT0M signal amplitude"}; | ||
| ConfigurableAxis binsMassLc{"binsMassLc", {200, 1.98, 2.58}, "inv. mass (p K #pi) (GeV/#it{c}^{2})"}; | ||
| ConfigurableAxis binsCentFt0m{"binsCentFt0m", {100, 0., 100.}, "Centrality percentile (FT0M)"}; | ||
|
|
||
| BinningType corrBinning{{binsZVtx, binsMultiplicity}, true}; | ||
|
|
||
|
|
@@ -286,6 +296,7 @@ struct HfCorrelatorLcHadrons { | |
| AxisSpec axisPoolBin = {binsPoolBin, "PoolBin"}; | ||
| AxisSpec axisRapidity = {100, -2, 2, "Rapidity"}; | ||
| AxisSpec axisSign = {2, -1, 1, "Sign"}; | ||
| AxisSpec axisCent = {binsCentFt0m, "Centrality"}; | ||
|
|
||
| registry.add("hPtCand", "Lc,Hadron candidates;candidate #it{p}_{T} (GeV/#it{c});entries", {HistType::kTH1F, {axisPtLc}}); | ||
| registry.add("hPtProng0", "Lc,Hadron candidates;prong 0 #it{p}_{T} (GeV/#it{c});entries", {HistType::kTH1F, {axisPtLc}}); | ||
|
|
@@ -301,9 +312,14 @@ struct HfCorrelatorLcHadrons { | |
| registry.add("hMultiplicity", "multiplicity;multiplicity;entries", {HistType::kTH1F, {{10000, 0., 10000.}}}); | ||
| registry.add("hMultFT0M", "multiplicity;multiplicity;entries", {HistType::kTH1F, {{10000, 0., 10000.}}}); | ||
| registry.add("hZvtx", "z vertex;z vertex;entries", {HistType::kTH1F, {{200, -20., 20.}}}); | ||
| registry.add("hCentFT0M", "Centrality FT0M; Centrality;entries", {HistType::kTH1D, {{100, 0., 100.}}}); | ||
| registry.add("hLcBin", "Lc selected in pool Bin;pool Bin;entries", {HistType::kTH1F, {{9, 0., 9.}}}); | ||
| registry.add("hTracksBin", "Tracks selected in pool Bin;pool Bin;entries", {HistType::kTH1F, {{9, 0., 9.}}}); | ||
| if (isMultiplicityDependent) { | ||
| registry.add("hMassLcVsPtvsMult", "Lc candidates;inv. mass (p K #pi) (GeV/#it{c}^{2});entries", {HistType::kTH3F, {{axisMassLc}, {axisPtLc}, {axisCent}}}); | ||
| } else { | ||
| registry.add("hMassLcVsPt", "Lc candidates;inv. mass (p K #pi) (GeV/#it{c}^{2});entries", {HistType::kTH2F, {{axisMassLc}, {axisPtLc}}}); | ||
| } | ||
|
Collaborator
There was a problem hiding this comment. I don't see any relation to multiplicity.
Collaborator
There was a problem hiding this comment. Why do you need
Contributor
Author
There was a problem hiding this comment. Okay, We have changed the name of the histogram to have
Collaborator
There was a problem hiding this comment. What is unsafe about projecting
Contributor
Author
There was a problem hiding this comment. Projecting along the centrality axis works fine for us too. We only kept the separate hMassLcVsPt upon Antonio’s suggestion, so that others using the same task without centrality wouldn’t need to change anything. If you think it's better to just keep one and project, we’re happy to update it — please suggest.
Collaborator
There was a problem hiding this comment. I suggest a single histogram.
Collaborator
There was a problem hiding this comment. Hi @vkucera I am not sure what is the recommended process in HF when adding new things to an exciting analysis task that is being used by others. During the discussion with coordinators, we agreed that while extending this task to perform a multiplicity based study, Tushar adds flags and additional histograms in a way that has minimal impact on current analyzers. Changing a 2D histogram to a 3D will disrupt others, which we would like to minimize. If you still insist on changing it, I would appreciate if you could share a reasoning to do it as well.
Collaborator
There was a problem hiding this comment. Hi @deepathoms @apalasciano , here is how I see the two scenarios:
IMO, the drawbacks of the split scenario and the benefits of the no-split scenario are far more important than protecting users from changing a histogram name in their local scripts. One can even keep the current histogram name. Concerning the projection, I don't see how it changes anything for the users. You said
but I don't understand what you mean here.
In general, I don't agree that keeping the framework interface and the output intact should be a top priority. The framework is evolving based on the analysis needs and we should be able to collaborate on the evolution of the common code, not fragment it and protect it from changes. We have introduced breaking changes even in the core code many times before. We communicated them to the community in an announcement and explained the consequences. It's just about communication. |
||
| registry.add("hMassLcData", "Lc candidates;inv. mass (p K #pi) (GeV/#it{c}^{2});entries", {HistType::kTH1F, {{axisMassLc}}}); | ||
| registry.add("hLcPoolBin", "Lc candidates pool bin", {HistType::kTH1F, {axisPoolBin}}); | ||
| registry.add("hTracksPoolBin", "Particles associated pool bin", {HistType::kTH1F, {axisPoolBin}}); | ||
|
|
@@ -346,7 +362,6 @@ struct HfCorrelatorLcHadrons { | |
| registry.add("hEtaMcGen", "Lc,Hadron particles - MC Gen", {HistType::kTH1F, {axisEta}}); | ||
| registry.add("hPhiMcGen", "Lc,Hadron particles - MC Gen", {HistType::kTH1F, {axisPhi}}); | ||
| registry.add("hMultFT0AMcGen", "Lc,Hadron multiplicity FT0A - MC Gen", {HistType::kTH1F, {axisMultiplicity}}); | ||
| corrBinning = {{binsZVtx, binsMultiplicity}, true}; | ||
| } | ||
|
|
||
|
|
@@ -374,6 +389,8 @@ struct HfCorrelatorLcHadrons { | |
| int gCollisionId = collision.globalIndex(); | ||
| int64_t timeStamp = bc.timestamp(); | ||
|
|
||
| float cent = collision.centFT0M(); | ||
|
|
||
| int poolBin = corrBinning.getBin(std::make_tuple(collision.posZ(), collision.multFT0M())); | ||
| int nTracks = 0; | ||
| if (collision.numContrib() > 1) { | ||
|
|
@@ -412,8 +429,13 @@ struct HfCorrelatorLcHadrons { | |
| registry.fill(HIST("hPhi"), RecoDecay::constrainAngle(candidate.phi(), -PIHalf)); | ||
| registry.fill(HIST("hY"), hfHelper.yLc(candidate)); | ||
| registry.fill(HIST("hLcBin"), poolBin); | ||
|
|
||
| if (candidate.isSelLcToPKPi() >= selectionFlagLc) { | ||
| if (isMultiplicityDependent) { | ||
| registry.fill(HIST("hMassLcVsPtvsMult"), hfHelper.invMassLcToPKPi(candidate), candidate.pt(), cent, efficiencyWeightLc); | ||
| } else { | ||
| registry.fill(HIST("hMassLcVsPt"), hfHelper.invMassLcToPKPi(candidate), candidate.pt(), efficiencyWeightLc); | ||
| } | ||
| registry.fill(HIST("hMassLcData"), hfHelper.invMassLcToPKPi(candidate), efficiencyWeightLc); | ||
| registry.fill(HIST("hSelectionStatusLcToPKPi"), candidate.isSelLcToPKPi()); | ||
| for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) { | ||
|
|
@@ -423,7 +445,11 @@ struct HfCorrelatorLcHadrons { | |
| entryLc(candidate.phi(), candidate.eta(), candidate.pt() * chargeLc, hfHelper.invMassLcToPKPi(candidate), poolBin, gCollisionId, timeStamp); | ||
| } | ||
| if (candidate.isSelLcToPiKP() >= selectionFlagLc) { | ||
| if (isMultiplicityDependent) { | ||
| registry.fill(HIST("hMassLcVsPtvsMult"), hfHelper.invMassLcToPKPi(candidate), candidate.pt(), cent, efficiencyWeightLc); | ||
| } else { | ||
| registry.fill(HIST("hMassLcVsPt"), hfHelper.invMassLcToPKPi(candidate), candidate.pt(), efficiencyWeightLc); | ||
| } | ||
| registry.fill(HIST("hMassLcData"), hfHelper.invMassLcToPiKP(candidate), efficiencyWeightLc); | ||
| registry.fill(HIST("hSelectionStatusLcToPiKP"), candidate.isSelLcToPiKP()); | ||
| for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) { | ||
|
|
@@ -463,7 +489,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| track.pt() * track.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(track.rapidity(MassProton) - hfHelper.yLc(candidate)); // only for proton as of now | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPKPi(candidate), false); | ||
| entryLcHadronGenInfo(false, false, 0); | ||
|
|
@@ -479,7 +506,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| track.pt() * track.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(track.rapidity(MassProton) - hfHelper.yLc(candidate)); // only for proton as of now | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPiKP(candidate), false); | ||
| entryLcHadronGenInfo(false, false, 0); | ||
|
|
@@ -503,6 +531,7 @@ struct HfCorrelatorLcHadrons { | |
| } // end outer Lc loop | ||
| registry.fill(HIST("hZvtx"), collision.posZ()); | ||
| registry.fill(HIST("hMultFT0M"), collision.multFT0M()); | ||
| registry.fill(HIST("hCentFT0M"), cent); | ||
| } | ||
| PROCESS_SWITCH(HfCorrelatorLcHadrons, processData, "Process data", true); | ||
|
|
||
|
|
@@ -667,6 +696,7 @@ struct HfCorrelatorLcHadrons { | |
| } | ||
| //} | ||
| } | ||
| float cent = 100.0; // will be updated later | ||
|
|
||
| // Lc-Hadron correlation dedicated section | ||
| // if the candidate is selected as Lc, search for Hadron ad evaluate correlations | ||
|
|
@@ -716,7 +746,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| track.pt() * track.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(track.rapidity(MassProton) - hfHelper.yLc(candidate)); // only for proton as of now | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPKPi(candidate), isLcSignal); | ||
| if (fillTrkPID) { | ||
|
|
@@ -746,7 +777,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| track.pt() * track.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(track.rapidity(MassProton) - hfHelper.yLc(candidate)); // only for proton as of now | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPiKP(candidate), isLcSignal); | ||
| if (fillTrkPID) { | ||
|
|
@@ -790,6 +822,7 @@ struct HfCorrelatorLcHadrons { | |
|
|
||
| bool isLcPrompt = false; | ||
| bool isLcNonPrompt = false; | ||
| float cent = 100.0; // will be updated later | ||
|
|
||
| // find leading particle | ||
| if (correlateLcWithLeadingParticle) { | ||
|
|
@@ -878,7 +911,8 @@ struct HfCorrelatorLcHadrons { | |
| particle.pt() * chargeLc / std::abs(chargeLc), | ||
| particleAssoc.pt() * chargeAssoc / std::abs(chargeAssoc), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(particleAssoc.y() - yL); | ||
| entryLcHadronRecoInfo(MassLambdaCPlus, true); | ||
| entryLcHadronGenInfo(isLcPrompt, particleAssoc.isPhysicalPrimary(), trackOrigin); | ||
|
|
@@ -911,6 +945,8 @@ struct HfCorrelatorLcHadrons { | |
| auto trackPos1 = trigLc.template prong0_as<TracksData>(); // positive daughter (negative for the antiparticles) | ||
| int8_t chargeLc = trackPos1.sign(); // charge of 1st prong will be the charge of Lc candidate | ||
|
|
||
| float cent = 100.0; // will be updated later | ||
|
|
||
| std::vector<float> outputMl = {-1., -1., -1.}; | ||
| // LcToPKPi and LcToPiKP division | ||
| if (trigLc.isSelLcToPKPi() >= selectionFlagLc) { | ||
|
|
@@ -919,7 +955,8 @@ struct HfCorrelatorLcHadrons { | |
| trigLc.pt() * chargeLc, | ||
| assocParticle.pt() * assocParticle.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(assocParticle.y() - hfHelper.yLc(trigLc)); | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPKPi(trigLc), false); | ||
| entryLcHadronGenInfo(false, false, 0); | ||
|
|
@@ -938,7 +975,8 @@ struct HfCorrelatorLcHadrons { | |
| trigLc.pt() * chargeLc, | ||
| assocParticle.pt() * assocParticle.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(assocParticle.y() - hfHelper.yLc(trigLc)); | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPiKP(trigLc), false); | ||
| entryLcHadronGenInfo(false, false, 0); | ||
|
|
@@ -1006,6 +1044,7 @@ struct HfCorrelatorLcHadrons { | |
| registry.fill(HIST("hZvtx"), c1.posZ()); | ||
| registry.fill(HIST("hTracksPoolBin"), poolBin); | ||
| registry.fill(HIST("hLcPoolBin"), poolBinLc); | ||
| float cent = 100.0; // will be updated later | ||
| for (const auto& [candidate, pAssoc] : o2::soa::combinations(o2::soa::CombinationsFullIndexPolicy(tracks1, tracks2))) { | ||
| if (std::abs(hfHelper.yLc(candidate)) > yCandMax || candidate.pt() < ptCandMin || candidate.pt() > ptCandMax) { | ||
| continue; | ||
|
|
@@ -1038,7 +1077,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| pAssoc.pt() * pAssoc.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(pAssoc.y() - hfHelper.yLc(candidate)); | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPKPi(candidate), isLcSignal); | ||
| entryLcHadronGenInfo(isLcPrompt, isPhysicalPrimary, trackOrigin); | ||
|
|
@@ -1057,7 +1097,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc, | ||
| pAssoc.pt() * pAssoc.sign(), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(pAssoc.y() - hfHelper.yLc(candidate)); | ||
| entryLcHadronRecoInfo(hfHelper.invMassLcToPiKP(candidate), isLcSignal); | ||
| entryLcHadronGenInfo(isLcPrompt, isPhysicalPrimary, trackOrigin); | ||
|
|
@@ -1105,6 +1146,7 @@ struct HfCorrelatorLcHadrons { | |
| } | ||
| int8_t chargeLc = pdg->GetParticle(candidate.pdgCode())->Charge(); // Retrieve charge | ||
| int8_t chargeAssoc = pdg->GetParticle(particleAssoc.pdgCode())->Charge(); // Retrieve charge | ||
| float cent = 100.0; // will be updated later | ||
|
|
||
| int trackOrigin = RecoDecay::getCharmHadronOrigin(mcParticles, particleAssoc, true); | ||
| bool isLcPrompt = candidate.originMcGen() == RecoDecay::OriginType::Prompt; | ||
|
|
@@ -1113,7 +1155,8 @@ struct HfCorrelatorLcHadrons { | |
| candidate.pt() * chargeLc / std::abs(chargeLc), | ||
| particleAssoc.pt() * chargeAssoc / std::abs(chargeAssoc), | ||
| poolBin, | ||
| correlationStatus, | ||
| cent); | ||
| entryLcHadronPairY(particleAssoc.y() - yL); | ||
| entryLcHadronRecoInfo(MassLambdaCPlus, true); | ||
| entryLcHadronGenInfo(isLcPrompt, particleAssoc.isPhysicalPrimary(), trackOrigin); | ||
|
|
||
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.