]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDrecoParam.h
bug fix and small changed in the macros
[u/mrichter/AliRoot.git] / TRD / AliTRDrecoParam.h
index d07996be91b6c767b46208d48a2b96230684a6ad..cc5f3691ce22f74401fe364181a5266c63ed018f 100644 (file)
@@ -30,16 +30,18 @@ public:
     kTRDreconstructionTasks = 3
   };
   enum ETRDflags {
-    kDriftGas,
-    kVertexConstraint,
-    kTailCancelation,
-    kImproveTracklet, 
-    kLUT, 
-    kGAUS,
-    kClusterSharing,
-    kSteerPID,
-    kEightSlices  
-   };
+    kDriftGas
+    ,kVertexConstraint
+    ,kTailCancelation
+    ,kImproveTracklet
+    ,kLUT
+    ,kGAUS
+    ,kClusterSharing
+    ,kSteerPID
+    ,kEightSlices
+    ,kCheckTimeConsistency
+    ,kLQ2D
+  };
   AliTRDrecoParam();
   AliTRDrecoParam(const AliTRDrecoParam &rec);
   ~AliTRDrecoParam() { }
@@ -54,6 +56,7 @@ public:
   Double_t GetNMeanClusters() const         { return fkNMeanClusters; }
   Double_t GetNSigmaClusters() const        { return fkNSigmaClusters; }
   Double_t GetFindableClusters() const      { return fkFindable; }
+  inline Int_t    GetPIDLQslices() const;
   Double_t GetMaxTheta() const              { return fkMaxTheta; }
   Double_t GetMaxPhi() const                { return fkMaxPhi;   }
   Double_t GetPlaneQualityThreshold() const { return fkPlaneQualityThreshold; }
@@ -69,19 +72,20 @@ public:
   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; };
-  Int_t    GetADCBaseline() const           { return fADCBaseline; }
+  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;}
-  Bool_t   IsArgon() const { return TESTBIT(fFlags, kDriftGas); }
-  Bool_t   IsXenon() const { return !TESTBIT(fFlags, kDriftGas); }
+  Int_t    GetNumberOfPresamples()  const   { return fNumberOfPresamples;}
+  Int_t    GetNumberOfPostsamples() const   { return fNumberOfPostsamples;}
+  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   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);}
@@ -93,15 +97,17 @@ public:
   static   AliTRDrecoParam *GetHighFluxParam();
   static   AliTRDrecoParam *GetCosmicTestParam();
 
-  void     SetArgon()                                         {SETBIT(fFlags, kDriftGas);}
-  void     SetClusterSharing()                                {SETBIT(fFlags, kClusterSharing);}
-  void     SetEightSlices()                                   {SETBIT(fFlags, kEightSlices);}
-  void     SetImproveTracklets()                              {SETBIT(fFlags, kImproveTracklet);}
-  void     SetLUT()                                           {SETBIT(fFlags, kLUT);}
-  void     SetGAUS()                                          {SETBIT(fFlags, kGAUS);}
-  void     SetPIDNeuralNetwork()                              {SETBIT(fFlags, kSteerPID);}
-  void     SetTailCancelation()                               {SETBIT(fFlags, kTailCancelation);}
-  void     SetXenon()                                         {CLRBIT(fFlags, kDriftGas);}
+  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     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;}
@@ -114,13 +120,14 @@ public:
   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     SetADCBaseline(Int_t baseline)                  { fADCBaseline = baseline; }
+  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);
@@ -133,7 +140,7 @@ private:
   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
 
@@ -142,7 +149,7 @@ private:
 
   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.
@@ -167,7 +174,6 @@ private:
   
   // Raw Reader Params
   TString   fRawStreamVersion;       // Raw Reader version
-  Int_t     fADCBaseline;            // ADC Baseline
 
   // Clusterization parameter
   Double_t  fMinMaxCutSigma;         // Threshold sigma noise pad middle
@@ -181,7 +187,7 @@ private:
   Int_t     fNumberOfPresamples;     // number of presamples 
   Int_t     fNumberOfPostsamples;     // number of postsamples 
 
-  ClassDef(AliTRDrecoParam, 9)       // Reconstruction parameters for TRD detector
+  ClassDef(AliTRDrecoParam, 11)       // Reconstruction parameters for TRD detector
 
 };
 
@@ -232,4 +238,12 @@ 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;
+}
+
 #endif