[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;

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
  virtual Long64_t Merge(TCollection *li);
  void     MakeResPlotsQTree(Int_t minEntries = 100, char* pathName = "plots");
  static void MakeQPosNormAll(TTree * chain, AliTPCClusterParam * param, Int_t maxPoints=1000000);
   void     MakeReport(Int_t stat, 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, char* pathName = "plots");
  void     MakeDeltaPlots(char* pathName = "plots");
  void     MakeChargeVsDriftLengthPlotsOld(char* pathName = "plots");
  void     MakeChargeVsDriftLengthPlots(char* pathName = "plots");
  void     FitResolutionNew(char* pathName = "plots");
  void     FitRMSNew(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;
	44
	45	using namespace std;
	46
c32da879	47	class AliTPCcalibTracks : public AliTPCcalibBase {
10757ee9	48	public :
10757ee9	49	AliTPCcalibTracks(); // default constructor
b42cf9ed	50	AliTPCcalibTracks(const AliTPCcalibTracks&calibTracks); // copy constructor
9c171b96	51	AliTPCcalibTracks(const Text_t name, const Text_t title, AliTPCClusterParam clusterParam, AliTPCcalibTracksCuts cuts, Int_t logLevel = 0);
b42cf9ed	52	AliTPCcalibTracks & operator=(const AliTPCcalibTracks& calibTracks);
9c171b96	53
	54	virtual ~AliTPCcalibTracks(); // destructor
	55
0ffd2ae1	56	virtual void Process(AliTPCseed *track); // to be called by the Selector
9c171b96	57	virtual Long64_t Merge(TCollection *li);
9c171b96	58	void MakeResPlotsQTree(Int_t minEntries = 100, char* pathName = "plots");
1bfaa9e9	59	static void MakeQPosNormAll(TTree * chain, AliTPCClusterParam * param, Int_t maxPoints=1000000);
10757ee9	60	void MakeReport(Int_t stat, char* pathName = "plots"); // calls all functions that procude pictures, results are written to pathName, stat is the minimal statistic threshold
9c171b96	61	//
10757ee9	62
9c171b96	63
	64	Int_t AcceptTrack(AliTPCseed * track);
	65	void FillResolutionHistoLocal(AliTPCseed * track); // the MAIN-FUNCTION, called for each track to fill the histograms, called by Process(...)
	66	static TH2D* MakeDiff(TH2D * hfit, TF2 * func);
	67
	68	static AliTPCcalibTracks* TestMerge(AliTPCcalibTracks ct, AliTPCClusterParam clusterParam, Int_t nCalTracks = 50);
	69
	70	void SetStyle() const;
	71	void Draw(Option_t* opt); // draw some exemplaric histograms for fast result-check
	72	void MakeAmpPlots(Int_t stat, char* pathName = "plots");
	73	void MakeDeltaPlots(char* pathName = "plots");
	74	void MakeChargeVsDriftLengthPlotsOld(char* pathName = "plots");
	75	void MakeChargeVsDriftLengthPlots(char* pathName = "plots");
	76	void FitResolutionNew(char* pathName = "plots");
	77	void FitRMSNew(char* pathName = "plots");
	78
	79	TObjArray* GetfArrayAmpRow() {return fArrayAmpRow;}
	80	TObjArray* GetfArrayAmp() {return fArrayAmp;}
	81	TObjArray* GetfArrayQDY() {return fArrayQDY;}
	82	TObjArray* GetfArrayQDZ() {return fArrayQDZ;}
	83	TObjArray* GetfArrayQRMSY() {return fArrayQRMSY;}
	84	TObjArray* GetfArrayQRMSZ() {return fArrayQRMSZ;}
	85	TObjArray* GetfArrayChargeVsDriftlength() {return fArrayChargeVsDriftlength;}
	86	TH1F* GetfDeltaY() {return fDeltaY;}
	87	TH1F* GetfDeltaZ() {return fDeltaZ;}
	88	TObjArray* GetfResolY() {return fResolY;}
	89	TObjArray* GetfResolZ() {return fResolZ;}
	90	TObjArray* GetfRMSY() {return fRMSY;}
	91	TObjArray* GetfRMSZ() {return fRMSZ;}
	92	TH1I* GetfHclus() {return fHclus;}
	93	TH1I* GetfRejectedTracksHisto() {return fRejectedTracksHisto;}
	94	TH1I* GetfHclusterPerPadrow() {return fHclusterPerPadrow;}
	95	TH1I* GetfHclusterPerPadrowRaw() {return fHclusterPerPadrowRaw;}
	96	TH2I* GetfClusterCutHisto() {return fClusterCutHisto;}
	97	AliTPCCalPad* GetfCalPadClusterPerPad() {return fCalPadClusterPerPad; }
	98	AliTPCCalPad* GetfCalPadClusterPerPadRaw() {return fCalPadClusterPerPadRaw;}
	99	AliTPCCalPadRegion* GetCalPadRegionchargeVsDriftlength() {return fcalPadRegionChargeVsDriftlength;}
	100	AliTPCcalibTracksCuts* GetCuts() {return fCuts;}
10757ee9	101	protected:
10757ee9	102
10757ee9	103	private:
	104	static Int_t GetBin(Float_t q, Int_t pad);
	105	static Int_t GetBin(Int_t iq, Int_t pad);
	106	static Float_t GetQ(Int_t bin);
	107	static Float_t GetPad(Int_t bin);
	108	void FillResolutionHistoLocalDebugPart(AliTPCseed track, AliTPCclusterMI cluster0, Int_t irow, Float_t angley, Float_t anglez, Int_t nclFound, Int_t kDelta);
9f0beaf7	109	AliTPCClusterParam *fClusterParam; // pointer to cluster parameterization
10757ee9	110	AliTPCROC *fROC; //!
	111	TObjArray *fArrayAmpRow; // array with amplitudes versus row for given sector
	112	TObjArray *fArrayAmp; // array with amplitude for sectors
	113	TObjArray *fArrayQDY; // q binned delta Y histograms
	114	TObjArray *fArrayQDZ; // q binned delta Z histograms
	115	TObjArray *fArrayQRMSY; // q binned delta Y histograms
	116	TObjArray *fArrayQRMSZ; // q binned delta Z histograms
	117	TObjArray *fArrayChargeVsDriftlength; // array of arrays of TProfiles with charge vs. driftlength for each padsize and sector
	118	AliTPCCalPadRegion *fcalPadRegionChargeVsDriftlength; // CalPadRegion, one TProfile for charge vs. driftlength for each padsize and sector
	119	TH1F *fDeltaY; // integrated delta y histo
	120	TH1F *fDeltaZ; // integrated delta z histo
	121	TObjArray *fResolY; // array of resolution histograms Y
	122	TObjArray *fResolZ; // array of resolution histograms Z
	123	TObjArray *fRMSY; // array of RMS histograms Y
	124	TObjArray *fRMSZ; // array of RMS histograms Z
	125	AliTPCcalibTracksCuts *fCuts; // object with cuts, that is passed to the constructor
	126	TH1I *fHclus; // number of clusters per track
	127	TH1I *fRejectedTracksHisto; // histogram of rejecteced tracks, the number coresponds to the failed cut
	128	TH1I *fHclusterPerPadrow; // histogram showing the number of clusters per padRow
	129	TH1I *fHclusterPerPadrowRaw;// histogram showing the number of clusters per padRow before cuts on clusters are applied
	130	TH2I *fClusterCutHisto; // histogram showing in which padRow the clusters were cutted by which criterium
	131	AliTPCCalPad *fCalPadClusterPerPad; // AliTPCCalPad showing the number of clusters per Pad
	132	AliTPCCalPad *fCalPadClusterPerPadRaw; // AliTPCCalPad showing the number of clusters per Pad before cuts on clusters are applied
9c171b96	133	ClassDef(AliTPCcalibTracks,1)
10757ee9	134
10757ee9	135	};
	136
	137	#endif