[u/mrichter/AliRoot.git] / THerwig / AliGenHerwig.h

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

/* $Id$ */

// Generator using HERWIG as an external generator
// The main HERWIG options are accessable for the user through this interface.
// Author Andreas Morsch
// andreas.morsch@cern.ch

#include "AliGenMC.h"
#include <TString.h>
#include <TArrayI.h>
#include <AliRndm.h>
#include <AliStructFuncType.h>

class THerwig6;
class TArrayI;
class TParticle;
class TClonesArray;


class AliGenHerwig : public AliGenMC

{
    enum {kNoTrigger, kHardProcesses, kDirectPhotons};

 public:
    AliGenHerwig();
    AliGenHerwig(Int_t npart);
    AliGenHerwig(const AliGenHerwig &Herwig);
    virtual ~AliGenHerwig();
    virtual void    Generate();
    virtual void    Init();
    virtual void    InitJimmy();
    // set centre of mass energy
    virtual void    SetBeamMomenta(Float_t p1=7000., Float_t p2=7000.)
	{fMomentum1 = p1; fMomentum2 = p2;}
    virtual void    SetProcess(Int_t proc)            {fProcess = proc;}
    virtual void    KeepFullEvent();
    virtual void    SetDecaysOff(Int_t flag=1)        {fDecaysOff = flag;}
    virtual void    SetTrigger(Int_t flag=kNoTrigger) {fTrigger   = flag;}
    virtual void    SetFlavor(Int_t flag=0)           {fFlavor    = flag;}
    virtual void    SetSelectAll(Int_t flag=0)        {fSelectAll = flag;}
    AliGenHerwig &  operator=(const AliGenHerwig & rhs);
    virtual void    SetStrucFunc(StrucFunc_t func = kCTEQ5L)
      {fStrucFunc = func;}
    virtual void    SetPtHardMin(Double_t pt) {fPtHardMin=pt;}
    virtual void    SetPtRMS(Double_t pt) {fPtRMS=pt;}
    virtual void    SetMaxPr(Int_t i) {fMaxPr=i;}
    virtual void    SetMaxErrors(Int_t i) {fMaxErrors=i;}
    virtual void    FinishRun();
    virtual void    FinishRunJimmy();
    virtual void    SetEnSoft(Double_t e) {fEnSoft=e;}

    virtual void    SetHardProcessFile(TString filename) {fFileName=TString(filename);};
    virtual void    SetEventListRange(Int_t eventFirst=-1, Int_t eventLast=-1);

 protected:
    Bool_t SelectFlavor(Int_t pid);

 protected:
    TString     fAutPDF;         // PDF group
    Int_t       fModPDF;         // PDF set
    StrucFunc_t fStrucFunc;      //Structure Function
    Int_t       fKeep;           // Flag to keep full event information
    Int_t       fDecaysOff;      // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
    Int_t       fTrigger;        // Trigger type
    Int_t       fSelectAll;      // Flag to write the full event
    Int_t       fFlavor;         // Selected particle flavor 4: charm+beauty 5: beauty
    Float_t     fEnergyCMS;      // Centre of mass energy
    Float_t     fMomentum1;      // Momentum of projectile
    Float_t     fMomentum2;      // Momentum of target
    Float_t     fKineBias;       // Bias from kinematic selection
    Int_t       fTrials;         // Number of trials
    Float_t     fXsection;       // Cross-section
    THerwig6    *fHerwig;        // Herwig
    Int_t       fProcess;        // Process number
    Double_t    fPtHardMin;      // lower pT-hard cut
    Double_t    fPtRMS;          // intrinsic pt of incoming hadrons
    Int_t       fMaxPr;          // maximum number of events to print out
    Int_t       fMaxErrors;      // maximum number of errors allowed
    Double_t    fEnSoft;          // change on soft energy distribution
    Int_t       fEv1Pr;          // first event to be printed
    Int_t       fEv2Pr;          // last event to be printed
    TString     fFileName;       //!Name of file to read from hard scattering

 private:
    // check if particle is selected as parent particle
    Bool_t ParentSelected(Int_t ip);
    // check if particle is selected as child particle
    Bool_t ChildSelected(Int_t ip);
    // adjust the weight from kinematic cuts
    void   AdjustWeights();
    // check seleted daughters
    Bool_t DaughtersSelection(TParticle* iparticle, TClonesArray* particles);
    // check if stable
    Bool_t Stable(TParticle*  particle);

    void InitPDF();

    ClassDef(AliGenHerwig,1) // AliGenerator interface to Herwig
};
#endif
Commit	Line	Data
618e1dc0	1	#ifndef ALIGENHERWIG_H
	2	#define ALIGENHERWIG_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	// Generator using HERWIG as an external generator
	9	// The main HERWIG options are accessable for the user through this interface.
c4ff35db	10	// Author Andreas Morsch
c4ff35db	11	// andreas.morsch@cern.ch
618e1dc0	12
	13	#include "AliGenMC.h"
	14	#include <TString.h>
	15	#include <TArrayI.h>
	16	#include <AliRndm.h>
	17	#include <AliStructFuncType.h>
	18
	19	class THerwig6;
	20	class TArrayI;
	21	class TParticle;
	22	class TClonesArray;
	23
	24
	25	class AliGenHerwig : public AliGenMC
	26
	27	{
	28	enum {kNoTrigger, kHardProcesses, kDirectPhotons};
	29
	30	public:
	31	AliGenHerwig();
	32	AliGenHerwig(Int_t npart);
	33	AliGenHerwig(const AliGenHerwig &Herwig);
	34	virtual ~AliGenHerwig();
	35	virtual void Generate();
	36	virtual void Init();
7677b281	37	virtual void InitJimmy();
618e1dc0	38	// set centre of mass energy
	39	virtual void SetBeamMomenta(Float_t p1=7000., Float_t p2=7000.)
	40	{fMomentum1 = p1; fMomentum2 = p2;}
7677b281	41	virtual void SetProcess(Int_t proc) {fProcess = proc;}
618e1dc0	42	virtual void KeepFullEvent();
	43	virtual void SetDecaysOff(Int_t flag=1) {fDecaysOff = flag;}
	44	virtual void SetTrigger(Int_t flag=kNoTrigger) {fTrigger = flag;}
7677b281	45	virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
7677b281	46	virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
618e1dc0	47	AliGenHerwig & operator=(const AliGenHerwig & rhs);
7677b281	48	virtual void SetStrucFunc(StrucFunc_t func = kCTEQ5L)
618e1dc0	49	{fStrucFunc = func;}
	50	virtual void SetPtHardMin(Double_t pt) {fPtHardMin=pt;}
	51	virtual void SetPtRMS(Double_t pt) {fPtRMS=pt;}
	52	virtual void SetMaxPr(Int_t i) {fMaxPr=i;}
	53	virtual void SetMaxErrors(Int_t i) {fMaxErrors=i;}
	54	virtual void FinishRun();
7677b281	55	virtual void FinishRunJimmy();
618e1dc0	56	virtual void SetEnSoft(Double_t e) {fEnSoft=e;}
7677b281	57
e2054d85	58	virtual void SetHardProcessFile(TString filename) {fFileName=TString(filename);};
	59	virtual void SetEventListRange(Int_t eventFirst=-1, Int_t eventLast=-1);
	60
618e1dc0	61	protected:
	62	Bool_t SelectFlavor(Int_t pid);
	63
	64	protected:
618e1dc0	65	TString fAutPDF; // PDF group
	66	Int_t fModPDF; // PDF set
	67	StrucFunc_t fStrucFunc; //Structure Function
	68	Int_t fKeep; // Flag to keep full event information
	69	Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
	70	Int_t fTrigger; // Trigger type
	71	Int_t fSelectAll; // Flag to write the full event
	72	Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
	73	Float_t fEnergyCMS; // Centre of mass energy
	74	Float_t fMomentum1; // Momentum of projectile
	75	Float_t fMomentum2; // Momentum of target
	76	Float_t fKineBias; // Bias from kinematic selection
	77	Int_t fTrials; // Number of trials
618e1dc0	78	Float_t fXsection; // Cross-section
	79	THerwig6 *fHerwig; // Herwig
	80	Int_t fProcess; // Process number
7677b281	81	Double_t fPtHardMin; // lower pT-hard cut
618e1dc0	82	Double_t fPtRMS; // intrinsic pt of incoming hadrons
	83	Int_t fMaxPr; // maximum number of events to print out
	84	Int_t fMaxErrors; // maximum number of errors allowed
	85	Double_t fEnSoft; // change on soft energy distribution
e2054d85	86	Int_t fEv1Pr; // first event to be printed
	87	Int_t fEv2Pr; // last event to be printed
	88	TString fFileName; //!Name of file to read from hard scattering
7677b281	89
618e1dc0	90	private:
	91	// check if particle is selected as parent particle
	92	Bool_t ParentSelected(Int_t ip);
	93	// check if particle is selected as child particle
	94	Bool_t ChildSelected(Int_t ip);
	95	// adjust the weight from kinematic cuts
	96	void AdjustWeights();
	97	// check seleted daughters
	98	Bool_t DaughtersSelection(TParticle* iparticle, TClonesArray* particles);
	99	// check if stable
	100	Bool_t Stable(TParticle* particle);
7677b281	101
618e1dc0	102	void InitPDF();
	103
	104	ClassDef(AliGenHerwig,1) // AliGenerator interface to Herwig
	105	};
	106	#endif
	107
	108
	109
	110
	111