[u/mrichter/AliRoot.git] / TRD / AliTRDCalibTask.h

#ifndef ALITRDCALIBTASK_H
#define ALITRDCALIBTASK_H

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD calibration task for offline calibration                             //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


class TList;
class TObject;
class TObjArray;
class TH2F;
class TH1F;
class TH1I;
class TProfile2D;
class TH2I;
class TTree;
class AliESDEvent;
class AliESDfriend;
class AliESDtrack;
class AliESDfriendTrack;
class AliTRDtrackV1;
class AliTRDCalibraFillHisto;
class AliTRDcluster;
class AliESDtrackCuts;
class AliTRDCalDet;

#include "TObjString.h"
#include "AliAnalysisTaskSE.h" 
#include "TMath.h"

class AliTRDCalibTask : public AliAnalysisTaskSE {
 public:
  AliTRDCalibTask(const char *name = "AliTRDCalibTask");
  virtual ~AliTRDCalibTask();
  
  //  virtual void   ConnectInputData(Option_t *);
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *);
  virtual void   Terminate(Option_t *);
  virtual Bool_t Load(const Char_t *filename);
  virtual Bool_t Load(TList *lister);
  void           Plot();
  virtual Long64_t  Merge(TCollection *li);
  void           AddTask(const AliTRDCalibTask * calibTask);
  TList          *GetList() const {return fListHist;};

  void SetHisto2d(Bool_t histo2d)                                   {fHisto2d=histo2d;};
  void SetVector2d(Bool_t vector2d)                                 {fVector2d=vector2d;};
  void SetVdriftLinear(Bool_t vdriftLinear)                         {fVdriftLinear = vdriftLinear;};
  void SetNbTimeBins(Int_t nbTimeBins)                              {fNbTimeBins=nbTimeBins;};  
  
  void SetNz(Short_t nz, Int_t i)                                      {fNz[i]=nz;};
  void SetNrphi(Short_t nrphi, Int_t i)                                {fNrphi[i]=nrphi;};
  
  void AddSelectedTriggerClass(const char*name)                        {fSelectedTrigger->Add(new TObjString(name));};
  void SetReject(Bool_t rejected)                                      {fRejected = rejected;};
  
  void SetESDtrackCuts(AliESDtrackCuts * const esdtrackCuts)           {fEsdTrackCuts = esdtrackCuts;};
  void SetRequirePrimaryVertex(Bool_t requirePrimaryVertex)            {fRequirePrimaryVertex = requirePrimaryVertex;};
  void SetUseTPCVertex()                                               {fVtxTPC=kTRUE ; fVtxSPD=kFALSE;} 
  void SetUseSPDVertex()                                               {fVtxTPC=kFALSE; fVtxSPD=kTRUE ;} 
  void SetMinNbOfContributors(Int_t minNbOfContributors)               {fMinNbContributors = minNbOfContributors;};  
  void SetRangePrimaryVertexZ(Double_t rangePrimaryVertexZ)            {fRangePrimaryVertexZ = TMath::Abs(rangePrimaryVertexZ);};  
  void SetVersionGainUsed(Int_t versionGainUsed)                       { fVersionGainUsed = versionGainUsed;   }
  void SetSubVersionGainUsed(Int_t subVersionGainUsed)                 { fSubVersionGainUsed = subVersionGainUsed;   }
  void SetVersionVdriftUsed(Int_t versionVdriftUsed)                   { fVersionVdriftUsed = versionVdriftUsed;   }
  void SetSubVersionVdriftUsed(Int_t subVersionVdriftUsed)             { fSubVersionVdriftUsed = subVersionVdriftUsed;   }
  
  void SetLow(Int_t low)                                               {fLow=low;};
  void SetHigh(Int_t high)                                             {fHigh=high;};
  void SetFillZero(Bool_t fillZero)                                    {fFillZero =  fillZero;};
  void SetNormalizeNbOfCluster(Bool_t normalizeNbOfCluster =  kTRUE)   {fNormalizeNbOfCluster = normalizeNbOfCluster;};
  void SetMaxCluster(Float_t maxCluster)                               {fMaxCluster =  maxCluster; }; 
  void SetNbMaxCluster(Short_t nbMaxCluster)                           {fNbMaxCluster =  nbMaxCluster; }; 
  void SetOfflineTracks()                                              {fOfflineTracks=kTRUE; fStandaloneTracks=kFALSE; };
  void SetStandaloneTracks()                                           {fStandaloneTracks=kTRUE; fOfflineTracks=kFALSE; };
  
  void SetCalDetGain(AliTRDCalDet * const calDetGain)                  {fCalDetGain = calDetGain;};  

  void SetMaxEvent(Int_t nbevents)                                     { fMaxEvent = nbevents; };
  void SetDebug(Int_t debug)                                           { fDebug = debug; };

 private:
  AliESDEvent  *fESD;                            //! ESD object
  AliESDfriend *fESDfriend;                      //! ESD friend
  const AliESDtrack *fkEsdTrack;                  //! ESD track
  AliESDfriendTrack *fFriendTrack;               //! ESD friend track
  TObject *fCalibObject;                         //! calibration objects attached to the ESD friend
  AliTRDtrackV1 *fTrdTrack;                      //! trdtrack
  AliTRDcluster *fCl;                            //! cluster
  
  TList       *fListHist;                        //! list of histograms

  AliTRDCalibraFillHisto *fTRDCalibraFillHisto;  //! calibration analyse object

  TH1I        *fNEvents;                         //! counter  
  
  TH1F        *fNbTRDTrack;                      //! nb ESD tracks with TRD clusters
  TH1F        *fNbTRDTrackOffline;               //! nb ESD tracks with TRD clusters
  TH1F        *fNbTRDTrackStandalone;            //! nb ESD tracks with TRD clusters
  TH2F        *fNbTPCTRDtrack;                   //! nb TPC and TRD tracks when problems
   
  TH1F        *fNbTimeBin;                       //! nb Time Bin
  TH1F        *fNbTimeBinOffline;                //! nb Time Bin offline
  TH1F        *fNbTimeBinStandalone;             //! nb Time Bin standalone
  TH1F        *fNbClusters;                      //! nb Clusters
  TH1F        *fNbClustersOffline;               //! nb Clusters offline
  TH1F        *fNbClustersStandalone;            //! nb Clusters standalone
  TH1F        *fNbTracklets;                     //! nb Tracklets
  TH1F        *fNbTrackletsOffline;              //! nb Tracklets offline
  TH1F        *fNbTrackletsStandalone;           //! nb Tracklets standalone
  
  TH2F        *fAbsoluteGain;                    //! Absolute Gain without AliESDfriend
  TH2I        *fCH2dSum;                         //! CH2d charge all
  TProfile2D  *fPH2dSum;                         //! PH2d PH all
  TH2I        *fCH2dSM;                          //! CH2d per SM
  TProfile2D  *fPH2dSM;                          //! PH2d per SM

  Bool_t      fHisto2d;                          // histo
  Bool_t      fVector2d;                         // vector
  Bool_t      fVdriftLinear;                     // vdrift Linear

  Int_t       fNbTimeBins;                       // number of timebins 

  Short_t     fNz[3];                            // Nz mode 
  Short_t     fNrphi[3];                         // Nrphi mode
   
  TObjArray   *fSelectedTrigger;                 // Trigger class names accepted/rejected
  Bool_t      fRejected;                         // Reject the selected trigger class
  
  AliESDtrackCuts *fEsdTrackCuts;                // Quality cut on the AliESDtrack
  Bool_t      fRequirePrimaryVertex;             // Primary Vertex
  Bool_t      fVtxTPC;                           // Flag for use of TPC vertex
  Bool_t      fVtxSPD;                           // Flag for use of SPD vertex
  Int_t       fMinNbContributors;                // Min number of contributors
  Double_t    fRangePrimaryVertexZ;              // Were the primary vertex is
  
  Int_t       fLow;                              // lower limit of nb of TRD clusters per tracklet
  Int_t       fHigh;                             // higher limit of nb of TRD clusters per tracklet
  Bool_t      fFillZero;                         // fill zero
  Bool_t      fNormalizeNbOfCluster;             // normalize with number of clusters (per default not) 
  Float_t     fRelativeScale;                    // relative scale for gas gain
  Float_t     fMaxCluster;                       // Maxcluster 
  Short_t     fNbMaxCluster;                     // Number of tb at the end
  Bool_t      fOfflineTracks;                    // Only Offline refitted tracks
  Bool_t      fStandaloneTracks;                 // Take only standalone tracks

  Int_t       fVersionGainUsed;                  // VersionGainUsed 
  Int_t       fSubVersionGainUsed;               // SubVersionGainUsed
  Int_t       fVersionVdriftUsed;                // VersionVdriftUsed 
  Int_t       fSubVersionVdriftUsed;             // SubVersionVdriftUsed

  AliTRDCalDet *fCalDetGain;                     // Calib object gain

  Int_t       fMaxEvent;                         // max events
  Int_t       fCounter;                          // max events
  Int_t       fDebug;                            // fDebug

  AliTRDCalibTask(const AliTRDCalibTask&); 
  AliTRDCalibTask& operator=(const AliTRDCalibTask&); 

  ClassDef(AliTRDCalibTask, 1); 
};

#endif
Commit	Line	Data
6d3c5222	1	#ifndef ALITRDCALIBTASK_H
	2	#define ALITRDCALIBTASK_H
	3
	4	///////////////////////////////////////////////////////////////////////////////
	5	// //
	6	// TRD calibration task for offline calibration //
	7	// //
	8	///////////////////////////////////////////////////////////////////////////////
	9
	10
	11	class TList;
	12	class TObject;
	13	class TObjArray;
	14	class TH2F;
	15	class TH1F;
	16	class TH1I;
	17	class TProfile2D;
	18	class TH2I;
	19	class TTree;
	20	class AliESDEvent;
	21	class AliESDfriend;
	22	class AliESDtrack;
	23	class AliESDfriendTrack;
	24	class AliTRDtrackV1;
	25	class AliTRDCalibraFillHisto;
	26	class AliTRDcluster;
	27	class AliESDtrackCuts;
	28	class AliTRDCalDet;
	29
	30	#include "TObjString.h"
01239968	31	#include "AliAnalysisTaskSE.h"
98937691	32	#include "TMath.h"
6d3c5222	33
01239968	34	class AliTRDCalibTask : public AliAnalysisTaskSE {
6d3c5222	35	public:
	36	AliTRDCalibTask(const char *name = "AliTRDCalibTask");
	37	virtual ~AliTRDCalibTask();
	38
01239968	39	// virtual void ConnectInputData(Option_t *);
	40	virtual void UserCreateOutputObjects();
	41	virtual void UserExec(Option_t *);
6d3c5222	42	virtual void Terminate(Option_t *);
6d3c5222	43	virtual Bool_t Load(const Char_t *filename);
57185ead	44	virtual Bool_t Load(TList *lister);
6d3c5222	45	void Plot();
57185ead	46	virtual Long64_t Merge(TCollection *li);
	47	void AddTask(const AliTRDCalibTask * calibTask);
	48	TList *GetList() const {return fListHist;};
6d3c5222	49
	50	void SetHisto2d(Bool_t histo2d) {fHisto2d=histo2d;};
	51	void SetVector2d(Bool_t vector2d) {fVector2d=vector2d;};
	52	void SetVdriftLinear(Bool_t vdriftLinear) {fVdriftLinear = vdriftLinear;};
	53	void SetNbTimeBins(Int_t nbTimeBins) {fNbTimeBins=nbTimeBins;};
	54
	55	void SetNz(Short_t nz, Int_t i) {fNz[i]=nz;};
	56	void SetNrphi(Short_t nrphi, Int_t i) {fNrphi[i]=nrphi;};
	57
	58	void AddSelectedTriggerClass(const char*name) {fSelectedTrigger->Add(new TObjString(name));};
	59	void SetReject(Bool_t rejected) {fRejected = rejected;};
	60
98937691	61	void SetESDtrackCuts(AliESDtrackCuts * const esdtrackCuts) {fEsdTrackCuts = esdtrackCuts;};
6d3c5222	62	void SetRequirePrimaryVertex(Bool_t requirePrimaryVertex) {fRequirePrimaryVertex = requirePrimaryVertex;};
	63	void SetUseTPCVertex() {fVtxTPC=kTRUE ; fVtxSPD=kFALSE;}
	64	void SetUseSPDVertex() {fVtxTPC=kFALSE; fVtxSPD=kTRUE ;}
	65	void SetMinNbOfContributors(Int_t minNbOfContributors) {fMinNbContributors = minNbOfContributors;};
98937691	66	void SetRangePrimaryVertexZ(Double_t rangePrimaryVertexZ) {fRangePrimaryVertexZ = TMath::Abs(rangePrimaryVertexZ);};
4c865c34	67	void SetVersionGainUsed(Int_t versionGainUsed) { fVersionGainUsed = versionGainUsed; }
	68	void SetSubVersionGainUsed(Int_t subVersionGainUsed) { fSubVersionGainUsed = subVersionGainUsed; }
	69	void SetVersionVdriftUsed(Int_t versionVdriftUsed) { fVersionVdriftUsed = versionVdriftUsed; }
	70	void SetSubVersionVdriftUsed(Int_t subVersionVdriftUsed) { fSubVersionVdriftUsed = subVersionVdriftUsed; }
6d3c5222	71
	72	void SetLow(Int_t low) {fLow=low;};
	73	void SetHigh(Int_t high) {fHigh=high;};
	74	void SetFillZero(Bool_t fillZero) {fFillZero = fillZero;};
	75	void SetNormalizeNbOfCluster(Bool_t normalizeNbOfCluster = kTRUE) {fNormalizeNbOfCluster = normalizeNbOfCluster;};
	76	void SetMaxCluster(Float_t maxCluster) {fMaxCluster = maxCluster; };
	77	void SetNbMaxCluster(Short_t nbMaxCluster) {fNbMaxCluster = nbMaxCluster; };
	78	void SetOfflineTracks() {fOfflineTracks=kTRUE; fStandaloneTracks=kFALSE; };
	79	void SetStandaloneTracks() {fStandaloneTracks=kTRUE; fOfflineTracks=kFALSE; };
4c865c34	80
4c865c34	81	void SetCalDetGain(AliTRDCalDet * const calDetGain) {fCalDetGain = calDetGain;};
6d3c5222	82
	83	void SetMaxEvent(Int_t nbevents) { fMaxEvent = nbevents; };
	84	void SetDebug(Int_t debug) { fDebug = debug; };
	85
	86	private:
	87	AliESDEvent *fESD; //! ESD object
	88	AliESDfriend *fESDfriend; //! ESD friend
	89	const AliESDtrack *fkEsdTrack; //! ESD track
	90	AliESDfriendTrack *fFriendTrack; //! ESD friend track
	91	TObject *fCalibObject; //! calibration objects attached to the ESD friend
	92	AliTRDtrackV1 *fTrdTrack; //! trdtrack
	93	AliTRDcluster *fCl; //! cluster
	94
	95	TList *fListHist; //! list of histograms
	96
	97	AliTRDCalibraFillHisto *fTRDCalibraFillHisto; //! calibration analyse object
	98
	99	TH1I *fNEvents; //! counter
	100
	101	TH1F *fNbTRDTrack; //! nb ESD tracks with TRD clusters
	102	TH1F *fNbTRDTrackOffline; //! nb ESD tracks with TRD clusters
	103	TH1F *fNbTRDTrackStandalone; //! nb ESD tracks with TRD clusters
	104	TH2F *fNbTPCTRDtrack; //! nb TPC and TRD tracks when problems
	105
	106	TH1F *fNbTimeBin; //! nb Time Bin
	107	TH1F *fNbTimeBinOffline; //! nb Time Bin offline
	108	TH1F *fNbTimeBinStandalone; //! nb Time Bin standalone
	109	TH1F *fNbClusters; //! nb Clusters
	110	TH1F *fNbClustersOffline; //! nb Clusters offline
	111	TH1F *fNbClustersStandalone; //! nb Clusters standalone
	112	TH1F *fNbTracklets; //! nb Tracklets
	113	TH1F *fNbTrackletsOffline; //! nb Tracklets offline
	114	TH1F *fNbTrackletsStandalone; //! nb Tracklets standalone
	115
98937691	116	TH2F *fAbsoluteGain; //! Absolute Gain without AliESDfriend
6d3c5222	117	TH2I *fCH2dSum; //! CH2d charge all
	118	TProfile2D *fPH2dSum; //! PH2d PH all
	119	TH2I *fCH2dSM; //! CH2d per SM
	120	TProfile2D *fPH2dSM; //! PH2d per SM
	121
98937691	122	Bool_t fHisto2d; // histo
	123	Bool_t fVector2d; // vector
	124	Bool_t fVdriftLinear; // vdrift Linear
6d3c5222	125
98937691	126	Int_t fNbTimeBins; // number of timebins
6d3c5222	127
	128	Short_t fNz[3]; // Nz mode
	129	Short_t fNrphi[3]; // Nrphi mode
	130
	131	TObjArray *fSelectedTrigger; // Trigger class names accepted/rejected
	132	Bool_t fRejected; // Reject the selected trigger class
	133
	134	AliESDtrackCuts *fEsdTrackCuts; // Quality cut on the AliESDtrack
	135	Bool_t fRequirePrimaryVertex; // Primary Vertex
	136	Bool_t fVtxTPC; // Flag for use of TPC vertex
	137	Bool_t fVtxSPD; // Flag for use of SPD vertex
	138	Int_t fMinNbContributors; // Min number of contributors
98937691	139	Double_t fRangePrimaryVertexZ; // Were the primary vertex is
6d3c5222	140
	141	Int_t fLow; // lower limit of nb of TRD clusters per tracklet
	142	Int_t fHigh; // higher limit of nb of TRD clusters per tracklet
	143	Bool_t fFillZero; // fill zero
	144	Bool_t fNormalizeNbOfCluster; // normalize with number of clusters (per default not)
	145	Float_t fRelativeScale; // relative scale for gas gain
	146	Float_t fMaxCluster; // Maxcluster
	147	Short_t fNbMaxCluster; // Number of tb at the end
	148	Bool_t fOfflineTracks; // Only Offline refitted tracks
	149	Bool_t fStandaloneTracks; // Take only standalone tracks
	150
4c865c34	151	Int_t fVersionGainUsed; // VersionGainUsed
	152	Int_t fSubVersionGainUsed; // SubVersionGainUsed
	153	Int_t fVersionVdriftUsed; // VersionVdriftUsed
	154	Int_t fSubVersionVdriftUsed; // SubVersionVdriftUsed
	155
6d3c5222	156	AliTRDCalDet *fCalDetGain; // Calib object gain
	157
	158	Int_t fMaxEvent; // max events
	159	Int_t fCounter; // max events
98937691	160	Int_t fDebug; // fDebug
6d3c5222	161
	162	AliTRDCalibTask(const AliTRDCalibTask&);
	163	AliTRDCalibTask& operator=(const AliTRDCalibTask&);
	164
	165	ClassDef(AliTRDCalibTask, 1);
	166	};
	167
21185a48	168	#endif
01239968	169