Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliAnalysisTaskReducedTree.h

diff --git a/PWGDQ/dielectron/AliAnalysisTaskReducedTree.h b/PWGDQ/dielectron/AliAnalysisTaskReducedTree.h
index d8076b24c22a4e8cb6754d6408623bbd720e3a86..f64210b897b0e93ce1b6f0fdf24efa4c7fc3a1d8 100644 (file)
--- a/PWGDQ/dielectron/AliAnalysisTaskReducedTree.h
+++ b/PWGDQ/dielectron/AliAnalysisTaskReducedTree.h
@@ -52,6 +52,8 @@ public:
   void SetK0sCuts(AliESDv0Cuts* const cuts) {fK0sCuts = cuts;}
   void SetLambdaCuts(AliESDv0Cuts* const cuts) {fLambdaCuts = cuts;}
   void SetGammaConvCuts(AliESDv0KineCuts* const cuts) {fGammaConvCuts = cuts;}
+  void SetV0OpenCuts(AliESDv0KineCuts* const cuts) {fV0OpenCuts = cuts;}
+  void SetV0StrongCuts(AliESDv0KineCuts* const cuts) {fV0StrongCuts = cuts;}
   void SetK0sMassRange(Double_t min=0.4, Double_t max=0.6) {fK0sMassRange[0]=min; fK0sMassRange[1]=max;}
   void SetLambdaMassRange(Double_t min=1.08, Double_t max=1.15) {fLambdaMassRange[0]=min; fLambdaMassRange[1]=max;}
   void SetGammaConvMassRange(Double_t min=0.0, Double_t max=0.1) {fGammaMassRange[0]=min; fGammaMassRange[1]=max;}
@@ -66,6 +68,8 @@ public:
   void SetFillLambda(Bool_t flag=kTRUE)           {fFillLambda = flag;}
   void SetFillALambda(Bool_t flag=kTRUE)          {fFillALambda = flag;}
   void SetFillCaloClusterInfo(Bool_t flag=kTRUE)  {fFillCaloClusterInfo = flag;}
+  void SetFillFMDSectorInfo(Bool_t flag=kFALSE)   {fFillFMDSectorInfo = flag;}
+  void SetFillFMDChannelInfo(Bool_t flag=kFALSE)  {fFillFMDChannelInfo = flag;}
   void SetFillFriendInfo(Bool_t flag=kTRUE)       {fFillFriendInfo = flag;}
   
   // Add dielectron objects to the list. These contain cuts and histogram definitions
@@ -88,6 +92,8 @@ public:
   Bool_t fFillLambda;                // fill the lambda V0s
   Bool_t fFillALambda;               // fill the anti-lambda V0s
   Bool_t fFillCaloClusterInfo;       // fill the calorimeter clusters
+  Bool_t fFillFMDSectorInfo;         // fill the FMD info for every sector
+  Bool_t fFillFMDChannelInfo;        // fill the FMD info for every channel
   Bool_t fFillFriendInfo;            // fill friend tree information
 
   AliAnalysisCuts *fEventFilter;     // event filter
@@ -98,9 +104,13 @@ public:
   AliESDv0Cuts *fLambdaCuts;         // v0 standard filter for Lambda0->p + pi
   AliESDv0KineCuts *fGammaConvCuts;  // v0 standard filter for gamma conversions
   AliAnalysisCuts *fK0sPionCuts;     // filter for pions from K0s
-  AliAnalysisCuts *fLambdaProtonCuts;// filter for protons from Lambda
-  AliAnalysisCuts *fLambdaPionCuts;  // filter for pions from Lambda
+  AliAnalysisCuts *fLambdaProtonCuts;   // filter for protons from Lambda
+  AliAnalysisCuts *fLambdaPionCuts;     // filter for pions from Lambda
   AliAnalysisCuts *fGammaElectronCuts;  // filter for electrons from gamma conversions
+  AliESDv0KineCuts *fV0OpenCuts;       // v0 strong filter for tagged V0s
+  AliESDv0KineCuts *fV0StrongCuts;     // v0 strong filter for tagged V0s
+  
+    
   Double_t fK0sMassRange[2];         // mass range for allowed K0s pairs
   Double_t fLambdaMassRange[2];      // mass range for allowed Lambda pairs
   Double_t fGammaMassRange[2];       // mass range for allowed Gamma conversion pairs
@@ -119,12 +129,14 @@ public:
   void FillDielectronPairInfo(AliDielectron* die, Short_t iDie);  // fill dielectron reduced pair information
   void FillV0PairInfo();                    // fill V0 reduced pair information
   AliReducedPair* FillV0PairInfo(AliESDv0* v0, Int_t id, AliESDtrack* legPos, AliESDtrack* legNeg, AliKFVertex* vtxKF, Bool_t chargesAreCorrect);
-  UChar_t EncodeTPCClusterMap(AliESDtrack* track);
+  UChar_t EncodeTPCClusterMap(AliVParticle* track, Bool_t isAOD);
   void FillCaloClusters();
+  void FillFMDInfo();
+  Int_t GetSPDTrackletMultiplicity(AliVEvent* event, Float_t lowEta, Float_t highEta);
   
   AliAnalysisTaskReducedTree(const AliAnalysisTaskReducedTree &c);
   AliAnalysisTaskReducedTree& operator= (const AliAnalysisTaskReducedTree &c);
 
-  ClassDef(AliAnalysisTaskReducedTree, 2); //Analysis Task for creating a reduced event information tree 
+  ClassDef(AliAnalysisTaskReducedTree, 3); //Analysis Task for creating a reduced event information tree 
 };
 #endif