Moved true N sigma plots earlier in the PID process. Added histograms to show PID contamination

Author: bghanley1995

PWGCF/TwoParticleCorrelations/TableProducer/identifiedBfFilter.cxx

@@ -12,6 +12,7 @@
 /// \file identifiedBfFilter.cxx
 /// \brief Filters collisions and tracks according to selection criteria
 /// \author bghanley1995@gmail.com
+
 #include "PWGCF/TwoParticleCorrelations/TableProducer/identifiedBfFilter.h"
 
 #include <cmath>
@@ -156,8 +157,8 @@ TH2F* fhCompatibleCollisionsZVtxRms = nullptr;
 TH2S* fhTruePIDMismatch = nullptr;
 TH1S* fhTruePIDCorrect = nullptr;
 
-TH2F* fhTrueNSigmaTPC[kIdBfNoOfSpecies] = {nullptr};
-TH2F* fhTrueNSigmaTOF[kIdBfNoOfSpecies] = {nullptr};
+std::vector<std::vector<TH2F*>> fhTrueNSigmaTPC = {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies,{o2::analysis::identifiedbffilter::kIdBfNoOfSpecies,nullptr}};
+std::vector<std::vector<TH2F*>> fhTrueNSigmaTOF = {o2::analysis::identifiedbffilter::kIdBfNoOfSpecies,{o2::analysis::identifiedbffilter::kIdBfNoOfSpecies,nullptr}};
 
 TH1F* fhTrueCentMultB = nullptr;
 TH1F* fhTrueCentMultA = nullptr;
@@ -1025,16 +1026,18 @@ struct IdentifiedBfFilterTracks {
                                      79, -39.5, 39.5);
       }
 
-      for (int sp = 0; sp < kIdBfNoOfSpecies; ++sp) {
-        fhTrueNSigmaTPC[sp] = new TH2F(TString::Format("fhTrueNSigmaTPC_%s", speciesName[sp]).Data(),
-                                       TString::Format("N #sigma from TPC vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp]).Data(),
-                                       48, -6, 6,
-                                       ptbins, ptlow, ptup);
-
-        fhTrueNSigmaTOF[sp] = new TH2F(TString::Format("fhTrueNSigmaTOF_%s", speciesName[sp]).Data(),
-                                       TString::Format("N #sigma from TOF vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp]).Data(),
-                                       48, -6, 6,
-                                       ptbins, ptlow, ptup);
+      for (int sp1 = 0; sp1 < kIdBfNoOfSpecies; ++sp1) {
+        for (int sp2 = 0; sp2 < kIdBfNoOfSpecies; ++sp2) {
+          fhTrueNSigmaTPC[sp1][sp2] = new TH2F(TString::Format("fhTrueNSigmaTPC%s_%s", speciesName[sp1], speciesName[sp2]).Data(),
+                                        TString::Format("N #sigma %s from TPC vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp1], speciesTitle[sp2]).Data(),
+                                        48, -6, 6,
+                                        ptbins, ptlow, ptup);
+
+          fhTrueNSigmaTOF[sp1][sp2] = new TH2F(TString::Format("fhTrueNSigmaTOF%s_%s", speciesName[sp1], speciesName[sp2]).Data(),
+                                        TString::Format("N #sigma %s from TOF vs P for generated %s;N #sigma;p (GeV/c)", speciesTitle[sp1], speciesTitle[sp2]).Data(),
+                                        48, -6, 6,
+                                        ptbins, ptlow, ptup);
+        }
       }
 
       /* add the hstograms to the output list */
@@ -1073,9 +1076,11 @@ struct IdentifiedBfFilterTracks {
         fOutputList->Add(fhTrueNPosNegA[sp]);
         fOutputList->Add(fhTrueDeltaNA[sp]);
       }
-      for (int sp = 0; sp < kIdBfNoOfSpecies; ++sp) {
-        fOutputList->Add(fhTrueNSigmaTPC[sp]);
-        fOutputList->Add(fhTrueNSigmaTOF[sp]);
+      for (int sp1 = 0; sp1 < kIdBfNoOfSpecies; ++sp1) {
+        for (int sp2 = 0; sp2 < kIdBfNoOfSpecies; ++sp2) {
+          fOutputList->Add(fhTrueNSigmaTPC[sp1][sp2]);
+          fOutputList->Add(fhTrueNSigmaTOF[sp1][sp2]);
+        }
       }
     }
     /* initialize access to the CCDB */
@@ -1380,7 +1385,6 @@ inline void IdentifiedBfFilterTracks::identifyRealNSigma(ParticleObject const& p
 
   MatchRecoGenSpecies realPID = kWrongSpecies;
   int pdgcode = std::fabs(particle.pdgCode());
-
   switch (pdgcode) {
     case pdgcodeEl:
       realPID = kIdBfElectron;
@@ -1402,8 +1406,14 @@ inline void IdentifiedBfFilterTracks::identifyRealNSigma(ParticleObject const& p
       break;
   }
   if (!(realPID < 0)) {
-    fhTrueNSigmaTPC[realPID]->Fill(tpcNSigma[realPID], tpcInnerParam);
-    fhTrueNSigmaTOF[realPID]->Fill(tofNSigma[realPID], tpcInnerParam);
+    fhTrueNSigmaTPC[kIdBfElectron][realPID]->Fill(tpcNSigma[kIdBfElectron], tpcInnerParam);
+    fhTrueNSigmaTOF[kIdBfElectron][realPID]->Fill(tofNSigma[kIdBfElectron], tpcInnerParam);
+    fhTrueNSigmaTPC[kIdBfPion][realPID]->Fill(tpcNSigma[kIdBfPion], tpcInnerParam);
+    fhTrueNSigmaTOF[kIdBfPion][realPID]->Fill(tofNSigma[kIdBfPion], tpcInnerParam);
+    fhTrueNSigmaTPC[kIdBfKaon][realPID]->Fill(tpcNSigma[kIdBfKaon], tpcInnerParam);
+    fhTrueNSigmaTOF[kIdBfKaon][realPID]->Fill(tofNSigma[kIdBfKaon], tpcInnerParam);
+    fhTrueNSigmaTPC[kIdBfProton][realPID]->Fill(tpcNSigma[kIdBfProton], tpcInnerParam);
+    fhTrueNSigmaTOF[kIdBfProton][realPID]->Fill(tofNSigma[kIdBfProton], tpcInnerParam);
   }
 }
 
@@ -1439,8 +1449,8 @@ void fillNSigmaHistos(TrackObject const& track)
 
   fhNSigmaTOF[kIdBfElectron]->Fill(actualTOFNSigma[kIdBfElectron], track.tpcInnerParam());
   fhNSigmaTOF[kIdBfPion]->Fill(actualTOFNSigma[kIdBfPion], track.tpcInnerParam());
-  fhNSigmaTOF[kIdBfKaon]->Fill(track.tofNSigmaKa(), track.tpcInnerParam());
-  fhNSigmaTOF[kIdBfProton]->Fill(track.tofNSigmaPr(), track.tpcInnerParam());
+  fhNSigmaTOF[kIdBfKaon]->Fill(actualTOFNSigma[kIdBfKaon], track.tpcInnerParam());
+  fhNSigmaTOF[kIdBfProton]->Fill(actualTOFNSigma[kIdBfProton], track.tpcInnerParam());
 
   fhNSigmaCombo[kIdBfElectron]->Fill(sqrtf(actualTOFNSigma[kIdBfElectron] * actualTOFNSigma[kIdBfElectron] + actualTPCNSigma[kIdBfElectron] * actualTPCNSigma[kIdBfElectron]), track.tpcInnerParam());
   fhNSigmaCombo[kIdBfPion]->Fill(sqrtf(actualTOFNSigma[kIdBfPion] * actualTOFNSigma[kIdBfPion] + actualTPCNSigma[kIdBfPion] * actualTPCNSigma[kIdBfPion]), track.tpcInnerParam());
@@ -1459,14 +1469,14 @@ inline MatchRecoGenSpecies IdentifiedBfFilterTracks::identifyTrack(TrackObject c
 
   fillNSigmaHistos(track);
 
-  std::vector<float> actualTPCNSigma;
+  std::vector<float> actualTPCNSigma(kIdBfNoOfSpecies,0.);
 
   actualTPCNSigma[kIdBfElectron] = track.tpcNSigmaEl();
   actualTPCNSigma[kIdBfPion] = track.tpcNSigmaPi();
   actualTPCNSigma[kIdBfKaon] = track.tpcNSigmaKa();
   actualTPCNSigma[kIdBfProton] = track.tpcNSigmaPr();
 
-  std::vector<float> actualTOFNSigma;
+  std::vector<float> actualTOFNSigma(kIdBfNoOfSpecies,0.);
 
   actualTOFNSigma[kIdBfElectron] = track.tofNSigmaEl();
   actualTOFNSigma[kIdBfPion] = track.tofNSigmaPi();