Be sure to load mapping when needed
[u/mrichter/AliRoot.git] / TRD / AliTRDrecoParam.h
index 69d1575..86325b1 100644 (file)
@@ -1,7 +1,7 @@
 #ifndef ALITRDRECOPARAM_H
 #define ALITRDRECOPARAM_H
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
- * See cxx source for full Copyright notice                               */
+* See cxx source for full Copyright notice                               */
 
 /* $Id$ */
 
 #include "AliDetectorRecoParam.h"
 #endif
 
+#ifndef ALITRDCALPID_H
+#include "Cal/AliTRDCalPID.h"
+#endif
+
+#ifndef ALITRDPIDRESPONSE_H
+#include "AliTRDPIDResponse.h"
+#endif
+
+class TString;
+
 class AliTRDrecoParam : public AliDetectorRecoParam
 {
-  public:
-       enum{
-         kNNslices = 8,
-         kLQslices = 3
-       };
-       
-       AliTRDrecoParam();
-       ~AliTRDrecoParam() { }
-
-       Double_t GetChi2Y() const                 { return fkChi2Y;    }
-       Double_t GetChi2Z() const                 { return fkChi2Z;    }
-       Bool_t   GetClusterSharing() const        { return fkClusterSharing;}
-       Double_t GetFindableClusters() const      { return fkFindable; }
-       Double_t GetMaxTheta() const              { return fkMaxTheta; }
-       Double_t GetMaxPhi() const                { return fkMaxPhi;   }
-       Int_t    GetNdEdxSlices() const           { return fkPIDMethod ? kNNslices : kLQslices;}
-       Int_t    GetPIDMethod() const             { return fkPIDMethod;}
-       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 GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
-
-       Double_t GetTrackLikelihood() const       { return fkTrackLikelihood;       }
-       Int_t    GetStreamLevel() const           { return fkStreamLevel;           }
-
-       Bool_t SeedingOn() const { return fSeedingOn; }
-       Bool_t IsVertexConstrained() const { return fVertexConstrained; }
-       
-        Double_t GetClusMaxThresh() const         { return fClusMaxThresh;   };
-        Double_t GetClusSigThresh() const         { return fClusSigThresh;   };
-        Int_t    GetTCnexp() const                { return fTCnexp;          };
-        Bool_t   LUTOn() const                    { return fLUTOn;           };
-        Bool_t   TCOn() const                     { return fTCOn;            };
-
-        Int_t    GetADCbaseline() const           { return fADCbaseline;     };
+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() { }
+
+  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 *GetHighFluxParam();
-        static   AliTRDrecoParam *GetCosmicTestParam();
-
-       void     SetClusterSharing(Bool_t share = kTRUE) { fkClusterSharing = share;  };
-       void     SetPIDMethod(Int_t pid = 1)             { fkPIDMethod = pid ? 1 : 0; };
-       void     SetSeedingOn(Bool_t seedingOn = kTRUE)  { fSeedingOn = seedingOn; }
-       void     SetVertexConstrained(Bool_t vertexConstrained = kTRUE) { fVertexConstrained = vertexConstrained; }
-       void     SetStreamLevel(Int_t streamLevel= 1) { fkStreamLevel = streamLevel; }
-        void     SetLUT(Int_t lutOn = 1)                 { fLUTOn           = lutOn;  };
-        void     SetClusMaxThresh(Float_t thresh)        { fClusMaxThresh   = thresh; };
-        void     SetClusSigThresh(Float_t thresh)        { fClusSigThresh   = thresh; };
-        void     SetTailCancelation(Int_t tcOn = 1)      { fTCOn            = tcOn;   };
-        void     SetNexponential(Int_t nexp)             { fTCnexp          = nexp;   };
-        void     SetADCbaseline(Int_t base)              { fADCbaseline     = base;   };
-
- private:
-
-       // Tracking parameters
-       Bool_t    fkClusterSharing;        // Toggle cluster sharing
-       Int_t     fkPIDMethod;             // PID method selector 0(LQ) 1(NN)
-       Double_t  fkMaxTheta;              // Maximum theta
-       Double_t  fkMaxPhi;                // Maximum phi
-
-       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;              // Ratio of clusters from a track in one chamber which are at minimum supposed to be found.
-       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  fkTrackLikelihood;       // Track likelihood for tracklets Rieman fit
-       Int_t     fkStreamLevel;                                         // Streaming Level in TRD Reconstruction
-       
-       Bool_t    fSeedingOn;                // Do stand alone tracking in the TRD
-       Bool_t    fVertexConstrained;      // Perform vertex constrained fit
-
-        // Clusterization parameter
-        Double_t  fClusMaxThresh;          // Threshold value for cluster maximum
-        Double_t  fClusSigThresh;          // Threshold value for cluster signal
-        Int_t     fLUTOn;                  // Switch for the lookup table method  
-        Int_t     fTCOn;                   // Switch for the tail cancelation
-        Int_t     fTCnexp;                 // Number of exponentials, digital filter
+  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
   
-       // ADC parameter
-        Int_t     fADCbaseline;            // ADC baseline to be subtracted
+  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
 
-       ClassDef(AliTRDrecoParam, 3)       // Reconstruction parameters for TRD detector
+  // 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