fix

[u/mrichter/AliRoot.git] / PMD / AliPMDDigitizer.h

#ifndef ALIPMDDIGITIZER_H
#define ALIPMDDIGITIZER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
//-----------------------------------------------------//
//                                                     //
//  Header File : PMDDigitization.h, Version 00        //
//                                                     //
//  Date   : September 20 2002                         //
//                                                     //
//-----------------------------------------------------//
// Author - B.K. Nandi
//

#include "AliDigitizer.h"

class TClonesArray;
class TFile;
class TObjArray;
class TParticle;
class TTree;
class TNtuple;

class AliLoader;
class AliRunLoader;
class AliRun;
class AliDetector;
class AliPMDhit;
class AliHit;
class AliHeader;
class AliDigitizationInput;
class AliCDBManager;
class AliCDBStorage;

class AliPMDcell;
class AliPMDsdigit;
class AliPMDdigit;
class AliPMDCalibData;
class AliPMDPedestal;

class AliPMDDigitizer:public AliDigitizer
{
 public:

  AliPMDDigitizer();
  AliPMDDigitizer(const AliPMDDigitizer &digitizer);  // copy constructor
  AliPMDDigitizer &operator=(const AliPMDDigitizer &digitizer); // assign op
  AliPMDDigitizer(AliDigitizationInput* digInput);
  virtual ~AliPMDDigitizer();

  void OpengAliceFile(const char *file, Option_t *option);

  void Hits2SDigits(Int_t ievt);
  void Hits2Digits(Int_t ievt);
  void SDigits2Digits(Int_t ievt);
  void Digitize(Option_t *option);
  void MergeSDigits(Int_t filenumber, Int_t troffset);
  void TrackAssignment2CPVCell();
  void TrackAssignment2Cell();
  void MeV2ADC(Float_t mev, Float_t & adc) const;
  void AddSDigit(Int_t trnumber, Int_t trpid, Int_t det, Int_t smnumber, 
		 Int_t irow, Int_t icol, Float_t adc);
  void AddDigit(Int_t trnumber, Int_t trpid, Int_t det, Int_t smnumber, 
		Int_t irow, Int_t icol, Float_t adc);
  void  SetZPosition(Float_t zpos);
  Float_t GetZPosition() const;
  void ResetCell();
  void ResetSDigit();
  void ResetDigit();
  void ResetCellADC();
  void UnLoad(Option_t * option);

  Float_t  Gain(Int_t det, Int_t smn, Int_t row, Int_t col) const;

  AliPMDCalibData *GetCalibGain() const;
  AliPMDPedestal  *GetCalibPed() const;

 protected:
  AliRunLoader *fRunLoader;  //! Pointer to Run Loader
  AliPMDhit    *fPMDHit;     //! Pointer to specific detector hits
  AliDetector  *fPMD;        //! Get pointers to Alice detectors 
                             // and Hits containers 
  AliLoader    *fPMDLoader;  //! Pointer to specific detector loader

  AliPMDCalibData *fCalibGain;  //! Gain calibration data
  AliPMDPedestal  *fCalibPed;   //! Pedestal calibration data

  TClonesArray *fSDigits;    //! List of summable digits
  TClonesArray *fDigits;     //! List of digits

  TObjArray     fCPVCell;     //! List of cpv cells   
  TObjArray     fCell;       //! List of pmd cells
  
  Int_t   fNsdigit;          // Summable digits counter
  Int_t   fNdigit;           // Digits counter
  Int_t   fDetNo;            // Detector Number (0:PRE, 1:CPV)
  Float_t fZPos;             // z-position of the detector

  static const Int_t fgkTotUM = 24; // Total Unit modules in one detector
  static const Int_t fgkRow   = 48; // Total number of rows in one unitmodule
  static const Int_t fgkCol   = 96; // Total number of cols in one unitmodule

  Float_t fCPV[fgkTotUM][fgkRow][fgkCol]; //! CPV Array containing total edep
  Float_t fPRE[fgkTotUM][fgkRow][fgkCol]; //! PRE Array containing total edep
  
  Int_t   fCPVCounter[fgkTotUM][fgkRow][fgkCol]; //! Number of times each cell
                                                 // is fired in CPV
  Int_t   fPRECounter[fgkTotUM][fgkRow][fgkCol]; //! Number of times each cell
                                                 // is fired in PMD

  Int_t   fCPVTrackNo[fgkTotUM][fgkRow][fgkCol]; //! CPV Array containing 
                                                 //  track number
  Int_t   fPRETrackNo[fgkTotUM][fgkRow][fgkCol]; //! PRE Array containing 
                                                 //  track number

  Int_t   fCPVTrackPid[fgkTotUM][fgkRow][fgkCol]; //! CPV Array containing 
                                                 //  track number
  Int_t   fPRETrackPid[fgkTotUM][fgkRow][fgkCol]; //! PRE Array containing 
                                                 //  track number

  
  ClassDef(AliPMDDigitizer,9)    // To digitize PMD Hits
};
#endif