[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 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
   
   static void     AddInfo(TChain *chain, char *fileName);        // add clusterParametrization as user info to the chain
   void            Process(AliTPCseed *track);  // to be called by the Selector
   
   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);

   Long64_t Merge(TCollection *li);
   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     MakeReport(Int_t stat, char* pathName = "plots");     // calls all functions that procude pictures, results are written to pathName, stat is the minimal statistic threshold
   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");
   void     MakeResPlotsQTree(Int_t minEntries = 100, char* pathName = "plots"); 
   // Thread-stuff:
   // void     MakeResPlotsQTreeThread(Int_t minEntries = 100, char* pathName = "plots"); 
   // static void* MakeResPlotsQTreeThreadFunction(void* arg); 
   
//protected:   
   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:         
   ClassDef(AliTPCcalibTracks,1)
  
   
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
   TLinearFitter *fFitterLinY1;   //!
   TLinearFitter *fFitterLinZ1;   //! 
   TLinearFitter *fFitterLinY2;   //! 
   TLinearFitter *fFitterLinZ2;   //!
   TLinearFitter *fFitterParY;    //! 
   TLinearFitter *fFitterParZ;    //!
   
};

#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;
	42	class TMutex;
	43
	44	using namespace std;
	45
c32da879	46	class AliTPCcalibTracks : public AliTPCcalibBase {
10757ee9	47	public :
10757ee9	48	AliTPCcalibTracks(); // default constructor
b42cf9ed	49	AliTPCcalibTracks(const AliTPCcalibTracks&calibTracks); // copy constructor
10757ee9	50	AliTPCcalibTracks(const Text_t name, const Text_t title, AliTPCClusterParam clusterParam, AliTPCcalibTracksCuts cuts, Int_t logLevel = 0);
b42cf9ed	51	AliTPCcalibTracks & operator=(const AliTPCcalibTracks& calibTracks);
b42cf9ed	52
10757ee9	53	virtual ~AliTPCcalibTracks(); // destructor
	54
	55	static void AddInfo(TChain chain, char fileName); // add clusterParametrization as user info to the chain
c32da879	56	void Process(AliTPCseed *track); // to be called by the Selector
10757ee9	57
	58	Int_t AcceptTrack(AliTPCseed * track);
	59	void FillResolutionHistoLocal(AliTPCseed * track); // the MAIN-FUNCTION, called for each track to fill the histograms, called by Process(...)
	60	static TH2D* MakeDiff(TH2D * hfit, TF2 * func);
	61
	62	Long64_t Merge(TCollection *li);
	63	static AliTPCcalibTracks* TestMerge(AliTPCcalibTracks ct, AliTPCClusterParam clusterParam, Int_t nCalTracks = 50);
	64
	65	void SetStyle() const;
	66	void Draw(Option_t* opt); // draw some exemplaric histograms for fast result-check
	67	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
	68	void MakeAmpPlots(Int_t stat, char* pathName = "plots");
	69	void MakeDeltaPlots(char* pathName = "plots");
	70	void MakeChargeVsDriftLengthPlotsOld(char* pathName = "plots");
	71	void MakeChargeVsDriftLengthPlots(char* pathName = "plots");
	72	void FitResolutionNew(char* pathName = "plots");
	73	void FitRMSNew(char* pathName = "plots");
	74	void MakeResPlotsQTree(Int_t minEntries = 100, char* pathName = "plots");
	75	// Thread-stuff:
	76	// void MakeResPlotsQTreeThread(Int_t minEntries = 100, char* pathName = "plots");
	77	// static void* MakeResPlotsQTreeThreadFunction(void* arg);
	78
	79	//protected:
	80	TObjArray* GetfArrayAmpRow() {return fArrayAmpRow;}
	81	TObjArray* GetfArrayAmp() {return fArrayAmp;}
	82	TObjArray* GetfArrayQDY() {return fArrayQDY;}
	83	TObjArray* GetfArrayQDZ() {return fArrayQDZ;}
	84	TObjArray* GetfArrayQRMSY() {return fArrayQRMSY;}
	85	TObjArray* GetfArrayQRMSZ() {return fArrayQRMSZ;}
	86	TObjArray* GetfArrayChargeVsDriftlength() {return fArrayChargeVsDriftlength;}
	87	TH1F* GetfDeltaY() {return fDeltaY;}
	88	TH1F* GetfDeltaZ() {return fDeltaZ;}
	89	TObjArray* GetfResolY() {return fResolY;}
	90	TObjArray* GetfResolZ() {return fResolZ;}
	91	TObjArray* GetfRMSY() {return fRMSY;}
	92	TObjArray* GetfRMSZ() {return fRMSZ;}
	93	TH1I* GetfHclus() {return fHclus;}
	94	TH1I* GetfRejectedTracksHisto() {return fRejectedTracksHisto;}
	95	TH1I* GetfHclusterPerPadrow() {return fHclusterPerPadrow;}
	96	TH1I* GetfHclusterPerPadrowRaw() {return fHclusterPerPadrowRaw;}
	97	TH2I* GetfClusterCutHisto() {return fClusterCutHisto;}
	98	AliTPCCalPad* GetfCalPadClusterPerPad() {return fCalPadClusterPerPad; }
	99	AliTPCCalPad* GetfCalPadClusterPerPadRaw() {return fCalPadClusterPerPadRaw;}
	100	AliTPCCalPadRegion* GetCalPadRegionchargeVsDriftlength() {return fcalPadRegionChargeVsDriftlength;}
	101	AliTPCcalibTracksCuts* GetCuts() {return fCuts;}
10757ee9	102
	103	protected:
	104	ClassDef(AliTPCcalibTracks,1)
	105
	106
	107	private:
	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);
	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
10757ee9	137	TLinearFitter *fFitterLinY1; //!
	138	TLinearFitter *fFitterLinZ1; //!
	139	TLinearFitter *fFitterLinY2; //!
	140	TLinearFitter *fFitterLinZ2; //!
	141	TLinearFitter *fFitterParY; //!
	142	TLinearFitter *fFitterParZ; //!
	143
10757ee9	144	};
	145
	146	#endif