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

#ifndef ALIANALYSISTASKEMCALCLUSTERIZEFAST_H
#define ALIANALYSISTASKEMCALCLUSTERIZEFAST_H

// $Id$

class TObjArray;
class TClonesArray;
class AliAODEvent;
class AliESDEvent;
class AliEMCALCalibData;
class AliCaloCalibPedestal;
class AliEMCALClusterizer;
class AliEMCALAfterBurnerUF;
class AliEMCALRecParam;
class AliEMCALRecoUtils;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEMCALClusterizeFast : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskEMCALClusterizeFast();
  AliAnalysisTaskEMCALClusterizeFast(const char *name);
  virtual ~AliAnalysisTaskEMCALClusterizeFast();
  
 public:
  virtual void           UserCreateOutputObjects();
  virtual void           UserExec(Option_t *option);

  Bool_t                 GetAttachClusters()                  const  { return fAttachClusters       ; }
  Bool_t                 GetRecalibrateOnly()                 const  { return fRecalibOnly          ; }
  Bool_t                 GetSubBackground()                   const  { return fSubBackground        ; }
  const TObjArray       *GetClusters()                        const  { return fClusterArr           ; }
  const TString         &GeometryName()                       const  { return fGeomName             ; }  
  AliEMCALRecParam      *GetRecParam()                        const  { return fRecParam             ; }
  AliEMCALRecoUtils     *GetRecoUtils()                       const  { return fRecoUtils            ; }
  AliEMCALCalibData     *GetCalibData()                       const  { return fCalibData            ; }
  AliCaloCalibPedestal  *GetPedData()                         const  { return fPedestalData         ; }
  TGeoHMatrix           *GetGeometryMatrix(Int_t i)           const  { return fGeomMatrix[i]        ; }
  void                   JustUnfold(Bool_t yesno)                    { fJustUnfold          = yesno ; }
  void                   LoadOwnGeometryMatrices(Bool_t b)           { fLoadGeomMatrices    = b     ; }
  void                   SetAODBranchName(const char *name)          { fOutputAODBrName     = name  ; }
  void                   SetAttachClusters(Bool_t b)                 { fAttachClusters      = b     ; }
  void                   SetCalibData(AliEMCALCalibData *d)          { fCalibData           = d     ; }
  void                   SetEMCALRecoUtils(AliEMCALRecoUtils *ru)    { fRecoUtils           = ru    ; }
  void                   SetGeometryMatrix(TGeoHMatrix* m, Int_t i)  { fGeomMatrix[i]       = m     ; }
  void                   SetGeometryName(const char *name)           { fGeomName            = name  ; }
  void                   SetLoadCalib(Bool_t b)                      { fLoadCalib           = b     ; }
  void                   SetLoadPed(Bool_t b)                        { fLoadPed             = b     ; }
  void                   SetOCDBPath(const char *path)               { fOCDBpath            = path  ; }
  void                   SetPedestalData(AliCaloCalibPedestal *d)    { fPedestalData        = d     ; }
  void                   SetRecalibrateCellsOnly(Bool_t b)           { fRecalibOnly         = b     ; }
  void                   SetSubBackground(Bool_t b)                  { fSubBackground       = b     ; }

 protected:
  void                   Clusterize();
  void                   FillDigitsArray();
  void                   Init();
  void                   RecPoints2Clusters(TClonesArray *clus);
  void                   SubBackground();
  void                   UpdateCells();
  void                   UpdateClusters();

  Int_t                  fRun;              //!run number
  TClonesArray          *fDigitsArr;        //!digits array
  TObjArray             *fClusterArr;       //!recpoints array
  AliEMCALRecParam      *fRecParam;         // reconstruction parameters container
  AliEMCALClusterizer   *fClusterizer;      //!clusterizer
  AliEMCALAfterBurnerUF *fUnfolder;         //!unfolding procedure
  Bool_t                 fJustUnfold;       // just unfold, do not recluster
  TString                fGeomName;         // name of geometry to use.
  Bool_t                 fGeomMatrixSet;    // set geometry matrices only once, for the first event.         
  Bool_t                 fLoadGeomMatrices; // matrices from configuration, not geometry.root nor ESDs/AODs
  TGeoHMatrix           *fGeomMatrix[12];   // geometry matrices with alignments
  TString                fOCDBpath;         // path with OCDB location
  AliEMCALCalibData     *fCalibData;        // EMCAL calib data
  AliCaloCalibPedestal  *fPedestalData;     // EMCAL pedestal
  TClonesArray          *fOutputAODBranch;  //!AOD Branch with output clusters  
  TString                fOutputAODBrName;  // output AOD branch name (none by default)
  AliEMCALRecoUtils     *fRecoUtils;        // access to factorized reconstruction algorithms
  Bool_t                 fLoadCalib;        // access calib object from OCDB (def=off)
  Bool_t                 fLoadPed;          // access ped object from OCDB (def=off)
  Bool_t                 fAttachClusters;   // attach clusters to input event (AOD or ESD)
  Bool_t                 fRecalibOnly;      // only recalibrate cells if true (def=off)
  Bool_t                 fSubBackground;    // subtract background if true (def=off)

 private:
  AliAnalysisTaskEMCALClusterizeFast(const AliAnalysisTaskEMCALClusterizeFast&);            // not implemented
  AliAnalysisTaskEMCALClusterizeFast &operator=(const AliAnalysisTaskEMCALClusterizeFast&); // not implemented

  ClassDef(AliAnalysisTaskEMCALClusterizeFast, 5);
};
#endif //ALIANALYSISTASKEMCALCLUSTERIZEFAST_H
Commit	Line	Data
2f7259cf	1	#ifndef ALIANALYSISTASKEMCALCLUSTERIZEFAST_H
	2	#define ALIANALYSISTASKEMCALCLUSTERIZEFAST_H
	3
	4	// $Id$
	5
	6	class TObjArray;
	7	class TClonesArray;
	8	class AliAODEvent;
	9	class AliESDEvent;
	10	class AliEMCALCalibData;
	11	class AliCaloCalibPedestal;
	12	class AliEMCALClusterizer;
	13	class AliEMCALAfterBurnerUF;
	14	class AliEMCALRecParam;
	15	class AliEMCALRecoUtils;
	16
	17	#include "AliAnalysisTaskSE.h"
	18
	19	class AliAnalysisTaskEMCALClusterizeFast : public AliAnalysisTaskSE {
	20	public:
2431b20f	21	AliAnalysisTaskEMCALClusterizeFast();
2431b20f	22	AliAnalysisTaskEMCALClusterizeFast(const char *name);
2f7259cf	23	virtual ~AliAnalysisTaskEMCALClusterizeFast();
	24
	25	public:
	26	virtual void UserCreateOutputObjects();
	27	virtual void UserExec(Option_t *option);
	28
dc503ae1	29	Bool_t GetAttachClusters() const { return fAttachClusters ; }
b0e354fe	30	Bool_t GetRecalibrateOnly() const { return fRecalibOnly ; }
95f69c66	31	Bool_t GetSubBackground() const { return fSubBackground ; }
2f7259cf	32	const TObjArray *GetClusters() const { return fClusterArr ; }
	33	const TString &GeometryName() const { return fGeomName ; }
	34	AliEMCALRecParam *GetRecParam() const { return fRecParam ; }
	35	AliEMCALRecoUtils *GetRecoUtils() const { return fRecoUtils ; }
dc503ae1	36	AliEMCALCalibData *GetCalibData() const { return fCalibData ; }
dc503ae1	37	AliCaloCalibPedestal *GetPedData() const { return fPedestalData ; }
94019119	38	TGeoHMatrix *GetGeometryMatrix(Int_t i) const { return fGeomMatrix[i] ; }
2f7259cf	39	void JustUnfold(Bool_t yesno) { fJustUnfold = yesno ; }
	40	void LoadOwnGeometryMatrices(Bool_t b) { fLoadGeomMatrices = b ; }
	41	void SetAODBranchName(const char *name) { fOutputAODBrName = name ; }
4cf9204b	42	void SetAttachClusters(Bool_t b) { fAttachClusters = b ; }
2f7259cf	43	void SetCalibData(AliEMCALCalibData *d) { fCalibData = d ; }
	44	void SetEMCALRecoUtils(AliEMCALRecoUtils *ru) { fRecoUtils = ru ; }
	45	void SetGeometryMatrix(TGeoHMatrix* m, Int_t i) { fGeomMatrix[i] = m ; }
	46	void SetGeometryName(const char *name) { fGeomName = name ; }
10d33986	47	void SetLoadCalib(Bool_t b) { fLoadCalib = b ; }
10d33986	48	void SetLoadPed(Bool_t b) { fLoadPed = b ; }
4cf9204b	49	void SetOCDBPath(const char *path) { fOCDBpath = path ; }
4cf9204b	50	void SetPedestalData(AliCaloCalibPedestal *d) { fPedestalData = d ; }
b0e354fe	51	void SetRecalibrateCellsOnly(Bool_t b) { fRecalibOnly = b ; }
95f69c66	52	void SetSubBackground(Bool_t b) { fSubBackground = b ; }
2f7259cf	53
5c5b84d6	54	protected:
95f69c66	55	void Clusterize();
95f69c66	56	void FillDigitsArray();
2f7259cf	57	void Init();
95f69c66	58	void RecPoints2Clusters(TClonesArray *clus);
	59	void SubBackground();
	60	void UpdateCells();
	61	void UpdateClusters();
2f7259cf	62
	63	Int_t fRun; //!run number
	64	TClonesArray *fDigitsArr; //!digits array
	65	TObjArray *fClusterArr; //!recpoints array
	66	AliEMCALRecParam *fRecParam; // reconstruction parameters container
	67	AliEMCALClusterizer *fClusterizer; //!clusterizer
	68	AliEMCALAfterBurnerUF *fUnfolder; //!unfolding procedure
	69	Bool_t fJustUnfold; // just unfold, do not recluster
	70	TString fGeomName; // name of geometry to use.
	71	Bool_t fGeomMatrixSet; // set geometry matrices only once, for the first event.
	72	Bool_t fLoadGeomMatrices; // matrices from configuration, not geometry.root nor ESDs/AODs
94019119	73	TGeoHMatrix *fGeomMatrix[12]; // geometry matrices with alignments
2f7259cf	74	TString fOCDBpath; // path with OCDB location
	75	AliEMCALCalibData *fCalibData; // EMCAL calib data
	76	AliCaloCalibPedestal *fPedestalData; // EMCAL pedestal
	77	TClonesArray *fOutputAODBranch; //!AOD Branch with output clusters
	78	TString fOutputAODBrName; // output AOD branch name (none by default)
	79	AliEMCALRecoUtils *fRecoUtils; // access to factorized reconstruction algorithms
10d33986	80	Bool_t fLoadCalib; // access calib object from OCDB (def=off)
10d33986	81	Bool_t fLoadPed; // access ped object from OCDB (def=off)
4cf9204b	82	Bool_t fAttachClusters; // attach clusters to input event (AOD or ESD)
b0e354fe	83	Bool_t fRecalibOnly; // only recalibrate cells if true (def=off)
95f69c66	84	Bool_t fSubBackground; // subtract background if true (def=off)
10d33986	85
5c5b84d6	86	private:
	87	AliAnalysisTaskEMCALClusterizeFast(const AliAnalysisTaskEMCALClusterizeFast&); // not implemented
	88	AliAnalysisTaskEMCALClusterizeFast &operator=(const AliAnalysisTaskEMCALClusterizeFast&); // not implemented
	89
95f69c66	90	ClassDef(AliAnalysisTaskEMCALClusterizeFast, 5);
2f7259cf	91	};
2f7259cf	92	#endif //ALIANALYSISTASKEMCALCLUSTERIZEFAST_H