PndGasGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                PndGasGenerator header file                    -----
// -----               Created 02/04/09  by F. Uhlig                   -----
// -------------------------------------------------------------------------

#ifndef PND_GASGENERATOR_H
#define PND_GASGENERATOR_H

#include "FairGenerator.h"
#include <iostream>

class FairPrimaryGenerator;
class TF1;

class PndGasGenerator : public FairGenerator
{
public:

  PndGasGenerator();

  PndGasGenerator(Int_t pdgid, Int_t mult=1);


  virtual ~PndGasGenerator() {};

  void SetPDGType      (Int_t pdg)  {fPDGType = pdg;  };

  void SetMultiplicity (Int_t mult) {fMult    = mult; };

  void SetPRange(Double32_t pmin=0   , Double32_t pmax=10)
    {fPMin=pmin; fPMax=pmax; fPRangeIsSet=kTRUE;}

  void SetPhiRange     (Double32_t phimin=0  , Double32_t phimax=360)
    {fPhiMin=phimin; fPhiMax=phimax;};

  void SetThetaRange   (Double32_t thetamin=0, Double32_t thetamax=90)
    {fThetaMin=thetamin; fThetaMax=thetamax; fThetaRangeIsSet=kTRUE;};

  void SetCosTheta   ()
    {fCosThetaIsSet=kTRUE;};

  void SetZRange  (Double32_t zmin=0, Double32_t zmax=10) 
    {fZMin=zmin; fZMax=zmax;}

  void SetRadius (Double32_t radius=0.2) {fRadius=radius;}

  //  void SetDensityProfile(TString in)

  void SetDebug(Bool_t debug=0) {fDebug = debug;}

  Bool_t Init();

  virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);

private:
  Int_t      fPDGType;             // Particle type (PDG encoding)
  Int_t      fMult;                // Multiplicity

  Double32_t fPDGMass;             // Particle mass [GeV]
  Double32_t fPhiMin, fPhiMax;     // Azimuth angle range [degree]
  Double32_t fPMin,   fPMax;       // Momentum range in lab system
  Double32_t fThetaMin, fThetaMax; // Polar angle range in lab system [degree]
  Double32_t fX, fY, fZ;           // Point vertex coordinates [cm]
  Double32_t fRadius;               // Radius of beam profile

  Bool_t     fThetaRangeIsSet;     // True if theta range is set
  Bool_t     fCosThetaIsSet;       // True if uniform distribution in cos(theta) is set (default -> uniform theta)
  Bool_t     fPRangeIsSet;         // True if abs.momentum range is set
  Double32_t fZMin,   fZMax;       // Minimum and maximum z position
  Bool_t     fDebug;               // Debug switch

  //  TString    fProfileFunction;      // function to generate density profile
  TF1*       fDensityProfile;      // density profile
  
  ClassDef(PndGasGenerator,1);

};


#endif