1ad73051f6fb176ba112347685a6875d9f5b3180
[u/mrichter/AliRoot.git] / TRD / AliTRDdigitizer.h
1 #ifndef ALITRDDIGITIZER_H
2 #define ALITRDDIGITIZER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ////////////////////////////////////////////////////////////////////////////
9 //                                                                        //
10 //  Produces digits from the hits information                             //
11 //                                                                        //
12 ////////////////////////////////////////////////////////////////////////////
13
14 #include "AliDigitizer.h"
15
16 class TFile;
17 class TF1;
18
19 class AliRunDigitizer;
20 class AliRunLoader;
21
22 class AliTRD;
23 class AliTRDdigitsManager;
24 class AliTRDgeometry;
25 class AliTRDarraySignal;
26 class AliTRDarrayADC;
27
28 class AliTRDdigitizer : public AliDigitizer {
29
30  public:
31
32   AliTRDdigitizer();
33   AliTRDdigitizer(const Text_t *name, const Text_t *title);
34   AliTRDdigitizer(AliRunDigitizer *manager, const Text_t *name, const Text_t *title);
35   AliTRDdigitizer(AliRunDigitizer *manager);
36   AliTRDdigitizer(const AliTRDdigitizer &d);
37   virtual             ~AliTRDdigitizer();
38   AliTRDdigitizer     &operator=(const AliTRDdigitizer &d);
39
40   virtual void         Copy(TObject &d) const;
41           Bool_t       InitDetector();
42           void         InitOutput(Int_t iEvent);
43   virtual void         Exec(const Option_t * const option = 0);  
44
45   virtual Bool_t       Open(const Char_t *file, Int_t nEvent = 0);
46   virtual Bool_t       Open(AliRunLoader * const runLoader, Int_t nEvent = 0);
47   virtual Bool_t       MakeBranch(TTree *tree) const;
48   virtual Bool_t       WriteDigits() const;
49
50   virtual void         AddSDigitsManager(AliTRDdigitsManager *manager);
51   virtual void         DeleteSDigitsManager();
52
53   virtual Bool_t       MakeDigits();
54
55           Bool_t       SortHits(Float_t **hits, Int_t *nhit);
56           Bool_t       ConvertHits(Int_t det, const Float_t * const hits, Int_t nhit, AliTRDarraySignal *signals);
57           Bool_t       ConvertSignals(Int_t det, AliTRDarraySignal *signals);
58
59           Bool_t       Digits2SDigits(AliTRDdigitsManager * const manDig, AliTRDdigitsManager * const manSDig);
60           Bool_t       SDigits2Digits();
61           Bool_t       MergeSDigits();
62           Bool_t       ConvertSDigits();
63
64           Bool_t       Signal2ADC(Int_t det, AliTRDarraySignal *signals);
65           Bool_t       Signal2SDigits(Int_t det, AliTRDarraySignal *signals);
66           Bool_t       CopyDictionary(Int_t det);
67           void         CompressOutputArrays(Int_t det);
68
69           void         SetCompress(Int_t c = 1)                    { fCompress        = c;   }
70           void         SetSDigits(Int_t v = 1)                     { fSDigits         = v;   }
71           void         SetEvent(Int_t v = 0)                       { fEvent           = v;   }
72           void         SetManager(AliTRDdigitsManager * const man) { fDigitsManager   = man; }
73           void         SetGeometry(AliTRDgeometry * const geo)     { fGeo             = geo; }
74           void         SetMergeSignalOnly(Bool_t m = kTRUE)        { fMergeSignalOnly = m;   }
75
76   AliTRDdigitsManager *Digits() const                              { return fDigitsManager;  }
77
78           Bool_t       GetCompress() const                         { return fCompress;       }
79           Bool_t       GetSDigits() const                          { return fSDigits;        }
80           Float_t      GetLorentzFactor(Float_t vdrift);
81
82           Int_t        Diffusion(Float_t vdrift, Double_t absdriftlength
83                                , Double_t &lRow, Double_t &lCol, Double_t &lTime);
84           Int_t        ExB(Float_t vdrift, Double_t driftlength, Double_t &lRow);
85           void         RunDigitalProcessing(Int_t det = 0);
86
87  protected:
88
89   virtual Bool_t       Init();
90
91   AliRunLoader        *fRunLoader;          //! Local pointer
92   AliTRDdigitsManager *fDigitsManager;      //! Manager for the output digits
93   AliTRDdigitsManager *fSDigitsManager;     //! Manager for the summed input s-digits
94   TList               *fSDigitsManagerList; //! List of managers of input s-digits
95   AliTRD              *fTRD;                //! TRD detector class
96   AliTRDgeometry      *fGeo;                //! TRD geometry
97
98           Int_t        fEvent;              //! Event number
99           Int_t       *fMasks;              //! Masks for the merging
100           Bool_t       fCompress;           //  Switch to keep only compressed data in memory
101           Bool_t       fSDigits;            //  Switch for the summable digits
102           Bool_t       fMergeSignalOnly;    //  Merge only detectors that contain a signal
103
104   ClassDef(AliTRDdigitizer,19)              //  Produces TRD-Digits
105
106 };
107 #endif