Re-structured digitizer

[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.h

diff --git a/TRD/AliTRDclusterizer.h b/TRD/AliTRDclusterizer.h
index ab3089c96c5337bc965c002f50599aa402b8c395..87d2822acf69cc435a9e8c30d0577b4dead9dd5a 100644 (file)
--- a/TRD/AliTRDclusterizer.h
+++ b/TRD/AliTRDclusterizer.h
@@ -5,12 +5,29 @@
 
 /* $Id$ */
 
+////////////////////////////////////////////////////////////////////////////
+//                                                                        //
+//  TRD cluster finder                                                    //
+//                                                                        //
+////////////////////////////////////////////////////////////////////////////
+
 #include <TNamed.h>
-#include <TFile.h>
 
-///////////////////////////////////////////////////////
-//  Finds and handles cluster                        //
-///////////////////////////////////////////////////////
+class TFile;
+class TTree;
+class TObjArray;
+
+class AliRunLoader;
+class AliRawReader;
+
+class AliTRD;
+class AliTRDcluster;
+class AliTRDdataArrayS;
+class AliTRDdataArrayF;
+class AliTRDdigitsManager;
+class AliTRDSignalIndex;
+class AliTRDtransform;
+class AliTRDCalROC;
 
 class AliTRDclusterizer : public TNamed {
 
@@ -18,36 +35,69 @@ class AliTRDclusterizer : public TNamed {
 
   AliTRDclusterizer();
   AliTRDclusterizer(const Text_t* name, const Text_t* title);
-  AliTRDclusterizer(AliTRDclusterizer &c);
-  virtual ~AliTRDclusterizer();
-  
-  inline  AliTRDclusterizer &operator=(AliTRDclusterizer &c);
-  virtual void    Copy(AliTRDclusterizer &c);
-  virtual void    Init();
-  virtual Bool_t  Open(const Char_t *name, Int_t nEvent = 0);
-  virtual Bool_t  MakeCluster() = 0;
-  virtual Bool_t  WriteCluster();
+  AliTRDclusterizer(const AliTRDclusterizer &c);
+  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   OpenOutput(TTree *clusterTree);
+
+  virtual Bool_t   ReadDigits();
+  virtual Bool_t   ReadDigits(AliRawReader *rawReader);
+  virtual Bool_t   ReadDigits(TTree *digitsTree);
+
+  virtual Bool_t   WriteClusters(Int_t det);
+          void     ResetRecPoints();
+  TObjArray       *RecPoints();
+
+  virtual Bool_t   Raw2Clusters(AliRawReader *rawReader);
+  virtual Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
+
+  virtual Bool_t   MakeClusters();
+  virtual Bool_t   MakeClusters(Int_t det);
+
+  virtual Bool_t   AddLabels(Int_t idet, Int_t firstClusterROC, Int_t nClusterROC);
+  virtual Bool_t   SetAddLabels(Bool_t kset) { fAddLabels = kset; 
+                                               return fAddLabels;  } // should we assign labels to clusters
+  virtual void     SetRawVersion(Int_t iver) { fRawVersion = iver; } // set the expected raw data version
 
  protected:
 
-  TFile   *fInputFile;             //! AliROOT input file
-  
-  Int_t    fEvent;                 //! Event number
+          void     DeConvExp(Double_t *source, Double_t *target
+                           , Int_t nTimeTotal, Int_t nexp);
+         void     TailCancelation(AliTRDdataArrayS *digitsIn
+                                 , AliTRDdataArrayF *digitsOut 
+                                 , AliTRDSignalIndex *indexesIn
+                                , AliTRDSignalIndex *indexesOut
+                                , Int_t nTimeTotal
+                                , Float_t ADCthreshold
+                                , AliTRDCalROC *calGainFactorROC
+                                , Float_t calGainFactorDetValue);
+  virtual Double_t Unfold(Double_t eps, Int_t plane, Double_t *padSignal);
+          Double_t GetCOG(Double_t signal[5]) const; 
 
-  ClassDef(AliTRDclusterizer,1)    // TRD-Cluster manager base class
+  virtual void     ResetHelperIndexes(AliTRDSignalIndex *indexesIn);
 
-};
+  AliRunLoader        *fRunLoader;           //! Run Loader
+  TTree               *fClusterTree;         //! Tree with the cluster
+  TObjArray           *fRecPoints;           //! Array of clusters
+
+  AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager
 
-//_____________________________________________________________________________
-AliTRDclusterizer &AliTRDclusterizer::operator=(AliTRDclusterizer &c)
-{
-  //
-  // Assignment operator
-  //
+  Bool_t               fAddLabels;           //  Should clusters have MC labels?
+  Int_t                fRawVersion;          //  Expected raw version of the data - default is 2
 
-  if (this != &c) c.Copy(*this);
-  return *this;
+  AliTRDSignalIndex   *fIndexesOut;          //! Helper indexes for clusterization
+  AliTRDSignalIndex   *fIndexesMaxima;       //! Helper indexes for clusterization
 
-}
+  AliTRDtransform     *fTransform;           //! Transforms the reconstructed space points
+
+  ClassDef(AliTRDclusterizer,5)              //  TRD clusterfinder
+
+};
 
 #endif