Code reworking to analize ESD and AOD (H.Qvigstad)

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSEDs.cxx

diff --git a/PWGHF/vertexingHF/AliAnalysisTaskSEDs.cxx b/PWGHF/vertexingHF/AliAnalysisTaskSEDs.cxx
index 507f98a2c467d5c7fa23242616a7e8683624c54b..489960b705c35ba80f704f33bd50d8e4cbaebda4 100644 (file)
--- a/PWGHF/vertexingHF/AliAnalysisTaskSEDs.cxx
+++ b/PWGHF/vertexingHF/AliAnalysisTaskSEDs.cxx
@@ -53,6 +53,8 @@ AliAnalysisTaskSEDs::AliAnalysisTaskSEDs():
   fOutput(0), 
   fHistNEvents(0),
   fPtVsMass(0),
+  fPtVsMassPhi(0),
+  fPtVsMassK0st(0),
   fYVsPt(0),
   fYVsPtSig(0),
   fNtupleDs(0),
@@ -109,6 +111,8 @@ AliAnalysisTaskSEDs::AliAnalysisTaskSEDs(const char *name,AliRDHFCutsDstoKKpi* a
   fOutput(0),
   fHistNEvents(0),
   fPtVsMass(0),
+  fPtVsMassPhi(0),
+  fPtVsMassK0st(0),
   fYVsPt(0),
   fYVsPtSig(0),
   fNtupleDs(0),
@@ -226,6 +230,8 @@ AliAnalysisTaskSEDs::~AliAnalysisTaskSEDs()
   }
 
   delete fPtVsMass;
+  delete fPtVsMassPhi;
+  delete fPtVsMassK0st;
   delete fYVsPt;
   delete fYVsPtSig;
   delete fNtupleDs;
@@ -392,10 +398,14 @@ void AliAnalysisTaskSEDs::UserCreateOutputObjects()
   fOutput->Add(fHistNEvents);
 
   fPtVsMass=new TH2F("hPtVsMass","PtVsMass (prod. cuts)",nInvMassBins,minMass,maxMass,40,0.,20.);
+  fPtVsMassPhi=new TH2F("hPtVsMassPhi","PtVsMass (phi selection)",nInvMassBins,minMass,maxMass,200,0.,20.);
+  fPtVsMassK0st=new TH2F("hPtVsMassK0st","PtVsMass (K0* selection)",nInvMassBins,minMass,maxMass,200,0.,20.);
   fYVsPt=new TH2F("hYVsPt","YvsPt (prod. cuts)",40,0.,20.,80,-2.,2.);
   fYVsPtSig=new TH2F("hYVsPtSig","YvsPt (MC, only sig., prod. cuts)",40,0.,20.,80,-2.,2.);
 
   fOutput->Add(fPtVsMass);
+  fOutput->Add(fPtVsMassPhi);
+  fOutput->Add(fPtVsMassK0st);
   fOutput->Add(fYVsPt);
   fOutput->Add(fYVsPtSig);
 
@@ -409,7 +419,7 @@ void AliAnalysisTaskSEDs::UserCreateOutputObjects()
   if(fFillNtuple>0){
     OpenFile(4); // 4 is the slot number of the ntuple
     
-    fNtupleDs = new TNtuple("fNtupleDs","Ds","labDs:retcode:pdgcode0:Pt0:Pt1:Pt2:PtRec:P0:P1:P2:PidTrackBit0:PidTrackBit1:PidTrackBit2:PointingAngle:PointingAngleXY:DecLeng:DecLengXY:NorDecLeng:NorDecLengXY:InvMassKKpi:InvMasspiKK:sigvert:d00:d01:d02:dca:d0square:InvMassPhiKKpi:InvMassPhipiKK:InvMassK0starKKpi:InvMassK0starpiKK:cosinePiDsFrameKKpi:cosinePiDsFramepiKK:cosineKPhiFrameKKpi:cosineKPhiFramepiKK:centrality"); 
+    fNtupleDs = new TNtuple("fNtupleDs","Ds","labDs:retcode:pdgcode0:Pt0:Pt1:Pt2:PtRec:P0:P1:P2:PidTrackBit0:PidTrackBit1:PidTrackBit2:PointingAngle:PointingAngleXY:DecLeng:DecLengXY:NorDecLeng:NorDecLengXY:InvMassKKpi:InvMasspiKK:sigvert:d00:d01:d02:dca:d0square:InvMassPhiKKpi:InvMassPhipiKK:InvMassK0starKKpi:InvMassK0starpiKK:cosinePiDsFrameKKpi:cosinePiDsFramepiKK:cosineKPhiFrameKKpi:cosineKPhiFramepiKK:centrality:runNumber"); 
     
   }
   
@@ -470,7 +480,7 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
   if(fAnalysisCuts->IsEventRejectedDueToCentrality())fHistNEvents->Fill(7);
   
   Float_t centrality=fAnalysisCuts->GetCentrality(aod);
-  //Int_t runNumber=aod->GetRunNumber();
+  Int_t runNumber=aod->GetRunNumber();
 
   if(!isEvSel)return;
   
@@ -533,7 +543,7 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
     
     Double_t ptCand = d->Pt();
     Int_t iPtBin=TMath::BinarySearch(fNPtBins,fPtLimits,(Float_t)ptCand);
-    Int_t retCodeAnalysisCuts=fAnalysisCuts->IsSelected(d,AliRDHFCuts::kCandidate,aod);
+    Int_t retCodeAnalysisCuts=fAnalysisCuts->IsSelected(d,AliRDHFCuts::kAll,aod);
     Double_t rapid=d->YDs(); 
     fYVsPt->Fill(ptCand,rapid);
 
@@ -606,8 +616,14 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
       Double_t invMass=d->InvMassDsKKpi();
       fMassHist[index]->Fill(invMass);
       fPtVsMass->Fill(invMass,ptCand);
-      if(isPhiKKpi) fMassHistPhi[index]->Fill(invMass); 
-      if(isK0starKKpi) fMassHistK0st[index]->Fill(invMass);
+      if(isPhiKKpi){
+       fMassHistPhi[index]->Fill(invMass); 
+       fPtVsMassPhi->Fill(invMass,ptCand);
+      }
+      if(isK0starKKpi){
+       fMassHistK0st[index]->Fill(invMass);
+       fPtVsMassK0st->Fill(invMass,ptCand);
+      }
       if(fReadMC  && indexMCKKpi!=-1){
        fMassHist[indexMCKKpi]->Fill(invMass);
        if(isPhiKKpi) fMassHistPhi[indexMCKKpi]->Fill(invMass);
@@ -618,8 +634,14 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
       Double_t invMass=d->InvMassDspiKK();
       fMassHist[index]->Fill(invMass);
       fPtVsMass->Fill(invMass,ptCand);
-      if(isPhipiKK) fMassHistPhi[index]->Fill(invMass);
-      if(isK0starpiKK) fMassHistK0st[index]->Fill(invMass);
+      if(isPhipiKK){ 
+       fMassHistPhi[index]->Fill(invMass);
+       fPtVsMassPhi->Fill(invMass,ptCand);
+      }
+      if(isK0starpiKK){
+       fMassHistK0st[index]->Fill(invMass);
+       fPtVsMassK0st->Fill(invMass,ptCand);
+      }
       if(fReadMC  && indexMCpiKK!=-1){
        fMassHist[indexMCpiKK]->Fill(invMass);
        if(isPhipiKK) fMassHistPhi[indexMCpiKK]->Fill(invMass);
@@ -699,7 +721,7 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
    
     }
     
-    Float_t tmp[36];
+    Float_t tmp[37];
     if(fFillNtuple>0){
       
       if ((fFillNtuple==1 && (isPhiKKpi || isPhipiKK)) || (fFillNtuple==2 && (isK0starKKpi || isK0starpiKK)) || (fFillNtuple==3 && (isKKpi || ispiKK))){
@@ -748,9 +770,8 @@ void AliAnalysisTaskSEDs::UserExec(Option_t */*option*/)
            tmp[33]=d->CosPiKPhiRFrameKKpi();      
            tmp[34]=d->CosPiKPhiRFramepiKK();   
            tmp[35]=(Float_t)(centrality);
+           tmp[36]=(Float_t)(runNumber);       
        
-       
-       
            fNtupleDs->Fill(tmp);
            PostData(4,fNtupleDs);
       }