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

#ifndef TRDdigitizer_h
#define TRDdigitizer_h
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

#include <TNamed.h>
#include <TFile.h>

#include "AliHit.h" 
#include "AliDigit.h"
#include "AliTRDconst.h"
#include "AliTRDgeometry.h"
#include "AliTRDsegmentArray.h"

///////////////////////////////////////////////////////
//  Produces digits from the hits information        //
///////////////////////////////////////////////////////

const Int_t kNDict = 3;

class AliTRDdigitizer : public TNamed {

 public:

  AliTRDdigitizer();
  AliTRDdigitizer(const Text_t* name, const Text_t* title);
  ~AliTRDdigitizer();

  virtual void        Init();
  virtual Bool_t      Open(const Char_t *name, Int_t nEvent = 0);
  virtual Bool_t      MakeBranch();
  virtual Bool_t      MakeDigits();
  virtual Bool_t      WriteDigits();

  virtual void        SetGasGain(Float_t gasgain)     { fGasGain       = gasgain;  };
  virtual void        SetNoise(Float_t noise)         { fNoise         = noise;    };
  virtual void        SetChipGain(Float_t chipgain)   { fChipGain      = chipgain; };
  virtual void        SetADCoutRange(Float_t range)   { fADCoutRange   = range;    };
  virtual void        SetADCinRange(Float_t range)    { fADCinRange    = range;    };
  virtual void        SetADCthreshold(Int_t thresh)   { fADCthreshold  = thresh;   };
  virtual void        SetDiffusion(Int_t diff_on = 1) { fDiffusionOn   = diff_on;  };
  virtual void        SetDiffusionT(Float_t diff)     { fDiffusionT    = diff;     };
  virtual void        SetDiffusionL(Float_t diff)     { fDiffusionL    = diff;     };
  virtual void        SetElAttach(Int_t el_on = 1)    { fElAttachOn    = el_on;    };
  virtual void        SetElAttachProp(Float_t prop)   { fElAttachProp  = prop;     };
  virtual void        SetExB(Int_t exb_on = 1)        { fExBOn         = exb_on;   };
  virtual void        SetLorentzAngle(Float_t angle)  { fLorentzAngle  = angle;    };

  AliTRDsegmentArray *DigitsArray()                   { return fDigitsArray;   };
  AliTRDsegmentArray *Dictionary(Int_t i)             { return fDictionary[i]; };

  virtual Float_t     GetGasGain()                    { return fGasGain;       };
  virtual Float_t     GetNoise()                      { return fNoise;         };
  virtual Float_t     GetChipGain()                   { return fChipGain;      };
  virtual Float_t     GetADCoutRange()                { return fADCoutRange;   };
  virtual Float_t     GetADCinRange()                 { return fADCinRange;    };
  virtual Int_t       GetADCthreshold()               { return fADCthreshold;  };
  virtual Float_t     GetDiffusionT()                 { return fDiffusionT;    };
  virtual Float_t     GetDiffusionL()                 { return fDiffusionL;    };
  virtual Float_t     GetElAttachProp()               { return fElAttachProp;  };
  virtual Float_t     GetLorentzAngle()               { return fLorentzAngle;  };

 protected:

  TFile              *fInputFile;       //! ALIROOT-filename

  AliTRDsegmentArray *fDigitsArray;     //! Array of detector segments containing the digits
  AliTRDsegmentArray *fDictionary[3];   //! Dictionary array, connecting MC tracks to the digits
  
  Int_t               fEvent;           //! Event number

  Float_t             fGasGain;         // Gas gain
  Float_t             fNoise;           // Electronics noise
  Float_t             fChipGain;        // Electronics gain
  Float_t             fADCoutRange;     // ADC output range (number of channels)
  Float_t             fADCinRange;      // ADC input range (input charge)
  Int_t               fADCthreshold;    // ADC threshold in ADC channel
  Int_t               fDiffusionOn;     // Switch for the diffusion
  Float_t             fDiffusionT;      // Diffusion in transverse direction
  Float_t             fDiffusionL;      // Diffusion in longitudinal direction
  Int_t               fElAttachOn;      // Switch for the electron attachment
  Float_t             fElAttachProp;    // Propability for electron attachment (for 1m)
  Int_t               fExBOn;           // Switch for the ExB effects
  Float_t             fLorentzAngle;    // Lorentz angle 
  Float_t             fLorentzFactor;   // Factor due to Lorentz force

 private:

  virtual Int_t       Diffusion(Float_t driftlength, Float_t *xyz);
  virtual Int_t       ExB(Float_t driftlength, Float_t *xyz);  
  virtual Float_t     PadResponse(Float_t x);
  
  ClassDef(AliTRDdigitizer,1)           // TRD-Digits manager

};

//_____________________________________________________________________________
class AliTRDdigit : public AliDigitNew {

 public:

  AliTRDdigit() {};
  AliTRDdigit(Int_t *digits);
  virtual ~AliTRDdigit() {};

  virtual Int_t GetAmp()      { return fAmplitude; };
  virtual Int_t GetDetector() { return fDetector;  };
  virtual Int_t GetRow()      { return fRow;       };
  virtual Int_t GetCol()      { return fCol;       };
  virtual Int_t GetTime()     { return fTime;      };

 protected:

  Int_t        fDetector;   // TRD detector number
  Int_t        fRow;        // Pad row number
  Int_t        fCol;        // Pad col number
  Int_t        fTime;       // Time bucket
  Int_t        fAmplitude;  // Signal amplitude

  ClassDef(AliTRDdigit,1)   // Digits for Transition Radiation Detector

};

#endif
Commit	Line	Data
f7336fa3	1	#ifndef TRDdigitizer_h
	2	#define TRDdigitizer_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	#include <TFile.h>
	10
	11	#include "AliHit.h"
	12	#include "AliDigit.h"
	13	#include "AliTRDconst.h"
	14	#include "AliTRDgeometry.h"
	15	#include "AliTRDsegmentArray.h"
	16
	17	///////////////////////////////////////////////////////
	18	// Produces digits from the hits information //
	19	///////////////////////////////////////////////////////
	20
	21	const Int_t kNDict = 3;
	22
	23	class AliTRDdigitizer : public TNamed {
	24
	25	public:
	26
	27	AliTRDdigitizer();
	28	AliTRDdigitizer(const Text_t* name, const Text_t* title);
	29	~AliTRDdigitizer();
	30
	31	virtual void Init();
	32	virtual Bool_t Open(const Char_t *name, Int_t nEvent = 0);
	33	virtual Bool_t MakeBranch();
	34	virtual Bool_t MakeDigits();
	35	virtual Bool_t WriteDigits();
	36
	37	virtual void SetGasGain(Float_t gasgain) { fGasGain = gasgain; };
	38	virtual void SetNoise(Float_t noise) { fNoise = noise; };
	39	virtual void SetChipGain(Float_t chipgain) { fChipGain = chipgain; };
	40	virtual void SetADCoutRange(Float_t range) { fADCoutRange = range; };
	41	virtual void SetADCinRange(Float_t range) { fADCinRange = range; };
	42	virtual void SetADCthreshold(Int_t thresh) { fADCthreshold = thresh; };
	43	virtual void SetDiffusion(Int_t diff_on = 1) { fDiffusionOn = diff_on; };
	44	virtual void SetDiffusionT(Float_t diff) { fDiffusionT = diff; };
	45	virtual void SetDiffusionL(Float_t diff) { fDiffusionL = diff; };
	46	virtual void SetElAttach(Int_t el_on = 1) { fElAttachOn = el_on; };
	47	virtual void SetElAttachProp(Float_t prop) { fElAttachProp = prop; };
	48	virtual void SetExB(Int_t exb_on = 1) { fExBOn = exb_on; };
	49	virtual void SetLorentzAngle(Float_t angle) { fLorentzAngle = angle; };
	50
	51	AliTRDsegmentArray *DigitsArray() { return fDigitsArray; };
	52	AliTRDsegmentArray *Dictionary(Int_t i) { return fDictionary[i]; };
	53
	54	virtual Float_t GetGasGain() { return fGasGain; };
	55	virtual Float_t GetNoise() { return fNoise; };
	56	virtual Float_t GetChipGain() { return fChipGain; };
	57	virtual Float_t GetADCoutRange() { return fADCoutRange; };
	58	virtual Float_t GetADCinRange() { return fADCinRange; };
	59	virtual Int_t GetADCthreshold() { return fADCthreshold; };
	60	virtual Float_t GetDiffusionT() { return fDiffusionT; };
	61	virtual Float_t GetDiffusionL() { return fDiffusionL; };
	62	virtual Float_t GetElAttachProp() { return fElAttachProp; };
	63	virtual Float_t GetLorentzAngle() { return fLorentzAngle; };
	64
65	protected:
66
67	TFile *fInputFile; //! ALIROOT-filename
68
69	AliTRDsegmentArray *fDigitsArray; //! Array of detector segments containing the digits
70	AliTRDsegmentArray *fDictionary[3]; //! Dictionary array, connecting MC tracks to the digits
71
72	Int_t fEvent; //! Event number
73
74	Float_t fGasGain; // Gas gain
75	Float_t fNoise; // Electronics noise
76	Float_t fChipGain; // Electronics gain
77	Float_t fADCoutRange; // ADC output range (number of channels)
78	Float_t fADCinRange; // ADC input range (input charge)
79	Int_t fADCthreshold; // ADC threshold in ADC channel
80	Int_t fDiffusionOn; // Switch for the diffusion
81	Float_t fDiffusionT; // Diffusion in transverse direction
82	Float_t fDiffusionL; // Diffusion in longitudinal direction
83	Int_t fElAttachOn; // Switch for the electron attachment
84	Float_t fElAttachProp; // Propability for electron attachment (for 1m)
85	Int_t fExBOn; // Switch for the ExB effects
86	Float_t fLorentzAngle; // Lorentz angle
87	Float_t fLorentzFactor; // Factor due to Lorentz force
88
89	private:
90
91	virtual Int_t Diffusion(Float_t driftlength, Float_t *xyz);
92	virtual Int_t ExB(Float_t driftlength, Float_t *xyz);
93	virtual Float_t PadResponse(Float_t x);
94
95	ClassDef(AliTRDdigitizer,1) // TRD-Digits manager
96
97	};
98
99	//_____________________________________________________________________________
100	class AliTRDdigit : public AliDigitNew {
101
102	public:
103
104	AliTRDdigit() {};
105	AliTRDdigit(Int_t *digits);
106	virtual ~AliTRDdigit() {};
107
108	virtual Int_t GetAmp() { return fAmplitude; };
109	virtual Int_t GetDetector() { return fDetector; };
110	virtual Int_t GetRow() { return fRow; };
111	virtual Int_t GetCol() { return fCol; };
112	virtual Int_t GetTime() { return fTime; };
113
114	protected:
115
116	Int_t fDetector; // TRD detector number
117	Int_t fRow; // Pad row number
118	Int_t fCol; // Pad col number
119	Int_t fTime; // Time bucket
120	Int_t fAmplitude; // Signal amplitude
121
122	ClassDef(AliTRDdigit,1) // Digits for Transition Radiation Detector
123
124	};
125
126	#endif