[u/mrichter/AliRoot.git] / PWG4 / CaloCalib / AliAnalysisTaskEMCALClusterize.h

#ifndef ALIANALYSISTASKEMCALCLUSTERIZE_H
#define ALIANALYSISTASKEMCALCLUSTERIZE_H

// This analysis provides a new list of clusters to be used in other analysis
// Author: Gustavo Conesa Balbastre,
//         Adapted from analysis class from Deepa Thomas

//Root
class TTree;
class TClonesArray;

//EMCAL
class AliEMCALGeometry;
class AliEMCALCalibData;
class AliCaloCalibPedestal;
class AliEMCALClusterizer;
class AliEMCALAfterBurnerUF;
class AliEMCALRecParam;
class AliEMCALRecoUtils;


#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEMCALClusterize : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskEMCALClusterize();
  AliAnalysisTaskEMCALClusterize(const char *name);
  virtual ~AliAnalysisTaskEMCALClusterize();

 private:  
  AliAnalysisTaskEMCALClusterize(const AliAnalysisTaskEMCALClusterize&); 
  AliAnalysisTaskEMCALClusterize& operator=(const AliAnalysisTaskEMCALClusterize&); // not implemented
  
 public:
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Init();
  virtual void   LocalInit()                                    { Init()                       ; }
  
  //OCDB
  Bool_t         AccessOCDB();
  void           SwitchOnAccessOCDB()                           { fAccessOCDB = kTRUE          ; }
  void           SwitchOffAccessOCDB()                          { fAccessOCDB = kFALSE         ; } 
  void           SetOCDBPath(const char *path)                  { fOCDBpath = path             ; }
  
  //Geometry methods
  void           SetGeometryName(TString &name)                 { fGeomName = name             ; }
  TString        GeometryName()                          const  { return fGeomName             ; }  
  void           SwitchOnLoadOwnGeometryMatrices()              { fLoadGeomMatrices = kTRUE    ; }
  void           SwitchOffLoadOwnGeometryMatrices()             { fLoadGeomMatrices = kFALSE   ; } 
  void           SetGeometryMatrixInSM(TGeoHMatrix* m, Int_t i) { fGeomMatrix[i]    = m        ; }

  //AOD methods
  void           SetAODBranchName(TString &name)                { fOutputAODBranchName = name  ; }
  void           FillAODFile(Bool_t yesno)                      { fFillAODFile         = yesno ; }
  void           FillAODCaloCells();
  void           FillAODHeader();
  void           SwitchOnFillAODHeader()                        { fFillAODHeader    = kTRUE    ; }
  void           SwitchOffFillAODHeader()                       { fFillAODHeader    = kFALSE   ; } 
  void           SwitchOnFillAODCaloCells()                     { fFillAODCaloCells = kTRUE    ; }
  void           SwitchOffFillAODCaloCells()                    { fFillAODCaloCells = kFALSE   ; } 
  
  //Algorithms settings
  void           JustUnfold(Bool_t yesno)                       { fJustUnfold          = yesno ; }
  AliEMCALRecParam * GetRecParam()                       const  { return fRecParam             ; }
  void           InitClusterization();
  
  void           SetEMCALRecoUtils(AliEMCALRecoUtils * ru)      { fRecoUtils           = ru    ; }
  AliEMCALRecoUtils* GetRecoUtils()                      const  { return fRecoUtils            ; }
  
  void           SetConfigFileName(TString name)                { fConfigName          = name  ; }
  void           SetMaxEvent(Int_t max)                         { fMaxEvent            = max   ; }
  
  void           SwitchOnTrackMatching()                        { fDoTrackMatching    = kTRUE  ; }
  void           SwitchOffTrackMatching()                       { fDoTrackMatching    = kFALSE ; } 

  void           SwitchOnCellEnergySelection()                  { fSelectCell         = kTRUE  ; }
  void           SwitchOffCellEnergySelection()                 { fSelectCell         = kFALSE ; } 

  
 private:
    
  virtual void  RecPoints2Clusters(TClonesArray *fdigitsArr, TObjArray *fRecPoints, TObjArray *clusArray);
  
  //Geometry  
  AliEMCALGeometry      *fGeom;             // EMCAL geometry
  TString                fGeomName;         // Name of geometry to use.
  TGeoHMatrix           *fGeomMatrix[10];   // Geometry matrices with alignments
  Bool_t                 fGeomMatrixSet;    // Set geometry matrices only once, for the first event.         
  Bool_t                 fLoadGeomMatrices; // Matrices set from configuration, not get from geometry.root or from ESDs/AODs

  //OCDB
  AliEMCALCalibData     *fCalibData;        // EMCAL calib data
  AliCaloCalibPedestal  *fPedestalData;     // EMCAL pedestal
  TString                fOCDBpath;         // Path with OCDB location
  Bool_t                 fAccessOCDB;       // Need to access info from OCDB (not really)   

  //Temporal arrays
  TClonesArray          *fDigitsArr;        //-> Digits array
  TObjArray             *fClusterArr;       //-> Recpoints array
  TObjArray             *fCaloClusterArr;   //-> CaloClusters array

  //Clusterizers 
  AliEMCALRecParam      *fRecParam;         // Reconstruction parameters container
  AliEMCALClusterizer   *fClusterizer;      //! EMCAL clusterizer
  AliEMCALAfterBurnerUF *fUnfolder;         //! Unfolding procedure
  Bool_t                 fJustUnfold;       // Just unfold, do not recluster
  
  //AOD
  TClonesArray          *fOutputAODBranch;  //! AOD Branch with output clusters  
  TString                fOutputAODBranchName;  // New of output AOD branch
  Bool_t                 fFillAODFile;      // Fill the output AOD file with the new clusters, 
                                            // if not they will be only available for the event they were generated
  Bool_t                 fFillAODHeader;    // Copy header to standard branch
  Bool_t                 fFillAODCaloCells; // Copy calocells to standard branch

  Int_t                  fRun;              //!run number
  
  AliEMCALRecoUtils*     fRecoUtils;        // Access to factorized reconstruction algorithms
  TString                fConfigName;       // Name of analysis configuration file
  
  Int_t                  fCellLabels[12672];       // Array with MC label to be passed to digit. 
  Int_t                  fCellSecondLabels[12672]; // Array with Second MC label to be passed to digit. 
  Double_t               fCellTime[12672];         // Array with cluster time to be passed to digit in case of AODs 

  Int_t                  fMaxEvent;         // Set a maximum event
  
  Bool_t                 fDoTrackMatching;  // On/Off the matching recalulation to speed up analysis in PbPb
  Bool_t                 fSelectCell;       // Reject cells from cluster if energy is too low and recalculate position/energy and other
  
  ClassDef(AliAnalysisTaskEMCALClusterize, 10);
};

#endif //ALIANALYSISTASKEMCALCLUSTERIZE_H
Commit	Line	Data
37d2296c	1	#ifndef ALIANALYSISTASKEMCALCLUSTERIZE_H
	2	#define ALIANALYSISTASKEMCALCLUSTERIZE_H
	3
	4	// This analysis provides a new list of clusters to be used in other analysis
	5	// Author: Gustavo Conesa Balbastre,
	6	// Adapted from analysis class from Deepa Thomas
	7
	8	//Root
	9	class TTree;
	10	class TClonesArray;
	11
	12	//EMCAL
	13	class AliEMCALGeometry;
	14	class AliEMCALCalibData;
	15	class AliCaloCalibPedestal;
	16	class AliEMCALClusterizer;
	17	class AliEMCALAfterBurnerUF;
	18	class AliEMCALRecParam;
	19	class AliEMCALRecoUtils;
	20
	21
	22	#include "AliAnalysisTaskSE.h"
	23
	24	class AliAnalysisTaskEMCALClusterize : public AliAnalysisTaskSE {
	25	public:
	26	AliAnalysisTaskEMCALClusterize();
	27	AliAnalysisTaskEMCALClusterize(const char *name);
	28	virtual ~AliAnalysisTaskEMCALClusterize();
	29
	30	private:
	31	AliAnalysisTaskEMCALClusterize(const AliAnalysisTaskEMCALClusterize&);
	32	AliAnalysisTaskEMCALClusterize& operator=(const AliAnalysisTaskEMCALClusterize&); // not implemented
	33
	34	public:
	35	virtual void UserCreateOutputObjects();
	36	virtual void UserExec(Option_t *option);
	37	virtual void Init();
	38	virtual void LocalInit() { Init() ; }
e9dd2d80	39
e9dd2d80	40	//OCDB
37d2296c	41	Bool_t AccessOCDB();
e9dd2d80	42	void SwitchOnAccessOCDB() { fAccessOCDB = kTRUE ; }
e9dd2d80	43	void SwitchOffAccessOCDB() { fAccessOCDB = kFALSE ; }
37d2296c	44	void SetOCDBPath(const char *path) { fOCDBpath = path ; }
	45
	46	//Geometry methods
	47	void SetGeometryName(TString &name) { fGeomName = name ; }
	48	TString GeometryName() const { return fGeomName ; }
	49	void SwitchOnLoadOwnGeometryMatrices() { fLoadGeomMatrices = kTRUE ; }
	50	void SwitchOffLoadOwnGeometryMatrices() { fLoadGeomMatrices = kFALSE ; }
	51	void SetGeometryMatrixInSM(TGeoHMatrix* m, Int_t i) { fGeomMatrix[i] = m ; }
	52
	53	//AOD methods
	54	void SetAODBranchName(TString &name) { fOutputAODBranchName = name ; }
	55	void FillAODFile(Bool_t yesno) { fFillAODFile = yesno ; }
1f77507b	56	void FillAODCaloCells();
	57	void FillAODHeader();
	58	void SwitchOnFillAODHeader() { fFillAODHeader = kTRUE ; }
	59	void SwitchOffFillAODHeader() { fFillAODHeader = kFALSE ; }
	60	void SwitchOnFillAODCaloCells() { fFillAODCaloCells = kTRUE ; }
	61	void SwitchOffFillAODCaloCells() { fFillAODCaloCells = kFALSE ; }
37d2296c	62
	63	//Algorithms settings
	64	void JustUnfold(Bool_t yesno) { fJustUnfold = yesno ; }
	65	AliEMCALRecParam * GetRecParam() const { return fRecParam ; }
	66	void InitClusterization();
	67
1f77507b	68	void SetEMCALRecoUtils(AliEMCALRecoUtils * ru) { fRecoUtils = ru ; }
37d2296c	69	AliEMCALRecoUtils* GetRecoUtils() const { return fRecoUtils ; }
37d2296c	70
e9dd2d80	71	void SetConfigFileName(TString name) { fConfigName = name ; }
e9dd2d80	72	void SetMaxEvent(Int_t max) { fMaxEvent = max ; }
37d2296c	73
cb3001b9	74	void SwitchOnTrackMatching() { fDoTrackMatching = kTRUE ; }
	75	void SwitchOffTrackMatching() { fDoTrackMatching = kFALSE ; }
	76
afaaef51	77	void SwitchOnCellEnergySelection() { fSelectCell = kTRUE ; }
	78	void SwitchOffCellEnergySelection() { fSelectCell = kFALSE ; }
	79
cb3001b9	80
37d2296c	81	private:
	82
	83	virtual void RecPoints2Clusters(TClonesArray fdigitsArr, TObjArray fRecPoints, TObjArray *clusArray);
	84
	85	//Geometry
	86	AliEMCALGeometry *fGeom; // EMCAL geometry
	87	TString fGeomName; // Name of geometry to use.
	88	TGeoHMatrix *fGeomMatrix[10]; // Geometry matrices with alignments
	89	Bool_t fGeomMatrixSet; // Set geometry matrices only once, for the first event.
	90	Bool_t fLoadGeomMatrices; // Matrices set from configuration, not get from geometry.root or from ESDs/AODs
	91
	92	//OCDB
	93	AliEMCALCalibData *fCalibData; // EMCAL calib data
	94	AliCaloCalibPedestal *fPedestalData; // EMCAL pedestal
	95	TString fOCDBpath; // Path with OCDB location
e9dd2d80	96	Bool_t fAccessOCDB; // Need to access info from OCDB (not really)
37d2296c	97
	98	//Temporal arrays
	99	TClonesArray *fDigitsArr; //-> Digits array
	100	TObjArray *fClusterArr; //-> Recpoints array
	101	TObjArray *fCaloClusterArr; //-> CaloClusters array
	102
	103	//Clusterizers
	104	AliEMCALRecParam *fRecParam; // Reconstruction parameters container
	105	AliEMCALClusterizer *fClusterizer; //! EMCAL clusterizer
	106	AliEMCALAfterBurnerUF *fUnfolder; //! Unfolding procedure
	107	Bool_t fJustUnfold; // Just unfold, do not recluster
	108
	109	//AOD
	110	TClonesArray *fOutputAODBranch; //! AOD Branch with output clusters
	111	TString fOutputAODBranchName; // New of output AOD branch
	112	Bool_t fFillAODFile; // Fill the output AOD file with the new clusters,
	113	// if not they will be only available for the event they were generated
1f77507b	114	Bool_t fFillAODHeader; // Copy header to standard branch
	115	Bool_t fFillAODCaloCells; // Copy calocells to standard branch
	116
37d2296c	117	Int_t fRun; //!run number
	118
	119	AliEMCALRecoUtils* fRecoUtils; // Access to factorized reconstruction algorithms
	120	TString fConfigName; // Name of analysis configuration file
	121
	122	Int_t fCellLabels[12672]; // Array with MC label to be passed to digit.
	123	Int_t fCellSecondLabels[12672]; // Array with Second MC label to be passed to digit.
a39f5b70	124	Double_t fCellTime[12672]; // Array with cluster time to be passed to digit in case of AODs
a39f5b70	125
1f77507b	126	Int_t fMaxEvent; // Set a maximum event
1f77507b	127
cb3001b9	128	Bool_t fDoTrackMatching; // On/Off the matching recalulation to speed up analysis in PbPb
afaaef51	129	Bool_t fSelectCell; // Reject cells from cluster if energy is too low and recalculate position/energy and other
cb3001b9	130
a39f5b70	131	ClassDef(AliAnalysisTaskEMCALClusterize, 10);
37d2296c	132	};
	133
	134	#endif //ALIANALYSISTASKEMCALCLUSTERIZE_H