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

[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / LambdaK0 / AliAnalysisTaskExtractV0pPb.cxx

diff --git a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractV0pPb.cxx b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractV0pPb.cxx
index 08f7f22c84c4de6e4b594a6fb1938a1d46c52925..05260308df20d0b55e9db21b0c66d63cb2162b09 100644 (file)
--- a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractV0pPb.cxx
+++ b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractV0pPb.cxx
@@ -100,6 +100,7 @@ AliAnalysisTaskExtractV0pPb::AliAnalysisTaskExtractV0pPb()
        fTreeVariableDcaV0ToPrimVertex(0),
        fTreeVariableDcaPosToPrimVertex(0),
        fTreeVariableDcaNegToPrimVertex(0),
+  fTreeVariableDCAV0ToPrimVertex(0),
        fTreeVariableV0CosineOfPointingAngle(0),
        fTreeVariableV0Radius(0),
        fTreeVariablePt(0),
@@ -153,6 +154,7 @@ AliAnalysisTaskExtractV0pPb::AliAnalysisTaskExtractV0pPb(const char *name)
        fTreeVariableDcaV0ToPrimVertex(0),
        fTreeVariableDcaPosToPrimVertex(0),
        fTreeVariableDcaNegToPrimVertex(0),
+  fTreeVariableDCAV0ToPrimVertex(0),
        fTreeVariableV0CosineOfPointingAngle(0),
        fTreeVariableV0Radius(0),
        fTreeVariablePt(0),
@@ -216,7 +218,7 @@ AliAnalysisTaskExtractV0pPb::~AliAnalysisTaskExtractV0pPb()
    }
    if (fTreeEvents){
       delete fTreeEvents;
-      fTree = 0x0;
+      fTreeEvents = 0x0;
    }
 }
 
@@ -247,6 +249,7 @@ void AliAnalysisTaskExtractV0pPb::UserCreateOutputObjects()
 /* 9*/ fTree->Branch("fTreeVariableInvMassK0s",&fTreeVariableInvMassK0s,"fTreeVariableInvMassK0s/F");
 /*10*/ fTree->Branch("fTreeVariableInvMassLambda",&fTreeVariableInvMassLambda,"fTreeVariableInvMassLambda/F");
 /*11*/ fTree->Branch("fTreeVariableInvMassAntiLambda",&fTreeVariableInvMassAntiLambda,"fTreeVariableInvMassAntiLambda/F");
+/*12*/ fTree->Branch("fTreeVariableDCAV0ToPrimVertex",&fTreeVariableDCAV0ToPrimVertex,"fTreeVariableDCAV0ToPrimVertex/F");
 /*12*/ fTree->Branch("fTreeVariableV0CosineOfPointingAngle",&fTreeVariableV0CosineOfPointingAngle,"fTreeVariableV0CosineOfPointingAngle/F");
 /*13*/ fTree->Branch("fTreeVariableAlphaV0",&fTreeVariableAlphaV0,"fTreeVariableAlphaV0/F");
 /*14*/ fTree->Branch("fTreeVariablePtArmV0",&fTreeVariablePtArmV0,"fTreeVariablePtArmV0/F");
@@ -268,7 +271,7 @@ void AliAnalysisTaskExtractV0pPb::UserCreateOutputObjects()
 //------------------------------------------------
 
    fTreeEvents = new TTree("fTreeEvents","Events");
-/*01*/ fTree->Branch("fTreeEventsCentrality",&fTreeEventsCentrality,"fTreeEventsCentrality/F");
+/*01*/ fTreeEvents->Branch("fTreeEventsCentrality",&fTreeEventsCentrality,"fTreeEventsCentrality/F");
 
 //------------------------------------------------
 // Particle Identification Setup
@@ -372,8 +375,7 @@ void AliAnalysisTaskExtractV0pPb::UserExec(Option_t *)
 
 
    //Set variable for filling tree afterwards!
-   //---> pp case......: GetReferenceMultiplicity
-   //---> Pb-Pb case...: Centrality by V0M
+   //---> Always V0A
   fTreeVariableCentrality = lMultiplicity;
   fTreeEventsCentrality = lMultiplicity;
  
@@ -471,6 +473,9 @@ void AliAnalysisTaskExtractV0pPb::UserExec(Option_t *)
   fHistCentralityVtxZ -> Fill ( fTreeVariableCentrality );  
   lMagneticField = lESDevent->GetMagneticField( );
 
+  //Fill Event Tree: Analysis Selection Level
+  fTreeEvents->Fill(); 
+
 //------------------------------------------------
 // MAIN LAMBDA LOOP STARTS HERE
 //------------------------------------------------
@@ -586,6 +591,7 @@ void AliAnalysisTaskExtractV0pPb::UserExec(Option_t *)
       lDcaV0Daughters = v0->GetDcaV0Daughters();
       lDcaV0ToPrimVertex = v0->GetD(tPrimaryVtxPosition[0],tPrimaryVtxPosition[1],tPrimaryVtxPosition[2]);
       lV0CosineOfPointingAngle = v0->GetV0CosineOfPointingAngle(tPrimaryVtxPosition[0],tPrimaryVtxPosition[1],tPrimaryVtxPosition[2]);
+      fTreeVariableDCAV0ToPrimVertex=lV0CosineOfPointingAngle;
       fTreeVariableV0CosineOfPointingAngle=lV0CosineOfPointingAngle;
 
       // Getting invariant mass infos directly from ESD