]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDclusterizer.h
Split dEdxUtils into dEdxBaseUtils, dEdxCalibUtils, dEdxReconUtils and one CalibHistA...
[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.h
index b4b7f2b97843eab8255aa56b9c64635b57fd645c..fe590dd71e5db3a589fa07ad9f4c2088c4b5aacd 100644 (file)
@@ -13,6 +13,8 @@
 
 #include <TNamed.h>
 
+#include "AliTRDgeometry.h"
+
 class TFile;
 class TTree;
 class TClonesArray;
@@ -23,6 +25,7 @@ class AliRawReader;
 class AliTRD;
 class AliTRDcluster;
 
+class AliTRDrawStream;
 class AliTRDarrayADC;
 class AliTRDarraySignal;
 class AliTRDdigitsManager;
@@ -30,6 +33,9 @@ class AliTRDSignalIndex;
 class AliTRDtransform;
 class AliTRDCalROC;
 class AliTRDReconstructor;
+class AliTRDCalSingleChamberStatus;
+class AliTRDrecoParam;
+class AliTRDCalOnlineGainTableROC;
 
 class AliTRDclusterizer : public TNamed 
 {
@@ -37,123 +43,150 @@ class AliTRDclusterizer : public TNamed
  public:
 
   // steering flags
+  // bits from 0-13 are reserved by ROOT (see TObject.h)
   enum{
-    kOwner = BIT(14)
+    kTrOwner = BIT(14)  //  toggle online tracklets ownership
+    ,kClOwner= BIT(15)  //  toggle cluster ownership
+    ,kLabels = BIT(16)  //  toggle MC labels for clusters
+    ,kSkipTrafo = BIT(17)  //  skip the coordinate transformation of clusters?
+    ,kLUT    = BIT(18)  //  using look up table for cluster's r-phi position
+    ,kGAUS   = BIT(19)  //  using gauss approx. for cluster's r-phi position
+    ,knewDM  = BIT(20)  //  was the digitsmanger created by raw2clusters?
+    ,kTracksOwner = BIT(21) //  toggle GTU track ownership
+    ,kRawSignal = BIT(22) //  toggle storage of raw signals in clusters
+  };
+
+  struct MaxStruct
+  {
+    Int_t       row;           // row of the current cluster candidate
+    Int_t       col;           // col of the current cluster candidate
+    Int_t       time;          // time -"-
+    Float_t     signals[3];    // signals of the maximum pad and it's twon neigbours
+    UChar_t     padStatus;     // padStatus of the current cluster candidate
+    Bool_t      fivePad;       // is this cluster candidate part of a 5 pad cluster (two overlaping clusters)?
+  MaxStruct():row(-1),col(0),time(0),padStatus(0),fivePad(kFALSE)
+      {memset(signals, 0, 3*sizeof(Float_t));}
+    MaxStruct &operator=(const MaxStruct &a)
+    {memcpy(this, &a, sizeof(MaxStruct)); return *this;}
   };
   
-  AliTRDclusterizer(AliTRDReconstructor *rec = 0x0);
-  AliTRDclusterizer(const Text_t* name, const Text_t* title, AliTRDReconstructor *rec = 0x0);
+  AliTRDclusterizer(const AliTRDReconstructor *const rec = 0x0);
+  AliTRDclusterizer(const Text_t* name, const Text_t* title, const AliTRDReconstructor *const rec = 0x0);
   AliTRDclusterizer(const AliTRDclusterizer &c);
   virtual         ~AliTRDclusterizer();
   AliTRDclusterizer &operator=(const AliTRDclusterizer &c);
 
-  virtual void     Copy(TObject &c) const;
+  void     Copy(TObject &c) const;
+
+  Bool_t   Open(const Char_t *name, Int_t nEvent = 0);
+  Bool_t   OpenInput(Int_t nEvent = 0);
+  Bool_t   OpenOutput();
+  Bool_t   OpenOutput(TTree *const clusterTree);
 
-  virtual Bool_t   Open(const Char_t *name, Int_t nEvent = 0);
-  virtual Bool_t   OpenInput(Int_t nEvent = 0);
-  virtual Bool_t   OpenOutput();
-  virtual Bool_t   OpenOutput(TTree *clusterTree);
+  Bool_t   ReadDigits();
+  Bool_t   ReadDigits(AliRawReader *rawReader);
+  Bool_t   ReadDigits(TTree *digitsTree);
 
-  virtual Bool_t   ReadDigits();
-  virtual Bool_t   ReadDigits(AliRawReader *rawReader);
-  virtual Bool_t   ReadDigits(TTree *digitsTree);
+  Bool_t   ReadTracklets();
+  Bool_t   ReadTracks();
 
-  virtual Bool_t   WriteClusters(Int_t det);
-          void     ResetRecPoints();
-  TClonesArray       *RecPoints();
-          Bool_t   WriteTracklets(Int_t det);
+  Bool_t   WriteClusters(Int_t det);
+  void     ResetRecPoints();
+  virtual TClonesArray    *RecPoints();
+  virtual TClonesArray    *TrackletsArray(const TString &trkltype = "");
+  virtual TClonesArray    *TracksArray();
 
-  virtual Bool_t   Raw2Clusters(AliRawReader *rawReader);
-  virtual Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
+  Bool_t   Raw2Clusters(AliRawReader *rawReader);
+  Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
 
-  virtual Bool_t   MakeClusters();
-  virtual Bool_t   MakeClusters(Int_t det);
+  Bool_t   MakeClusters();
+  Bool_t   MakeClusters(Int_t det);
 
-  virtual Bool_t   AddLabels(const Int_t idet, const Int_t firstClusterROC, const Int_t nClusterROC);
-  virtual Bool_t   SetAddLabels(const Bool_t kset) { fAddLabels = kset; return fAddLabels;  } // should we assign labels to clusters
-  virtual void     SetRawVersion(const Int_t iver) { fRawVersion = iver; } // set the expected raw data version
+  Bool_t   AddLabels();
+  Bool_t   SetUseLabels(const Bool_t kset) { SetBit(kLabels, kset); return TestBit(kLabels);  } // should we assign labels to clusters
+  void     SetRawVersion(const Int_t iver) { fRawVersion = iver; } // set the expected raw data version
   void             SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
   static UChar_t   GetStatus(Short_t &signal);
+  Int_t            GetAddedClusters() const {return fNoOfClusters;}
+  Int_t            GetNTimeBins() const {return fTimeTotal;}
 
-  Bool_t           IsClustersOwner() const {return TestBit(kOwner);}
-  void             SetClustersOwner(Bool_t own=kTRUE) {SetBit(kOwner, own); if(!own) fRecPoints = 0x0;}
+  Bool_t           IsClustersOwner() const {return TestBit(kClOwner);}
+  virtual void     SetClustersOwner(Bool_t own=kTRUE) {SetBit(kClOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
+  void             SetTrackletsOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fTracklets = 0x0; } }
+  void             SetTracksOwner(Bool_t own=kTRUE) {SetBit(kTracksOwner, own); if(!own) {fTracks = 0x0; } }
+  void             SetSkipTransform(Bool_t b=kTRUE) {SetBit(kSkipTrafo, b); }
+  void             SetStoreRawSignals(Bool_t b=kTRUE) {SetBit(kRawSignal, b); }
 
- protected:
+  UInt_t   GetTriggerFlags(const Int_t sector) const { return fTrgFlags[sector]; }
 
-  void             DeConvExp (const Double_t *const source, Double_t *const target
-                            ,const Int_t nTimeTotal, const Int_t nexp);
-  void             TailCancelation();
+protected:
 
-  virtual Double_t Unfold(Double_t eps, Int_t layer, Double_t *padSignal) const;
-          Double_t GetCOG(Double_t signal[5]) const; 
-  void             FillLUT();
-          Double_t LUTposition(Int_t ilayer, Double_t ampL, Double_t ampC, Double_t ampR) const;
+  void             ApplyTCTM(Short_t *const arr, const Int_t nTime, const Int_t nexp);
+  void             DeConvExp (Short_t *const arr, const Int_t nTime, const Int_t nexp);
+  void             ConvExp(Short_t *const arr, const Int_t nTime);
+  void             TailCancelation(const AliTRDrecoParam* const recoParam);
+
+  Float_t          Unfold(Double_t eps, Int_t layer, const Double_t *const padSignal) const;
   
-  void             SetPadStatus(const UChar_t status, UChar_t &encoding);
+  void             SetPadStatus(const UChar_t status, UChar_t &encoding) const;
   UChar_t          GetPadStatus(UChar_t encoding) const;
   Int_t            GetCorruption(UChar_t encoding) const;
 
-  Bool_t           IsMaximum(const Int_t row, const Int_t col, const Int_t time, 
-                            UChar_t &pasStatus, Double_t *const Signals);
-  Bool_t           IsFivePadCluster(const Int_t row, const Int_t col, const Int_t time, 
-                                   Double_t *SignalsThisMax, Double_t *SignalsNeighbourMax, Double_t &ratio);
-  void             CreateCluster(const Int_t row, const Int_t col, const Int_t time, 
-                                const Double_t *const clusterSignal, const UChar_t padStatus);
-
-  const AliTRDReconstructor *fReconstructor;       //! reconstructor
-  AliRunLoader        *fRunLoader;           //! Run Loader
-  TTree               *fClusterTree;         //! Tree with the cluster
-  TClonesArray        *fRecPoints;           //! Array of clusters
-
-  TTree               *fTrackletTree;        //! Tree for tracklets
-
-  AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager
-
-  UInt_t              **fTrackletContainer;  //! tracklet container
-
-  Bool_t               fAddLabels;           //  Should clusters have MC labels?
-  Int_t                fRawVersion;          //  Expected raw version of the data - default is 2
-
-  AliTRDtransform     *fTransform;           //! Transforms the reconstructed space points
-
-  Int_t                fLUTbin;              //  Number of bins of the LUT
-  Double_t            *fLUT;                 //! The lookup table
-
-  AliTRDarrayADC      *fDigitsIn;
-  AliTRDSignalIndex   *fIndexes;
-  Float_t              fADCthresh;            // ADC thresholds: There is no ADC threshold anymore, and simParam should not be used in clusterizer. KO
-  Float_t              fMaxThresh;            // Threshold value for the maximum
-  Float_t              fSigThresh;            // Threshold value for the digit signal
-  Float_t              fMinMaxCutSigma;       // Threshold value for the maximum (cut noise)
-  Float_t              fMinLeftRightCutSigma; // Threshold value for the sum pad (cut noise)
-  Int_t                fLayer;                // Current layer of the detector
-  Int_t                fDet;                  // Current detecor
-  UShort_t             fVolid;                // Volume ID
-  Int_t                fColMax;               // Number of Colums in one detector
-  Int_t                fTimeTotal;            // Number of time bins
-  AliTRDCalROC        *fCalGainFactorROC;     // Calibration object with pad wise values for the gain factors
-  Float_t              fCalGainFactorDetValue;// Calibration value for chamber wise noise
-  AliTRDCalROC        *fCalNoiseROC;          // Calibration object with pad wise values for the noise
-  Float_t              fCalNoiseDetValue;     // Calibration value for chamber wise noise
-  AliTRDarraySignal   *fDigitsOut;
-  Int_t                fClusterROC;           // The index to the first cluster of a given ROC
-  Int_t                firstClusterROC;       // The number of cluster in a given ROC
-
-  struct ClusterizerStruct
-  {
-    Int_t       Row;
-    Int_t       Col;
-    Int_t       Time;
-    UChar_t     padStatus;
-    Double_t    Signals[3];
-    ClusterizerStruct():Row(0),Col(0),Time(0),padStatus(0)
-      {}
-    ClusterizerStruct &operator=(const ClusterizerStruct &a)
-      {Row=a.Row; Col=a.Col; Time=a.Time; padStatus=a.padStatus;
-       memcpy(Signals, a.Signals, 3*sizeof(Double_t)); return *this;}
-  };
-
-  ClassDef(AliTRDclusterizer,6)              //  TRD clusterfinder
+  Bool_t           IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Float_t *const Signals);       //for const correctness reasons not const parameters are given separately
+  Bool_t           FivePadCluster(MaxStruct &ThisMax, MaxStruct &NeighbourMax);
+  void             CreateCluster(const MaxStruct &Max); 
+
+  virtual void     AddClusterToArray(AliTRDcluster* cluster);
+
+  const AliTRDReconstructor *fReconstructor;     //! reconstructor
+  AliRunLoader        *fRunLoader;                //! Run Loader
+  TTree               *fClusterTree;              //! Tree with the cluster
+  TClonesArray        *fRecPoints;                //! Array of clusters
+  TClonesArray        *fTracklets;                //! Array of online tracklets
+  TClonesArray        *fTracks;                   //! Array of GTU tracks
+
+  TTree               *fTrackletTree;             //! Tree for tracklets
+
+  AliTRDdigitsManager *fDigitsManager;            //! TRD digits manager
+
+  UInt_t              **fTrackletContainer;       //! tracklet container
+                                                 // legacy code to avoid breakint AliHLTTRDClusterizer
+                                                 // but it's useless
+
+  Int_t                fRawVersion;               //  Expected raw version of the data - default is 2
+
+  AliTRDtransform     *fTransform;                //! Transforms the reconstructed space points
+
+  AliTRDarrayADC      *fDigits;                   // Array holding the digits
+  AliTRDarrayADC      *fDigitsRaw;                // Array holding the raw digits if requested
+  AliTRDSignalIndex   *fIndexes;                  // Array holding the indexes to the digits
+  Float_t              fMaxThresh;                // Threshold value for the maximum
+  Float_t              fMaxThreshTest;            // Threshold value for the maximum to test agains
+  Float_t              fSigThresh;                // Threshold value for the digit signal
+  Float_t              fMinMaxCutSigma;           // Threshold value for the maximum (cut noise)
+  Float_t              fMinLeftRightCutSigma;     // Threshold value for the sum pad (cut noise)
+  Int_t                fLayer;                    // Current layer of the detector
+  Int_t                fDet;                      // Current detecor
+  UShort_t             fVolid;                    // Volume ID
+  Int_t                fColMax;                   // Number of Colums in one detector
+  Int_t                fTimeTotal;                // Number of time bins
+  AliTRDCalROC        *fCalGainFactorROC;         // Calibration object with pad wise values for the gain factors
+  Float_t              fCalGainFactorDetValue;    // Calibration value for chamber wise noise
+  AliTRDCalROC        *fCalNoiseROC;              // Calibration object with pad wise values for the noise
+  Float_t              fCalNoiseDetValue;         // Calibration value for chamber wise noise
+  AliTRDCalSingleChamberStatus *fCalPadStatusROC; // Calibration object with the pad status
+  AliTRDCalOnlineGainTableROC *fCalOnlGainROC;     // Calibration table of online gain factors
+  Int_t                fClusterROC;               // The index to the first cluster of a given ROC
+  Int_t                firstClusterROC;           // The number of cluster in a given ROC
+  Int_t                fNoOfClusters;             // Number of Clusters already processed and still owned by the clusterizer
+  Int_t                fBaseline;                 // Baseline of the ADC values
+  AliTRDrawStream     *fRawStream;                // Raw data streamer
+  UInt_t               fTrgFlags[AliTRDgeometry::kNsector]; // trigger flags
+
+private:
+  inline void      CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);
+
+  ClassDef(AliTRDclusterizer,6)                   //  TRD clusterfinder
 
 };