PWGHF: Add candidate and track counter histograms. Subscribe to PID tables only when needed. (#4723)

vkucera · web-flow · commit 4a12ca0c8ed2 · 2020-10-30T08:51:36.000+01:00
diff --git a/Analysis/DataModel/include/Analysis/HFSecondaryVertex.h b/Analysis/DataModel/include/Analysis/HFSecondaryVertex.h
@@ -37,8 +37,9 @@ DECLARE_SOA_TABLE(HFSelTrack, "AOD", "HFSELTRACK",
                   hf_seltrack::DCAPrim0,
                   hf_seltrack::DCAPrim1);
 
-using BigTracks = soa::Join<Tracks, TracksCov, TracksExtra, HFSelTrack, pidRespTPC, pidRespTOF>;
+using BigTracks = soa::Join<Tracks, TracksCov, TracksExtra, HFSelTrack>;
 using BigTracksMC = soa::Join<BigTracks, McTrackLabels>;
+using BigTracksPID = soa::Join<BigTracks, pidRespTPC, pidRespTOF>;
 
 // FIXME: this is a workaround until we get the index columns to work with joins.
 
diff --git a/Analysis/Tasks/PWGHF/HFD0CandidateSelector.cxx b/Analysis/Tasks/PWGHF/HFD0CandidateSelector.cxx
@@ -316,16 +316,16 @@ struct HFD0CandidateSelector {
     }
   }
 
-  void process(aod::HfCandProng2 const& hfCandProng2s, aod::BigTracks const& tracks)
+  void process(aod::HfCandProng2 const& hfCandProng2s, aod::BigTracksPID const& tracks)
   {
     int statusD0, statusD0bar; // final selection flag : 0-rejected  1-accepted
     bool topolD0, topolD0bar;
     int pidD0, pidD0bar, pionPlus, pionMinus, kaonPlus, kaonMinus;
 
     for (auto& hfCandProng2 : hfCandProng2s) { //looping over 2 prong candidates
 
-      auto trackPos = hfCandProng2.index0(); //positive daughter
-      auto trackNeg = hfCandProng2.index1(); //negative daughter
+      auto trackPos = hfCandProng2.index0_as<aod::BigTracksPID>(); //positive daughter
+      auto trackNeg = hfCandProng2.index1_as<aod::BigTracksPID>(); //negative daughter
 
       statusD0 = 0;
       statusD0bar = 0;
diff --git a/Analysis/Tasks/PWGHF/HFTrackIndexSkimsCreator.cxx b/Analysis/Tasks/PWGHF/HFTrackIndexSkimsCreator.cxx
@@ -169,6 +169,15 @@ struct HFTrackIndexSkimsCreator {
   OutputObj<TH1F> hvtx3_z{TH1F("hvtx3_z", "3-prong candidates;#it{z}_{sec. vtx.} (cm);entries", 1000, -20., 20.)};
   OutputObj<TH1F> hmass3{TH1F("hmass3", "3-prong candidates;inv. mass (#pi K #pi) (GeV/#it{c}^{2});entries", 500, 1.6, 2.1)};
 
+  /*
+  // Counter histograms
+  OutputObj<TH2F> hNCand2ProngVsNTracks{TH2F("hNCand2ProngVsNTracks", "2-prong candidates;# of tracks;# of candidates;entries", 1000, 0., 1000., 1000, 0., 10000.)};
+  OutputObj<TH2F> hNCand3ProngVsNTracks{TH2F("hNCand3ProngVsNTracks", "3-prong candidates;# of tracks;# of candidates;entries", 1000, 0., 1000., 1000, 0., 10000.)};
+  OutputObj<TH1F> hNCand2Prong{TH1F("hNCand2Prong", "2-prong candidates;# of candidates;entries", 1000, 0., 10000.)};
+  OutputObj<TH1F> hNCand3Prong{TH1F("hNCand3Prong", "3-prong candidates;# of candidates;entries", 1000, 0., 10000.)};
+  OutputObj<TH1F> hNTracks{TH1F("hNTracks", ";# of tracks;entries", 1000, 0., 1000.)};
+  */
+
   Filter filterSelectTracks = (aod::hf_seltrack::isSel2Prong == 1);
   using SelectedTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TracksCov, aod::TracksExtra, aod::HFSelTrack>>;
   // FIXME
@@ -214,6 +223,9 @@ struct HFTrackIndexSkimsCreator {
     df3.setMinRelChi2Change(d_minrelchi2change);
     df3.setUseAbsDCA(true);
 
+    //auto nCand2 = rowTrackIndexProng2.lastIndex();
+    //auto nCand3 = rowTrackIndexProng3.lastIndex();
+
     // first loop over positive tracks
     //for (auto trackPos1 = tracksPos.begin(); trackPos1 != tracksPos.end(); ++trackPos1) {
     for (auto trackPos1 = tracks.begin(); trackPos1 != tracks.end(); ++trackPos1) {
@@ -471,6 +483,16 @@ struct HFTrackIndexSkimsCreator {
         }
       }
     }
+    /*
+    auto nTracks = tracks.size(); // number of tracks in this collision
+    nCand2 = rowTrackIndexProng2.lastIndex() - nCand2; // number of 2-prong candidates in this collision
+    nCand3 = rowTrackIndexProng3.lastIndex() - nCand3; // number of 3-prong candidates in this collision
+    hNTracks->Fill(nTracks);
+    hNCand2Prong->Fill(nCand2);
+    hNCand3Prong->Fill(nCand3);
+    hNCand2ProngVsNTracks->Fill(nTracks, nCand2);
+    hNCand3ProngVsNTracks->Fill(nTracks, nCand3);
+    */
   }
 };
 

Original file line number	Diff line number	Diff line change
`@@ -316,16 +316,16 @@ struct HFD0CandidateSelector {`
`316`	`316`	`}`
`317`	`317`	`}`
`318`	`318`
`319`		`- void process(aod::HfCandProng2 const& hfCandProng2s, aod::BigTracks const& tracks)`
	`319`	`+ void process(aod::HfCandProng2 const& hfCandProng2s, aod::BigTracksPID const& tracks)`
`320`	`320`	`{`
`321`	`321`	`int statusD0, statusD0bar; // final selection flag : 0-rejected 1-accepted`
`322`	`322`	`bool topolD0, topolD0bar;`
`323`	`323`	`int pidD0, pidD0bar, pionPlus, pionMinus, kaonPlus, kaonMinus;`
`324`	`324`
`325`	`325`	`for (auto& hfCandProng2 : hfCandProng2s) { //looping over 2 prong candidates`
`326`	`326`
`327`		`- auto trackPos = hfCandProng2.index0(); //positive daughter`
`328`		`- auto trackNeg = hfCandProng2.index1(); //negative daughter`
	`327`	`+ auto trackPos = hfCandProng2.index0_as<aod::BigTracksPID>(); //positive daughter`
	`328`	`+ auto trackNeg = hfCandProng2.index1_as<aod::BigTracksPID>(); //negative daughter`
`329`	`329`
`330`	`330`	`statusD0 = 0;`
`331`	`331`	`statusD0bar = 0;`