Fix Coverity
[u/mrichter/AliRoot.git] / PWG4 / CaloCalib / AliAnalysisTaskCaloFilter.h
1 #ifndef ALIANALYSISTASKCALOFILTER_H
2 #define ALIANALYSISTASKCALOFILTER_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id: AliAnalysisTaskCaloFilter.h  $ */
8
9 //////////////////////////////////////////////////////////
10 // Filter the ESDCaloClusters and ESDCaloCells of EMCAL,
11 // PHOS or both, creating the corresponing AODCaloClusters
12 // and AODCaloCells.
13 // Keep also the AODHeader information and the vertex.
14 // Needed for calorimeter calibration.
15 // Copy of AliAnalysisTaskESDfilter.
16 // Author: Gustavo Conesa Balbastre (INFN - Frascati)
17 //////////////////////////////////////////////////////////
18
19 class TList;
20
21 #include "AliAnalysisTaskSE.h"
22 class AliEMCALRecoUtils;
23 class AliEMCALGeometry;
24 class AliESDtrackCuts;
25 class AliTriggerAnalysis;
26 class TNtuple;
27 class AliAnalysisTaskCaloFilter : public AliAnalysisTaskSE
28 {
29  public:
30   AliAnalysisTaskCaloFilter();
31   AliAnalysisTaskCaloFilter(const char* name);
32   virtual ~AliAnalysisTaskCaloFilter() ;
33   
34 private:
35   AliAnalysisTaskCaloFilter(const AliAnalysisTaskCaloFilter&);
36   AliAnalysisTaskCaloFilter& operator=(const AliAnalysisTaskCaloFilter&);
37   
38 public:
39   //General analysis frame methods
40   virtual void   UserCreateOutputObjects();
41   virtual void   Init();
42   virtual void   LocalInit() { Init() ; }
43   virtual void   UserExec(Option_t *option);
44   virtual void   Terminate(Option_t *option);
45       
46   //Geometry methods
47   void SetEMCALGeometryName(TString name)                  { fEMCALGeoName = name        ; }
48   TString EMCALGeometryName()                       const  { return fEMCALGeoName        ; }
49   void SwitchOnLoadOwnEMCALGeometryMatrices()              { fLoadEMCALMatrices = kTRUE  ; }
50   void SwitchOffLoadOwnEMCALGeometryMatrices()             { fLoadEMCALMatrices = kFALSE ; }
51   void SetEMCALGeometryMatrixInSM(TGeoHMatrix* m, Int_t i) { fEMCALMatrix[i]    = m      ; }
52   //void SwitchOnLoadOwnPHOSGeometryMatrices()               { fLoadPHOSMatrices = kTRUE  ; }
53   //void SwitchOffLoadOwnPHOSGeometryMatrices()              { fLoadPHOSMatrices = kFALSE ; }
54   //void SetPHOSGeometryMatrixInSM(TGeoHMatrix* m, Int_t i)  { fPHOSMatrix[i]    = m      ; }
55   
56   //Task settings
57   void    FillAODFile(Bool_t yesno)               { fFillAODFile = yesno    ; }
58   enum    caloFilter {kBoth = 0, kEMCAL = 1, kPHOS=2};
59   void    SetCaloFilter(Int_t calo)               { fCaloFilter = calo      ; }
60   TString GetCaloFilter()                  const  { return fCaloFilter      ; }  
61   
62   void SetEMCALRecoUtils(AliEMCALRecoUtils * ru)  { fEMCALRecoUtils = ru    ; }
63   AliEMCALRecoUtils* GetEMCALRecoUtils()   const  { return fEMCALRecoUtils  ; }
64
65   void    SwitchOnClusterCorrection()             { fCorrect = kTRUE        ; }
66   void    SwitchOffClusterCorrection()            { fCorrect = kFALSE       ; }
67   
68   //Event selection
69   AliESDtrackCuts* GetTrackCuts()          const  { return fESDtrackCuts    ; }
70   void    SetTrackCuts(AliESDtrackCuts * cuts)    { fESDtrackCuts = cuts    ; }           
71   Float_t GetTrackMultiplicityEtaCut()     const  { return fTrackMultEtaCut ; }
72   void    SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta  ; }         
73   virtual Bool_t CheckForPrimaryVertex();
74
75   void    PrintInfo();
76   
77   void    SetConfigFileName(TString name)         { fConfigName = name      ; }
78   
79 private:
80   
81   //TList* fCuts ;      //! List with analysis cuts
82   Int_t               fCaloFilter;        // Calorimeter to filter
83   Int_t               fCorrect;           // Recalibrate or recalculate different cluster parameters
84   //EMCAL specific
85   AliEMCALGeometry  * fEMCALGeo;          //! EMCAL geometry
86   TString             fEMCALGeoName;      // Name of geometry to use.
87   AliEMCALRecoUtils * fEMCALRecoUtils;    // Pointer to EMCAL utilities for clusterization
88
89   AliESDtrackCuts   * fESDtrackCuts;      // Track cut  
90   AliTriggerAnalysis* fTriggerAnalysis;   // Access to trigger selection algorithm for V0AND calculation
91   Float_t             fTrackMultEtaCut;   // Track multiplicity eta cut
92   
93   //Geometry
94   Bool_t              fLoadEMCALMatrices; // Matrices set from configuration, not get from geometry.root or from ESDs/AODs
95   TGeoHMatrix       * fEMCALMatrix[10];   // Geometry matrices with alignments
96   //Bool_t            fLoadPHOSMatrices;  // Matrices set from configuration, not get from geometry.root or from ESDs/AODs
97   //TGeoHMatrix *     fPHOSMatrix[5];     // Geometry matrices with alignments
98   Bool_t              fGeoMatrixSet;      // Set geometry matrices only once, for the first event.   
99
100   TNtuple           * fEventNtuple;       // NTuple with event parameters 
101   
102   TString             fConfigName;        // Name of analysis configuration file
103   Bool_t              fFillAODFile;       // Fill the output AOD file with clusters 
104   
105   ClassDef(AliAnalysisTaskCaloFilter, 6); // Analysis task for standard ESD filtering
106 };
107
108 #endif