1717#include " PWGCF/Femto3D/DataModel/singletrackselector.h"
1818
1919#include " Common/Core/RecoDecay.h"
20+ #include " Common/Core/TrackSelection.h"
21+ #include " Common/Core/trackUtilities.h"
22+ #include " Common/DataModel/Centrality.h"
23+ #include " Common/DataModel/EventSelection.h"
2024#include " Common/DataModel/Multiplicity.h"
25+ #include " Common/DataModel/TrackSelectionTables.h"
2126
2227#include " CCDB/BasicCCDBManager.h"
2328#include " CCDB/CcdbApi.h"
29+ #include " CommonConstants/PhysicsConstants.h"
2430#include " Framework/ASoA.h"
31+ #include " Framework/ASoAHelpers.h"
2532#include " Framework/AnalysisDataModel.h"
2633#include " Framework/AnalysisTask.h"
2734#include " Framework/DataTypes.h"
2835#include " Framework/Expressions.h"
2936#include " Framework/HistogramRegistry.h"
3037#include " Framework/O2DatabasePDGPlugin.h"
3138#include " Framework/StaticFor.h"
39+ #include " Framework/StepTHn.h"
3240#include " Framework/runDataProcessing.h"
3341#include " MathUtils/Utils.h"
34-
35- #include " Common/Core/TrackSelection.h"
36- #include " Common/Core/trackUtilities.h"
37- #include " Common/DataModel/Centrality.h"
38- #include " Common/DataModel/EventSelection.h"
39- #include " Common/DataModel/TrackSelectionTables.h"
40-
41- #include " CommonConstants/PhysicsConstants.h"
42- #include " Framework/ASoAHelpers.h"
43- #include " Framework/AnalysisDataModel.h"
44- #include " Framework/StepTHn.h"
4542#include " ReconstructionDataFormats/Track.h"
4643
4744#include " TGrid.h"
@@ -138,8 +135,8 @@ struct hadronnucleicorrelation {
138135
139136 using trkType = const FilteredTracks::iterator*;
140137 using trkTypeMC = const FilteredTracksMC::iterator*;
141- // typedef std::shared_ptr<FilteredCollisions::iterator> colType;
142- // typedef std::shared_ptr<SimCollisions::iterator> MCcolType;
138+ // typedef std::shared_ptr<FilteredCollisions::iterator> colType;
139+ // typedef std::shared_ptr<SimCollisions::iterator> MCcolType;
143140
144141 std::unique_ptr<o2::aod::singletrackselector::FemtoPair<trkType>> Pair = std::make_unique<o2::aod::singletrackselector::FemtoPair<trkType>>();
145142 std::unique_ptr<o2::aod::singletrackselector::FemtoPair<trkTypeMC>> PairMC = std::make_unique<o2::aod::singletrackselector::FemtoPair<trkTypeMC>>();
@@ -638,7 +635,6 @@ struct hadronnucleicorrelation {
638635 } // nBinspT loop
639636 }
640637
641-
642638 void GetCorrection (o2::framework::Service<o2::ccdb::BasicCCDBManager> const & ccdbObj, TString filepath, TString histname)
643639 {
644640 TList* l = ccdbObj->get <TList>(filepath.Data ());
@@ -676,7 +672,7 @@ struct hadronnucleicorrelation {
676672 {
677673
678674 if (std::abs (collision.posZ ()) > cutzvertex)
679- return ;
675+ return ;
680676
681677 registry.fill (HIST (" hNEvents" ), 0.5 );
682678 registry.fill (HIST (" hMult" ), collision.mult ());
@@ -717,28 +713,28 @@ struct hadronnucleicorrelation {
717713 QA.fill (HIST (" QA/hnSigmaITSVsPt_Pr" ), track.pt () * track.sign (), track.itsNSigmaPr ());
718714 QA.fill (HIST (" QA/hnSigmaITSVsPt_De" ), track.pt () * track.sign (), track.itsNSigmaDe ());
719715
720- if (IsProton (track, +1 )){
716+ if (IsProton (track, +1 )) {
721717 QA.fill (HIST (" QA/hEtaAntiPr" ), track.eta ());
722718 QA.fill (HIST (" QA/hPhiAntiPr" ), track.phi ());
723719 QA.fill (HIST (" QA/hnSigmaTOFVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.tofNSigmaPr ());
724720 QA.fill (HIST (" QA/hnSigmaTPCVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.tpcNSigmaPr ());
725721 QA.fill (HIST (" QA/hnSigmaITSVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.itsNSigmaPr ());
726722 }
727- if (IsProton (track, -1 )){
723+ if (IsProton (track, -1 )) {
728724 QA.fill (HIST (" QA/hEtaPr" ), track.eta ());
729725 QA.fill (HIST (" QA/hPhiPr" ), track.phi ());
730726 QA.fill (HIST (" QA/hnSigmaTOFVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.tofNSigmaPr ());
731727 QA.fill (HIST (" QA/hnSigmaTPCVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.tpcNSigmaPr ());
732728 QA.fill (HIST (" QA/hnSigmaITSVsPt_Pr_AfterSel" ), track.pt () * track.sign (), track.itsNSigmaPr ());
733729 }
734- if (IsDeuteron (track, +1 )){
730+ if (IsDeuteron (track, +1 )) {
735731 QA.fill (HIST (" QA/hEtaAntiDe" ), track.eta ());
736732 QA.fill (HIST (" QA/hPhiAntiDe" ), track.phi ());
737733 QA.fill (HIST (" QA/hnSigmaTOFVsPt_De_AfterSel" ), track.pt () * track.sign (), track.tofNSigmaDe ());
738734 QA.fill (HIST (" QA/hnSigmaTPCVsPt_De_AfterSel" ), track.pt () * track.sign (), track.tpcNSigmaDe ());
739735 QA.fill (HIST (" QA/hnSigmaITSVsPt_De_AfterSel" ), track.pt () * track.sign (), track.itsNSigmaDe ());
740736 }
741- if (IsDeuteron (track, -1 )){
737+ if (IsDeuteron (track, -1 )) {
742738 QA.fill (HIST (" QA/hEtaDe" ), track.eta ());
743739 QA.fill (HIST (" QA/hPhiDe" ), track.phi ());
744740 QA.fill (HIST (" QA/hnSigmaTOFVsPt_De_AfterSel" ), track.pt () * track.sign (), track.tofNSigmaDe ());
@@ -748,7 +744,7 @@ struct hadronnucleicorrelation {
748744 }
749745 }
750746
751- if (mode == 5 || mode == 6 ) { // Identical particle combinations
747+ if (mode == 5 || mode == 6 ) { // Identical particle combinations
752748
753749 for (const auto & [part0, part1] : combinations (CombinationsStrictlyUpperIndexPolicy (tracks, tracks))) {
754750
@@ -766,15 +762,15 @@ struct hadronnucleicorrelation {
766762 if (!applyDCAcut (part1))
767763 continue ;
768764
769- // mode 6
770- if (mode == 6 ){
765+ // mode 6
766+ if (mode == 6 ) {
771767 if (!IsProton (part0, +1 ))
772768 continue ;
773769 if (!IsProton (part1, +1 ))
774770 continue ;
775771 }
776- // mode 5
777- if (mode == 5 ){
772+ // mode 5
773+ if (mode == 5 ) {
778774 if (!IsProton (part0, -1 ))
779775 continue ;
780776 if (!IsProton (part1, -1 ))
@@ -802,7 +798,7 @@ struct hadronnucleicorrelation {
802798 if (!applyDCAcut (part1))
803799 continue ;
804800
805- // modes 0,1,2,3,4,7
801+ // modes 0,1,2,3,4,7
806802 if (mode == 0 ) {
807803 if (!IsDeuteron (part0, -1 ))
808804 continue ;
@@ -846,13 +842,12 @@ struct hadronnucleicorrelation {
846842 }
847843 PROCESS_SWITCH (hadronnucleicorrelation, processSameEvent, " processSameEvent" , true );
848844
849-
850845 void processMixedEvent (FilteredCollisions const & collisions, FilteredTracks const & tracks)
851846 {
852847
853848 for (const auto & [collision1, collision2] : soa::selfCombinations (colBinning, 5 , -1 , collisions, collisions)) {
854849
855- // LOGF(info, "Mixed event collisions: (%d, %d) zvtx (%.1f, %.1f) mult (%d, %d)", collision1.globalIndex(), collision2.globalIndex(), collision1.posZ(), collision2.posZ(), collision1.mult(), collision2.mult());
850+ // LOGF(info, "Mixed event collisions: (%d, %d) zvtx (%.1f, %.1f) mult (%d, %d)", collision1.globalIndex(), collision2.globalIndex(), collision1.posZ(), collision2.posZ(), collision1.mult(), collision2.mult());
856851
857852 auto groupPartsOne = tracks.sliceByCached (o2::aod::singletrackselector::singleCollSelId, collision1.globalIndex (), cache);
858853 auto groupPartsTwo = tracks.sliceByCached (o2::aod::singletrackselector::singleCollSelId, collision2.globalIndex (), cache);
@@ -911,19 +906,19 @@ struct hadronnucleicorrelation {
911906 if (!IsProton (part1, +1 ))
912907 continue ;
913908 }
914- if (mode == 5 ){
909+ if (mode == 5 ) {
915910 if (!IsProton (part0, -1 ))
916911 continue ;
917912 if (!IsProton (part1, -1 ))
918913 continue ;
919914 }
920- if (mode == 6 ){
915+ if (mode == 6 ) {
921916 if (!IsProton (part0, +1 ))
922917 continue ;
923918 if (!IsProton (part1, +1 ))
924919 continue ;
925920 }
926- if (mode == 7 ){
921+ if (mode == 7 ) {
927922 if (!IsProton (part0, +1 ))
928923 continue ;
929924 if (!IsProton (part1, -1 ))
@@ -1310,7 +1305,7 @@ struct hadronnucleicorrelation {
13101305 {
13111306
13121307 if (std::abs (mcCollision.posZ ()) > cutzvertex)
1313- return ;
1308+ return ;
13141309
13151310 registry.fill (HIST (" Generated/hNEventsMC" ), 0.5 );
13161311
@@ -1364,19 +1359,19 @@ struct hadronnucleicorrelation {
13641359 }
13651360 }
13661361
1367- if (mode == 5 || mode == 6 ) { // Identical particle combinations
1362+ if (mode == 5 || mode == 6 ) { // Identical particle combinations
13681363
13691364 for (const auto & [part0, part1] : combinations (CombinationsStrictlyUpperIndexPolicy (mcParticles, mcParticles))) {
13701365
1371- // mode 6
1372- if (mode == 6 ){
1366+ // mode 6
1367+ if (mode == 6 ) {
13731368 if (part0.pdgCode () != PDG_t::kProton )
13741369 continue ;
13751370 if (part1.pdgCode () != PDG_t::kProton )
13761371 continue ;
13771372 }
1378- // mode 5
1379- if (mode == 5 ){
1373+ // mode 5
1374+ if (mode == 5 ) {
13801375 if (part0.pdgCode () != -PDG_t::kProton )
13811376 continue ;
13821377 if (part1.pdgCode () != -PDG_t::kProton )
@@ -1438,7 +1433,7 @@ struct hadronnucleicorrelation {
14381433
14391434 for (const auto & [collision1, collision2] : soa::selfCombinations (colBinningGen, 5 , -1 , mcCollisions, mcCollisions)) {
14401435
1441- // LOGF(info, "Mixed event collisions: (%d, %d) zvtx (%.1f, %.1f) mult (%d, %d)", collision1.globalIndex(), collision2.globalIndex(), collision1.posZ(), collision2.posZ(), collision1.mult(), collision2.mult());
1436+ // LOGF(info, "Mixed event collisions: (%d, %d) zvtx (%.1f, %.1f) mult (%d, %d)", collision1.globalIndex(), collision2.globalIndex(), collision1.posZ(), collision2.posZ(), collision1.mult(), collision2.mult());
14421437
14431438 auto groupPartsOne = mcParticles.sliceByCached (o2::aod::mcparticle::mcCollisionId, collision1.globalIndex (), cache);
14441439 auto groupPartsTwo = mcParticles.sliceByCached (o2::aod::mcparticle::mcCollisionId, collision2.globalIndex (), cache);
@@ -1475,13 +1470,13 @@ struct hadronnucleicorrelation {
14751470 if (part1.pdgCode () != PDG_t::kProton )
14761471 continue ;
14771472 }
1478- if (mode == 5 ){
1473+ if (mode == 5 ) {
14791474 if (part0.pdgCode () != -PDG_t::kProton )
14801475 continue ;
14811476 if (part1.pdgCode () != -PDG_t::kProton )
14821477 continue ;
14831478 }
1484- if (mode == 6 ){
1479+ if (mode == 6 ) {
14851480 if (part0.pdgCode () != PDG_t::kProton )
14861481 continue ;
14871482 if (part1.pdgCode () != PDG_t::kProton )
0 commit comments