[u/mrichter/AliRoot.git] / EVGEN / AliGenMC.h

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

/* $Id$ */


// Base class for generators using external MC generators.
// For example AliGenPythia using Pythia.
// Provides basic functionality: setting of kinematic cuts on 
// decay products and particle selection.
// andreas.morsch@cern.ch

class TClonesArray;
class TParticle;
#include <TArrayI.h>   
#include <TString.h>

class AliGeometry;
class AliGenEventHeader;

#include "AliDecayer.h"
#include "AliGenerator.h"

class AliGenMC : public AliGenerator
{
 public:
    AliGenMC();
    AliGenMC(Int_t npart);
    virtual ~AliGenMC();
    virtual void Init();
    virtual void SetForceDecay(Decay_t decay = kAll) {fForceDecay = decay;}
    virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;}
    virtual void SetChildMomentumRange(Float_t pmin = 0, Float_t pmax = 1.e10)
	{fChildPMin = pmin; fChildPMax = pmax;}
    virtual void SetChildPtRange(Float_t ptmin = 0, Float_t ptmax = 20.)
	{fChildPtMin = ptmin; fChildPtMax = ptmax;}
    virtual void SetChildPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
	{fChildPhiMin = TMath::Pi()*phimin/180;
	fChildPhiMax  = TMath::Pi()*phimax/180;}
    virtual void SetChildThetaRange(Float_t thetamin = 0, Float_t thetamax = 180)
	{fChildThetaMin = TMath::Pi()*thetamin/180;
	fChildThetaMax  = TMath::Pi()*thetamax/180;}
    virtual void SetChildYRange(Float_t ymin = -12, Float_t ymax = 12)
	{fChildYMin = ymin;
	fChildYMax  = ymax;}
    virtual void SetMaximumLifetime(Float_t time = 1.e-15) {fMaxLifeTime = time;}
   
    virtual void SetGeometryAcceptance(AliGeometry * GeometryAcceptance=0) {fGeometryAcceptance = GeometryAcceptance;}

    virtual void SetPdgCodeParticleforAcceptanceCut(Int_t PdgCodeParticleforAcceptanceCut=0) {fPdgCodeParticleforAcceptanceCut = PdgCodeParticleforAcceptanceCut;}

    virtual void SetNumberOfAcceptedParticles(Int_t NumberOfAcceptedParticles=2) {fNumberOfAcceptedParticles = NumberOfAcceptedParticles;}
    
    virtual Bool_t CheckAcceptanceGeometry(Int_t np, TClonesArray* particles);
    virtual void   SetProjectile(TString proj="P", Int_t a = 1, Int_t z = 1)
	{fProjectile = proj; fAProjectile = a; fZProjectile = z;}    
    virtual void   SetTarget(TString tar="P", Int_t a = 1, Int_t z = 1)
	{fTarget = tar; fATarget = a; fZTarget = z;}
    virtual void   SetCrossingAngle(Float_t phiX, Float_t phiY) {fXingAngleX = phiX; fXingAngleY = phiY;}
    virtual void Boost();
    virtual void AddHeader(AliGenEventHeader* header);
 protected:
    // check if particle is selected as parent particle
    Bool_t ParentSelected(Int_t ip) const;
    // check if particle is selected as child particle
    Bool_t ChildSelected(Int_t ip) const;
    // all kinematic selection cuts go here 
    Bool_t KinematicSelection(TParticle *particle, Int_t flag) const;
    Int_t  CheckPDGCode(Int_t pdgcode) const;

 protected:
    TClonesArray* fParticles;   //!Particle  List
    TArrayI     fParentSelect;  //!Parent particles to be selected 
    TArrayI     fChildSelect;   //!Decay products to be selected
    Int_t       fCutOnChild;    // Cuts on decay products (children)  are enabled/disabled
    Float_t     fChildPtMin;    // Children minimum pT
    Float_t     fChildPtMax;    // Children maximum pT
    Float_t     fChildPMin;     // Children minimum p
    Float_t     fChildPMax;     // Children maximum p
    Float_t     fChildPhiMin;   // Children minimum phi
    Float_t     fChildPhiMax;   // Children maximum phi
    Float_t     fChildThetaMin; // Children minimum theta
    Float_t     fChildThetaMax; // Children maximum theta
    Float_t     fChildYMin;     // Children minimum y
    Float_t     fChildYMax;     // Children maximum y
    Float_t     fXingAngleX;    // Crossing angle X
    Float_t     fXingAngleY;    // Crossing angle Y    
    Decay_t     fForceDecay;    // Decay channel forced
    Float_t     fMaxLifeTime;   // Maximum lifetime for unstable particles
    Int_t       fAProjectile;   // Projectile A
    Int_t       fZProjectile;   // Projectile Z
    Int_t       fATarget;       // Target A
    Int_t       fZTarget;       // Target Z
    TString     fProjectile;    // Projectile
    TString     fTarget;        // Target
    Double_t    fDyBoost;       // dy for boost into lab frame
    AliGeometry * fGeometryAcceptance; // Geometry to which particles must be simulated
    Int_t       fPdgCodeParticleforAcceptanceCut;  // Abs(PDG Code) of the particle to which the GeometryAcceptance must be applied
    Int_t       fNumberOfAcceptedParticles;  // Number of accepted particles in GeometryAcceptance with the right Abs(PdgCode) 
    Int_t       fNprimaries;                 // Number of produced and stored particles
    
 private:
    AliGenMC(const AliGenMC &MC);
    AliGenMC & operator=(const AliGenMC & rhs);
    
    ClassDef(AliGenMC,5)       // AliGenerator implementation for generators using MC methods
};
#endif
Commit	Line	Data
	1	#ifndef ALIGENMC_H
	2	#define ALIGENMC_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
	9	// Base class for generators using external MC generators.
	10	// For example AliGenPythia using Pythia.
	11	// Provides basic functionality: setting of kinematic cuts on
	12	// decay products and particle selection.
	13	// andreas.morsch@cern.ch
	14
	15	class TClonesArray;
	16	class TParticle;
	17	#include <TArrayI.h>
	18	#include <TString.h>
	19
	20	class AliGeometry;
	21	class AliGenEventHeader;
	22
	23	#include "AliDecayer.h"
	24	#include "AliGenerator.h"
	25
	26	class AliGenMC : public AliGenerator
	27	{
	28	public:
	29	AliGenMC();
	30	AliGenMC(Int_t npart);
	31	virtual ~AliGenMC();
	32	virtual void Init();
	33	virtual void SetForceDecay(Decay_t decay = kAll) {fForceDecay = decay;}
	34	virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;}
	35	virtual void SetChildMomentumRange(Float_t pmin = 0, Float_t pmax = 1.e10)
	36	{fChildPMin = pmin; fChildPMax = pmax;}
	37	virtual void SetChildPtRange(Float_t ptmin = 0, Float_t ptmax = 20.)
	38	{fChildPtMin = ptmin; fChildPtMax = ptmax;}
	39	virtual void SetChildPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
	40	{fChildPhiMin = TMath::Pi()*phimin/180;
	41	fChildPhiMax = TMath::Pi()*phimax/180;}
	42	virtual void SetChildThetaRange(Float_t thetamin = 0, Float_t thetamax = 180)
	43	{fChildThetaMin = TMath::Pi()*thetamin/180;
	44	fChildThetaMax = TMath::Pi()*thetamax/180;}
	45	virtual void SetChildYRange(Float_t ymin = -12, Float_t ymax = 12)
	46	{fChildYMin = ymin;
	47	fChildYMax = ymax;}
	48	virtual void SetMaximumLifetime(Float_t time = 1.e-15) {fMaxLifeTime = time;}
	49
	50	virtual void SetGeometryAcceptance(AliGeometry * GeometryAcceptance=0) {fGeometryAcceptance = GeometryAcceptance;}
	51
	52	virtual void SetPdgCodeParticleforAcceptanceCut(Int_t PdgCodeParticleforAcceptanceCut=0) {fPdgCodeParticleforAcceptanceCut = PdgCodeParticleforAcceptanceCut;}
	53
	54	virtual void SetNumberOfAcceptedParticles(Int_t NumberOfAcceptedParticles=2) {fNumberOfAcceptedParticles = NumberOfAcceptedParticles;}
	55
	56	virtual Bool_t CheckAcceptanceGeometry(Int_t np, TClonesArray* particles);
	57	virtual void SetProjectile(TString proj="P", Int_t a = 1, Int_t z = 1)
	58	{fProjectile = proj; fAProjectile = a; fZProjectile = z;}
	59	virtual void SetTarget(TString tar="P", Int_t a = 1, Int_t z = 1)
	60	{fTarget = tar; fATarget = a; fZTarget = z;}
	61	virtual void SetCrossingAngle(Float_t phiX, Float_t phiY) {fXingAngleX = phiX; fXingAngleY = phiY;}
	62	virtual void Boost();
	63	virtual void AddHeader(AliGenEventHeader* header);
	64	protected:
	65	// check if particle is selected as parent particle
	66	Bool_t ParentSelected(Int_t ip) const;
	67	// check if particle is selected as child particle
	68	Bool_t ChildSelected(Int_t ip) const;
	69	// all kinematic selection cuts go here
	70	Bool_t KinematicSelection(TParticle *particle, Int_t flag) const;
	71	Int_t CheckPDGCode(Int_t pdgcode) const;
	72
	73	protected:
	74	TClonesArray* fParticles; //!Particle List
	75	TArrayI fParentSelect; //!Parent particles to be selected
	76	TArrayI fChildSelect; //!Decay products to be selected
	77	Int_t fCutOnChild; // Cuts on decay products (children) are enabled/disabled
	78	Float_t fChildPtMin; // Children minimum pT
	79	Float_t fChildPtMax; // Children maximum pT
	80	Float_t fChildPMin; // Children minimum p
	81	Float_t fChildPMax; // Children maximum p
	82	Float_t fChildPhiMin; // Children minimum phi
	83	Float_t fChildPhiMax; // Children maximum phi
	84	Float_t fChildThetaMin; // Children minimum theta
	85	Float_t fChildThetaMax; // Children maximum theta
	86	Float_t fChildYMin; // Children minimum y
	87	Float_t fChildYMax; // Children maximum y
	88	Float_t fXingAngleX; // Crossing angle X
	89	Float_t fXingAngleY; // Crossing angle Y
	90	Decay_t fForceDecay; // Decay channel forced
	91	Float_t fMaxLifeTime; // Maximum lifetime for unstable particles
	92	Int_t fAProjectile; // Projectile A
	93	Int_t fZProjectile; // Projectile Z
	94	Int_t fATarget; // Target A
	95	Int_t fZTarget; // Target Z
	96	TString fProjectile; // Projectile
	97	TString fTarget; // Target
	98	Double_t fDyBoost; // dy for boost into lab frame
	99	AliGeometry * fGeometryAcceptance; // Geometry to which particles must be simulated
	100	Int_t fPdgCodeParticleforAcceptanceCut; // Abs(PDG Code) of the particle to which the GeometryAcceptance must be applied
	101	Int_t fNumberOfAcceptedParticles; // Number of accepted particles in GeometryAcceptance with the right Abs(PdgCode)
	102	Int_t fNprimaries; // Number of produced and stored particles
	103
	104	private:
	105	AliGenMC(const AliGenMC &MC);
	106	AliGenMC & operator=(const AliGenMC & rhs);
	107
	108	ClassDef(AliGenMC,5) // AliGenerator implementation for generators using MC methods
	109	};
	110	#endif
	111
	112
	113
	114
	115