Bug fix for the case of more than 30 timebins
[u/mrichter/AliRoot.git] / TRD / AliTRDdigitsManager.h
1 #ifndef ALITRDDIGITSMANAGER_H
2 #define ALITRDDIGITSMANAGER_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8
9 /////////////////////////////////////////////////////////////
10 //  Manages the TRD digits                                 //
11 /////////////////////////////////////////////////////////////
12
13 #include <TObject.h>
14
15 class TFile;
16 class TTree;
17 class TBranch;  
18 class AliTRDdigit;
19 class AliTRDSignalIndex;
20 class AliTRDarrayADC;  
21 class AliTRDarraySignal; 
22 class AliTRDarrayDictionary;
23 class AliTRDdigitsParam;
24
25 class AliTRDdigitsManager : public TObject {
26
27  public:
28
29   enum { kNDict = 3 };
30
31   AliTRDdigitsManager(Bool_t rawRec = kFALSE);  //if true digitsmanager uses only one entry in the TObjectArrays
32   AliTRDdigitsManager(const AliTRDdigitsManager &m);
33   virtual ~AliTRDdigitsManager();
34   AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);
35
36   virtual void                Copy(TObject &m) const;
37
38   virtual void                CreateArrays();
39   virtual void                ResetArrays();
40   virtual void                ResetArrays(Int_t det);
41   virtual Bool_t              BuildIndexes(Int_t det);
42
43   virtual Bool_t              MakeBranch(TTree * const tree);
44   virtual Bool_t              ReadDigits(TTree * consttree);
45   virtual Bool_t              WriteDigits();
46
47   virtual void                SetEvent(Int_t evt)             { fEvent           = evt;  };
48   virtual void                SetSDigits(Int_t v = 1)         { fHasSDigits      = v;    };
49   virtual void                SetUseDictionaries(Bool_t kval) { fUseDictionaries = kval; };
50
51   virtual Bool_t              UsesDictionaries() const        { return fUseDictionaries; };
52   virtual Bool_t              HasSDigits() const              { return fHasSDigits;      };
53   static  Int_t               NDict()                         { return fgkNDict;         }; 
54
55   virtual TObjArray          *GetDigits() const               { return fDigits;          };  
56   virtual TObjArray          *GetDictionary(Int_t i) const    { return fDict[i];         }; 
57
58   AliTRDdigit                *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det) const;
59   Int_t                       GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det) const;
60   
61   AliTRDarrayADC             *GetDigits(Int_t det)  const;
62   AliTRDarraySignal          *GetSDigits(Int_t det) const;    
63   AliTRDarrayDictionary      *GetDictionary(Int_t det, Int_t i) const;  
64   AliTRDdigitsParam          *GetDigitsParam() const          { return fDigitsParam;     };
65   AliTRDSignalIndex          *GetIndexes(Int_t det);
66   TObjArray                  *GetIndexes() const              { return fSignalIndexes;   };
67
68   void                        RemoveDigits(Int_t det);
69   void                        RemoveDictionaries(Int_t det);
70   void                        RemoveIndexes(Int_t det);
71   void                        ClearIndexes(Int_t det);
72   
73   Int_t                       GetTrack(Int_t track, AliTRDdigit * const digit) const;
74   Short_t                     GetDigitAmp(Int_t row, Int_t col, Int_t time, Int_t det) const;
75   UChar_t                     GetPadStatus(Int_t row, Int_t col, Int_t time, Int_t det) const;
76
77   Bool_t                      LoadArrayDigits();
78   Bool_t                      LoadArrayDict();
79   Bool_t                      LoadDigitsParam();
80   Bool_t                      StoreArrayDigits();
81   Bool_t                      StoreArrayDict();
82   Bool_t                      StoreDigitsParam();
83
84  protected:
85   
86   static const Int_t  fgkNDict;            //  Number of track dictionary arrays
87   Int_t               fEvent;              //  Event number
88   TTree              *fTree;               //! Tree for the digits arrays
89   TObjArray          *fDigits;             //  Digits data array               
90   TObjArray          *fDict[kNDict];       //  Track dictionary data array   
91   Bool_t              fHasSDigits;         //  Switch for the summable digits
92   TObjArray          *fSignalIndexes;      //  Provides access to the active pads and tbins
93   Bool_t              fUseDictionaries;    //  Use dictionaries or not (case of real data)
94   Int_t               fDets;               //  No of Detectors
95   Bool_t              fRawRec;             //  Reconstruct from raw files? If its kTRUE then the TObjArrays have only one entry.
96   AliTRDdigitsParam  *fDigitsParam;        //  Parameters of the digits
97
98   ClassDef(AliTRDdigitsManager,8)          //  Manages the TRD digits
99
100 };
101 #endif