Add a boolean flag to enable a more lightweight operation

mode. In this new mode, a big chunk of the output (debugging
info normally not needed for analysis) will be suppressed
to make outputs smaller and generally more manageable.

diff --git a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.cxx b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.cxx
index fd97893b178a1f6ccd607a15912637bf3b6679db..eccbc15a9e9f50ac3f1a4d2879a1a4d2a4f59a20 100644 (file)
--- a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.cxx
+++ b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.cxx
@@ -103,6 +103,7 @@ AliAnalysisTaskExtractPerformanceV0::AliAnalysisTaskExtractPerformanceV0()
    fkTakeAllTracks ( kFALSE ),
    fpArapidityShift ( 0.465 ),
   fCentralityEstimator("V0M"),
+  fkLightWeight  ( kFALSE ),  
 //------------------------------------------------
 // Tree Variables 
 
@@ -293,6 +294,7 @@ AliAnalysisTaskExtractPerformanceV0::AliAnalysisTaskExtractPerformanceV0(const c
    fkTakeAllTracks ( kFALSE ),
    fpArapidityShift ( 0.465 ),
   fCentralityEstimator("V0M"),
+  fkLightWeight  ( kFALSE ),  
 //------------------------------------------------
 // Tree Variables 
 
@@ -517,7 +519,7 @@ void AliAnalysisTaskExtractPerformanceV0::UserCreateOutputObjects()
 
 //-----------BASIC-INFO---------------------------
 /* 1*/   fTree->Branch("fTreeVariablePrimaryStatus",&fTreeVariablePrimaryStatus,"fTreeVariablePrimaryStatus/I");       
-/* 1*/   fTree->Branch("fTreeVariablePrimaryStatusMother",&fTreeVariablePrimaryStatusMother,"fTreeVariablePrimaryStatusMother/I");     
+/* 2*/   fTree->Branch("fTreeVariablePrimaryStatusMother",&fTreeVariablePrimaryStatusMother,"fTreeVariablePrimaryStatusMother/I");     
 /* 2*/   fTree->Branch("fTreeVariableChi2V0",&fTreeVariableChi2V0,"Chi2V0/F");
 /* 3*/   fTree->Branch("fTreeVariableDcaV0Daughters",&fTreeVariableDcaV0Daughters,"fTreeVariableDcaV0Daughters/F");
 /* 4*/   fTree->Branch("fTreeVariableDcaPosToPrimVertex",&fTreeVariableDcaPosToPrimVertex,"fTreeVariableDcaPosToPrimVertex/F");
@@ -558,14 +560,20 @@ void AliAnalysisTaskExtractPerformanceV0::UserCreateOutputObjects()
 /*34*/   fTree->Branch("fTreeVariableNegEta",&fTreeVariableNegEta,"fTreeVariableNegEta/F");
 /*35*/   fTree->Branch("fTreeVariablePosEta",&fTreeVariablePosEta,"fTreeVariablePosEta/F");
 /*36*/   fTree->Branch("fTreeVariableV0CreationRadius",&fTreeVariableV0CreationRadius,"fTreeVariableV0CreationRadius/F");
+  
+    if ( fkLightWeight == kFALSE ){ // these are debugging branches!
 /*37*/   fTree->Branch("fTreeVariableIndexStatus",&fTreeVariableIndexStatus,"fTreeVariableIndexStatus/I");
 /*38*/   fTree->Branch("fTreeVariableIndexStatusMother",&fTreeVariableIndexStatusMother,"fTreeVariableIndexStatusMother/I");
+    }
 
 /*39*/          fTree->Branch("fTreeVariableRunNumber",&fTreeVariableRunNumber,"fTreeVariableRunNumber/I");
 /*40*/   fTree->Branch("fTreeVariableEventNumber",&fTreeVariableEventNumber,"fTreeVariableEventNumber/l");
 
+    if ( fkLightWeight == kFALSE ){ // these are debugging branches!
 /*34*/   fTree->Branch("fTreeVariableVertexZ",&fTreeVariableVertexZ,"fTreeVariableVertexZ/F");
-
+    }
+      
+    if ( fkLightWeight == kFALSE ){ // these are debugging branches!
 //-----------FOR CTAU DEBUGGING: Full Phase Space + Decay Position Info 
         fTree->Branch("fTreeVariableV0x",&fTreeVariableV0x,"fTreeVariableV0x/F");
         fTree->Branch("fTreeVariableV0y",&fTreeVariableV0y,"fTreeVariableV0y/F");
@@ -592,11 +600,14 @@ void AliAnalysisTaskExtractPerformanceV0::UserCreateOutputObjects()
         fTree->Branch("fTreeVariableMCPVx",&fTreeVariableMCPVx,"fTreeVariableMCPVx/F");
         fTree->Branch("fTreeVariableMCPVy",&fTreeVariableMCPVy,"fTreeVariableMCPVy/F");
         fTree->Branch("fTreeVariableMCPVz",&fTreeVariableMCPVz,"fTreeVariableMCPVz/F");
+    }
 
         fTree->Branch("fTreeVariableIsNonInjected",&fTreeVariableIsNonInjected,"fTreeVariableIsNonInjected/O"); //O for bOOlean...
   
+  if ( fkLightWeight == kFALSE ){ // these are debugging branches!
         fTree->Branch("fTreeVariableNegTrackStatus",&fTreeVariableNegTrackStatus,"fTreeVariableNegTrackStatus/l");
         fTree->Branch("fTreeVariablePosTrackStatus",&fTreeVariablePosTrackStatus,"fTreeVariablePosTrackStatus/l");
+  }
   
 //------------------------------------------------
 // Particle Identification Setup

diff --git a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.h b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.h
index 97038988c1c4e13abdbd4a2bc1896a29ecc93482..cc1c7fe4fb0bceb16cb00280fdb7455e95abb054 100644 (file)
--- a/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.h
+++ b/PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractPerformanceV0.h
@@ -60,6 +60,7 @@ class AliAnalysisTaskExtractPerformanceV0 : public AliAnalysisTaskSE {
   void SetTakeAllTracks       (Bool_t lTakeAllTracks = kTRUE ) { fkTakeAllTracks = lTakeAllTracks; }
   void SetpARapidityShift     (Double_t lRapShift = 0.465 ) { fpArapidityShift = lRapShift; }
   void SetCentralityEstimator (TString lCentralityEstimator = "V0M" ) { fCentralityEstimator = lCentralityEstimator; }
+  void SetLightWeightAnalysis (Bool_t lLightWeight = kTRUE) {fkLightWeight = lLightWeight;  }
   
  private:
         // Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
@@ -80,6 +81,9 @@ class AliAnalysisTaskExtractPerformanceV0 : public AliAnalysisTaskSE {
   Bool_t fkTakeAllTracks; // if true, no TPC crossed rows and ratio cut
   Double_t fpArapidityShift; //pA rapidity shift (should be 0.465, usually)
   TString fCentralityEstimator; //Centrality Estimator String value (default V0M)
+  
+  Bool_t fkLightWeight; //if true, skip a number of debugging information branches in TTree
+                        //(to make resulting tree output significantly smaller!
 
 //===========================================================================================
 //   Variables for Tree, V0s