[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 "AliTRDCalPID.h"
#endif

#ifndef ALITRDPIDRESPONSE_H
#include "AliTRDPIDResponse.h"
#endif

class TString;

class AliTRDrecoParam : public AliDetectorRecoParam
{
public:
  enum ETRDReconstructionTask{
    kClusterizer = 0,
    kTracker = 1,
    kPID = 2,
    kTRDreconstructionTasks = 3
  };
  enum ETRDflags {
    kDriftGas
    ,kVertexConstraint
    ,kTailCancelation
    ,kImproveTracklet
    ,kLUT
    ,kGAUS
    ,kClusterSharing
    ,kSteerPID
    ,kEightSlices
    ,kCheckTimeConsistency
    ,kLQ2D
  };
  AliTRDrecoParam();
  AliTRDrecoParam(const AliTRDrecoParam &rec);
  AliTRDrecoParam& operator=(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; }
  inline Int_t    GetPIDLQslices() const;
  inline AliTRDPIDResponse::ETRDPIDMethod GetPIDmethod() const;
  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 GetRoadzMultiplicator() const    { return fkRoadzMultiplicator; }
  Double_t GetTrackLikelihood() const       { return fkTrackLikelihood;       }
  inline void GetSysCovMatrix(Double_t *sys) const;  
  inline void GetTCParams(Double_t *par) const;
  inline Int_t GetStreamLevel(ETRDReconstructionTask task) const;
  const TString *GetRawStreamVersion() const{ return &fRawStreamVersion; };
  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;}
  Int_t    GetNumberOfSeedConfigs() const   { return fNumberOfConfigs;}
  Int_t    GetRecEveryNTB() const           { return fRecEveryNTB; }
  Bool_t   IsArgon() const                  { return TESTBIT(fFlags, kDriftGas); }
  Bool_t   IsCheckTimeConsistency() const   { return kCheckTimeConsistency;}
  Bool_t   IsOverPtThreshold(Double_t pt) const {return Bool_t(pt>fkPtThreshold);}
  Bool_t   IsXenon() const                  { return !TESTBIT(fFlags, kDriftGas); }
  Bool_t   IsPIDNeuralNetwork() const       { return TESTBIT(fFlags, kSteerPID);}
  Bool_t   IsVertexConstrained() const      { return TESTBIT(fFlags, kVertexConstraint); }
  Bool_t   IsEightSlices() const            { return TESTBIT(fFlags, kEightSlices);}
  Bool_t   HasImproveTracklets() const      { return TESTBIT(fFlags, kImproveTracklet);}
  Bool_t   UseClusterSharing() const        { return TESTBIT(fFlags, kClusterSharing);}
  Bool_t   UseLUT() const                   { return TESTBIT(fFlags, kLUT);}
  Bool_t   UseGAUS() const                  { return TESTBIT(fFlags, kGAUS);}
  Bool_t   UseTailCancelation() const       { return TESTBIT(fFlags, kTailCancelation); }
        
  static   AliTRDrecoParam *GetLowFluxParam();
  static   AliTRDrecoParam *GetLowFluxHLTParam();
  static   AliTRDrecoParam *GetHighFluxParam();
  static   AliTRDrecoParam *GetHighFluxHLTParam();
  static   AliTRDrecoParam *GetCosmicTestParam();

  void     SetArgon(Bool_t b = kTRUE)                         {if(b) SETBIT(fFlags, kDriftGas); else CLRBIT(fFlags, kDriftGas);}
  void     SetCheckTimeConsistency(Bool_t b = kTRUE)          {if(b) SETBIT(fFlags, kCheckTimeConsistency); else CLRBIT(fFlags, kCheckTimeConsistency);}
  void     SetClusterSharing(Bool_t b = kTRUE)                {if(b) SETBIT(fFlags, kClusterSharing); else CLRBIT(fFlags, kClusterSharing);}
  void     SetEightSlices(Bool_t b = kTRUE)                   {if(b) SETBIT(fFlags, kEightSlices); else CLRBIT(fFlags, kEightSlices);}
  void     SetImproveTracklets(Bool_t b = kTRUE)              {if(b) SETBIT(fFlags, kImproveTracklet); else CLRBIT(fFlags, kImproveTracklet);}
  void     SetLUT(Bool_t b=kTRUE)                             {if(b) SETBIT(fFlags, kLUT); else CLRBIT(fFlags, kLUT);}
  void     SetGAUS(Bool_t b=kTRUE)                            {if(b) SETBIT(fFlags, kGAUS); else CLRBIT(fFlags, kGAUS);}
  void     SetPIDNeuralNetwork(Bool_t b=kTRUE)                {if(b) SETBIT(fFlags, kSteerPID); else CLRBIT(fFlags, kSteerPID);}
  inline void  SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method);
  void     SetPIDLQslices(Int_t s);
  void     SetTailCancelation(Bool_t b=kTRUE)                 {if(b) SETBIT(fFlags, kTailCancelation); else CLRBIT(fFlags, kTailCancelation);}
  void     SetXenon(Bool_t b = kTRUE)                         {if(b) CLRBIT(fFlags, kDriftGas); else SETBIT(fFlags, kDriftGas);}
  void     SetVertexConstrained()                             {SETBIT(fFlags, kVertexConstraint);}
  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     SetRawStreamVersion(const Char_t *version)         {fRawStreamVersion = version; }
  void     SetRoadzMultiplicator(Double_t mult)               {fkRoadzMultiplicator = mult; } 
  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; };
  inline void SetPIDThreshold(Double_t *pid);
  void     SetPtThreshold(Double_t pt) {fkPtThreshold = pt;}
  void     SetNexponential(Int_t nexp)                        { fTCnexp          = nexp;   };
  inline void SetTCParams(Double_t *par);
  inline void SetStreamLevel(ETRDReconstructionTask task, Int_t level);
  inline void SetSysCovMatrix(Double_t *sys);
  void     SetNumberOfPresamples(Int_t n)                     { fNumberOfPresamples = n;}
  void     SetNumberOfPostsamples(Int_t n)                    { fNumberOfPostsamples = n;}
  void     SetRecEveryTwoTB()                                 { fRecEveryNTB = 2; fkNMeanClusters = 10; }

private:
  // Physics reference values for TRD
  Double_t  fkdNchdy;                // dNch/dy
  Double_t  fkMaxTheta;              // Maximum theta
  Double_t  fkMaxPhi;                // Maximum phi - momentum cut
  // Tracker params 
  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  fkPtThreshold;           // pt threshold for using TRD points for updating Kalaman track
  Double_t  fkPlaneQualityThreshold; // Quality threshold
  Double_t  fkRoadzMultiplicator;    // Multiplicator for the Roads in z 
  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];   // PID Thresholds for Electron candidate decision
  Int_t     fNumberOfConfigs;        // Used number of seed configurations

  // Reconstruction Options for TRD reconstruction
  Int_t     fStreamLevel[kTRDreconstructionTasks]; // Stream Level
  Long64_t  fFlags;                  // option Flags

  // Raw Reader Params
  TString   fRawStreamVersion;       // Raw Reader version

  // 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
  Double_t  fTCParams[8];            // Tail Cancellation parameters for drift gases 
  Int_t     fRecEveryNTB;            // Reconstruct each nth timebin

  // ADC parameter
  Int_t     fNumberOfPresamples;     // number of presamples 
  Int_t     fNumberOfPostsamples;     // number of postsamples 

  ClassDef(AliTRDrecoParam, 12)       // 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));
}

//___________________________________________________
inline void AliTRDrecoParam::SetStreamLevel(ETRDReconstructionTask task, Int_t level){
  if(task >= kTRDreconstructionTasks) return;
  fStreamLevel[static_cast<Int_t>(task)] = level;
}

//___________________________________________________
inline Int_t AliTRDrecoParam::GetStreamLevel(ETRDReconstructionTask task) const{
  if(task >= kTRDreconstructionTasks) return 0;
  return fStreamLevel[static_cast<Int_t>(task)];
}

//___________________________________________________
inline void AliTRDrecoParam::GetTCParams(Double_t *par) const
{
  if(!par) return;
  if(IsArgon()) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
  else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
}

//___________________________________________________
inline void AliTRDrecoParam::SetTCParams(Double_t *par)
{
  if(!par) return;
  memcpy(fTCParams, par, 8*sizeof(Double_t));
}

//___________________________________________________
inline Int_t AliTRDrecoParam::GetPIDLQslices() const
{
  if(IsPIDNeuralNetwork()) return -1;
  return TESTBIT(fFlags, kLQ2D) ? 2 : 1;
}

//___________________________________________________
inline AliTRDPIDResponse::ETRDPIDMethod AliTRDrecoParam::GetPIDmethod() const
{
  AliTRDPIDResponse::ETRDPIDMethod method = AliTRDPIDResponse::kLQ1D;
  if(IsPIDNeuralNetwork()) method = AliTRDPIDResponse::kNN;
  else if(TESTBIT(fFlags, kLQ2D)) method = AliTRDPIDResponse::kLQ2D;
  return method;
}

//___________________________________________________
inline void  AliTRDrecoParam::SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method)
{
  switch(method){
  case AliTRDPIDResponse::kLQ2D:
    CLRBIT(fFlags, kSteerPID); 
    SETBIT(fFlags, kLQ2D);
    break;
  case AliTRDPIDResponse::kNN:
    SETBIT(fFlags, kSteerPID); 
    break;
  case AliTRDPIDResponse::kLQ1D:
  default:
    CLRBIT(fFlags, kSteerPID); 
    CLRBIT(fFlags, kLQ2D);
    break;
  }
}

#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
9dcc64cc	17
22a4ab0c	18	#ifndef ALITRDCALPID_H
97587798	19	#include "AliTRDCalPID.h"
22a4ab0c	20	#endif
e4f2f73d	21
9dcc64cc	22	#ifndef ALITRDPIDRESPONSE_H
	23	#include "AliTRDPIDResponse.h"
	24	#endif
	25
a2fbb6ec	26	class TString;
a2fbb6ec	27
e4f2f73d	28	class AliTRDrecoParam : public AliDetectorRecoParam
e4f2f73d	29	{
3c66288b	30	public:
a2fbb6ec	31	enum ETRDReconstructionTask{
	32	kClusterizer = 0,
	33	kTracker = 1,
	34	kPID = 2,
	35	kTRDreconstructionTasks = 3
	36	};
	37	enum ETRDflags {
11d80e40	38	kDriftGas
	39	,kVertexConstraint
	40	,kTailCancelation
	41	,kImproveTracklet
	42	,kLUT
	43	,kGAUS
	44	,kClusterSharing
	45	,kSteerPID
	46	,kEightSlices
	47	,kCheckTimeConsistency
	48	,kLQ2D
828c6f80	49	};
4ba1d6ae	50	AliTRDrecoParam();
a7ac01d2	51	AliTRDrecoParam(const AliTRDrecoParam &rec);
f09f3167	52	AliTRDrecoParam& operator=(const AliTRDrecoParam &rec);
4ba1d6ae	53	~AliTRDrecoParam() { }
	54
	55	Double_t GetChi2Y() const { return fkChi2Y; }
	56	Double_t GetChi2Z() const { return fkChi2Z; }
5a2e200c	57	Double_t GetChi2YSlope() const { return fkChi2YSlope; }
	58	Double_t GetChi2ZSlope() const { return fkChi2ZSlope; }
	59	Double_t GetChi2YCut() const { return fkChi2YCut; }
	60	Double_t GetPhiSlope() const { return fkPhiSlope; }
8ae98148	61	Float_t GetNClusters() const;
	62	Double_t GetNMeanClusters() const { return fkNMeanClusters; }
	63	Double_t GetNSigmaClusters() const { return fkNSigmaClusters; }
4ba1d6ae	64	Double_t GetFindableClusters() const { return fkFindable; }
11d80e40	65	inline Int_t GetPIDLQslices() const;
9dcc64cc	66	inline AliTRDPIDResponse::ETRDPIDMethod GetPIDmethod() const;
4ba1d6ae	67	Double_t GetMaxTheta() const { return fkMaxTheta; }
4ba1d6ae	68	Double_t GetMaxPhi() const { return fkMaxPhi; }
3afdab72	69	Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
56b32fd7	70	Double_t GetPIDThreshold(Float_t /p/) const { return 0.;}
4ba1d6ae	71	Double_t GetRoad0y() const { return fkRoad0y; }
4ba1d6ae	72	Double_t GetRoad0z() const { return fkRoad0z; }
4ba1d6ae	73	Double_t GetRoad1y() const { return fkRoad1y; }
4ba1d6ae	74	Double_t GetRoad1z() const { return fkRoad1z; }
4ba1d6ae	75	Double_t GetRoad2y() const { return fkRoad2y; }
4ba1d6ae	76	Double_t GetRoad2z() const { return fkRoad2z; }
566bf887	77	Double_t GetRoadzMultiplicator() const { return fkRoadzMultiplicator; }
4ba1d6ae	78	Double_t GetTrackLikelihood() const { return fkTrackLikelihood; }
7e88424f	79	inline void GetSysCovMatrix(Double_t *sys) const;
a2fbb6ec	80	inline void GetTCParams(Double_t *par) const;
a2fbb6ec	81	inline Int_t GetStreamLevel(ETRDReconstructionTask task) const;
828c6f80	82	const TString *GetRawStreamVersion() const{ return &fRawStreamVersion; };
4ba1d6ae	83	Double_t GetMinMaxCutSigma() const { return fMinMaxCutSigma; };
	84	Double_t GetMinLeftRightCutSigma() const { return fMinLeftRightCutSigma; };
	85	Double_t GetClusMaxThresh() const { return fClusMaxThresh; };
	86	Double_t GetClusSigThresh() const { return fClusSigThresh; };
	87	Int_t GetTCnexp() const { return fTCnexp; };
828c6f80	88	Int_t GetNumberOfPresamples() const { return fNumberOfPresamples;}
828c6f80	89	Int_t GetNumberOfPostsamples() const { return fNumberOfPostsamples;}
8c499dbf	90	Int_t GetNumberOfSeedConfigs() const { return fNumberOfConfigs;}
8c499dbf	91	Int_t GetRecEveryNTB() const { return fRecEveryNTB; }
828c6f80	92	Bool_t IsArgon() const { return TESTBIT(fFlags, kDriftGas); }
828c6f80	93	Bool_t IsCheckTimeConsistency() const { return kCheckTimeConsistency;}
b06a50a5	94	Bool_t IsOverPtThreshold(Double_t pt) const {return Bool_t(pt>fkPtThreshold);}
828c6f80	95	Bool_t IsXenon() const { return !TESTBIT(fFlags, kDriftGas); }
a2fbb6ec	96	Bool_t IsPIDNeuralNetwork() const { return TESTBIT(fFlags, kSteerPID);}
828c6f80	97	Bool_t IsVertexConstrained() const { return TESTBIT(fFlags, kVertexConstraint); }
a2fbb6ec	98	Bool_t IsEightSlices() const { return TESTBIT(fFlags, kEightSlices);}
	99	Bool_t HasImproveTracklets() const { return TESTBIT(fFlags, kImproveTracklet);}
	100	Bool_t UseClusterSharing() const { return TESTBIT(fFlags, kClusterSharing);}
	101	Bool_t UseLUT() const { return TESTBIT(fFlags, kLUT);}
	102	Bool_t UseGAUS() const { return TESTBIT(fFlags, kGAUS);}
	103	Bool_t UseTailCancelation() const { return TESTBIT(fFlags, kTailCancelation); }
3c66288b	104
4ba1d6ae	105	static AliTRDrecoParam *GetLowFluxParam();
8c499dbf	106	static AliTRDrecoParam *GetLowFluxHLTParam();
4ba1d6ae	107	static AliTRDrecoParam *GetHighFluxParam();
8c499dbf	108	static AliTRDrecoParam *GetHighFluxHLTParam();
4ba1d6ae	109	static AliTRDrecoParam *GetCosmicTestParam();
4ba1d6ae	110
828c6f80	111	void SetArgon(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kDriftGas); else CLRBIT(fFlags, kDriftGas);}
	112	void SetCheckTimeConsistency(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kCheckTimeConsistency); else CLRBIT(fFlags, kCheckTimeConsistency);}
	113	void SetClusterSharing(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kClusterSharing); else CLRBIT(fFlags, kClusterSharing);}
	114	void SetEightSlices(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kEightSlices); else CLRBIT(fFlags, kEightSlices);}
	115	void SetImproveTracklets(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kImproveTracklet); else CLRBIT(fFlags, kImproveTracklet);}
8c25d014	116	void SetLUT(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kLUT); else CLRBIT(fFlags, kLUT);}
	117	void SetGAUS(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kGAUS); else CLRBIT(fFlags, kGAUS);}
	118	void SetPIDNeuralNetwork(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kSteerPID); else CLRBIT(fFlags, kSteerPID);}
9dcc64cc	119	inline void SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method);
11d80e40	120	void SetPIDLQslices(Int_t s);
8c25d014	121	void SetTailCancelation(Bool_t b=kTRUE) {if(b) SETBIT(fFlags, kTailCancelation); else CLRBIT(fFlags, kTailCancelation);}
828c6f80	122	void SetXenon(Bool_t b = kTRUE) {if(b) CLRBIT(fFlags, kDriftGas); else SETBIT(fFlags, kDriftGas);}
a2fbb6ec	123	void SetVertexConstrained() {SETBIT(fFlags, kVertexConstraint);}
	124	void SetMaxTheta(Double_t maxTheta) {fkMaxTheta = maxTheta;}
	125	void SetMaxPhi(Double_t maxPhi) {fkMaxPhi = maxPhi;}
	126	void SetFindableClusters(Double_t r) {fkFindable = r;}
	127	void SetChi2Y(Double_t chi2) {fkChi2Y = chi2;}
	128	void SetChi2Z(Double_t chi2) {fkChi2Z = chi2;}
	129	void SetChi2YSlope(Double_t chi2YSlope) {fkChi2YSlope = chi2YSlope;}
	130	void SetChi2ZSlope(Double_t chi2ZSlope) {fkChi2ZSlope = chi2ZSlope;}
	131	void SetChi2YCut(Double_t chi2Cut) {fkChi2YCut = chi2Cut; }
	132	void SetPhiSlope(Double_t phiSlope) {fkPhiSlope = phiSlope;}
	133	void SetNMeanClusters(Double_t meanNclusters) {fkNMeanClusters = meanNclusters;}
	134	void SetNSigmaClusters(Double_t sigmaNclusters) {fkNSigmaClusters = sigmaNclusters;}
240db9c7	135	void SetRawStreamVersion(const Char_t *version) {fRawStreamVersion = version; }
240db9c7	136	void SetRoadzMultiplicator(Double_t mult) {fkRoadzMultiplicator = mult; }
fb872574	137	void SetMinMaxCutSigma(Float_t minMaxCutSigma) { fMinMaxCutSigma = minMaxCutSigma; }
4ba1d6ae	138	void SetMinLeftRightCutSigma(Float_t minLeftRightCutSigma) { fMinLeftRightCutSigma = minLeftRightCutSigma; };
	139	void SetClusMaxThresh(Float_t thresh) { fClusMaxThresh = thresh; };
	140	void SetClusSigThresh(Float_t thresh) { fClusSigThresh = thresh; };
22a4ab0c	141	inline void SetPIDThreshold(Double_t *pid);
b06a50a5	142	void SetPtThreshold(Double_t pt) {fkPtThreshold = pt;}
4ba1d6ae	143	void SetNexponential(Int_t nexp) { fTCnexp = nexp; };
a2fbb6ec	144	inline void SetTCParams(Double_t *par);
a2fbb6ec	145	inline void SetStreamLevel(ETRDReconstructionTask task, Int_t level);
3afdab72	146	inline void SetSysCovMatrix(Double_t *sys);
fb872574	147	void SetNumberOfPresamples(Int_t n) { fNumberOfPresamples = n;}
fb872574	148	void SetNumberOfPostsamples(Int_t n) { fNumberOfPostsamples = n;}
b33337a6	149	void SetRecEveryTwoTB() { fRecEveryNTB = 2; fkNMeanClusters = 10; }
4ba1d6ae	150
3c66288b	151	private:
8ae98148	152	// Physics reference values for TRD
8ae98148	153	Double_t fkdNchdy; // dNch/dy
4ba1d6ae	154	Double_t fkMaxTheta; // Maximum theta
8ae98148	155	Double_t fkMaxPhi; // Maximum phi - momentum cut
b06a50a5	156	// Tracker params
4ba1d6ae	157	Double_t fkRoad0y; // Road for middle cluster
	158	Double_t fkRoad0z; // Road for middle cluster
	159
	160	Double_t fkRoad1y; // Road in y for seeded cluster
	161	Double_t fkRoad1z; // Road in z for seeded cluster
	162
	163	Double_t fkRoad2y; // Road in y for extrapolated cluster
	164	Double_t fkRoad2z; // Road in z for extrapolated cluster
b06a50a5	165	Double_t fkPtThreshold; // pt threshold for using TRD points for updating Kalaman track
4ba1d6ae	166	Double_t fkPlaneQualityThreshold; // Quality threshold
566bf887	167	Double_t fkRoadzMultiplicator; // Multiplicator for the Roads in z
8ae98148	168	Double_t fkFindable; // minimum ratio of clusters per tracklet supposed to be attached.
4ba1d6ae	169	Double_t fkChi2Z; // Max chi2 on the z direction for seeding clusters fit
4ba1d6ae	170	Double_t fkChi2Y; // Max chi2 on the y direction for seeding clusters Rieman fit
5a2e200c	171	Double_t fkChi2YSlope; // Slope of the chi2-distribution in y-direction
5a2e200c	172	Double_t fkChi2ZSlope; // Slope of the chi2-distribution in z-direction
8c499dbf	173	Double_t fkChi2YCut; // Cut on the Chi2 in y-direction in the likelihood filter
5a2e200c	174	Double_t fkPhiSlope; // Slope of the distribution of the deviation between track angle and tracklet angle
8ae98148	175	Double_t fkNMeanClusters; // Mean number of clusters per tracklet
	176	Double_t fkNSigmaClusters; // Sigma of the number of clusters per tracklet
	177	Double_t fkNClusterNoise; // ratio of noisy clusters to the true one
	178	Double_t fkNMeanTracklets; // Mean number of tracklets per track
4ba1d6ae	179	Double_t fkTrackLikelihood; // Track likelihood for tracklets Rieman fit
4ba1d6ae	180
3afdab72	181	Double_t fSysCovMatrix[5]; // Systematic uncertainty from calibration and alignment for each tracklet
4d6aee34	182	Double_t fPIDThreshold[AliTRDCalPID::kNMom]; // PID Thresholds for Electron candidate decision
8c499dbf	183	Int_t fNumberOfConfigs; // Used number of seed configurations
e4f2f73d	184
a2fbb6ec	185	// Reconstruction Options for TRD reconstruction
	186	Int_t fStreamLevel[kTRDreconstructionTasks]; // Stream Level
	187	Long64_t fFlags; // option Flags
9dcc64cc	188
a2fbb6ec	189	// Raw Reader Params
a2fbb6ec	190	TString fRawStreamVersion; // Raw Reader version
a2fbb6ec	191
a7ac01d2	192	// Clusterization parameter
4ba1d6ae	193	Double_t fMinMaxCutSigma; // Threshold sigma noise pad middle
	194	Double_t fMinLeftRightCutSigma; // Threshold sigma noise sum pad
	195	Double_t fClusMaxThresh; // Threshold value for cluster maximum
	196	Double_t fClusSigThresh; // Threshold value for cluster signal
4ba1d6ae	197	Int_t fTCnexp; // Number of exponentials, digital filter
a2fbb6ec	198	Double_t fTCParams[8]; // Tail Cancellation parameters for drift gases
8c499dbf	199	Int_t fRecEveryNTB; // Reconstruct each nth timebin
8c499dbf	200
4ba1d6ae	201	// ADC parameter
4e459a9d	202	Int_t fNumberOfPresamples; // number of presamples
4e459a9d	203	Int_t fNumberOfPostsamples; // number of postsamples
9716329b	204
7acbfbbd	205	ClassDef(AliTRDrecoParam, 12) // Reconstruction parameters for TRD detector
e4f2f73d	206
e4f2f73d	207	};
3afdab72	208
3afdab72	209	//___________________________________________________
7e88424f	210	inline void AliTRDrecoParam::GetSysCovMatrix(Double_t *sys) const
3afdab72	211	{
	212	if(!sys) return;
	213	memcpy(sys, fSysCovMatrix, 5*sizeof(Double_t));
	214	}
	215
	216	//___________________________________________________
	217	inline void AliTRDrecoParam::SetSysCovMatrix(Double_t *sys)
	218	{
	219	if(!sys) return;
	220	memcpy(fSysCovMatrix, sys, 5*sizeof(Double_t));
	221	}
	222
22a4ab0c	223	//___________________________________________________
	224	inline void AliTRDrecoParam::SetPIDThreshold(Double_t *pid)
	225	{
	226	if(!pid) return;
	227	memcpy(fPIDThreshold, pid, AliTRDCalPID::kNMom*sizeof(Double_t));
	228	}
3c66288b	229
a2fbb6ec	230	//___________________________________________________
	231	inline void AliTRDrecoParam::SetStreamLevel(ETRDReconstructionTask task, Int_t level){
	232	if(task >= kTRDreconstructionTasks) return;
	233	fStreamLevel[static_cast<Int_t>(task)] = level;
	234	}
	235
	236	//___________________________________________________
	237	inline Int_t AliTRDrecoParam::GetStreamLevel(ETRDReconstructionTask task) const{
	238	if(task >= kTRDreconstructionTasks) return 0;
	239	return fStreamLevel[static_cast<Int_t>(task)];
	240	}
	241
	242	//___________________________________________________
	243	inline void AliTRDrecoParam::GetTCParams(Double_t *par) const
	244	{
	245	if(!par) return;
	246	if(IsArgon()) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
	247	else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
	248	}
3c66288b	249
a2fbb6ec	250	//___________________________________________________
	251	inline void AliTRDrecoParam::SetTCParams(Double_t *par)
	252	{
	253	if(!par) return;
	254	memcpy(fTCParams, par, 8*sizeof(Double_t));
	255	}
11d80e40	256
	257	//___________________________________________________
	258	inline Int_t AliTRDrecoParam::GetPIDLQslices() const
	259	{
	260	if(IsPIDNeuralNetwork()) return -1;
	261	return TESTBIT(fFlags, kLQ2D) ? 2 : 1;
	262	}
	263
9dcc64cc	264	//___________________________________________________
	265	inline AliTRDPIDResponse::ETRDPIDMethod AliTRDrecoParam::GetPIDmethod() const
	266	{
	267	AliTRDPIDResponse::ETRDPIDMethod method = AliTRDPIDResponse::kLQ1D;
	268	if(IsPIDNeuralNetwork()) method = AliTRDPIDResponse::kNN;
	269	else if(TESTBIT(fFlags, kLQ2D)) method = AliTRDPIDResponse::kLQ2D;
	270	return method;
	271	}
	272
	273	//___________________________________________________
	274	inline void AliTRDrecoParam::SetPIDmethod(AliTRDPIDResponse::ETRDPIDMethod method)
	275	{
	276	switch(method){
	277	case AliTRDPIDResponse::kLQ2D:
	278	CLRBIT(fFlags, kSteerPID);
	279	SETBIT(fFlags, kLQ2D);
	280	break;
	281	case AliTRDPIDResponse::kNN:
	282	SETBIT(fFlags, kSteerPID);
	283	break;
	284	case AliTRDPIDResponse::kLQ1D:
	285	default:
	286	CLRBIT(fFlags, kSteerPID);
	287	CLRBIT(fFlags, kLQ2D);
	288	break;
	289	}
	290	}
	291
e4f2f73d	292	#endif