Merge branch 'master' of https://git.cern.ch/reps/AliRoot

[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / Cascades / AliAnalysisTaskStrangenessVsMultiplicityMC.cxx

diff --git a/PWGLF/STRANGENESS/Cascades/AliAnalysisTaskStrangenessVsMultiplicityMC.cxx b/PWGLF/STRANGENESS/Cascades/AliAnalysisTaskStrangenessVsMultiplicityMC.cxx
index c9d2da526d6e397b68703d731b593ab2d11c2174..757a78fd8bcce862da35bea7a8bc762f5f747a6f 100644 (file)
--- a/PWGLF/STRANGENESS/Cascades/AliAnalysisTaskStrangenessVsMultiplicityMC.cxx
+++ b/PWGLF/STRANGENESS/Cascades/AliAnalysisTaskStrangenessVsMultiplicityMC.cxx
@@ -167,6 +167,7 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
   fTreeVariableCentV0AEq(0),
   fTreeVariableCentV0CEq(0),
   fTreeVariableCentV0MEq(0),
+  fTreeVariableCustomCentV0M(0),
   fTreeVariableAmpV0A(0),
   fTreeVariableAmpV0C(0),
   fTreeVariableAmpV0AEq(0),
@@ -220,6 +221,7 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
        fTreeCascVarCentV0AEq(0),
        fTreeCascVarCentV0CEq(0),
        fTreeCascVarCentV0MEq(0),
+       fTreeCascVarCustomCentV0M(0),
        fTreeCascVarAmpV0A(0),
        fTreeCascVarAmpV0C(0),
        fTreeCascVarAmpV0AEq(0),
@@ -307,7 +309,9 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
        fHistPtVsAmpV0MEq_GenXiMinus(0),
        fHistPtVsAmpV0MEq_GenXiPlus(0),
        fHistPtVsAmpV0MEq_GenOmegaMinus(0),
-       fHistPtVsAmpV0MEq_GenOmegaPlus(0)  
+       fHistPtVsAmpV0MEq_GenOmegaPlus(0),
+       fHistVZEROResponseStudy(0),
+  fHistVZEROResponseStudyTotal(0)
 
 //------------------------------------------------
 // Tree Variables 
@@ -386,6 +390,7 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
   fTreeVariableCentV0AEq(0),
   fTreeVariableCentV0CEq(0),
   fTreeVariableCentV0MEq(0),
+  fTreeVariableCustomCentV0M(0),
   fTreeVariableAmpV0A(0),
   fTreeVariableAmpV0C(0),
   fTreeVariableAmpV0AEq(0),
@@ -439,6 +444,7 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
        fTreeCascVarCentV0AEq(0),
        fTreeCascVarCentV0CEq(0),
        fTreeCascVarCentV0MEq(0),
+  fTreeCascVarCustomCentV0M(0),
        fTreeCascVarAmpV0A(0),
        fTreeCascVarAmpV0C(0),
        fTreeCascVarAmpV0AEq(0),
@@ -526,7 +532,9 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
        fHistPtVsAmpV0MEq_GenXiMinus(0),
        fHistPtVsAmpV0MEq_GenXiPlus(0),
        fHistPtVsAmpV0MEq_GenOmegaMinus(0),
-       fHistPtVsAmpV0MEq_GenOmegaPlus(0)
+       fHistPtVsAmpV0MEq_GenOmegaPlus(0),
+       fHistVZEROResponseStudy(0),
+  fHistVZEROResponseStudyTotal(0)
 {
 
   //Re-vertex: Will only apply for cascade candidates
@@ -534,6 +542,7 @@ AliAnalysisTaskStrangenessVsMultiplicityMC::AliAnalysisTaskStrangenessVsMultipli
   fV0VertexerSels[0] =  33.  ;  // max allowed chi2
   fV0VertexerSels[1] =   0.02;  // min allowed impact parameter for the 1st daughter (LHC09a4 : 0.05)
   fV0VertexerSels[2] =   0.02;  // min allowed impact parameter for the 2nd daughter (LHC09a4 : 0.05)
+  fV0VertexerSels[3] =   2.0 ;  // max allowed DCA between the daughter tracks       (LHC09a4 : 0.5)
   fV0VertexerSels[4] =   0.95;  // min allowed cosine of V0's pointing angle         (LHC09a4 : 0.99)
   fV0VertexerSels[5] =   1.0 ;  // min radius of the fiducial volume                 (LHC09a4 : 0.2)
   fV0VertexerSels[6] = 200.  ;  // max radius of the fiducial volume                 (LHC09a4 : 100.0)
@@ -682,6 +691,7 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserCreateOutputObjects()
   fTreeV0->Branch("fTreeVariableCentV0AEq",&fTreeVariableCentV0AEq,"fTreeVariableCentV0AEq/F");
   fTreeV0->Branch("fTreeVariableCentV0CEq",&fTreeVariableCentV0CEq,"fTreeVariableCentV0CEq/F");
   fTreeV0->Branch("fTreeVariableCentV0MEq",&fTreeVariableCentV0MEq,"fTreeVariableCentV0MEq/F");
+  fTreeV0->Branch("fTreeVariableCustomCentV0M",&fTreeVariableCustomCentV0M,"fTreeVariableCustomCentV0M/F");
   fTreeV0->Branch("fTreeVariableAmpV0A",&fTreeVariableAmpV0A,"fTreeVariableAmpV0A/F");
   fTreeV0->Branch("fTreeVariableAmpV0C",&fTreeVariableAmpV0C,"fTreeVariableAmpV0C/F");
   fTreeV0->Branch("fTreeVariableAmpV0AEq",&fTreeVariableAmpV0AEq,"fTreeVariableAmpV0AEq/F");
@@ -739,6 +749,7 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserCreateOutputObjects()
   fTreeCascade->Branch("fTreeCascVarCentV0AEq",&fTreeCascVarCentV0AEq,"fTreeCascVarCentV0AEq/F");
   fTreeCascade->Branch("fTreeCascVarCentV0CEq",&fTreeCascVarCentV0CEq,"fTreeCascVarCentV0CEq/F");
   fTreeCascade->Branch("fTreeCascVarCentV0MEq",&fTreeCascVarCentV0MEq,"fTreeCascVarCentV0MEq/F");
+  fTreeCascade->Branch("fTreeCascVarCustomCentV0M",&fTreeCascVarCustomCentV0M,"fTreeCascVarCustomCentV0M/F");
   fTreeCascade->Branch("fTreeCascVarAmpV0A",&fTreeCascVarAmpV0A,"fTreeCascVarAmpV0A/F");
   fTreeCascade->Branch("fTreeCascVarAmpV0C",&fTreeCascVarAmpV0C,"fTreeCascVarAmpV0C/F");
   fTreeCascade->Branch("fTreeCascVarAmpV0AEq",&fTreeCascVarAmpV0AEq,"fTreeCascVarAmpV0AEq/F");
@@ -1045,6 +1056,16 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserCreateOutputObjects()
     "fHistPtVsAmpV0MEq_GenOmegaPlus",       "Generated;p_{T} (GeV/c); Mult",200,0,20,lAmplitudeBins,0,lMaxAmplitude);   
     fListHist->Add(fHistPtVsAmpV0MEq_GenOmegaPlus); }  
 
+  if(! fHistVZEROResponseStudy ) {
+    fHistVZEROResponseStudy    = new TH2D( 
+    "fHistVZEROResponseStudy",       "Generated;p_{T} (GeV/c); Mult",200,0,20,lAmplitudeBins,0,lMaxAmplitude);   
+    fListHist->Add(fHistVZEROResponseStudy); }  
+    
+  if(! fHistVZEROResponseStudyTotal ) {
+    fHistVZEROResponseStudyTotal    = new TH2D( 
+    "fHistVZEROResponseStudyTotal",       "Generated;p_{T} (GeV/c); Mult",5000,0,500,lAmplitudeBins,0,lMaxAmplitude);   
+    fListHist->Add(fHistVZEROResponseStudyTotal); } 
+
    //List of Histograms: Normal
    PostData(1, fListHist);
 
@@ -1236,6 +1257,9 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserExec(Option_t *)
   Long_t lNchVZEROA = 0; 
   Long_t lNchVZEROC = 0; 
 
+  Float_t lPtOfParticleInsideVZEROA = -1; 
+  Float_t lPOfParticleInsideVZEROA = -1; 
+
   //----- Loop on Stack ----------------------------------------------------------------
   for (Int_t iCurrentLabelStack = 0;  iCurrentLabelStack < (lMCstack->GetNtrack()); iCurrentLabelStack++) 
   {// This is the begining of the loop on tracks
@@ -1252,9 +1276,11 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserExec(Option_t *)
       if( TMath::Abs(geta) < 0.5 ) lNchEta5++; 
       if( TMath::Abs(geta) < 0.8 ) lNchEta8++; 
       if( 2.8 < geta && geta < 5.1 ) lNchVZEROA++; 
+      if( 2.8 < geta && geta < 5.1 ) lPtOfParticleInsideVZEROA = particleOne->Pt(); 
+      if( 2.8 < geta && geta < 5.1 ) lPOfParticleInsideVZEROA = particleOne->P(); 
       if(-3.7 < geta && geta <-1.7 ) lNchVZEROC++; 
   }//End of loop on tracks
-
+  
   //Attribution 
   fTrueMultEta5 = lNchEta5; 
   fTrueMultEta8 = lNchEta8; 
@@ -1290,6 +1316,9 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserExec(Option_t *)
   fAmplitude_V0A = multV0ACorr; 
   fAmplitude_V0C = multV0CCorr; 
 
+  if( fTrueMultVZEROA == 1 ) fHistVZEROResponseStudy->Fill( lPtOfParticleInsideVZEROA , fAmplitude_V0A ); 
+  if( fTrueMultVZEROA == 1 ) fHistVZEROResponseStudyTotal->Fill( lPOfParticleInsideVZEROA , fAmplitude_V0A ); 
+
   // Equalized signals // From AliCentralitySelectionTask
   for(Int_t iCh = 4; iCh < 7; ++iCh) {
     Double_t mult = lESDevent->GetVZEROEqMultiplicity(iCh);
@@ -1324,6 +1353,9 @@ void AliAnalysisTaskStrangenessVsMultiplicityMC::UserExec(Option_t *)
   fCustomCentrality_V0M = fPPVsMultUtils -> GetMultiplicityPercentile(lESDevent, "V0M");
   fCustomCentrality_V0MEq = fPPVsMultUtils -> GetMultiplicityPercentile(lESDevent, "V0MEq");
   
+  fTreeVariableCustomCentV0M = fCustomCentrality_V0M; 
+  fTreeCascVarCustomCentV0M  = fCustomCentrality_V0M; 
+  
   //Event-level fill 
   fTreeEvent->Fill();