few changes by Thomas Kuhr

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

diff --git a/TRD/AliTRDdigitsManager.h b/TRD/AliTRDdigitsManager.h
index 335806824139d1a519e097d9a49c04cc8000d5fe..21418ac08f7de692033b901d9edeb3ee944c37db 100644 (file)
--- a/TRD/AliTRDdigitsManager.h
+++ b/TRD/AliTRDdigitsManager.h
@@ -13,6 +13,7 @@
 #include <TObject.h>
 
 class TFile;
+class TTree;
 
 class AliTRDsegmentArray;
 class AliTRDdataArrayI;
@@ -29,12 +30,19 @@ class AliTRDdigitsManager : public TObject {
   virtual ~AliTRDdigitsManager();
   AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);
 
+  virtual void                CreateArrays();
+  virtual void                ResetArrays();
   virtual void                Copy(TObject &m);
-  virtual Bool_t              MakeBranch(char *file=0);
-  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;       }; 
@@ -49,17 +57,24 @@ class AliTRDdigitsManager : public TObject {
           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:
 
   static const Int_t  fgkNDict;            //  Number of track dictionary arrays
 
-  AliTRDsegmentArray *fDigits;             //! Digits data Array
+  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
 
 };