]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDclusterizer.h
Making online tracklets usable in offline reconstruction
[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.h
index 0b01ccf25a5f4fbc78f9b7b1cf7f6977ee62bac0..f6adbf39c4d521682c4ec8edd801b90e6cf248b4 100644 (file)
 
 /* $Id$ */
 
+////////////////////////////////////////////////////////////////////////////
+//                                                                        //
+//  TRD cluster finder                                                    //
+//                                                                        //
+////////////////////////////////////////////////////////////////////////////
+
 #include <TNamed.h>
-#include <TObjArray.h>
 
 class TFile;
 class TTree;
+class TClonesArray;
+
 class AliRunLoader;
-class AliTRDparameter;
+class AliRawReader;
+
 class AliTRD;
 class AliTRDcluster;
-///////////////////////////////////////////////////////
-//  Finds and handles cluster                        //
-///////////////////////////////////////////////////////
 
-class AliTRDclusterizer : public TNamed {
+class AliTRDarrayADC;
+class AliTRDarraySignal;
+class AliTRDdigitsManager;
+class AliTRDSignalIndex;
+class AliTRDtransform;
+class AliTRDCalROC;
+class AliTRDReconstructor;
+class AliTRDCalSingleChamberStatus;
+
+class AliTRDclusterizer : public TNamed 
+{
 
  public:
 
-  AliTRDclusterizer();
-  AliTRDclusterizer(const Text_t* name, const Text_t* title);
+  // steering flags
+  enum{
+    kTrOwner= BIT(13)  //  toggle online tracklets ownership
+   ,kClOwner= BIT(14)  //  toggle cluster ownership
+   ,kLabels = BIT(15)  //  toggle MC labels for clusters
+   ,kHLT    = BIT(16)  //  HLT mode
+   ,kLUT    = BIT(17)  //  using look up table for cluster's r-phi position
+   ,kGAUS   = BIT(18)  //  using gauss approx. for cluster's r-phi position
+  };
+
+  struct MaxStruct
+  {
+    Int_t       Row;
+    Int_t       Col;
+    Int_t       Time;
+    UChar_t     padStatus;
+    Short_t     Signals[3];
+    Bool_t      FivePad;
+  MaxStruct():Row(-1),Col(0),Time(0),padStatus(0),FivePad(kFALSE)
+      {memset(Signals, 0, 3*sizeof(Short_t));}
+    MaxStruct &operator=(const MaxStruct &a)
+    {Row=a.Row; Col=a.Col; Time=a.Time; padStatus=a.padStatus; FivePad=a.FivePad;
+     memcpy(Signals, a.Signals, 3*sizeof(Short_t)); return *this;}
+  };
+  
+  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();
+  virtual         ~AliTRDclusterizer();
   AliTRDclusterizer &operator=(const AliTRDclusterizer &c);
 
-  virtual void    Copy(TObject &c) const;
-  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  MakeClusters() = 0;
-  virtual Bool_t  WriteClusters(Int_t det);
-  virtual void    SetParameter(AliTRDparameter *par)      { fPar           = par; };
-  void            SetVerbose(Int_t v = 1)                 { fVerbose       = v;   };
+  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 *clusterTree);
+
+  Bool_t   ReadDigits();
+  Bool_t   ReadDigits(AliRawReader *rawReader);
+  Bool_t   ReadDigits(TTree *digitsTree);
+
+  Bool_t   WriteClusters(Int_t det);
+  void     ResetRecPoints();
+  virtual TClonesArray    *RecPoints();
+  TClonesArray    *TrackletsArray();
+  Bool_t   WriteTracklets(Int_t det);
 
-  AliTRDparameter *GetParameter()                   const { return fPar;          };
+  Bool_t   Raw2Clusters(AliRawReader *rawReader);
+  Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
 
-  TObjArray*      RecPoints() {if (!fRecPoints) fRecPoints = new TObjArray(400); return fRecPoints;}
-  virtual AliTRDcluster  * AddCluster(Double_t *pos, Int_t timebin, Int_t det, Double_t amp, Int_t *tracks
-                            , Double_t *sig, Int_t iType, Float_t center = 0);
-  void            ResetRecPoints() {if (fRecPoints) fRecPoints->Delete();}
+  Bool_t   MakeClusters();
+  Bool_t   MakeClusters(Int_t det);
 
- protected:
+  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() {return fNoOfClusters;}
 
-   Double_t CalcXposFromTimebin(Float_t timebin, Int_t idet, Int_t col, Int_t row);
-       
-  AliRunLoader    *fRunLoader;     //! Run Loader
+  Bool_t   IsClustersOwner() const {return TestBit(kClOwner);}
+  virtual void     SetClustersOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
+  void     SetTrackletsOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fTracklets = 0x0; } }
+
+protected:
+
+  void             DeConvExp (const Double_t *const source, Double_t *const target
+                            ,const Int_t nTimeTotal, const Int_t nexp);
+  void             TailCancelation();
+
+  Float_t  Unfold(Double_t eps, Int_t layer, Double_t *padSignal) const;
   
-  TTree           *fClusterTree;   //! Tree with the cluster
-  AliTRDparameter *fPar;           //  TRD digitization parameter object
+  void             SetPadStatus(const UChar_t status, UChar_t &encoding);
+  UChar_t          GetPadStatus(UChar_t encoding) const;
+  Int_t            GetCorruption(UChar_t encoding) const;
+
+  Bool_t           IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Short_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); 
+  inline void      CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);
+  virtual void     AddClusterToArray(AliTRDcluster *cluster);
+  inline void      AddTrackletsToArray();
+
+  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
+
+  TTree               *fTrackletTree;        //! Tree for tracklets
+
+  AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager
+
+  UInt_t              **fTrackletContainer;  //! tracklet container
+
+  Int_t                fRawVersion;          //  Expected raw version of the data - default is 2
+
+  AliTRDtransform     *fTransform;           //! Transforms the reconstructed space points
 
-  TObjArray*       fRecPoints;     //! Array of clusters
-  Int_t            fVerbose;       //  Sets the verbose level
+  AliTRDarrayADC      *fDigits;               // Array holding the digits
+  AliTRDSignalIndex   *fIndexes;              // Array holding the indexes to the digits
+  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
+  AliTRDCalSingleChamberStatus *fCalPadStatusROC; // Calibration object with the pad status
+  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
 
-  ClassDef(AliTRDclusterizer,3)    //  TRD-Cluster manager base class
+  ClassDef(AliTRDclusterizer,6)               //  TRD clusterfinder
 
 };