[u/mrichter/AliRoot.git] / TPC / AliTPCcalibTracks.h

#ifndef AliTPCCALIBTRACKS_H
#define AliTPCCALIBTRACKS_H


///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//     Class to analyse tracks for calibration                               //
//     to be used as a component in AliTPCSelectorTracks                     //
//     In the constructor you have to specify name and title                 //
//     to get the Object out of a file.                                      //
//     The parameter 'clusterParam', a AliTPCClusterParam object             //
//      (needed for TPC cluster error and shape parameterization)            //
//     Normally you get this object out of the file 'TPCClusterParam.root'   //
//     In the parameter 'cuts' the cuts are specified, that decide           //
//     weather a track will be accepted for calibration or not.              //
//                                                                           //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


#include <AliTPCcalibBase.h>
class TF2;
class TH3F;
class TH1F;
class TH1I;
class TH2I;
class TH2D;
class TCollection;
class TTreeSRedirector;
class TLinearFitter;
class AliTPCClusterParam;
class TTreeSRedirector;
class AliTPCROC;
class AliTPCseed;
class AliESDtrack;
class AliTPCclusterMI;
class AliTPCcalibTracksCuts;
class AliTPCCalPadRegion;
class AliTPCCalPad;
class TChain;
class TTree;
class TMutex;
class AliESDEvent;

using namespace std;

class AliTPCcalibTracks : public AliTPCcalibBase {
public :
   AliTPCcalibTracks();                         // default constructor
  AliTPCcalibTracks(const AliTPCcalibTracks&calibTracks); // copy constructor
  AliTPCcalibTracks(const Text_t *name, const Text_t *title, AliTPCClusterParam *clusterParam, AliTPCcalibTracksCuts* cuts, Int_t logLevel = 0);
  AliTPCcalibTracks & operator=(const AliTPCcalibTracks& calibTracks);
  
  virtual ~AliTPCcalibTracks();                // destructor
  
  virtual void            Process(AliTPCseed *track);  // to be called by the Selector
  void     Process(AliESDEvent *event) {AliTPCcalibBase::Process(event);};
  void     Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
  virtual Long64_t Merge(TCollection *li);
  void     MakeResPlotsQTree(Int_t minEntries = 100, const char* pathName = "plots");
  static void MakeQPosNormAll(TTree * chain, AliTPCClusterParam * param, Int_t maxPoints=1000000);
   void     MakeReport(Int_t stat, const char* pathName = "plots");     // calls all functions that procude pictures, results are written to pathName, stat is the minimal statistic threshold
  //

   
  Int_t           AcceptTrack(AliTPCseed * track);
  void            FillResolutionHistoLocal(AliTPCseed * track);  // the MAIN-FUNCTION, called for each track to fill the histograms, called by Process(...)
  static  TH2D*   MakeDiff(TH2D * hfit, TF2 * func);
  
  static AliTPCcalibTracks* TestMerge(AliTPCcalibTracks *ct, AliTPCClusterParam *clusterParam, Int_t nCalTracks = 50);
  
  void     SetStyle() const;
  void     Draw(Option_t* opt);                                  // draw some exemplaric histograms for fast result-check
  void     MakeAmpPlots(Int_t stat, const char* pathName = "plots");
  void     MakeDeltaPlots(const char* pathName = "plots");
  void     MakeChargeVsDriftLengthPlotsOld(const char* pathName = "plots");
  void     MakeChargeVsDriftLengthPlots(const char* pathName = "plots");
  void     FitResolutionNew(const char* pathName = "plots");
  void     FitRMSNew(const char* pathName = "plots");
  
  TObjArray* GetfArrayAmpRow() {return fArrayAmpRow;}
  TObjArray* GetfArrayAmp() {return fArrayAmp;}
  TObjArray* GetfArrayQDY() {return fArrayQDY;}
  TObjArray* GetfArrayQDZ() {return fArrayQDZ;}
  TObjArray* GetfArrayQRMSY() {return fArrayQRMSY;}
  TObjArray* GetfArrayQRMSZ() {return fArrayQRMSZ;}
  TObjArray* GetfArrayChargeVsDriftlength() {return fArrayChargeVsDriftlength;}
  TH1F*      GetfDeltaY() {return fDeltaY;}
  TH1F*      GetfDeltaZ() {return fDeltaZ;}
  TObjArray* GetfResolY() {return fResolY;}
  TObjArray* GetfResolZ() {return fResolZ;}
  TObjArray* GetfRMSY() {return fRMSY;}
  TObjArray* GetfRMSZ() {return fRMSZ;}
  TH1I*      GetfHclus() {return fHclus;}
  TH1I*      GetfRejectedTracksHisto() {return fRejectedTracksHisto;}
  TH1I*      GetfHclusterPerPadrow() {return fHclusterPerPadrow;}
  TH1I*      GetfHclusterPerPadrowRaw() {return fHclusterPerPadrowRaw;}
  TH2I*      GetfClusterCutHisto() {return fClusterCutHisto;}
  AliTPCCalPad*          GetfCalPadClusterPerPad() {return fCalPadClusterPerPad; }
  AliTPCCalPad*          GetfCalPadClusterPerPadRaw() {return fCalPadClusterPerPadRaw;}
  AliTPCCalPadRegion*    GetCalPadRegionchargeVsDriftlength() {return fcalPadRegionChargeVsDriftlength;}
  AliTPCcalibTracksCuts* GetCuts() {return fCuts;}
protected:         
  
private:

   static Int_t   GetBin(Float_t q, Int_t pad);
   static Int_t   GetBin(Int_t  iq, Int_t pad);
   static Float_t GetQ(Int_t bin);
   static Float_t GetPad(Int_t bin);
   void FillResolutionHistoLocalDebugPart(AliTPCseed *track, AliTPCclusterMI *cluster0, Int_t irow, Float_t  angley, Float_t  anglez, Int_t nclFound, Int_t kDelta);
   AliTPCClusterParam *fClusterParam; // pointer to cluster parameterization
   AliTPCROC *fROC;          //!
   TObjArray *fArrayAmpRow; // array with amplitudes versus row for given sector 
   TObjArray *fArrayAmp;    // array with amplitude for sectors
   TObjArray *fArrayQDY;    // q binned delta Y histograms
   TObjArray *fArrayQDZ;    // q binned delta Z histograms 
   TObjArray *fArrayQRMSY;  // q binned delta Y histograms
   TObjArray *fArrayQRMSZ;  // q binned delta Z histograms 
   TObjArray *fArrayChargeVsDriftlength; // array of arrays of TProfiles with charge vs. driftlength for each padsize and sector
   AliTPCCalPadRegion *fcalPadRegionChargeVsDriftlength;  // CalPadRegion, one TProfile for charge vs. driftlength for each padsize and sector
   TH1F      *fDeltaY;      // integrated delta y histo
   TH1F      *fDeltaZ;      // integrated delta z histo
   TObjArray *fResolY;      // array of resolution histograms Y
   TObjArray *fResolZ;      // array of resolution histograms Z
   TObjArray *fRMSY;        // array of RMS histograms Y
   TObjArray *fRMSZ;        // array of RMS histograms Z
   AliTPCcalibTracksCuts *fCuts; // object with cuts, that is passed to the constructor
   TH1I      *fHclus;       // number of clusters per track
   TH1I      *fRejectedTracksHisto; // histogram of rejecteced tracks, the number coresponds to the failed cut
   TH1I      *fHclusterPerPadrow;   // histogram showing the number of clusters per padRow
   TH1I      *fHclusterPerPadrowRaw;// histogram showing the number of clusters per padRow before cuts on clusters are applied
   TH2I      *fClusterCutHisto;     // histogram showing in which padRow the clusters were cutted by which criterium
   AliTPCCalPad *fCalPadClusterPerPad;    // AliTPCCalPad showing the number of clusters per Pad
   AliTPCCalPad *fCalPadClusterPerPadRaw; // AliTPCCalPad showing the number of clusters per Pad before cuts on clusters are applied
   ClassDef(AliTPCcalibTracks,1)
   
};

#endif
Commit	Line	Data
10757ee9	1	#ifndef AliTPCCALIBTRACKS_H
	2	#define AliTPCCALIBTRACKS_H
	3
	4
	5	///////////////////////////////////////////////////////////////////////////////
	6	// //
	7	// Class to analyse tracks for calibration //
	8	// to be used as a component in AliTPCSelectorTracks //
	9	// In the constructor you have to specify name and title //
	10	// to get the Object out of a file. //
	11	// The parameter 'clusterParam', a AliTPCClusterParam object //
	12	// (needed for TPC cluster error and shape parameterization) //
	13	// Normally you get this object out of the file 'TPCClusterParam.root' //
	14	// In the parameter 'cuts' the cuts are specified, that decide //
	15	// weather a track will be accepted for calibration or not. //
	16	// //
	17	// //
	18	///////////////////////////////////////////////////////////////////////////////
	19
	20
	21
c32da879	22	#include <AliTPCcalibBase.h>
10757ee9	23	class TF2;
	24	class TH3F;
	25	class TH1F;
	26	class TH1I;
	27	class TH2I;
	28	class TH2D;
	29	class TCollection;
	30	class TTreeSRedirector;
	31	class TLinearFitter;
	32	class AliTPCClusterParam;
	33	class TTreeSRedirector;
	34	class AliTPCROC;
	35	class AliTPCseed;
	36	class AliESDtrack;
	37	class AliTPCclusterMI;
	38	class AliTPCcalibTracksCuts;
	39	class AliTPCCalPadRegion;
	40	class AliTPCCalPad;
	41	class TChain;
9c171b96	42	class TTree;
10757ee9	43	class TMutex;
d11c5a19	44	class AliESDEvent;
10757ee9	45
	46	using namespace std;
	47
c32da879	48	class AliTPCcalibTracks : public AliTPCcalibBase {
10757ee9	49	public :
10757ee9	50	AliTPCcalibTracks(); // default constructor
b42cf9ed	51	AliTPCcalibTracks(const AliTPCcalibTracks&calibTracks); // copy constructor
9c171b96	52	AliTPCcalibTracks(const Text_t name, const Text_t title, AliTPCClusterParam clusterParam, AliTPCcalibTracksCuts cuts, Int_t logLevel = 0);
b42cf9ed	53	AliTPCcalibTracks & operator=(const AliTPCcalibTracks& calibTracks);
9c171b96	54
	55	virtual ~AliTPCcalibTracks(); // destructor
	56
0ffd2ae1	57	virtual void Process(AliTPCseed *track); // to be called by the Selector
d11c5a19	58	void Process(AliESDEvent *event) {AliTPCcalibBase::Process(event);};
d11c5a19	59	void Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
9c171b96	60	virtual Long64_t Merge(TCollection *li);
f78da5ae	61	void MakeResPlotsQTree(Int_t minEntries = 100, const char* pathName = "plots");
1bfaa9e9	62	static void MakeQPosNormAll(TTree * chain, AliTPCClusterParam * param, Int_t maxPoints=1000000);
f78da5ae	63	void MakeReport(Int_t stat, const char* pathName = "plots"); // calls all functions that procude pictures, results are written to pathName, stat is the minimal statistic threshold
9c171b96	64	//
10757ee9	65
9c171b96	66
	67	Int_t AcceptTrack(AliTPCseed * track);
	68	void FillResolutionHistoLocal(AliTPCseed * track); // the MAIN-FUNCTION, called for each track to fill the histograms, called by Process(...)
	69	static TH2D* MakeDiff(TH2D * hfit, TF2 * func);
	70
	71	static AliTPCcalibTracks* TestMerge(AliTPCcalibTracks ct, AliTPCClusterParam clusterParam, Int_t nCalTracks = 50);
	72
	73	void SetStyle() const;
	74	void Draw(Option_t* opt); // draw some exemplaric histograms for fast result-check
f78da5ae	75	void MakeAmpPlots(Int_t stat, const char* pathName = "plots");
	76	void MakeDeltaPlots(const char* pathName = "plots");
	77	void MakeChargeVsDriftLengthPlotsOld(const char* pathName = "plots");
	78	void MakeChargeVsDriftLengthPlots(const char* pathName = "plots");
	79	void FitResolutionNew(const char* pathName = "plots");
	80	void FitRMSNew(const char* pathName = "plots");
9c171b96	81
	82	TObjArray* GetfArrayAmpRow() {return fArrayAmpRow;}
	83	TObjArray* GetfArrayAmp() {return fArrayAmp;}
	84	TObjArray* GetfArrayQDY() {return fArrayQDY;}
	85	TObjArray* GetfArrayQDZ() {return fArrayQDZ;}
	86	TObjArray* GetfArrayQRMSY() {return fArrayQRMSY;}
	87	TObjArray* GetfArrayQRMSZ() {return fArrayQRMSZ;}
	88	TObjArray* GetfArrayChargeVsDriftlength() {return fArrayChargeVsDriftlength;}
	89	TH1F* GetfDeltaY() {return fDeltaY;}
	90	TH1F* GetfDeltaZ() {return fDeltaZ;}
	91	TObjArray* GetfResolY() {return fResolY;}
	92	TObjArray* GetfResolZ() {return fResolZ;}
	93	TObjArray* GetfRMSY() {return fRMSY;}
	94	TObjArray* GetfRMSZ() {return fRMSZ;}
	95	TH1I* GetfHclus() {return fHclus;}
	96	TH1I* GetfRejectedTracksHisto() {return fRejectedTracksHisto;}
	97	TH1I* GetfHclusterPerPadrow() {return fHclusterPerPadrow;}
	98	TH1I* GetfHclusterPerPadrowRaw() {return fHclusterPerPadrowRaw;}
	99	TH2I* GetfClusterCutHisto() {return fClusterCutHisto;}
	100	AliTPCCalPad* GetfCalPadClusterPerPad() {return fCalPadClusterPerPad; }
	101	AliTPCCalPad* GetfCalPadClusterPerPadRaw() {return fCalPadClusterPerPadRaw;}
	102	AliTPCCalPadRegion* GetCalPadRegionchargeVsDriftlength() {return fcalPadRegionChargeVsDriftlength;}
	103	AliTPCcalibTracksCuts* GetCuts() {return fCuts;}
10757ee9	104	protected:
10757ee9	105
10757ee9	106	private:
d11c5a19	107
10757ee9	108	static Int_t GetBin(Float_t q, Int_t pad);
	109	static Int_t GetBin(Int_t iq, Int_t pad);
	110	static Float_t GetQ(Int_t bin);
	111	static Float_t GetPad(Int_t bin);
	112	void FillResolutionHistoLocalDebugPart(AliTPCseed track, AliTPCclusterMI cluster0, Int_t irow, Float_t angley, Float_t anglez, Int_t nclFound, Int_t kDelta);
9f0beaf7	113	AliTPCClusterParam *fClusterParam; // pointer to cluster parameterization
10757ee9	114	AliTPCROC *fROC; //!
	115	TObjArray *fArrayAmpRow; // array with amplitudes versus row for given sector
	116	TObjArray *fArrayAmp; // array with amplitude for sectors
	117	TObjArray *fArrayQDY; // q binned delta Y histograms
	118	TObjArray *fArrayQDZ; // q binned delta Z histograms
	119	TObjArray *fArrayQRMSY; // q binned delta Y histograms
	120	TObjArray *fArrayQRMSZ; // q binned delta Z histograms
	121	TObjArray *fArrayChargeVsDriftlength; // array of arrays of TProfiles with charge vs. driftlength for each padsize and sector
	122	AliTPCCalPadRegion *fcalPadRegionChargeVsDriftlength; // CalPadRegion, one TProfile for charge vs. driftlength for each padsize and sector
	123	TH1F *fDeltaY; // integrated delta y histo
	124	TH1F *fDeltaZ; // integrated delta z histo
	125	TObjArray *fResolY; // array of resolution histograms Y
	126	TObjArray *fResolZ; // array of resolution histograms Z
	127	TObjArray *fRMSY; // array of RMS histograms Y
	128	TObjArray *fRMSZ; // array of RMS histograms Z
	129	AliTPCcalibTracksCuts *fCuts; // object with cuts, that is passed to the constructor
	130	TH1I *fHclus; // number of clusters per track
	131	TH1I *fRejectedTracksHisto; // histogram of rejecteced tracks, the number coresponds to the failed cut
	132	TH1I *fHclusterPerPadrow; // histogram showing the number of clusters per padRow
	133	TH1I *fHclusterPerPadrowRaw;// histogram showing the number of clusters per padRow before cuts on clusters are applied
	134	TH2I *fClusterCutHisto; // histogram showing in which padRow the clusters were cutted by which criterium
	135	AliTPCCalPad *fCalPadClusterPerPad; // AliTPCCalPad showing the number of clusters per Pad
	136	AliTPCCalPad *fCalPadClusterPerPadRaw; // AliTPCCalPad showing the number of clusters per Pad before cuts on clusters are applied
9c171b96	137	ClassDef(AliTPCcalibTracks,1)
10757ee9	138
10757ee9	139	};
	140
	141	#endif