[u/mrichter/AliRoot.git] / ACORDE / AliGenACORDE.h

#ifndef ALIGENACORDE_H
#define ALIGENACORDE_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

#include "AliGenerator.h"
#include "AliACORDEConstants.h"

class TClonesArray;
class TF1;

class AliGenACORDE : public AliGenerator {
 public:
  AliGenACORDE();
  AliGenACORDE(Int_t npart);
  virtual ~AliGenACORDE();

  virtual void Init();
  virtual void Generate();
  virtual void SetPart(Int_t part) {fIpart = part;}

  void SetMode(ECRMode mode) {fCRMode = mode;}
  const TString*  GetMode() const {return fCRModeName;}

  void SetZenithalAngleRange(Float_t min,Float_t max=0) {fZenithMin=min;fZenithMax=max;}
  void SetAzimuthalAngleRange(Float_t min, Float_t max=0) {fAzimuthMin=min;fAzimuthMax=max;}

  void SetGridRange(Int_t nx,Float_t xwidth, Int_t nz, Float_t zwidth);
  Float_t GetMomentumResolution() const {return fPResolution;}

  void SetMomentumDistrubutionFunction(TF1 *func) {fMomentumDist=func;}
  void SetZenithalDistributionFunction(TF1 *func) {fZenithDist = func;}
  void SetMomentumResolution(Float_t res=1.) {fPResolution=res;}

  Float_t GetMomentum() const;
  Float_t GetZenithAngle(Float_t mom) const;


  void SetNumberOfParticles(Int_t nParticles) {fNParticles = nParticles;}

  // The following methods are for testing pourpuses
  TF1* GetMomentumDistibution() const {return fMomentumDist;}
  TF1* GetUnfoldedDistribution() const {return fUnfoldedMomentumDist;}

  TClonesArray* GetArray() const {return fPDist;}

 protected:
  void InitApWeightFactors();
  void InitMomentumGeneration();
  void InitZenithalAngleGeneration();
  void GenerateOneMuonBundle();
  void GenerateOneSingleMuon(Bool_t withFlatMomentum=kFALSE);

 private:
  AliGenACORDE(const AliGenACORDE& gen);
  AliGenACORDE& operator= (const AliGenACORDE& gen);

  Int_t    fIpart;              //! Particle type.
  ECRMode  fCRMode;             //! Cosmic muons generation method flag
  TString* fCRModeName;         //! Cosmic muons generation mode name

  Float_t  fXwidth;             //! X width of the grid
  Int_t    fNx;                 //! Number of divisions in x
  Float_t  fZwidth;             //! Z widht of the  grid
  Int_t    fNz;                 //! Number of divisions in z
  Bool_t   fMuonGrid;           //! Flag for method (Muon-bundles) checkout

  Float_t  fZenithMin;          //! Minimum zenithal angle.
  Float_t  fZenithMax;          //! Maximum zenithal angle.

  Float_t  fAzimuthMin;         //! Minimum azimuthal angle.
  Float_t  fAzimuthMax;         //! Maximum azimuthal angle.

  Float_t  fPRange;             //! Cosmic muon momentum range width in GeVs.
  Float_t  fPResolution;        //! Momentum resolution in GeVs.

  TArrayF* fAp;                 //! a(p) correction factors for the ang. dist.

  TF1*     fMomentumDist;       //! Function to generate the momentum dist.
  TF1*     fUnfoldedMomentumDist; //!
  TF1*     fZenithDist;         //! Function to generate the zenith angle dist.

  TClonesArray* fPDist;         //! Array of fZenithDist, to be used by a(p).

  Int_t fNParticles;		//! Fixed number of particles per event

  ClassDef(AliGenACORDE, 1) // Generator for AliACORDE class
};

typedef AliGenACORDE AliGenCRT; // for backward compatibility

#endif // ALIGENACORDE_H
Commit	Line	Data
b86e74f5	1	#ifndef ALIGENACORDE_H
	2	#define ALIGENACORDE_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	#include "AliGenerator.h"
	9	#include "AliACORDEConstants.h"
	10
	11	class TClonesArray;
	12	class TF1;
	13
	14	class AliGenACORDE : public AliGenerator {
	15	public:
	16	AliGenACORDE();
	17	AliGenACORDE(Int_t npart);
b86e74f5	18	virtual ~AliGenACORDE();
b86e74f5	19
b86e74f5	20	virtual void Init();
	21	virtual void Generate();
	22	virtual void SetPart(Int_t part) {fIpart = part;}
	23
	24	void SetMode(ECRMode mode) {fCRMode = mode;}
	25	const TString* GetMode() const {return fCRModeName;}
	26
	27	void SetZenithalAngleRange(Float_t min,Float_t max=0) {fZenithMin=min;fZenithMax=max;}
	28	void SetAzimuthalAngleRange(Float_t min, Float_t max=0) {fAzimuthMin=min;fAzimuthMax=max;}
	29
	30	void SetGridRange(Int_t nx,Float_t xwidth, Int_t nz, Float_t zwidth);
	31	Float_t GetMomentumResolution() const {return fPResolution;}
	32
	33	void SetMomentumDistrubutionFunction(TF1 *func) {fMomentumDist=func;}
	34	void SetZenithalDistributionFunction(TF1 *func) {fZenithDist = func;}
	35	void SetMomentumResolution(Float_t res=1.) {fPResolution=res;}
	36
	37	Float_t GetMomentum() const;
	38	Float_t GetZenithAngle(Float_t mom) const;
	39
1881b46e	40
	41	void SetNumberOfParticles(Int_t nParticles) {fNParticles = nParticles;}
	42
b86e74f5	43	// The following methods are for testing pourpuses
	44	TF1* GetMomentumDistibution() const {return fMomentumDist;}
	45	TF1* GetUnfoldedDistribution() const {return fUnfoldedMomentumDist;}
	46
	47	TClonesArray* GetArray() const {return fPDist;}
	48
	49	protected:
	50	void InitApWeightFactors();
	51	void InitMomentumGeneration();
	52	void InitZenithalAngleGeneration();
	53	void GenerateOneMuonBundle();
	54	void GenerateOneSingleMuon(Bool_t withFlatMomentum=kFALSE);
	55
	56	private:
93a2041b	57	AliGenACORDE(const AliGenACORDE& gen);
	58	AliGenACORDE& operator= (const AliGenACORDE& gen);
	59
b86e74f5	60	Int_t fIpart; //! Particle type.
	61	ECRMode fCRMode; //! Cosmic muons generation method flag
	62	TString* fCRModeName; //! Cosmic muons generation mode name
	63
	64	Float_t fXwidth; //! X width of the grid
	65	Int_t fNx; //! Number of divisions in x
	66	Float_t fZwidth; //! Z widht of the grid
	67	Int_t fNz; //! Number of divisions in z
	68	Bool_t fMuonGrid; //! Flag for method (Muon-bundles) checkout
	69
	70	Float_t fZenithMin; //! Minimum zenithal angle.
	71	Float_t fZenithMax; //! Maximum zenithal angle.
	72
	73	Float_t fAzimuthMin; //! Minimum azimuthal angle.
	74	Float_t fAzimuthMax; //! Maximum azimuthal angle.
	75
	76	Float_t fPRange; //! Cosmic muon momentum range width in GeVs.
	77	Float_t fPResolution; //! Momentum resolution in GeVs.
	78
	79	TArrayF* fAp; //! a(p) correction factors for the ang. dist.
	80
	81	TF1* fMomentumDist; //! Function to generate the momentum dist.
	82	TF1* fUnfoldedMomentumDist; //!
	83	TF1* fZenithDist; //! Function to generate the zenith angle dist.
	84
	85	TClonesArray* fPDist; //! Array of fZenithDist, to be used by a(p).
	86
1881b46e	87	Int_t fNParticles; //! Fixed number of particles per event
1881b46e	88
b86e74f5	89	ClassDef(AliGenACORDE, 1) // Generator for AliACORDE class
b86e74f5	90	};
2814d35e	91
	92	typedef AliGenACORDE AliGenCRT; // for backward compatibility
	93
b86e74f5	94	#endif // ALIGENACORDE_H