charge dependence of contamination (Alis Rodriguez Manso <alisrm@nikhef.nl>)

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.cxx b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.cxx
index 1f40e9296e8b263d9fe220b0dec6390e2a0339af..fb83612d1175d5348e4580075d04553ad87b7073 100644 (file)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.cxx
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.cxx
@@ -40,8 +40,10 @@ AliAnalysisTaskEffContBF::AliAnalysisTaskEffContBF(const char *name)
     fHistCentrality(0),
     fHistNMult(0), 
     fHistVz(0), 
-    fHistContaminationSecondaries(0),
-    fHistContaminationPrimaries(0),
+    fHistContaminationSecondariesPlus(0),
+    fHistContaminationSecondariesMinus(0), //
+    fHistContaminationPrimariesPlus(0),
+    fHistContaminationPrimariesMinus(0), //
     fHistGeneratedEtaPtPhiPlus(0), 
     fHistSurvivedEtaPtPhiPlus(0),
     fHistGeneratedEtaPtPhiMinus(0),
@@ -159,12 +161,18 @@ void AliAnalysisTaskEffContBF::UserCreateOutputObjects() {
   fQAList->Add(fHistVz);
 
   //Contamination for Secondaries 
-  fHistContaminationSecondaries = new TH3D("fHistContaminationSecondaries","Secondaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
-  fOutputList->Add(fHistContaminationSecondaries);
+  fHistContaminationSecondariesPlus = new TH3D("fHistContaminationSecondariesPlus","Secondaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
+  fOutputList->Add(fHistContaminationSecondariesPlus);
+
+  fHistContaminationSecondariesMinus = new TH3D("fHistContaminationSecondariesMinus","Secondaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
+  fOutputList->Add(fHistContaminationSecondariesMinus);
 
   //Contamination for Primaries
-  fHistContaminationPrimaries = new TH3D("fHistContaminationPrimaries","Primaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
-  fOutputList->Add(fHistContaminationPrimaries);
+  fHistContaminationPrimariesPlus = new TH3D("fHistContaminationPrimariesPlus","Primaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
+  fOutputList->Add(fHistContaminationPrimariesPlus);
+
+  fHistContaminationPrimariesMinus = new TH3D("fHistContaminationPrimariesMinus","Primaries;#eta;p_{T} (GeV/c);#varphi",etaBin,nArrayEta,ptBin,nArrayPt,phiBin,nArrayPhi);
+  fOutputList->Add(fHistContaminationPrimariesMinus);
   
   //eta vs pt for MC positives
   fHistGeneratedEtaPtPhiPlus = new TH3D("fHistGeneratedEtaPtPhiPlus",
@@ -379,15 +387,26 @@ void AliAnalysisTaskEffContBF::UserExec(Option_t *) {
                    Int_t label = TMath::Abs(track->GetLabel());
                    if(label > nMCParticles) continue;
                    AliAODMCParticle *AODmcTrack = (AliAODMCParticle*) mcEvent->GetTrack(label); 
+                   Short_t gAODmcCharge = AODmcTrack->Charge();////
                    //fHistContaminationPrimaries->Fill(track->Eta(),track->Pt(),phiDeg);
                    //if (!(AODmcTrack->IsPhysicalPrimary())) {
                    //fHistContaminationSecondaries->Fill(track->Eta(),track->Pt(),phiDeg);
                    //}
                    if (AODmcTrack->IsPhysicalPrimary()) {
-                     fHistContaminationPrimaries->Fill(track->Eta(),track->Pt(),phiDeg);
+                     if(gAODmcCharge > 0){
+                       fHistContaminationPrimariesPlus->Fill(track->Eta(),track->Pt(),phiDeg);
+                     }
+                     if(gAODmcCharge < 0){
+                       fHistContaminationPrimariesMinus->Fill(track->Eta(),track->Pt(),phiDeg);
+                     }
                    }
                    else{
-                     fHistContaminationSecondaries->Fill(track->Eta(),track->Pt(),phiDeg);
+                     if(gAODmcCharge > 0){
+                       fHistContaminationSecondariesPlus->Fill(track->Eta(),track->Pt(),phiDeg);
+                     }
+                     if(gAODmcCharge < 0){
+                       fHistContaminationSecondariesMinus->Fill(track->Eta(),track->Pt(),phiDeg);
+                     }
                    }
                  }
                  //++++++++++++++++++CONTAMINATION++++++++++++++++++//

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.h b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.h
index 334d88500bbba1d30a8df47c0af7828874df0bc9..71a83fe40886db7afb7d673af1fea9bc456918e7 100644 (file)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.h
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskEffContBF.h
@@ -30,8 +30,10 @@ class AliAnalysisTaskEffContBF : public AliAnalysisTaskSE {
     fHistCentrality(0), 
     fHistNMult(0), 
     fHistVz(0),
-    fHistContaminationSecondaries(0),
-    fHistContaminationPrimaries(0),
+    fHistContaminationSecondariesPlus(0),
+    fHistContaminationSecondariesMinus(0),
+    fHistContaminationPrimariesPlus(0),
+    fHistContaminationPrimariesMinus(0),
     fHistGeneratedEtaPtPhiPlus(0),
     fHistSurvivedEtaPtPhiPlus(0),
     fHistGeneratedEtaPtPhiMinus(0),
@@ -158,8 +160,11 @@ class AliAnalysisTaskEffContBF : public AliAnalysisTaskSE {
   TH1F        *fHistVz;//!
 
   // output histograms
-  TH3D        *fHistContaminationSecondaries;//!
-  TH3D        *fHistContaminationPrimaries;//!
+  TH3D        *fHistContaminationSecondariesPlus;//!
+  TH3D        *fHistContaminationSecondariesMinus;//!
+  TH3D        *fHistContaminationPrimariesPlus;//!
+  TH3D        *fHistContaminationPrimariesMinus;//!
+  
   // output histograms (single particles)
   TH3D        *fHistGeneratedEtaPtPhiPlus;//!correction map for positives (generated)
   TH3D        *fHistSurvivedEtaPtPhiPlus;//!correction map positives (survived)