[u/mrichter/AliRoot.git] / TOF / AliTOFCalibTask.h

#ifndef ALITOFCALIBTASK_H
#define ALITOFCALIBTASK_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

////////////////////////////////////////////
//    task for TOF calibration            //
//         C.Zampolli                     //
////////////////////////////////////////////

/* $Id$ */

#define TOFCHANNELS       160000   // number TOF channels 
#define CHENTRIES            500   // number of entries per TOF channel per run
                                   // (to be divided by 5 to get the 
                                   // real number of entries), bigarray 
#define CHENTRIESSMALL       300   // number of entries per TOF channel per run
                                   // (to be divided by 3 to get the 
                                   // real number of entries), smallarray 
#define MAXCHENTRIESSMALL  10000   // max number of entries per TOF channel 
                                   // (to be divided by 3 to get the 
                                   // real number of entries), smallarray 
				   
#define LOWERMOMBOUND        1.0   // [GeV/c] default value Pb-Pb
#define UPPERMOMBOUND        1.8   // [GeV/c] default value Pb-Pb
#define MINTIME              5E3   // min time of flight value (ns)
#define NIDX                   5   // number of values stored 
                                   // in big/smallarray per ESD track
#define NIDXSMALL              3   // number of values stored 
                                   // after Comb PID per ESD track
#define DELTAIDXTOT            0   // index for ToT in bigarray
#define DELTAIDXTIME           1   // index for TOF time in big/smallarray
#define DELTAIDXEXTIMEPI       2   // index for Exp Time Pi in bigarray
#define DELTAIDXEXTIMEKA       3   // index for Exp Time Ka in bigarray
#define DELTAIDXEXTIMEPR       4   // index for Exp Time Pr in bigarray
#define DELTAIDXPID            2   // index for Exp Time after Comb PID 
                                   // in smallarray
#define TRACKERROR       90*1E-3   // error on the tracks for 
                                   // Combinatorial PID (ns)
#define MEANENTRIES           15   // Mean number of entries per channel 
                                   // to perform calibration

#include "AliAnalysisTask.h"  

class TTree;
class AliESDtrack ;  
class TFile; 
class TH1F ;
class TH1I ;
class TH1D ;
class TH2F ;
class AliESD ; 

class AliTOFCalibTask : public AliAnalysisTask {

public:
  AliTOFCalibTask(const char *name) ; //ctor
  AliTOFCalibTask(const AliTOFCalibTask & calibtask); // copy constructor
  AliTOFCalibTask& operator=(const AliTOFCalibTask & calibtask); // assignment operator
  virtual ~AliTOFCalibTask(); //dtor
  virtual void Exec(Option_t * opt="") ;
  virtual void ConnectInputData(Option_t *) ;
  virtual void CreateOutputObjects();
  virtual void Terminate(Option_t * opt = "") ;

 public: 
  class AliTOFArrayTask : public TObject {
  public:
    AliTOFArrayTask(): TObject(),fSize(0),fArray(0x0){}
    AliTOFArrayTask(Int_t size) :
      TObject(),
      fSize(size),
      fArray(new TArrayF*[size]) {
    } 
    AliTOFArrayTask(const AliTOFArrayTask & source):
      TObject(),fSize(0),fArray(0x0){ // copy constructor
      this->fSize= source.fSize;
      this->fArray= source.fArray;
    };

    AliTOFArrayTask& operator=(const AliTOFArrayTask & source) { // assignment operator
      this->fSize= source.fSize;
      this->fArray= source.fArray;
      return *this;
    }

    Int_t GetSize() const {return fSize;}
    void AddArray(Int_t pos, TArrayF * parr) {
      if (pos>-1 && pos < fSize)fArray[pos] = parr;}
    TArrayF *  GetArray(Int_t pos) const {
      TArrayF * parr = 0x0; 
      if  (pos>-1 && pos < fSize)parr = fArray[pos];
      return parr;
    }
    virtual ~AliTOFArrayTask() {
      delete [] fArray;
    }
    
  private:
    
    Int_t fSize;       // Size of the array of TArrayFs
    TArrayF ** fArray; //[fSize]};
    
  };

private:
  Bool_t Select(AliESDtrack *t);
  Bool_t CombPID(Float_t *smallarray, Int_t size);
  void BookHistos();
  void DrawHistos();
  Int_t Calibrate(Option_t *optionSave="", Option_t *optionFit="RQ");
  Int_t Calibrate(Int_t nch,Int_t *ich, Option_t *optionSave="", Option_t *optionFit="RQ");
  Int_t Calibrate(Int_t ichmin, Int_t ichmax, Option_t *optionSave="", Option_t *optionFit="RQ");
  Int_t Calibrate(Int_t ich, Option_t *optionSave="", Option_t *optionFit="RQ");
  Int_t CalibrateFromProfile(Int_t ich, Option_t *optionSave="", Option_t *optionFit="RQ");
  TH1F* Profile(Int_t i);
  Int_t FindBins (TH1F* h, Double_t *bins) const;

  const Char_t *fdir;              // initial directory
  TTree* fChain ;            //!pointer to the analyzed TTree or TChain
  //leaf types
  AliESD* fESD ;              //! Declaration of leave types
  Int_t fMinEntries;            // minimum number of entries to steer the task
  Int_t fIch;                   // TOF channel number
  Float_t fToT;                 // Time over Threshold, ns
  Float_t fTime;                // TOF time, ns
  Float_t fExpTimePi;           // exp time, Pions, ns
  Float_t fExpTimeKa;           // exp time, Kaons, ns
  Float_t fExpTimePr;           // exp time, Protons, ns
  TTree* ftree;                 // tree for calibration
  Float_t fMinTime;             // min TOF time for track selection; not used
  Float_t** fbigarray;          // big array for calibration
  Int_t* findexarray;           // array for entry index in each channel 
  Int_t fnESD;                  // number of analyzed ESD tracks 
  Int_t fnESDselected;          // number of selected ESD tracks 
  Int_t fnESDkTOFout;           // number of ESD tracks with kTOFout
  Int_t fnESDkTIME;             // number of ESD tracks with kTIME
  Int_t fnESDassTOFcl;          // number of ESD tracks with assigned TOF cluster
  Int_t fnESDTIMEcut;           // number of ESD tracks with TOF time < 17 ns
  Int_t fnESDTRDcut;            // number of ESD tracks with TRD ok

  // Histos
  TH1F* fhToT;                  // ToT histo
  TH1F* fhTime;                 // Time histo
  TH1F* fhExpTimePi;            // Exp Time Pi histo                  
  TH1F* fhExpTimeKa;            // Exp Time Ka histo                  
  TH1F* fhExpTimePr;            // Exp Time Pr histo                  
  TH1I* fhPID;                  // PID histo                  
  TH1D* fhch;                   // TOF channel histo               

  TObjArray * fOutputContainer ; //! output data container

  ClassDef(AliTOFCalibTask, 1); //  TOF Calib task 
};
#endif // ALITOFCALIBTASK_H
Commit	Line	Data
1ed77580	1	#ifndef ALITOFCALIBTASK_H
	2	#define ALITOFCALIBTASK_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////
	8	// task for TOF calibration //
	9	// C.Zampolli //
	10	////////////////////////////////////////////
	11
	12	/* $Id$ */
	13
	14	#define TOFCHANNELS 160000 // number TOF channels
	15	#define CHENTRIES 500 // number of entries per TOF channel per run
	16	// (to be divided by 5 to get the
	17	// real number of entries), bigarray
	18	#define CHENTRIESSMALL 300 // number of entries per TOF channel per run
	19	// (to be divided by 3 to get the
	20	// real number of entries), smallarray
	21	#define MAXCHENTRIESSMALL 10000 // max number of entries per TOF channel
	22	// (to be divided by 3 to get the
	23	// real number of entries), smallarray
	24
	25	#define LOWERMOMBOUND 1.0 // [GeV/c] default value Pb-Pb
	26	#define UPPERMOMBOUND 1.8 // [GeV/c] default value Pb-Pb
	27	#define MINTIME 5E3 // min time of flight value (ns)
	28	#define NIDX 5 // number of values stored
	29	// in big/smallarray per ESD track
	30	#define NIDXSMALL 3 // number of values stored
	31	// after Comb PID per ESD track
	32	#define DELTAIDXTOT 0 // index for ToT in bigarray
	33	#define DELTAIDXTIME 1 // index for TOF time in big/smallarray
	34	#define DELTAIDXEXTIMEPI 2 // index for Exp Time Pi in bigarray
	35	#define DELTAIDXEXTIMEKA 3 // index for Exp Time Ka in bigarray
	36	#define DELTAIDXEXTIMEPR 4 // index for Exp Time Pr in bigarray
	37	#define DELTAIDXPID 2 // index for Exp Time after Comb PID
	38	// in smallarray
	39	#define TRACKERROR 90*1E-3 // error on the tracks for
	40	// Combinatorial PID (ns)
	41	#define MEANENTRIES 15 // Mean number of entries per channel
	42	// to perform calibration
	43
	44	#include "AliAnalysisTask.h"
	45
	46	class TTree;
	47	class AliESDtrack ;
	48	class TFile;
	49	class TH1F ;
	50	class TH1I ;
	51	class TH1D ;
	52	class TH2F ;
	53	class AliESD ;
	54
	55	class AliTOFCalibTask : public AliAnalysisTask {
	56
	57	public:
	58	AliTOFCalibTask(const char *name) ; //ctor
	59	AliTOFCalibTask(const AliTOFCalibTask & calibtask); // copy constructor
	60	AliTOFCalibTask& operator=(const AliTOFCalibTask & calibtask); // assignment operator
	61	virtual ~AliTOFCalibTask(); //dtor
	62	virtual void Exec(Option_t * opt="") ;
	63	virtual void ConnectInputData(Option_t *) ;
	64	virtual void CreateOutputObjects();
65	virtual void Terminate(Option_t * opt = "") ;
66
67	public:
68	class AliTOFArrayTask : public TObject {
69	public:
70	AliTOFArrayTask(): TObject(),fSize(0),fArray(0x0){}
71	AliTOFArrayTask(Int_t size) :
72	TObject(),
73	fSize(size),
74	fArray(new TArrayF*[size]) {
75	}
76	AliTOFArrayTask(const AliTOFArrayTask & source):
77	TObject(),fSize(0),fArray(0x0){ // copy constructor
78	this->fSize= source.fSize;
79	this->fArray= source.fArray;
80	};
81
82	AliTOFArrayTask& operator=(const AliTOFArrayTask & source) { // assignment operator
83	this->fSize= source.fSize;
84	this->fArray= source.fArray;
85	return *this;
86	}
87
88	Int_t GetSize() const {return fSize;}
89	void AddArray(Int_t pos, TArrayF * parr) {
90	if (pos>-1 && pos < fSize)fArray[pos] = parr;}
91	TArrayF * GetArray(Int_t pos) const {
92	TArrayF * parr = 0x0;
93	if (pos>-1 && pos < fSize)parr = fArray[pos];
94	return parr;
95	}
96	virtual ~AliTOFArrayTask() {
97	delete [] fArray;
98	}
99
100	private:
101
102	Int_t fSize; // Size of the array of TArrayFs
103	TArrayF ** fArray; //[fSize]};
104
105	};
106
107	private:
108	Bool_t Select(AliESDtrack *t);
109	Bool_t CombPID(Float_t *smallarray, Int_t size);
110	void BookHistos();
111	void DrawHistos();
112	Int_t Calibrate(Option_t optionSave="", Option_t optionFit="RQ");
113	Int_t Calibrate(Int_t nch,Int_t ich, Option_t optionSave="", Option_t *optionFit="RQ");
114	Int_t Calibrate(Int_t ichmin, Int_t ichmax, Option_t optionSave="", Option_t optionFit="RQ");
115	Int_t Calibrate(Int_t ich, Option_t optionSave="", Option_t optionFit="RQ");
116	Int_t CalibrateFromProfile(Int_t ich, Option_t optionSave="", Option_t optionFit="RQ");
117	TH1F* Profile(Int_t i);
118	Int_t FindBins (TH1F* h, Double_t *bins) const;
119
120	const Char_t *fdir; // initial directory
121	TTree* fChain ; //!pointer to the analyzed TTree or TChain
122	//leaf types
123	AliESD* fESD ; //! Declaration of leave types
124	Int_t fMinEntries; // minimum number of entries to steer the task
125	Int_t fIch; // TOF channel number
126	Float_t fToT; // Time over Threshold, ns
127	Float_t fTime; // TOF time, ns
128	Float_t fExpTimePi; // exp time, Pions, ns
129	Float_t fExpTimeKa; // exp time, Kaons, ns
130	Float_t fExpTimePr; // exp time, Protons, ns
131	TTree* ftree; // tree for calibration
132	Float_t fMinTime; // min TOF time for track selection; not used
133	Float_t** fbigarray; // big array for calibration
134	Int_t* findexarray; // array for entry index in each channel
135	Int_t fnESD; // number of analyzed ESD tracks
136	Int_t fnESDselected; // number of selected ESD tracks
137	Int_t fnESDkTOFout; // number of ESD tracks with kTOFout
138	Int_t fnESDkTIME; // number of ESD tracks with kTIME
139	Int_t fnESDassTOFcl; // number of ESD tracks with assigned TOF cluster
140	Int_t fnESDTIMEcut; // number of ESD tracks with TOF time < 17 ns
141	Int_t fnESDTRDcut; // number of ESD tracks with TRD ok
142
143	// Histos
144	TH1F* fhToT; // ToT histo
145	TH1F* fhTime; // Time histo
146	TH1F* fhExpTimePi; // Exp Time Pi histo
147	TH1F* fhExpTimeKa; // Exp Time Ka histo
148	TH1F* fhExpTimePr; // Exp Time Pr histo
149	TH1I* fhPID; // PID histo
150	TH1D* fhch; // TOF channel histo
151
152	TObjArray * fOutputContainer ; //! output data container
153
154	ClassDef(AliTOFCalibTask, 1); // TOF Calib task
155	};
156	#endif // ALITOFCALIBTASK_H
157