63b076c97bcc3c251cab26c900648c52c3d4c966
[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
24 class AliTRDdigitsManager : public TObject {
25
26  public:
27
28   enum { kNDict = 3 };
29
30   AliTRDdigitsManager(Bool_t rawRec = kFALSE);  //if true digitsmanager uses only one entry in the TObjectArrays
31   AliTRDdigitsManager(const AliTRDdigitsManager &m);
32   virtual ~AliTRDdigitsManager();
33   AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);
34
35   virtual void                Copy(TObject &m) const;
36
37   virtual void                CreateArrays();
38   virtual void                ResetArrays();
39   virtual void                ResetArrays(Int_t det);
40   virtual Bool_t              BuildIndexes(Int_t det);
41
42   virtual Bool_t              MakeBranch(TTree * const tree);
43   virtual Bool_t              ReadDigits(TTree * consttree);
44   virtual Bool_t              WriteDigits();
45
46   virtual void                SetEvent(Int_t evt)             { fEvent           = evt;  };
47   virtual void                SetSDigits(Int_t v = 1)         { fHasSDigits      = v;    };
48   virtual void                SetUseDictionaries(Bool_t kval) { fUseDictionaries = kval; };
49
50   virtual Bool_t              UsesDictionaries() const        { return fUseDictionaries; };
51   virtual Bool_t              HasSDigits() const              { return fHasSDigits;      };
52   static  Int_t               NDict()                         { return fgkNDict;         }; 
53
54   virtual TObjArray          *GetDigits() const               { return fDigits;          };  
55   virtual TObjArray          *GetDictionary(Int_t i) const    { return fDict[i];         }; 
56
57   AliTRDdigit                *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det) const;
58   Int_t                       GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det) const;
59   
60   AliTRDarrayADC             *GetDigits(Int_t det)  const;
61   AliTRDarraySignal          *GetSDigits(Int_t det) const;    
62   AliTRDarrayDictionary      *GetDictionary(Int_t det, Int_t i) const;  
63   
64   AliTRDSignalIndex          *GetIndexes(Int_t det);
65   TObjArray                  *GetIndexes() const              { return fSignalIndexes;   };
66
67   void                        RemoveDigits(Int_t det);
68   void                        RemoveDictionaries(Int_t det);
69   void                        RemoveIndexes(Int_t det);
70   void                        ClearIndexes(Int_t det);
71   
72   Int_t                       GetTrack(Int_t track, AliTRDdigit * const digit) const;
73   Short_t                     GetDigitAmp(Int_t row, Int_t col, Int_t time, Int_t det) const;
74   UChar_t                     GetPadStatus(Int_t row, Int_t col, Int_t time, Int_t det) const;
75
76   Bool_t                      LoadArray(TObjArray * const object, const Char_t *branchname, TTree * const tree=0);  
77   Bool_t                      LoadArrayDict(TObjArray * const object, const Char_t *branchname, TTree * const tree=0);  
78   Bool_t                      StoreArray(TObjArray * const array1, const Char_t *branchname, TTree * const tree=0); 
79   Bool_t                      StoreArrayDict(TObjArray * const array3, const Char_t *branchname, TTree * const tree=0); 
80   
81  protected:
82   
83   static const Int_t  fgkNDict;            //  Number of track dictionary arrays
84   Int_t               fEvent;              //  Event number
85   TTree              *fTree;               //! Tree for the digits arrays
86   TObjArray          *fDigits;             //  Digits data array               
87   TObjArray          *fDict[kNDict];       //  Track dictionary data array   
88   Bool_t              fHasSDigits;         //  Switch for the summable digits
89   TObjArray          *fSignalIndexes;      //  Provides access to the active pads and tbins
90   Bool_t              fUseDictionaries;    //  Use dictionaries or not (case of real data)
91   TTree              *fTreeD;              //  Tree with detector objects
92   TBranch            *fBranch;             //  Branchaddress
93   Int_t               fDets;               //  No of Detectors
94   Bool_t              fRawRec;             //  Reconstruct from raw files? If its kTRUE then the TObjArrays have only one entry.
95
96   ClassDef(AliTRDdigitsManager,7)          //  Manages the TRD digits
97
98 };
99 #endif