few changes by Thomas Kuhr

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

diff --git a/TRD/AliTRDdigitsManager.h b/TRD/AliTRDdigitsManager.h
index 3170692562c2b38854cfedec9df75b9e42e3d710..21418ac08f7de692033b901d9edeb3ee944c37db 100644 (file)
--- a/TRD/AliTRDdigitsManager.h
+++ b/TRD/AliTRDdigitsManager.h
@@ -10,49 +10,71 @@
 //  Manages the TRD digits                                 //
 /////////////////////////////////////////////////////////////
 
-#include "TObject.h"
+#include <TObject.h>
 
-#include "AliTRDsegmentArray.h"
-#include "AliTRDdataArrayI.h"
-#include "AliTRDdigit.h"
+class TFile;
+class TTree;
 
-const Int_t  kNDict = 3;
+class AliTRDsegmentArray;
+class AliTRDdataArrayI;
+class AliTRDdigit;
 
 class AliTRDdigitsManager : public TObject {
 
  public:
 
+  enum { kNDict = 3 };
+
   AliTRDdigitsManager();
   AliTRDdigitsManager(const AliTRDdigitsManager &m);
   virtual ~AliTRDdigitsManager();
   AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);
 
+  virtual void                CreateArrays();
+  virtual void                ResetArrays();
   virtual void                Copy(TObject &m);
-  virtual Bool_t              MakeBranch();
-  virtual Bool_t              ReadDigits();
+
+  virtual Bool_t              MakeBranch(TTree *tree);
+          
+  virtual Bool_t              ReadDigits(TTree *tree);
   virtual Bool_t              WriteDigits();
 
   virtual void                SetRaw();
+  virtual void                SetEvent(Int_t evt)          { fEvent   = evt; };
+  virtual void                SetDebug(Int_t v = 1)        { fDebug   = v;   };
+  virtual void                SetSDigits(Int_t v = 1)      { fSDigits = v;   };
+
+  virtual Bool_t              IsRaw() const                { return fIsRaw;         };
+  static  Int_t               NDict()                      { return fgkNDict;       }; 
 
-  virtual Bool_t              IsRaw()                { return fIsRaw;         };
-  virtual AliTRDsegmentArray *GetDigits()            { return fDigits;        };
-  virtual AliTRDsegmentArray *GetDictionary(Int_t i) { return fDictionary[i]; };
+  virtual AliTRDsegmentArray *GetDigits() const            { return fDigits;        };
+  virtual AliTRDsegmentArray *GetDictionary(Int_t i) const { return fDictionary[i]; };
 
-          AliTRDdigit        *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det);
-          Int_t               GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det);
+          AliTRDdigit        *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det) const;
+          Int_t               GetTrack(Int_t track, Int_t row, Int_t col
+                                     , Int_t time, Int_t det) const;
 
-  inline  AliTRDdataArrayI   *GetDigits(Int_t det);
-  inline  AliTRDdataArrayI   *GetDictionary(Int_t det, Int_t i);
-  inline  Int_t               GetTrack(Int_t track, AliTRDdigit *Digit);
+          AliTRDdataArrayI   *GetDigits(Int_t det) const;
+          AliTRDdataArrayI   *GetDictionary(Int_t det, Int_t i) const;
+          Int_t               GetTrack(Int_t track, AliTRDdigit *Digit) const;
+          Short_t             GetDigitAmp(Int_t row, Int_t col, Int_t time, Int_t det) const;
 
  protected:
 
-  AliTRDsegmentArray *fDigits;             //! Digits data Array
+  static const Int_t  fgkNDict;            //  Number of track dictionary arrays
+
+  Int_t               fEvent;              //  Event number
+
+  TTree              *fTree;               //! Tree for the digits arrays
+
+  AliTRDsegmentArray *fDigits;             //! Digits data array
   AliTRDsegmentArray *fDictionary[kNDict]; //! Track dictionary data array
 
   Bool_t              fIsRaw;              //  Flag indicating raw digits
+  Bool_t              fSDigits;            //  Switch for the summable digits
+  Int_t               fDebug;              //  Debug flag
 
-  ClassDef(AliTRDdigitsManager,1)          //  Manages the TRD digits
+  ClassDef(AliTRDdigitsManager,4)          //  Manages the TRD digits
 
 };