Update of digitization. Moved digit branch definition to AliTRD
[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 #include <TNamed.h>
9
10 // Time response function of the preamp
11 Double_t TRFlandau(Double_t *x, Double_t *par);
12
13 class TFile;
14 class TF1;
15
16 class AliTRD;
17 class AliTRDdigitsManager;
18 class AliTRDgeometry;
19
20 ///////////////////////////////////////////////////////
21 //  Produces digits from the hits information        //
22 ///////////////////////////////////////////////////////
23
24 class AliTRDdigitizer : public TNamed {
25
26  public:
27
28   AliTRDdigitizer();
29   AliTRDdigitizer(const Text_t* name, const Text_t* title);
30   AliTRDdigitizer(const AliTRDdigitizer &d);
31   virtual ~AliTRDdigitizer();
32   AliTRDdigitizer &operator=(const AliTRDdigitizer &d);
33
34   virtual void         Copy(TObject &d);
35   virtual void         Init();
36   virtual Bool_t       Open(const Char_t *name, Int_t nEvent = 0);
37   virtual Bool_t       MakeDigits();
38   virtual void         ReInit();
39   virtual Bool_t       SumSDigits();
40   virtual Bool_t       WriteDigits();
41   virtual Bool_t       InitDetector();
42
43   virtual void         SetGasGain(Float_t gasgain)      { fGasGain        = gasgain;  };
44   virtual void         SetNoise(Float_t noise)          { fNoise          = noise;    };
45   virtual void         SetChipGain(Float_t chipgain)    { fChipGain       = chipgain; };
46   virtual void         SetADCoutRange(Float_t range)    { fADCoutRange    = range;    };
47   virtual void         SetADCinRange(Float_t range)     { fADCinRange     = range;    };
48   virtual void         SetADCthreshold(Int_t thresh)    { fADCthreshold   = thresh;   };
49   virtual void         SetDiffusion(Int_t diffOn = 1)   { fDiffusionOn    = diffOn;   };
50   virtual void         SetDiffusionT(Float_t diff)      { fDiffusionT     = diff;     };
51   virtual void         SetDiffusionL(Float_t diff)      { fDiffusionL     = diff;     };
52   virtual void         SetElAttach(Int_t elOn = 1)      { fElAttachOn     = elOn;     };
53   virtual void         SetElAttachProp(Float_t prop)    { fElAttachProp   = prop;     };
54   virtual void         SetExB(Int_t exbOn = 1)          { fExBOn          = exbOn;    };
55   virtual void         SetOmegaTau(Float_t ot)          { fOmegaTau       = ot;      
56                                                           ReInit();                   };
57   virtual void         SetPadResponse(Int_t prfOn = 1)  { fPRFOn          = prfOn;    };
58   virtual void         SetPRF(TF1 *prf);
59   virtual void         SetTimeResponse(Int_t trfOn = 1) { fTRFOn          = trfOn;   
60                                                           ReInit();                   };
61   virtual void         SetTRF(TF1 *trf);
62   virtual void         SetDriftVelocity(Float_t v)      { fDriftVelocity  = v;       
63                                                           ReInit();                   };
64   virtual void         SetPadCoupling(Float_t v)        { fPadCoupling    = v;        };
65   virtual void         SetTimeCoupling(Float_t v)       { fTimeCoupling   = v;        };
66   virtual void         SetCompress(Int_t c = 1)         { fCompress       = c;        };
67   virtual void         SetVerbose(Int_t v = 1)          { fVerbose        = v;        };
68   virtual void         SetSDigits(Int_t v = 1)          { fSDigits        = v;        };
69
70   AliTRDdigitsManager *Digits() const                   { return fDigits;            };
71
72   virtual Float_t      GetGasGain() const               { return fGasGain;           };
73   virtual Float_t      GetNoise() const                 { return fNoise;             };
74   virtual Float_t      GetChipGain() const              { return fChipGain;          };
75   virtual Float_t      GetADCoutRange() const           { return fADCoutRange;       };
76   virtual Float_t      GetADCinRange() const            { return fADCinRange;        };
77   virtual Int_t        GetADCthreshold() const          { return fADCthreshold;      };
78   virtual Float_t      GetDiffusionT() const            { return fDiffusionT;        };
79   virtual Float_t      GetDiffusionL() const            { return fDiffusionL;        };
80   virtual Float_t      GetElAttachProp() const          { return fElAttachProp;      };
81   virtual Float_t      GetOmegaTau() const              { return fOmegaTau;          };
82   virtual TF1         *GetPadResponse() const           { return fPRF;               };
83   virtual TF1         *GetTimeResponse() const          { return fTRF;               };
84   virtual Float_t      GetDriftVelocity() const         { return fDriftVelocity;     };
85   virtual Float_t      GetPadCoupling() const           { return fPadCoupling;       };
86   virtual Float_t      GetTimeCoupling() const          { return fTimeCoupling;      };
87   virtual Bool_t       GetCompress() const              { return fCompress;          };
88   virtual Bool_t       GetSDigits() const               { return fSDigits;           };
89   virtual Float_t      GetTimeBinWidth() const          { return fTimeBinWidth;      };
90
91  protected:
92
93   TFile               *fInputFile;       //! ALIROOT-filename
94   AliTRDdigitsManager *fDigits;          //! TRD digits manager
95   AliTRD              *fTRD;             //! TRD detector class
96   AliTRDgeometry      *fGeo;             //! TRD geometry
97   
98   Int_t                fEvent;           //! Event number
99
100   Float_t              fGasGain;         // Gas gain
101   Float_t              fNoise;           // Electronics noise
102   Float_t              fChipGain;        // Electronics gain
103   Float_t              fADCoutRange;     // ADC output range (number of channels)
104   Float_t              fADCinRange;      // ADC input range (input charge)
105   Float_t              fSinRange;        // Input range for summable digits 
106   Float_t              fSoutRange;       // Output range for summable digits 
107   Int_t                fADCthreshold;    // ADC threshold in ADC channel
108   Int_t                fDiffusionOn;     // Switch for the diffusion
109   Float_t              fDiffusionT;      // Diffusion in transverse direction
110   Float_t              fDiffusionL;      // Diffusion in longitudinal direction
111   Int_t                fElAttachOn;      // Switch for the electron attachment
112   Float_t              fElAttachProp;    // Propability for electron attachment (for 1m)
113   Int_t                fExBOn;           // Switch for the ExB effects
114   Float_t              fOmegaTau;        // Tangens of the Lorentz angle 
115   Float_t              fLorentzFactor;   // Factor due to Lorentz force
116   Int_t                fPRFOn;           // Switch for the pad response
117   TF1                 *fPRF;             // Pad response function
118   Float_t             *fPRFsmp;          //!Sampled pad response
119   Int_t                fPRFbin;          // Number of bins for the PRF
120   Float_t              fPRFlo;           // Lower boundary of the PRF
121   Float_t              fPRFhi;           // Higher boundary of the PRF
122   Float_t              fPRFwid;          // Bin width of the sampled PRF
123   Int_t                fPRFpad;          // Distance to next pad in PRF
124   Int_t                fTRFOn;           // Switch for the time response
125   TF1                 *fTRF;             // Time response function of the shaper
126   Float_t             *fTRFint;          //!Integrated time response
127   Int_t                fTRFbin;          // Number of bins for the TRF
128   Float_t              fTRFlo;           // Lower boundary of the TRF
129   Float_t              fTRFhi;           // Higher boundary of the TRF
130   Float_t              fTRFwid;          // Bin width of the integrated TRF
131   Float_t              fDriftVelocity;   // Drift velocity (cm / mus)
132   Float_t              fTimeBinWidth;    // Time bin width in ns
133   Float_t              fPadCoupling;     // Pad coupling factor
134   Float_t              fTimeCoupling;    // Time coupling factor (image charge of moving ions)
135   Bool_t               fCompress;        // Switch to keep only compressed data in memory
136   Int_t                fVerbose;         // Sets the verbose level
137   Bool_t               fSDigits;         // Switch for the summable digits
138
139  private:
140
141   virtual Int_t        Diffusion(Float_t driftlength, Float_t *xyz);
142   virtual Int_t        ExB(Float_t driftlength, Float_t *xyz);  
143   virtual Int_t        PadResponse(Float_t signal, Float_t dist, Float_t *pad);
144   virtual Float_t      TimeResponse(Float_t time);  
145   virtual Bool_t       CheckDetector(Int_t plane, Int_t chamber, Int_t sector);
146   virtual void         SamplePRF();
147   virtual void         SampleTRF();
148
149   ClassDef(AliTRDdigitizer,3)            // Produces TRD-Digits
150
151 };
152
153 #endif