[u/mrichter/AliRoot.git] / EPOS / TEpos.h

/*
 *###################################################################
 *#        EPOS 1.67     K. WERNER, T. PIEROG, S. PORTEBOEUF.       #
 *#                      Contact: werner@subatech.in2p3.fr          #
 *###################################################################
 *
 * TEpos.h
 * 
 * Wraper class for interfacing EPOS model, derived from ROOT's TGenerator.
 * It generates temporary input file for the model, providing user with
 * ability to add his/hers own lines to the input.
 * Output is read directly from common blocks.
 *
 *      Author: Piotr Ostrowski, postrow@if.pw.edu.pl
 */


#ifndef TEPOS_H
#define TEPOS_H

#ifndef ROOT_TGenerator
#include "TGenerator.h"
#endif

#include <vector>
class TObjArray;

class TEpos : public TGenerator {


public:

   TEpos();

   virtual            ~TEpos();

   virtual void        Initialize();

   virtual void        GenerateEvent();

   virtual Int_t       ImportParticles(TClonesArray *particles, Option_t *option="");
   virtual TObjArray*  ImportParticles(Option_t *option="");

   //Geters for event header

   Float_t GetBimevt(); //   bimevt ........ absolute value of impact parameter
   Float_t GetPhievt(); //   phievt ........ angle of impact parameter
   Int_t GetKolevt();   //   kolevt ........ number of collisions
   Int_t GetKoievt();   //   koievt ........ number of inelastic collisions
   Float_t GetPmxevt(); //   pmxevt ........ reference momentum
   Float_t GetEgyevt(); //   egyevt ........ pp cm energy (hadron) or string energy (lepton)
   Int_t GetNpjevt();   //   npjevt ........ number of primary projectile participants
   Int_t GetNtgevt();   //   ntgevt ........ number of primary target participants
   Int_t GetNpnevt();   //   npnevt ........ number of primary projectile neutron spectators
   Int_t GetNppevt();   //   nppevt ........ number of primary projectile proton spectators
   Int_t GetNtnevt();   //   ntnevt ........ number of primary target neutron spectators
   Int_t GetNtpevt();   //   ntpevt ........ number of primary target proton spectators
   Int_t GetJpnevt();   //   jpnevt ........ number of absolute projectile neutron spectators
   Int_t GetJppevt();   //   jppevt ........ number of absolute projectile proton spectators
   Int_t GetJtnevt();   //   jtnevt ........ number of absolute target neutron spectators
   Int_t GetJtpevt();   //   jtpevt ........ number of absolute target proton spectators
   Float_t GetXbjevt(); //   xbjevt ........ bjorken x for dis
   Float_t GetQsqevt(); //   qsqevt ........ q**2 for dis
   Int_t GetNglevt();   //   nglevt ........ number of collisions acc to  Glauber
   Float_t GetZppevt(); //   zppevt ........ average Z-parton-proj
   Float_t GetZptevt(); //   zptevt ........ average Z-parton-targ

   //Parameters for the generator:
   void SetLaproj(Int_t laproj) { fLaproj = laproj; }
   void SetMaproj(Int_t maproj) { fMaproj = maproj; }
   void SetLatarg(Int_t latarg) { fLatarg = latarg; }
   void SetMatarg(Int_t matarg) { fMatarg = matarg; }

   void SetBminim(Float_t bminim) { fBminim = bminim; }
   void SetBmaxim(Float_t bmaxim) { fBmaxim = bmaxim; }

   void SetPhimin(Float_t phimin) { fPhimin = phimin; }
   void SetPhimax(Float_t phimax) { fPhimax = phimax; }

   void SetEcms(Float_t ecms) { fEcms = ecms; }
   void SetSplitting(Bool_t splitting) { fSplitting = splitting; }

   void AddNoDecay(Int_t nodecay);
   void AddExtraInputLine(const char *);

   Int_t GetLaproj() const { return fLaproj; }
   Int_t GetMaproj() const { return fMaproj; }
   Int_t GetLatarg() const { return fLatarg; }
   Int_t GetMatarg() const { return fMatarg; }

   Double_t GetBminim() const { return fBminim; }
   Double_t GetBmaxim() const { return fBmaxim; }

   Double_t GetPhimin() const { return fPhimin; }
   Double_t GetPhimax() const { return fPhimax; }

   Double_t GetEcms() const { return fEcms; }
   Bool_t GetSplitting() const { return fSplitting; }

protected:
   virtual void GenerateInputFile();
   virtual const char * GetInputFileName() const { return "/tmp/epos.input"; }

   Int_t fLaproj;	//atomic number of projectile
   Int_t fMaproj;	//mass number of projectile
   Int_t fLatarg;	//atomic number of target
   Int_t fMatarg;	//mass number of target
   Float_t fBminim;     //lower limit for impact parameter
   Float_t fBmaxim;     //upper limit for impact parameter
   Float_t fPhimin;     //lower limit for reaction plane angle
   Float_t fPhimax;     //upper limit for reaction plane angle
   Float_t fEcms;       //energy in CMS 
   Bool_t fSplitting;   //consideration of parton splitting and fussion

   std::vector<Int_t> fNoDecays;		// list of user supplied ISAJET codes of particles that will not be decayed by the model
   std::vector<const char * > fExtraInputLines; // list of user supplied lines to be added to EPOS input


private:
   ClassDef(TEpos,1)  //Interface to EPOS Event Generator
};

#endif
Commit	Line	Data
9ef1c2d9	1	/*
	2	*###################################################################
	3	*# EPOS 1.67 K. WERNER, T. PIEROG, S. PORTEBOEUF. #
	4	*# Contact: werner@subatech.in2p3.fr #
	5	*###################################################################
	6	*
	7	* TEpos.h
	8	*
	9	* Wraper class for interfacing EPOS model, derived from ROOT's TGenerator.
	10	* It generates temporary input file for the model, providing user with
	11	* ability to add his/hers own lines to the input.
	12	* Output is read directly from common blocks.
	13	*
	14	* Author: Piotr Ostrowski, postrow@if.pw.edu.pl
	15	*/
	16
	17
	18	#ifndef TEPOS_H
	19	#define TEPOS_H
	20
	21	#ifndef ROOT_TGenerator
	22	#include "TGenerator.h"
	23	#endif
	24
	25	#include <vector>
	26	class TObjArray;
	27
	28	class TEpos : public TGenerator {
	29
	30
	31	public:
	32
	33	TEpos();
	34
	35	virtual ~TEpos();
	36
	37	virtual void Initialize();
	38
	39	virtual void GenerateEvent();
	40
	41	virtual Int_t ImportParticles(TClonesArray particles, Option_t option="");
	42	virtual TObjArray* ImportParticles(Option_t *option="");
	43
	44	//Geters for event header
	45
	46	Float_t GetBimevt(); // bimevt ........ absolute value of impact parameter
	47	Float_t GetPhievt(); // phievt ........ angle of impact parameter
	48	Int_t GetKolevt(); // kolevt ........ number of collisions
	49	Int_t GetKoievt(); // koievt ........ number of inelastic collisions
	50	Float_t GetPmxevt(); // pmxevt ........ reference momentum
	51	Float_t GetEgyevt(); // egyevt ........ pp cm energy (hadron) or string energy (lepton)
	52	Int_t GetNpjevt(); // npjevt ........ number of primary projectile participants
	53	Int_t GetNtgevt(); // ntgevt ........ number of primary target participants
	54	Int_t GetNpnevt(); // npnevt ........ number of primary projectile neutron spectators
	55	Int_t GetNppevt(); // nppevt ........ number of primary projectile proton spectators
	56	Int_t GetNtnevt(); // ntnevt ........ number of primary target neutron spectators
	57	Int_t GetNtpevt(); // ntpevt ........ number of primary target proton spectators
	58	Int_t GetJpnevt(); // jpnevt ........ number of absolute projectile neutron spectators
	59	Int_t GetJppevt(); // jppevt ........ number of absolute projectile proton spectators
	60	Int_t GetJtnevt(); // jtnevt ........ number of absolute target neutron spectators
	61	Int_t GetJtpevt(); // jtpevt ........ number of absolute target proton spectators
	62	Float_t GetXbjevt(); // xbjevt ........ bjorken x for dis
	63	Float_t GetQsqevt(); // qsqevt ........ q**2 for dis
	64	Int_t GetNglevt(); // nglevt ........ number of collisions acc to Glauber
65	Float_t GetZppevt(); // zppevt ........ average Z-parton-proj
66	Float_t GetZptevt(); // zptevt ........ average Z-parton-targ
67
68	//Parameters for the generator:
69	void SetLaproj(Int_t laproj) { fLaproj = laproj; }
70	void SetMaproj(Int_t maproj) { fMaproj = maproj; }
71	void SetLatarg(Int_t latarg) { fLatarg = latarg; }
72	void SetMatarg(Int_t matarg) { fMatarg = matarg; }
73
74	void SetBminim(Float_t bminim) { fBminim = bminim; }
75	void SetBmaxim(Float_t bmaxim) { fBmaxim = bmaxim; }
76
77	void SetPhimin(Float_t phimin) { fPhimin = phimin; }
78	void SetPhimax(Float_t phimax) { fPhimax = phimax; }
79
80	void SetEcms(Float_t ecms) { fEcms = ecms; }
81	void SetSplitting(Bool_t splitting) { fSplitting = splitting; }
82
83	void AddNoDecay(Int_t nodecay);
84	void AddExtraInputLine(const char *);
85
86	Int_t GetLaproj() const { return fLaproj; }
87	Int_t GetMaproj() const { return fMaproj; }
88	Int_t GetLatarg() const { return fLatarg; }
89	Int_t GetMatarg() const { return fMatarg; }
90
91	Double_t GetBminim() const { return fBminim; }
92	Double_t GetBmaxim() const { return fBmaxim; }
93
94	Double_t GetPhimin() const { return fPhimin; }
95	Double_t GetPhimax() const { return fPhimax; }
96
97	Double_t GetEcms() const { return fEcms; }
98	Bool_t GetSplitting() const { return fSplitting; }
99
100	protected:
101	virtual void GenerateInputFile();
102	virtual const char * GetInputFileName() const { return "/tmp/epos.input"; }
103
104	Int_t fLaproj; //atomic number of projectile
105	Int_t fMaproj; //mass number of projectile
106	Int_t fLatarg; //atomic number of target
107	Int_t fMatarg; //mass number of target
108	Float_t fBminim; //lower limit for impact parameter
109	Float_t fBmaxim; //upper limit for impact parameter
110	Float_t fPhimin; //lower limit for reaction plane angle
111	Float_t fPhimax; //upper limit for reaction plane angle
112	Float_t fEcms; //energy in CMS
113	Bool_t fSplitting; //consideration of parton splitting and fussion
114
115	std::vector<Int_t> fNoDecays; // list of user supplied ISAJET codes of particles that will not be decayed by the model
116	std::vector<const char * > fExtraInputLines; // list of user supplied lines to be added to EPOS input
117
118
119	private:
120	ClassDef(TEpos,1) //Interface to EPOS Event Generator
121	};
122
123	#endif
124