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

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

/* $Id$ */

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

#include "AliDigitizer.h"

class TFile;
class TF1;

class AliRunDigitizer;
class AliRunLoader;

class AliTRD;
class AliTRDdigitsManager;
class AliTRDgeometry;

class AliTRDdigitizer : public AliDigitizer {

 public:

  AliTRDdigitizer();
  AliTRDdigitizer(const Text_t *name, const Text_t *title);
  AliTRDdigitizer(AliRunDigitizer *manager, const Text_t *name, const Text_t *title);
  AliTRDdigitizer(AliRunDigitizer *manager);
  AliTRDdigitizer(const AliTRDdigitizer &d);
  virtual         ~AliTRDdigitizer();
  AliTRDdigitizer &operator=(const AliTRDdigitizer &d);

  virtual void     Copy(TObject &d) const;
  virtual Bool_t   InitDetector();
          void     InitOutput(Int_t iEvent);
  virtual void     Exec(Option_t *option = 0);  
  virtual Bool_t   Open(const Char_t *file, Int_t nEvent = 0);
  virtual Bool_t   Open(AliRunLoader *runLoader, Int_t nEvent = 0);
  virtual Bool_t   MakeBranch(TTree *tree) const;
  virtual Bool_t   MakeDigits();

  virtual void     AddSDigitsManager(AliTRDdigitsManager *manager);
  virtual void     DeleteSDigitsManager();
  virtual Bool_t   ConvertSDigits();
  virtual Bool_t   MergeSDigits();
  virtual Bool_t   SDigits2Digits();
  virtual Bool_t   WriteDigits() const;

  virtual void     SetCompress(Int_t c = 1)             { fCompress        = c;   }
  virtual void     SetSDigits(Int_t v = 1)              { fSDigits         = v;   }
  virtual void     SetSDigitsScale(Float_t s)           { fSDigitsScale    = s;   }
  virtual void     SetEvent(Int_t v = 0)                { fEvent           = v;   }
  virtual void     SetManager(AliTRDdigitsManager *man) { fDigitsManager   = man; }
  virtual void     SetGeometry(AliTRDgeometry *geo)     { fGeo             = geo; }
  virtual void     SetMergeSignalOnly(Bool_t m = kTRUE) { fMergeSignalOnly = m;   }

          Bool_t   GetCompress() const                  { return fCompress;       }
          Bool_t   GetSDigits() const                   { return fSDigits;        }
          Float_t  GetSDigitsScale() const              { return fSDigitsScale;   }
          Float_t  GetDiffusionT(Float_t vdrift);
          Float_t  GetDiffusionL(Float_t vdrift);
          Float_t  GetLorentzFactor(Float_t vdrift);

  virtual Double_t TimeStruct(Float_t vdrift, Double_t time, Double_t z);
  virtual Int_t    Diffusion(Float_t vdrift, Double_t driftlength, Double_t *xyz);
  virtual Int_t    ExB(Float_t vdrift, Double_t driftlength, Double_t *xyz);  

  AliTRDdigitsManager *Digits() const                   { return fDigitsManager;  }
  
 protected:

          void     SampleTimeStruct(Float_t vdrift);
          void     RecalcDiffusion(Float_t vdrift);

  AliRunLoader        *fRunLoader;          //! Local pointer
  AliTRDdigitsManager *fDigitsManager;      //! Manager for the output digits
  AliTRDdigitsManager *fSDigitsManager;     //! Manager for the summed input s-digits
  TList               *fSDigitsManagerList; //! List of managers of input s-digits
  AliTRD              *fTRD;                //! TRD detector class
  AliTRDgeometry      *fGeo;                //! TRD geometry

          Int_t    fEvent;                  //! Event number
          Int_t   *fMasks;                  //! Masks for the merging
          Bool_t   fCompress;               //  Switch to keep only compressed data in memory
          Bool_t   fSDigits;                //  Switch for the summable digits
          Float_t  fSDigitsScale;           //  Scale factor for the summable digits 
          Bool_t   fMergeSignalOnly;        //  Merge only detectors that contain a signal
  
          Float_t  fDiffLastVdrift;         //  The structures are valid for fLastVdrift (caching)
          Float_t  fDiffusionT;             //  Transverse drift coefficient
          Float_t  fDiffusionL;             //  Longitudinal drift coefficient
          Float_t  fOmegaTau;               //  Tangens of the Lorentz angle
          Float_t  fLorentzFactor;          //  Factor due to Lorentz force

          Float_t  fTimeLastVdrift;         //  The structures are valid for fLastVdrift (caching)
          Float_t *fTimeStruct1;            //! Time Structure of Drift Cells
          Float_t *fTimeStruct2;            //! Time Structure of Drift Cells
          Float_t  fVDlo;                   //  Lower drift velocity, for interpolation
          Float_t  fVDhi;                   //  Higher drift velocity, for interpolation
  
 private:
    
  virtual Bool_t   Init();

  ClassDef(AliTRDdigitizer,14)              //  Produces TRD-Digits

};

#endif
Commit	Line	Data
	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
	26	class AliTRDdigitizer : public AliDigitizer {
	27
	28	public:
	29
	30	AliTRDdigitizer();
	31	AliTRDdigitizer(const Text_t name, const Text_t title);
	32	AliTRDdigitizer(AliRunDigitizer manager, const Text_t name, const Text_t *title);
	33	AliTRDdigitizer(AliRunDigitizer *manager);
	34	AliTRDdigitizer(const AliTRDdigitizer &d);
	35	virtual ~AliTRDdigitizer();
	36	AliTRDdigitizer &operator=(const AliTRDdigitizer &d);
	37
	38	virtual void Copy(TObject &d) const;
	39	virtual Bool_t InitDetector();
	40	void InitOutput(Int_t iEvent);
	41	virtual void Exec(Option_t *option = 0);
	42	virtual Bool_t Open(const Char_t *file, Int_t nEvent = 0);
	43	virtual Bool_t Open(AliRunLoader *runLoader, Int_t nEvent = 0);
	44	virtual Bool_t MakeBranch(TTree *tree) const;
	45	virtual Bool_t MakeDigits();
	46
	47	virtual void AddSDigitsManager(AliTRDdigitsManager *manager);
	48	virtual void DeleteSDigitsManager();
	49	virtual Bool_t ConvertSDigits();
	50	virtual Bool_t MergeSDigits();
	51	virtual Bool_t SDigits2Digits();
	52	virtual Bool_t WriteDigits() const;
	53
	54	virtual void SetCompress(Int_t c = 1) { fCompress = c; }
	55	virtual void SetSDigits(Int_t v = 1) { fSDigits = v; }
	56	virtual void SetSDigitsScale(Float_t s) { fSDigitsScale = s; }
	57	virtual void SetEvent(Int_t v = 0) { fEvent = v; }
	58	virtual void SetManager(AliTRDdigitsManager *man) { fDigitsManager = man; }
	59	virtual void SetGeometry(AliTRDgeometry *geo) { fGeo = geo; }
	60	virtual void SetMergeSignalOnly(Bool_t m = kTRUE) { fMergeSignalOnly = m; }
	61
	62	Bool_t GetCompress() const { return fCompress; }
	63	Bool_t GetSDigits() const { return fSDigits; }
	64	Float_t GetSDigitsScale() const { return fSDigitsScale; }
	65	Float_t GetDiffusionT(Float_t vdrift);
	66	Float_t GetDiffusionL(Float_t vdrift);
	67	Float_t GetLorentzFactor(Float_t vdrift);
	68
	69	virtual Double_t TimeStruct(Float_t vdrift, Double_t time, Double_t z);
	70	virtual Int_t Diffusion(Float_t vdrift, Double_t driftlength, Double_t *xyz);
	71	virtual Int_t ExB(Float_t vdrift, Double_t driftlength, Double_t *xyz);
	72
	73	AliTRDdigitsManager *Digits() const { return fDigitsManager; }
	74
	75	protected:
	76
	77	void SampleTimeStruct(Float_t vdrift);
	78	void RecalcDiffusion(Float_t vdrift);
	79
	80	AliRunLoader *fRunLoader; //! Local pointer
	81	AliTRDdigitsManager *fDigitsManager; //! Manager for the output digits
	82	AliTRDdigitsManager *fSDigitsManager; //! Manager for the summed input s-digits
	83	TList *fSDigitsManagerList; //! List of managers of input s-digits
	84	AliTRD *fTRD; //! TRD detector class
	85	AliTRDgeometry *fGeo; //! TRD geometry
	86
	87	Int_t fEvent; //! Event number
	88	Int_t *fMasks; //! Masks for the merging
	89	Bool_t fCompress; // Switch to keep only compressed data in memory
	90	Bool_t fSDigits; // Switch for the summable digits
	91	Float_t fSDigitsScale; // Scale factor for the summable digits
	92	Bool_t fMergeSignalOnly; // Merge only detectors that contain a signal
	93
	94	Float_t fDiffLastVdrift; // The structures are valid for fLastVdrift (caching)
	95	Float_t fDiffusionT; // Transverse drift coefficient
	96	Float_t fDiffusionL; // Longitudinal drift coefficient
	97	Float_t fOmegaTau; // Tangens of the Lorentz angle
	98	Float_t fLorentzFactor; // Factor due to Lorentz force
	99
	100	Float_t fTimeLastVdrift; // The structures are valid for fLastVdrift (caching)
	101	Float_t *fTimeStruct1; //! Time Structure of Drift Cells
	102	Float_t *fTimeStruct2; //! Time Structure of Drift Cells
	103	Float_t fVDlo; // Lower drift velocity, for interpolation
	104	Float_t fVDhi; // Higher drift velocity, for interpolation
	105
	106	private:
	107
	108	virtual Bool_t Init();
	109
	110	ClassDef(AliTRDdigitizer,14) // Produces TRD-Digits
	111
	112	};
	113
	114	#endif