TRD/AliTRDrecoParam.h

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