[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 AliTRDfeeParam;

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 absdriftlength
                           , Double_t &lRow, Double_t &lCol, Double_t &lTime);
  virtual Int_t    ExB(Float_t vdrift, Double_t driftlength, Double_t &lRow);

  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
  AliTRDfeeParam      *fFee;                //  Fee Parameters

          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,15)              //  Produces TRD-Digits

};

#endif
Commit	Line	Data
dd9a6ee3	1	#ifndef ALITRDDIGITIZER_H
dd9a6ee3	2	#define ALITRDDIGITIZER_H
f7336fa3	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
4d18a639	8	////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// Produces digits from the hits information //
	11	// //
	12	////////////////////////////////////////////////////////////////////////////
	13
85cbec76	14	#include "AliDigitizer.h"
f7336fa3	15
793ff80c	16	class TFile;
	17	class TF1;
	18
85cbec76	19	class AliRunDigitizer;
88cb7938	20	class AliRunLoader;
85cbec76	21
793ff80c	22	class AliTRD;
6f1e466d	23	class AliTRDdigitsManager;
793ff80c	24	class AliTRDgeometry;
ecf39416	25	class AliTRDfeeParam;
f7336fa3	26
85cbec76	27	class AliTRDdigitizer : public AliDigitizer {
f7336fa3	28
	29	public:
	30
	31	AliTRDdigitizer();
4d18a639	32	AliTRDdigitizer(const Text_t name, const Text_t title);
4d18a639	33	AliTRDdigitizer(AliRunDigitizer manager, const Text_t name, const Text_t *title);
bfc40adc	34	AliTRDdigitizer(AliRunDigitizer *manager);
dd9a6ee3	35	AliTRDdigitizer(const AliTRDdigitizer &d);
4d18a639	36	virtual ~AliTRDdigitizer();
dd9a6ee3	37	AliTRDdigitizer &operator=(const AliTRDdigitizer &d);
f7336fa3	38
4d18a639	39	virtual void Copy(TObject &d) const;
	40	virtual Bool_t InitDetector();
	41	void InitOutput(Int_t iEvent);
	42	virtual void Exec(Option_t *option = 0);
	43	virtual Bool_t Open(const Char_t *file, Int_t nEvent = 0);
95867fd1	44	virtual Bool_t Open(AliRunLoader *runLoader, Int_t nEvent = 0);
4d18a639	45	virtual Bool_t MakeBranch(TTree *tree) const;
	46	virtual Bool_t MakeDigits();
	47
	48	virtual void AddSDigitsManager(AliTRDdigitsManager *manager);
	49	virtual void DeleteSDigitsManager();
	50	virtual Bool_t ConvertSDigits();
	51	virtual Bool_t MergeSDigits();
	52	virtual Bool_t SDigits2Digits();
	53	virtual Bool_t WriteDigits() const;
	54
	55	virtual void SetCompress(Int_t c = 1) { fCompress = c; }
	56	virtual void SetSDigits(Int_t v = 1) { fSDigits = v; }
	57	virtual void SetSDigitsScale(Float_t s) { fSDigitsScale = s; }
	58	virtual void SetEvent(Int_t v = 0) { fEvent = v; }
	59	virtual void SetManager(AliTRDdigitsManager *man) { fDigitsManager = man; }
	60	virtual void SetGeometry(AliTRDgeometry *geo) { fGeo = geo; }
	61	virtual void SetMergeSignalOnly(Bool_t m = kTRUE) { fMergeSignalOnly = m; }
	62
	63	Bool_t GetCompress() const { return fCompress; }
	64	Bool_t GetSDigits() const { return fSDigits; }
	65	Float_t GetSDigitsScale() const { return fSDigitsScale; }
	66	Float_t GetDiffusionT(Float_t vdrift);
	67	Float_t GetDiffusionL(Float_t vdrift);
	68	Float_t GetLorentzFactor(Float_t vdrift);
	69
	70	virtual Double_t TimeStruct(Float_t vdrift, Double_t time, Double_t z);
4329977a	71	virtual Int_t Diffusion(Float_t vdrift, Double_t absdriftlength
	72	, Double_t &lRow, Double_t &lCol, Double_t &lTime);
	73	virtual Int_t ExB(Float_t vdrift, Double_t driftlength, Double_t &lRow);
4d18a639	74
4d18a639	75	AliTRDdigitsManager *Digits() const { return fDigitsManager; }
3551db50	76
3becff3c	77	protected:
f7336fa3	78
4d18a639	79	void SampleTimeStruct(Float_t vdrift);
	80	void RecalcDiffusion(Float_t vdrift);
	81
f4611fed	82	AliRunLoader *fRunLoader; //! Local pointer
abaf1f1d	83	AliTRDdigitsManager *fDigitsManager; //! Manager for the output digits
	84	AliTRDdigitsManager *fSDigitsManager; //! Manager for the summed input s-digits
	85	TList *fSDigitsManagerList; //! List of managers of input s-digits
	86	AliTRD *fTRD; //! TRD detector class
	87	AliTRDgeometry *fGeo; //! TRD geometry
ecf39416	88	AliTRDfeeParam *fFee; // Fee Parameters
4d18a639	89
	90	Int_t fEvent; //! Event number
	91	Int_t *fMasks; //! Masks for the merging
	92	Bool_t fCompress; // Switch to keep only compressed data in memory
	93	Bool_t fSDigits; // Switch for the summable digits
	94	Float_t fSDigitsScale; // Scale factor for the summable digits
	95	Bool_t fMergeSignalOnly; // Merge only detectors that contain a signal
3551db50	96
4d18a639	97	Float_t fDiffLastVdrift; // The structures are valid for fLastVdrift (caching)
	98	Float_t fDiffusionT; // Transverse drift coefficient
	99	Float_t fDiffusionL; // Longitudinal drift coefficient
	100	Float_t fOmegaTau; // Tangens of the Lorentz angle
	101	Float_t fLorentzFactor; // Factor due to Lorentz force
	102
	103	Float_t fTimeLastVdrift; // The structures are valid for fLastVdrift (caching)
324b8ec7	104	Float_t *fTimeStruct1; //! Time Structure of Drift Cells
324b8ec7	105	Float_t *fTimeStruct2; //! Time Structure of Drift Cells
4d18a639	106	Float_t fVDlo; // Lower drift velocity, for interpolation
4d18a639	107	Float_t fVDhi; // Higher drift velocity, for interpolation
3551db50	108
3becff3c	109	private:
3551db50	110
4d18a639	111	virtual Bool_t Init();
793ff80c	112
5e4083b7	113	ClassDef(AliTRDdigitizer,15) // Produces TRD-Digits
f7336fa3	114
	115	};
	116
	117	#endif