#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