[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 AliTRDdataArrayF;
class AliTRDsegmentArray;

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;
          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       WriteDigits() const;

  virtual void         AddSDigitsManager(AliTRDdigitsManager *manager);
  virtual void         DeleteSDigitsManager();

  virtual Bool_t       MakeDigits();

          Bool_t       SortHits(Float_t **hits, Int_t *nhit);
          Bool_t       ConvertHits(Int_t det, Float_t *hits, Int_t nhit, AliTRDdataArrayF *signals);
          Bool_t       ConvertSignals(Int_t det, AliTRDdataArrayF *signals);

          Bool_t       SDigits2Digits();
          Bool_t       MergeSDigits();
          Bool_t       ConvertSDigits();

          Bool_t       Signal2ADC(Int_t det, AliTRDdataArrayF *signals);
          Bool_t       Signal2SDigits(Int_t det, AliTRDdataArrayF *signals);
          Bool_t       CopyDictionary(Int_t det);
          void         CompressOutputArrays(Int_t det);

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

  AliTRDdigitsManager *Digits() const                       { return fDigitsManager;  }

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

          Double_t     TimeStruct(Float_t vdrift, Double_t time, Double_t z);
          Int_t        Diffusion(Float_t vdrift, Double_t absdriftlength
                               , Double_t &lRow, Double_t &lCol, Double_t &lTime);
          Int_t        ExB(Float_t vdrift, Double_t driftlength, Double_t &lRow);
  
 protected:

  virtual Bool_t       Init();
          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
          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

  ClassDef(AliTRDdigitizer,16)              //  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;
625f5260	26	class AliTRDdataArrayF;
625f5260	27	class AliTRDsegmentArray;
f7336fa3	28
85cbec76	29	class AliTRDdigitizer : public AliDigitizer {
f7336fa3	30
	31	public:
	32
	33	AliTRDdigitizer();
4d18a639	34	AliTRDdigitizer(const Text_t name, const Text_t title);
4d18a639	35	AliTRDdigitizer(AliRunDigitizer manager, const Text_t name, const Text_t *title);
bfc40adc	36	AliTRDdigitizer(AliRunDigitizer *manager);
dd9a6ee3	37	AliTRDdigitizer(const AliTRDdigitizer &d);
625f5260	38	virtual ~AliTRDdigitizer();
	39	AliTRDdigitizer &operator=(const AliTRDdigitizer &d);
	40
	41	virtual void Copy(TObject &d) const;
	42	Bool_t InitDetector();
	43	void InitOutput(Int_t iEvent);
	44	virtual void Exec(Option_t *option = 0);
	45
	46	virtual Bool_t Open(const Char_t *file, Int_t nEvent = 0);
	47	virtual Bool_t Open(AliRunLoader *runLoader, Int_t nEvent = 0);
	48	virtual Bool_t MakeBranch(TTree *tree) const;
	49	virtual Bool_t WriteDigits() const;
	50
	51	virtual void AddSDigitsManager(AliTRDdigitsManager *manager);
	52	virtual void DeleteSDigitsManager();
	53
	54	virtual Bool_t MakeDigits();
	55
	56	Bool_t SortHits(Float_t *hits, Int_t nhit);
	57	Bool_t ConvertHits(Int_t det, Float_t hits, Int_t nhit, AliTRDdataArrayF signals);
	58	Bool_t ConvertSignals(Int_t det, AliTRDdataArrayF *signals);
	59
	60	Bool_t SDigits2Digits();
	61	Bool_t MergeSDigits();
	62	Bool_t ConvertSDigits();
	63
	64	Bool_t Signal2ADC(Int_t det, AliTRDdataArrayF *signals);
	65	Bool_t Signal2SDigits(Int_t det, AliTRDdataArrayF *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 *man) { fDigitsManager = man; }
	73	void SetGeometry(AliTRDgeometry *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 GetDiffusionT(Float_t vdrift);
	81	Float_t GetDiffusionL(Float_t vdrift);
	82	Float_t GetLorentzFactor(Float_t vdrift);
	83
	84	Double_t TimeStruct(Float_t vdrift, Double_t time, Double_t z);
	85	Int_t Diffusion(Float_t vdrift, Double_t absdriftlength
	86	, Double_t &lRow, Double_t &lCol, Double_t &lTime);
	87	Int_t ExB(Float_t vdrift, Double_t driftlength, Double_t &lRow);
3551db50	88
3becff3c	89	protected:
f7336fa3	90
625f5260	91	virtual Bool_t Init();
	92	void SampleTimeStruct(Float_t vdrift);
	93	void RecalcDiffusion(Float_t vdrift);
4d18a639	94
f4611fed	95	AliRunLoader *fRunLoader; //! Local pointer
abaf1f1d	96	AliTRDdigitsManager *fDigitsManager; //! Manager for the output digits
	97	AliTRDdigitsManager *fSDigitsManager; //! Manager for the summed input s-digits
	98	TList *fSDigitsManagerList; //! List of managers of input s-digits
	99	AliTRD *fTRD; //! TRD detector class
	100	AliTRDgeometry *fGeo; //! TRD geometry
ecf39416	101	AliTRDfeeParam *fFee; // Fee Parameters
4d18a639	102
625f5260	103	Int_t fEvent; //! Event number
	104	Int_t *fMasks; //! Masks for the merging
	105	Bool_t fCompress; // Switch to keep only compressed data in memory
	106	Bool_t fSDigits; // Switch for the summable digits
	107	Bool_t fMergeSignalOnly; // Merge only detectors that contain a signal
793ff80c	108
625f5260	109	Float_t fDiffLastVdrift; // The structures are valid for fLastVdrift (caching)
	110	Float_t fDiffusionT; // Transverse drift coefficient
	111	Float_t fDiffusionL; // Longitudinal drift coefficient
	112	Float_t fOmegaTau; // Tangens of the Lorentz angle
	113	Float_t fLorentzFactor; // Factor due to Lorentz force
f7336fa3	114
625f5260	115	Float_t fTimeLastVdrift; // The structures are valid for fLastVdrift (caching)
	116	Float_t *fTimeStruct1; //! Time Structure of Drift Cells
	117	Float_t *fTimeStruct2; //! Time Structure of Drift Cells
	118	Float_t fVDlo; // Lower drift velocity, for interpolation
	119	Float_t fVDhi; // Higher drift velocity, for interpolation
f7336fa3	120
625f5260	121	ClassDef(AliTRDdigitizer,16) // Produces TRD-Digits
	122
	123	};
f7336fa3	124	#endif