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

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

/* $Id$ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  Parameter class for the TRD reconstruction                            //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#ifndef ALIDETECTORRECOPARAM_H
#include "AliDetectorRecoParam.h"
#endif
#ifndef ALITRDCALPID_H
#include "Cal/AliTRDCalPID.h"
#endif

class AliTRDrecoParam : public AliDetectorRecoParam
{
public:
  AliTRDrecoParam();
  AliTRDrecoParam(const AliTRDrecoParam &rec);
  ~AliTRDrecoParam() { }

  Double_t GetChi2Y() const                 { return fkChi2Y;    }
  Double_t GetChi2Z() const                 { return fkChi2Z;    }
  Double_t GetChi2YSlope() const            { return fkChi2YSlope; }
  Double_t GetChi2ZSlope() const            { return fkChi2ZSlope; }
	Double_t GetChi2YCut() const              { return fkChi2YCut; }
  Double_t GetPhiSlope() const              { return fkPhiSlope;   }
  Float_t  GetNClusters() const;
  Double_t GetNMeanClusters() const         { return fkNMeanClusters; }
  Double_t GetNSigmaClusters() const        { return fkNSigmaClusters; }
  Double_t GetFindableClusters() const      { return fkFindable; }
  Double_t GetMaxTheta() const              { return fkMaxTheta; }
  Double_t GetMaxPhi() const                { return fkMaxPhi;   }
  Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
  Double_t GetPIDThreshold(Float_t /*p*/) const { return 0.;}
  Double_t GetRoad0y() const                { return fkRoad0y;   }
  Double_t GetRoad0z() const                { return fkRoad0z;   }
  Double_t GetRoad1y() const                { return fkRoad1y;   }
  Double_t GetRoad1z() const                { return fkRoad1z;   }
  Double_t GetRoad2y() const                { return fkRoad2y;   }
  Double_t GetRoad2z() const                { return fkRoad2z;   }
  Double_t GetTrackLikelihood() const       { return fkTrackLikelihood;       }
  inline void GetSysCovMatrix(Double_t *sys) const;  
  Double_t GetMinMaxCutSigma() const        { return fMinMaxCutSigma;     };
  Double_t GetMinLeftRightCutSigma() const  { return fMinLeftRightCutSigma;  };
  Double_t GetClusMaxThresh() const         { return fClusMaxThresh;   };
  Double_t GetClusSigThresh() const         { return fClusSigThresh;   };
  Int_t    GetTCnexp() const                { return fTCnexp;          };
  Int_t     GetNumberOfPresamples()  const {return fNumberOfPresamples;}
  Int_t    GetNumberOfPostsamples() const {return fNumberOfPostsamples;}

        
  static   AliTRDrecoParam *GetLowFluxParam();
  static   AliTRDrecoParam *GetHighFluxParam();
  static   AliTRDrecoParam *GetCosmicTestParam();

  Bool_t   IsClusterSharing() const         { return TestBit(kClusterSharing);}
  Bool_t   IsLUT() const                    { return TestBit(kLUT);}
  Bool_t   IsTailCancelation() const        { return TestBit(kTC);}
  Bool_t   IsVertexConstrained() const      { return TestBit(kVertexConstrained); }
  Bool_t   HasImproveTracklets() const       { return TestBit(kImproveTracklet); }

  void     SetMaxTheta(Double_t maxTheta) {fkMaxTheta = maxTheta;}
  void     SetMaxPhi(Double_t maxPhi) {fkMaxPhi = maxPhi;}
  void     SetFindableClusters(Double_t r) {fkFindable = r;}
  void     SetChi2Y(Double_t chi2) {fkChi2Y = chi2;}
  void     SetChi2Z(Double_t chi2) {fkChi2Z = chi2;}
  void     SetChi2YSlope(Double_t chi2YSlope) {fkChi2YSlope = chi2YSlope;}
  void     SetChi2ZSlope(Double_t chi2ZSlope) {fkChi2ZSlope = chi2ZSlope;}
	void	   SetChi2YCut(Double_t chi2Cut) {fkChi2YCut = chi2Cut; }
  void     SetPhiSlope(Double_t phiSlope) {fkPhiSlope = phiSlope;}
  void     SetNMeanClusters(Double_t meanNclusters) {fkNMeanClusters = meanNclusters;}
  void     SetNSigmaClusters(Double_t sigmaNclusters) {fkNSigmaClusters = sigmaNclusters;} 
  void     SetClusterSharing(Bool_t share = kTRUE)            { SetBit(kClusterSharing, share);}
  void     SetImproveTracklets(Bool_t improve = kTRUE)         { SetBit(kImproveTracklet, improve);}
  void     SetVertexConstrained(Bool_t vc = kTRUE)            { SetBit(kVertexConstrained, vc); }
  void     SetLUT(Bool_t lut = kTRUE)                         { SetBit(kLUT, lut);};
  void     SetMinMaxCutSigma(Float_t minMaxCutSigma)          { fMinMaxCutSigma   = minMaxCutSigma; }
  void     SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma   = minLeftRightCutSigma; };
  void     SetClusMaxThresh(Float_t thresh)                   { fClusMaxThresh   = thresh; };
  void     SetClusSigThresh(Float_t thresh)                   { fClusSigThresh   = thresh; };
  void     SetTailCancelation(Bool_t tc = kTRUE)              { SetBit(kTC, tc);  };
  inline void SetPIDThreshold(Double_t *pid);
  void     SetNexponential(Int_t nexp)                        { fTCnexp          = nexp;   };
  inline void SetSysCovMatrix(Double_t *sys);
  void     SetNumberOfPresamples(Int_t n)                     { fNumberOfPresamples = n;}
  void     SetNumberOfPostsamples(Int_t n)                    { fNumberOfPostsamples = n;}

private:
  enum{
    kTC                = BIT(14) // tail cancelation
   ,kLUT               = BIT(15) // look up table for cluster position determination 
   ,kClusterSharing    = BIT(16) // Toggle cluster sharing
   ,kVertexConstrained = BIT(17) // Perform vertex constrained fit
   ,kImproveTracklet   = BIT(18) // Improve tracklet in the SA TRD track finder 
  };
  // Physics reference values for TRD
  Double_t  fkdNchdy;                // dNch/dy
  Double_t  fkMaxTheta;              // Maximum theta
  Double_t  fkMaxPhi;                // Maximum phi - momentum cut

  Double_t  fkRoad0y;                // Road for middle cluster
  Double_t  fkRoad0z;                // Road for middle cluster

  Double_t  fkRoad1y;                // Road in y for seeded cluster
  Double_t  fkRoad1z;                // Road in z for seeded cluster

  Double_t  fkRoad2y;                // Road in y for extrapolated cluster
  Double_t  fkRoad2z;                // Road in z for extrapolated cluster
  
  Double_t  fkPlaneQualityThreshold; // Quality threshold
  Double_t  fkFindable;              // minimum ratio of clusters per tracklet supposed to be attached.
  Double_t  fkChi2Z;                 // Max chi2 on the z direction for seeding clusters fit
  Double_t  fkChi2Y;                 // Max chi2 on the y direction for seeding clusters Rieman fit
  Double_t  fkChi2YSlope;            // Slope of the chi2-distribution in y-direction
  Double_t  fkChi2ZSlope;            // Slope of the chi2-distribution in z-direction
	Double_t  fkChi2YCut;							 // Cut on the Chi2 in y-direction in the likelihood filter
  Double_t  fkPhiSlope;              // Slope of the distribution of the deviation between track angle and tracklet angle
  Double_t  fkNMeanClusters;         // Mean number of clusters per tracklet
  Double_t  fkNSigmaClusters;        // Sigma of the number of clusters per tracklet
  Double_t  fkNClusterNoise;         // ratio of noisy clusters to the true one
  Double_t  fkNMeanTracklets;        // Mean number of tracklets per track
  Double_t  fkTrackLikelihood;       // Track likelihood for tracklets Rieman fit
  
  Double_t  fSysCovMatrix[5];        // Systematic uncertainty from calibration and alignment for each tracklet
  Double_t  fPIDThreshold[AliTRDCalPID::kNMom];

  // Clusterization parameter
  Double_t  fMinMaxCutSigma;         // Threshold sigma noise pad middle
  Double_t  fMinLeftRightCutSigma;   // Threshold sigma noise sum pad
  Double_t  fClusMaxThresh;          // Threshold value for cluster maximum
  Double_t  fClusSigThresh;          // Threshold value for cluster signal
  Int_t     fTCnexp;                 // Number of exponentials, digital filter
  
  // ADC parameter
  Int_t     fNumberOfPresamples;     // number of presamples 
  Int_t     fNumberOfPostsamples;     // number of postsamples 

  ClassDef(AliTRDrecoParam, 7)       // Reconstruction parameters for TRD detector

};

//___________________________________________________
inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
{
  if(!sys) return;
  memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
}

//___________________________________________________
inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
{
  if(!sys) return;
  memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
}

//___________________________________________________
inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
{
  if(!pid) return;
  memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
}


#endif
Commit	Line	Data
e4f2f73d	1	#ifndef ALITRDRECOPARAM_H
	2	#define ALITRDRECOPARAM_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4ba1d6ae	4	* See cxx source for full Copyright notice */
e4f2f73d	5
	6	/* $Id$ */
	7
	8	////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// Parameter class for the TRD reconstruction //
	11	// //
	12	////////////////////////////////////////////////////////////////////////////
	13
	14	#ifndef ALIDETECTORRECOPARAM_H
	15	#include "AliDetectorRecoParam.h"
	16	#endif
22a4ab0c	17	#ifndef ALITRDCALPID_H
	18	#include "Cal/AliTRDCalPID.h"
	19	#endif
e4f2f73d	20
	21	class AliTRDrecoParam : public AliDetectorRecoParam
	22	{
3c66288b	23	public:
4ba1d6ae	24	AliTRDrecoParam();
a7ac01d2	25	AliTRDrecoParam(const AliTRDrecoParam &rec);
4ba1d6ae	26	~AliTRDrecoParam() { }
	27
	28	Double_t GetChi2Y() const { return fkChi2Y; }
	29	Double_t GetChi2Z() const { return fkChi2Z; }
5a2e200c	30	Double_t GetChi2YSlope() const { return fkChi2YSlope; }
	31	Double_t GetChi2ZSlope() const { return fkChi2ZSlope; }
	32	Double_t GetChi2YCut() const { return fkChi2YCut; }
	33	Double_t GetPhiSlope() const { return fkPhiSlope; }
8ae98148	34	Float_t GetNClusters() const;
	35	Double_t GetNMeanClusters() const { return fkNMeanClusters; }
	36	Double_t GetNSigmaClusters() const { return fkNSigmaClusters; }
4ba1d6ae	37	Double_t GetFindableClusters() const { return fkFindable; }
	38	Double_t GetMaxTheta() const { return fkMaxTheta; }
	39	Double_t GetMaxPhi() const { return fkMaxPhi; }
3afdab72	40	Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
56b32fd7	41	Double_t GetPIDThreshold(Float_t /p/) const { return 0.;}
4ba1d6ae	42	Double_t GetRoad0y() const { return fkRoad0y; }
4ba1d6ae	43	Double_t GetRoad0z() const { return fkRoad0z; }
4ba1d6ae	44	Double_t GetRoad1y() const { return fkRoad1y; }
4ba1d6ae	45	Double_t GetRoad1z() const { return fkRoad1z; }
4ba1d6ae	46	Double_t GetRoad2y() const { return fkRoad2y; }
4ba1d6ae	47	Double_t GetRoad2z() const { return fkRoad2z; }
4ba1d6ae	48	Double_t GetTrackLikelihood() const { return fkTrackLikelihood; }
7e88424f	49	inline void GetSysCovMatrix(Double_t *sys) const;
4ba1d6ae	50	Double_t GetMinMaxCutSigma() const { return fMinMaxCutSigma; };
	51	Double_t GetMinLeftRightCutSigma() const { return fMinLeftRightCutSigma; };
	52	Double_t GetClusMaxThresh() const { return fClusMaxThresh; };
	53	Double_t GetClusSigThresh() const { return fClusSigThresh; };
	54	Int_t GetTCnexp() const { return fTCnexp; };
3c66288b	55	Int_t GetNumberOfPresamples() const {return fNumberOfPresamples;}
3c66288b	56	Int_t GetNumberOfPostsamples() const {return fNumberOfPostsamples;}
4ba1d6ae	57
3c66288b	58
4ba1d6ae	59	static AliTRDrecoParam *GetLowFluxParam();
	60	static AliTRDrecoParam *GetHighFluxParam();
	61	static AliTRDrecoParam *GetCosmicTestParam();
	62
a7ac01d2	63	Bool_t IsClusterSharing() const { return TestBit(kClusterSharing);}
a7ac01d2	64	Bool_t IsLUT() const { return TestBit(kLUT);}
a7ac01d2	65	Bool_t IsTailCancelation() const { return TestBit(kTC);}
a7ac01d2	66	Bool_t IsVertexConstrained() const { return TestBit(kVertexConstrained); }
fb872574	67	Bool_t HasImproveTracklets() const { return TestBit(kImproveTracklet); }
3c66288b	68
91834b8d	69	void SetMaxTheta(Double_t maxTheta) {fkMaxTheta = maxTheta;}
91834b8d	70	void SetMaxPhi(Double_t maxPhi) {fkMaxPhi = maxPhi;}
4e459a9d	71	void SetFindableClusters(Double_t r) {fkFindable = r;}
0c19b2c9	72	void SetChi2Y(Double_t chi2) {fkChi2Y = chi2;}
0c19b2c9	73	void SetChi2Z(Double_t chi2) {fkChi2Z = chi2;}
5a2e200c	74	void SetChi2YSlope(Double_t chi2YSlope) {fkChi2YSlope = chi2YSlope;}
	75	void SetChi2ZSlope(Double_t chi2ZSlope) {fkChi2ZSlope = chi2ZSlope;}
	76	void SetChi2YCut(Double_t chi2Cut) {fkChi2YCut = chi2Cut; }
	77	void SetPhiSlope(Double_t phiSlope) {fkPhiSlope = phiSlope;}
8ae98148	78	void SetNMeanClusters(Double_t meanNclusters) {fkNMeanClusters = meanNclusters;}
8ae98148	79	void SetNSigmaClusters(Double_t sigmaNclusters) {fkNSigmaClusters = sigmaNclusters;}
fb872574	80	void SetClusterSharing(Bool_t share = kTRUE) { SetBit(kClusterSharing, share);}
	81	void SetImproveTracklets(Bool_t improve = kTRUE) { SetBit(kImproveTracklet, improve);}
	82	void SetVertexConstrained(Bool_t vc = kTRUE) { SetBit(kVertexConstrained, vc); }
	83	void SetLUT(Bool_t lut = kTRUE) { SetBit(kLUT, lut);};
	84	void SetMinMaxCutSigma(Float_t minMaxCutSigma) { fMinMaxCutSigma = minMaxCutSigma; }
4ba1d6ae	85	void SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma = minLeftRightCutSigma; };
	86	void SetClusMaxThresh(Float_t thresh) { fClusMaxThresh = thresh; };
	87	void SetClusSigThresh(Float_t thresh) { fClusSigThresh = thresh; };
fb872574	88	void SetTailCancelation(Bool_t tc = kTRUE) { SetBit(kTC, tc); };
22a4ab0c	89	inline void SetPIDThreshold(Double_t *pid);
4ba1d6ae	90	void SetNexponential(Int_t nexp) { fTCnexp = nexp; };
3afdab72	91	inline void SetSysCovMatrix(Double_t *sys);
fb872574	92	void SetNumberOfPresamples(Int_t n) { fNumberOfPresamples = n;}
fb872574	93	void SetNumberOfPostsamples(Int_t n) { fNumberOfPostsamples = n;}
4ba1d6ae	94
3c66288b	95	private:
a7ac01d2	96	enum{
fb872574	97	kTC = BIT(14) // tail cancelation
	98	,kLUT = BIT(15) // look up table for cluster position determination
	99	,kClusterSharing = BIT(16) // Toggle cluster sharing
	100	,kVertexConstrained = BIT(17) // Perform vertex constrained fit
	101	,kImproveTracklet = BIT(18) // Improve tracklet in the SA TRD track finder
a7ac01d2	102	};
8ae98148	103	// Physics reference values for TRD
8ae98148	104	Double_t fkdNchdy; // dNch/dy
4ba1d6ae	105	Double_t fkMaxTheta; // Maximum theta
8ae98148	106	Double_t fkMaxPhi; // Maximum phi - momentum cut
4ba1d6ae	107
	108	Double_t fkRoad0y; // Road for middle cluster
	109	Double_t fkRoad0z; // Road for middle cluster
	110
	111	Double_t fkRoad1y; // Road in y for seeded cluster
	112	Double_t fkRoad1z; // Road in z for seeded cluster
	113
	114	Double_t fkRoad2y; // Road in y for extrapolated cluster
	115	Double_t fkRoad2z; // Road in z for extrapolated cluster
	116
	117	Double_t fkPlaneQualityThreshold; // Quality threshold
8ae98148	118	Double_t fkFindable; // minimum ratio of clusters per tracklet supposed to be attached.
4ba1d6ae	119	Double_t fkChi2Z; // Max chi2 on the z direction for seeding clusters fit
4ba1d6ae	120	Double_t fkChi2Y; // Max chi2 on the y direction for seeding clusters Rieman fit
5a2e200c	121	Double_t fkChi2YSlope; // Slope of the chi2-distribution in y-direction
	122	Double_t fkChi2ZSlope; // Slope of the chi2-distribution in z-direction
	123	Double_t fkChi2YCut; // Cut on the Chi2 in y-direction in the likelihood filter
	124	Double_t fkPhiSlope; // Slope of the distribution of the deviation between track angle and tracklet angle
8ae98148	125	Double_t fkNMeanClusters; // Mean number of clusters per tracklet
	126	Double_t fkNSigmaClusters; // Sigma of the number of clusters per tracklet
	127	Double_t fkNClusterNoise; // ratio of noisy clusters to the true one
	128	Double_t fkNMeanTracklets; // Mean number of tracklets per track
4ba1d6ae	129	Double_t fkTrackLikelihood; // Track likelihood for tracklets Rieman fit
4ba1d6ae	130
3afdab72	131	Double_t fSysCovMatrix[5]; // Systematic uncertainty from calibration and alignment for each tracklet
22a4ab0c	132	Double_t fPIDThreshold[AliTRDCalPID::kNMom];
e4f2f73d	133
a7ac01d2	134	// Clusterization parameter
4ba1d6ae	135	Double_t fMinMaxCutSigma; // Threshold sigma noise pad middle
	136	Double_t fMinLeftRightCutSigma; // Threshold sigma noise sum pad
	137	Double_t fClusMaxThresh; // Threshold value for cluster maximum
	138	Double_t fClusSigThresh; // Threshold value for cluster signal
4ba1d6ae	139	Int_t fTCnexp; // Number of exponentials, digital filter
fc546d21	140
4ba1d6ae	141	// ADC parameter
4e459a9d	142	Int_t fNumberOfPresamples; // number of presamples
4e459a9d	143	Int_t fNumberOfPostsamples; // number of postsamples
9716329b	144
5a2e200c	145	ClassDef(AliTRDrecoParam, 7) // Reconstruction parameters for TRD detector
e4f2f73d	146
e4f2f73d	147	};
3afdab72	148
3afdab72	149	//___________________________________________________
7e88424f	150	inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
3afdab72	151	{
	152	if(!sys) return;
	153	memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
	154	}
	155
	156	//___________________________________________________
	157	inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
	158	{
	159	if(!sys) return;
	160	memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
	161	}
	162
22a4ab0c	163	//___________________________________________________
	164	inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
	165	{
	166	if(!pid) return;
	167	memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
	168	}
3c66288b	169
3c66288b	170
e4f2f73d	171	#endif