added protection

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

diff --git a/TRD/AliTRDclusterizer.h b/TRD/AliTRDclusterizer.h
index 955dc5144a42592e407b2a3135141e9ad05c24aa..87d2822acf69cc435a9e8c30d0577b4dead9dd5a 100644 (file)
--- a/TRD/AliTRDclusterizer.h
+++ b/TRD/AliTRDclusterizer.h
@@ -7,7 +7,7 @@
 
 ////////////////////////////////////////////////////////////////////////////
 //                                                                        //
-//  TRD cluster finder base class                                         //
+//  TRD cluster finder                                                    //
 //                                                                        //
 ////////////////////////////////////////////////////////////////////////////
 
@@ -18,9 +18,16 @@ 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 {
 
@@ -33,30 +40,63 @@ class AliTRDclusterizer : public TNamed {
   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   MakeClusters() = 0;
+
+  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();
 
-          TObjArray     *RecPoints();
-  virtual AliTRDcluster *AddCluster(Double_t *pos, Int_t timebin, Int_t det
-                                  , Double_t amp, Int_t *tracks
-                                 , Double_t *sig, Int_t iType
-                                  , Int_t col, UShort_t volid
-                                  , Float_t center = 0);
+  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:
 
-          Double_t CalcXposFromTimebin(Float_t timebin, Int_t idet, Int_t col, Int_t row);
-       
-          AliRunLoader    *fRunLoader;     //! Run Loader
-          TTree           *fClusterTree;   //! Tree with the cluster
-          TObjArray       *fRecPoints;     //! Array of clusters
+          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; 
+
+  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
+
+  Bool_t               fAddLabels;           //  Should clusters have MC labels?
+  Int_t                fRawVersion;          //  Expected raw version of the data - default is 2
+
+  AliTRDSignalIndex   *fIndexesOut;          //! Helper indexes for clusterization
+  AliTRDSignalIndex   *fIndexesMaxima;       //! Helper indexes for clusterization
+
+  AliTRDtransform     *fTransform;           //! Transforms the reconstructed space points
 
-  ClassDef(AliTRDclusterizer,4)            //  TRD-Cluster manager base class
+  ClassDef(AliTRDclusterizer,5)              //  TRD clusterfinder
 
 };