#ifndef ALITRDRECOPARAM_H
#define ALITRDRECOPARAM_H
#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 GetChi2Cut() const { return fChi2Cut; }
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; }
Int_t GetPIDLQslices() const;
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 ) 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; }
void GetSysCovMatrix(Double_t *sys) const;
void GetTCParams(Double_t *par) const;
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; }
Double_t GetCorrDZDXbiasRC(Bool_t dzdx) const { return fdzdxCorrRCbias[dzdx];}
Double_t GetCorrDZDX(Bool_t rc) const { return fdzdxCorrFactor[rc];}
Double_t GetCorrDZDXxcross() const { return fdzdxXcrossFactor;}
void GetYcorrTailCancel(Int_t it, Double_t par[3]) const;
Double_t GetS2Ycorr(Bool_t rc, Bool_t chg) const { return fS2Ycorr[2*rc+chg];}
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);}
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 SetChi2Cut(Double_t chi2Cut) {fChi2Cut = chi2Cut; }
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; };
void SetPIDThreshold(Double_t *pid);
void SetPtThreshold(Double_t pt) {fkPtThreshold = pt;}
void SetNexponential(Int_t nexp) { fTCnexp = nexp; };
void SetTCParams(Double_t *par);
void SetTrackletParams(Double_t *par=NULL);
void SetStreamLevel(ETRDReconstructionTask task, Int_t level);
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:
Double_t fkdNchdy;
Double_t fkMaxTheta;
Double_t fkMaxPhi;
Double_t fkRoad0y;
Double_t fkRoad0z;
Double_t fkRoad1y;
Double_t fkRoad1z;
Double_t fkRoad2y;
Double_t fkRoad2z;
Double_t fkPtThreshold;
Double_t fkPlaneQualityThreshold;
Double_t fkRoadzMultiplicator;
Double_t fkFindable;
Double_t fkChi2Z;
Double_t fkChi2Y;
Double_t fkChi2YSlope;
Double_t fkChi2ZSlope;
Double_t fChi2Cut;
Double_t fkChi2YCut;
Double_t fkPhiSlope;
Double_t fkNMeanClusters;
Double_t fkNSigmaClusters;
Double_t fkNClusterNoise;
Double_t fkNMeanTracklets;
Double_t fkTrackLikelihood;
Double_t fSysCovMatrix[5];
Double_t fPIDThreshold[AliTRDCalPID::kNMom];
Int_t fNumberOfConfigs;
Int_t fStreamLevel[kTRDreconstructionTasks];
Long64_t fFlags;
TString fRawStreamVersion;
Double_t fdzdxCorrFactor[2];
Double_t fdzdxCorrRCbias[2];
Double_t fdzdxXcrossFactor;
Double_t fYcorrTailCancel[4][3];
Double_t fS2Ycorr[4];
Double_t fMinMaxCutSigma;
Double_t fMinLeftRightCutSigma;
Double_t fClusMaxThresh;
Double_t fClusSigThresh;
Int_t fTCnexp;
Double_t fTCParams[8];
Int_t fRecEveryNTB;
Int_t fNumberOfPresamples;
Int_t fNumberOfPostsamples;
ClassDef(AliTRDrecoParam, 13)
};
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 void AliTRDrecoParam::GetYcorrTailCancel(Int_t it, Double_t par[3]) const
{
if(it<0||it>3) return;
memcpy(par, fYcorrTailCancel[it], 3*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