1) Fixed typo. 2) Added DCA histograms. (Misha Veldhoen <Misha.Veldhoen@cern.ch>)

[u/mrichter/AliRoot.git] / PWGCF / Correlations / DPhi / DiHadronPID / AliAODEventCutsDiHadronPID.cxx

diff --git a/PWGCF/Correlations/DPhi/DiHadronPID/AliAODEventCutsDiHadronPID.cxx b/PWGCF/Correlations/DPhi/DiHadronPID/AliAODEventCutsDiHadronPID.cxx
index 1582467e0b1890ddb354bbd463ccd1892eacf560..f5ee2f15813373f183e76636cfa9303cca37b516 100644 (file)
--- a/PWGCF/Correlations/DPhi/DiHadronPID/AliAODEventCutsDiHadronPID.cxx
+++ b/PWGCF/Correlations/DPhi/DiHadronPID/AliAODEventCutsDiHadronPID.cxx
@@ -50,6 +50,7 @@ AliAODEventCutsDiHadronPID::AliAODEventCutsDiHadronPID():
        fMinRefMult(0),
        fTestTrigger(kFALSE),
        fTestCentrality(kFALSE),
+       fTestContributorsOrSPDVertex(kFALSE),
        fTestVertexZ(kFALSE),
        fTestMinRefMult(kFALSE),
        fSelectedEventQAHistos(0x0),
@@ -85,6 +86,7 @@ AliAODEventCutsDiHadronPID::AliAODEventCutsDiHadronPID(const char* name):
        fMinRefMult(0), 
        fTestTrigger(kFALSE),
        fTestCentrality(kFALSE),
+       fTestContributorsOrSPDVertex(kFALSE),
        fTestVertexZ(kFALSE),
        fTestMinRefMult(kFALSE),        
        fSelectedEventQAHistos(0x0),
@@ -299,6 +301,14 @@ Bool_t AliAODEventCutsDiHadronPID::IsSelected(AliAODEvent* event) {
        if (TMath::Abs(vtxz) > fMaxVertexZ) select = kFALSE; 
        }
 
+       // Test number of contributors. (very loose cut...)
+       if (fTestContributorsOrSPDVertex) {
+               Int_t nContributors = CurrentPrimaryVertex->GetNContributors();
+               if (nContributors < 1) {
+                       if (CurrentPrimaryVertex->GetType() != AliAODVertex::kMainSPD ) {select = kFALSE;}
+               }  
+       }
+
        // Get the event header.
        AliAODHeader* CurrentHeader = event->GetHeader();