|
36 | 36 | #include "ReconstructionDataFormats/Vertex.h" |
37 | 37 |
|
38 | 38 | #include "Math/Vector4D.h" |
| 39 | +#include "TDatabasePDG.h" |
| 40 | +#include "TParticlePDG.h" |
39 | 41 |
|
40 | 42 | #include <cmath> |
41 | 43 | #include <memory> |
42 | 44 | #include <string> |
43 | 45 | #include <vector> |
44 | | -#include "TDatabasePDG.h" |
45 | | -#include "TParticlePDG.h" |
46 | 46 | // #include "PWGHF/Core/PDG.h" |
47 | 47 | #include "PWGLF/DataModel/LFNonPromptCascadeTables.h" |
48 | 48 | #include "PWGLF/DataModel/LFStrangenessTables.h" |
@@ -186,7 +186,6 @@ struct NonPromptCascadeTask { |
186 | 186 | using CollisionCandidatesRun3MC = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::FT0Mults, aod::FV0Mults, aod::CentFT0Cs, aod::CentFV0As, aod::CentFT0Ms, aod::MultsGlobal>; |
187 | 187 | using CollisionsWithLabel = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::MultsGlobal>; |
188 | 188 |
|
189 | | - |
190 | 189 | Preslice<TracksExtData> perCollision = aod::track::collisionId; |
191 | 190 | Preslice<TracksExtMC> perCollisionMC = aod::track::collisionId; |
192 | 191 |
|
@@ -244,7 +243,6 @@ struct NonPromptCascadeTask { |
244 | 243 | AxisSpec nTracksAxis = {100, 0., 100., "NTracksGlobal"}; |
245 | 244 | AxisSpec nTracksAxisMC = {100, 0., 100., "NTracksMC"}; |
246 | 245 |
|
247 | | - |
248 | 246 | void initCCDB(aod::BCsWithTimestamps::iterator const& bc) |
249 | 247 | { |
250 | 248 | if (mRunNumber == bc.runNumber()) { |
@@ -292,7 +290,7 @@ struct NonPromptCascadeTask { |
292 | 290 | mRegistry.add("hMultFV0VshNTracks", "hMultFV0VshNTracks", HistType::kTH2F, {nTracksAxis, multAxisFV0}); |
293 | 291 | mRegistry.add("hNTracksVsCentFV0A", "hNTracksVsCentFV0A", HistType::kTH2F, {nTracksAxis, centAxis}); |
294 | 292 | mRegistry.add("hNTracksMCVsTracksReco", "hNTracksMCVsTracksReco", HistType::kTH2F, {nTracksAxisMC, nTracksAxis}); |
295 | | - mRegistry.add("hNTracksMCNotInReco","hNTracksMCNotInReco", HistType::kTH1F, {nTracksAxisMC}); |
| 293 | + mRegistry.add("hNTracksMCNotInReco", "hNTracksMCNotInReco", HistType::kTH1F, {nTracksAxisMC}); |
296 | 294 | for (size_t iBin{0}; iBin < cutsNames.size(); ++iBin) { |
297 | 295 | cutsOmega->GetYaxis()->SetBinLabel(iBin + 1, cutsNames[iBin].c_str()); |
298 | 296 | cutsXi->GetYaxis()->SetBinLabel(iBin + 1, cutsNames[iBin].c_str()); |
@@ -711,7 +709,7 @@ struct NonPromptCascadeTask { |
711 | 709 |
|
712 | 710 | void processGenParticles(aod::McParticles const& mcParticles, aod::McCollisions const& collisions) |
713 | 711 | { |
714 | | - //fillMultHistos<aod::McCollisions>(collisions) |
| 712 | + // fillMultHistos<aod::McCollisions>(collisions) |
715 | 713 | for (const auto& p : mcParticles) { |
716 | 714 | auto absCode = std::abs(p.pdgCode()); |
717 | 715 | if (absCode != 3312 && absCode != 3334) { |
@@ -761,49 +759,49 @@ struct NonPromptCascadeTask { |
761 | 759 | } |
762 | 760 | PROCESS_SWITCH(NonPromptCascadeTask, processCascadesData, "process cascades: Data analysis", false); |
763 | 761 |
|
764 | | - int getMCMult(aod::McParticles const& mcParticles, int mcCollId ) |
| 762 | + int getMCMult(aod::McParticles const& mcParticles, int mcCollId) |
765 | 763 | { |
766 | 764 | int mult = 0; |
767 | 765 | for (auto const& mcp : mcParticles) { |
768 | | - if (mcp.mcCollisionId() == mcCollId) { |
| 766 | + if (mcp.mcCollisionId() == mcCollId) { |
769 | 767 | // multiplicity definition: |
770 | 768 | bool accept = mcp.isPhysicalPrimary(); |
771 | 769 | accept *= (mcp.eta() < 0.5) && (mcp.eta() > -0.5); |
772 | 770 | int q = 0; |
773 | 771 | auto pdgEntry = TDatabasePDG::Instance()->GetParticle(mcp.pdgCode()); |
774 | | - if(pdgEntry){ |
| 772 | + if (pdgEntry) { |
775 | 773 | q = int(std::round(pdgEntry->Charge() / 3.0)); |
776 | 774 | } else { |
777 | | - //LOG(warn) << "No pdg assuming neutral"; |
| 775 | + // LOG(warn) << "No pdg assuming neutral"; |
778 | 776 | } |
779 | | - accept *= (q != 0); |
| 777 | + accept *= (q != 0); |
780 | 778 | if (accept) { |
781 | 779 | ++mult; |
782 | 780 | } |
783 | 781 | } |
784 | 782 | } |
785 | 783 | return mult; |
786 | 784 | } |
787 | | - void processNegMC(CollisionsWithLabel const& colls, aod::McCollisions const& mcCollisions, aod::McParticles const& mcParticles ) |
| 785 | + void processNegMC(CollisionsWithLabel const& colls, aod::McCollisions const& mcCollisions, aod::McParticles const& mcParticles) |
788 | 786 | { |
789 | | - //std::cout << "ProcNegMC" << std::endl; |
790 | | - std::vector<int> mcReconstructed(mcCollisions.size(),0); |
791 | | - for(auto const& col: colls){ |
792 | | - int mcCollId = col.mcCollisionId(); // col.template mcCollision_as<aod::McCollisions>(); |
793 | | - //auto mc = col.mcCollision(); |
794 | | - //int mcId = mc.globalIndex(); |
795 | | - //std::cout << "globalIndex:" << mcId << " colID:" << mcCollId << std::endl; |
| 787 | + // std::cout << "ProcNegMC" << std::endl; |
| 788 | + std::vector<int> mcReconstructed(mcCollisions.size(), 0); |
| 789 | + for (auto const& col : colls) { |
| 790 | + int mcCollId = col.mcCollisionId(); // col.template mcCollision_as<aod::McCollisions>(); |
| 791 | + // auto mc = col.mcCollision(); |
| 792 | + // int mcId = mc.globalIndex(); |
| 793 | + // std::cout << "globalIndex:" << mcId << " colID:" << mcCollId << std::endl; |
796 | 794 | int mult = getMCMult(mcParticles, mcCollId); |
797 | 795 | mcReconstructed[mcCollId] = 1; |
798 | 796 | mRegistry.fill(HIST("hNTracksMCVsTracksReco"), mult, col.multNTracksGlobal()); |
799 | 797 | } |
800 | | - for(auto const& mc: mcCollisions){ |
| 798 | + for (auto const& mc : mcCollisions) { |
801 | 799 | int gindex = mc.globalIndex(); |
802 | | - //std::cout << "mc globalIndex:" << gindex << std::endl; |
803 | | - if(!mcReconstructed[gindex]){ |
| 800 | + // std::cout << "mc globalIndex:" << gindex << std::endl; |
| 801 | + if (!mcReconstructed[gindex]) { |
804 | 802 | int mult = getMCMult(mcParticles, gindex); |
805 | | - //std::cout << "===> unreconstructed:" << mult << std::endl; |
806 | | - mRegistry.fill(HIST( "hNTracksMCNotInReco"), mult); |
| 803 | + // std::cout << "===> unreconstructed:" << mult << std::endl; |
| 804 | + mRegistry.fill(HIST("hNTracksMCNotInReco"), mult); |
807 | 805 | } |
808 | 806 | } |
809 | 807 | } |
|
0 commit comments