[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();
    // 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 = kGRVHO) 
      {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    SetEnSoft(Double_t e) {fEnSoft=e;}
 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
      
 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();
	37	// set centre of mass energy
	38	virtual void SetBeamMomenta(Float_t p1=7000., Float_t p2=7000.)
	39	{fMomentum1 = p1; fMomentum2 = p2;}
93831bcf	40	virtual void SetProcess(Int_t proc) {fProcess = proc;}
618e1dc0	41	virtual void KeepFullEvent();
	42	virtual void SetDecaysOff(Int_t flag=1) {fDecaysOff = flag;}
	43	virtual void SetTrigger(Int_t flag=kNoTrigger) {fTrigger = flag;}
	44	virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
	45	virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
	46	AliGenHerwig & operator=(const AliGenHerwig & rhs);
	47	virtual void SetStrucFunc(StrucFunc_t func = kGRVHO)
	48	{fStrucFunc = func;}
	49	virtual void SetPtHardMin(Double_t pt) {fPtHardMin=pt;}
	50	virtual void SetPtRMS(Double_t pt) {fPtRMS=pt;}
	51	virtual void SetMaxPr(Int_t i) {fMaxPr=i;}
	52	virtual void SetMaxErrors(Int_t i) {fMaxErrors=i;}
	53	virtual void FinishRun();
	54	virtual void SetEnSoft(Double_t e) {fEnSoft=e;}
	55	protected:
	56	Bool_t SelectFlavor(Int_t pid);
	57
	58	protected:
618e1dc0	59	TString fAutPDF; // PDF group
	60	Int_t fModPDF; // PDF set
	61	StrucFunc_t fStrucFunc; //Structure Function
	62	Int_t fKeep; // Flag to keep full event information
	63	Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
	64	Int_t fTrigger; // Trigger type
	65	Int_t fSelectAll; // Flag to write the full event
	66	Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
	67	Float_t fEnergyCMS; // Centre of mass energy
	68	Float_t fMomentum1; // Momentum of projectile
	69	Float_t fMomentum2; // Momentum of target
	70	Float_t fKineBias; // Bias from kinematic selection
	71	Int_t fTrials; // Number of trials
618e1dc0	72	Float_t fXsection; // Cross-section
	73	THerwig6 *fHerwig; // Herwig
	74	Int_t fProcess; // Process number
	75	Double_t fPtHardMin; // lower pT-hard cut
	76	Double_t fPtRMS; // intrinsic pt of incoming hadrons
	77	Int_t fMaxPr; // maximum number of events to print out
	78	Int_t fMaxErrors; // maximum number of errors allowed
	79	Double_t fEnSoft; // change on soft energy distribution
	80
	81	private:
	82	// check if particle is selected as parent particle
	83	Bool_t ParentSelected(Int_t ip);
	84	// check if particle is selected as child particle
	85	Bool_t ChildSelected(Int_t ip);
	86	// adjust the weight from kinematic cuts
	87	void AdjustWeights();
	88	// check seleted daughters
	89	Bool_t DaughtersSelection(TParticle* iparticle, TClonesArray* particles);
	90	// check if stable
	91	Bool_t Stable(TParticle* particle);
	92
	93	void InitPDF();
	94
	95	ClassDef(AliGenHerwig,1) // AliGenerator interface to Herwig
	96	};
	97	#endif
	98
	99
	100
	101
	102