THerwig/AliGenHerwig.h

   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.
  10
  11
  12 #include "AliGenMC.h"
  13 #include <TString.h>
  14 #include <TArrayI.h>
  15 #include <AliRndm.h>
  16 #include <AliStructFuncType.h>
  17
  18 class THerwig6;
  19 class TArrayI;
  20 class TParticle;
  21 class TClonesArray;
  22
  23
  24 class AliGenHerwig : public AliGenMC
  25
  26 {
  27     enum {kNoTrigger, kHardProcesses, kDirectPhotons};
  28
  29  public:
  30     AliGenHerwig();
  31     AliGenHerwig(Int_t npart);
  32     AliGenHerwig(const AliGenHerwig &Herwig);
  33     virtual ~AliGenHerwig();
  34     virtual void    Generate();
  35     virtual void    Init();
  36     // set centre of mass energy
  37     virtual void    SetBeamMomenta(Float_t p1=7000., Float_t p2=7000.)
  38         {fMomentum1 = p1; fMomentum2 = p2;}
  39     virtual void    SetProjectile(TString proj="P")  {fProjectile = proj;}
  40     virtual void    SetTarget(TString tar="P")       {fTarget = tar;}
  41     virtual void    SetProcess(Int_t proc)       {fProcess = proc;}
  42     virtual void    KeepFullEvent();
  43     virtual void    SetDecaysOff(Int_t flag=1)        {fDecaysOff = flag;}
  44     virtual void    SetTrigger(Int_t flag=kNoTrigger) {fTrigger   = flag;}
  45     virtual void    SetFlavor(Int_t flag=0)           {fFlavor    = flag;}
  46     virtual void    SetSelectAll(Int_t flag=0)        {fSelectAll = flag;}
  47     AliGenHerwig &  operator=(const AliGenHerwig & rhs);
  48     virtual void    SetStrucFunc(StrucFunc_t func = kGRVHO)
  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();
  55     virtual void    SetEnSoft(Double_t e) {fEnSoft=e;}
  56  protected:
  57     Bool_t SelectFlavor(Int_t pid);
  58
  59  protected:
  60     TString     fProjectile;     // Projectile
  61     TString     fTarget;         // Target
  62     TString     fAutPDF;         // PDF group
  63     Int_t       fModPDF;         // PDF set
  64     StrucFunc_t fStrucFunc;      //Structure Function
  65     Int_t       fKeep;           // Flag to keep full event information
  66     Int_t       fDecaysOff;      // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
  67     Int_t       fTrigger;        // Trigger type
  68     Int_t       fSelectAll;      // Flag to write the full event
  69     Int_t       fFlavor;         // Selected particle flavor 4: charm+beauty 5: beauty
  70     Float_t     fEnergyCMS;      // Centre of mass energy
  71     Float_t     fMomentum1;      // Momentum of projectile
  72     Float_t     fMomentum2;      // Momentum of target
  73     Float_t     fKineBias;       // Bias from kinematic selection
  74     Int_t       fTrials;         // Number of trials
  75     TArrayI     fParentSelect;   // Parent particles to be selected
  76     TArrayI     fChildSelect;    // Decay products to be selected
  77     Float_t     fXsection;       // Cross-section
  78     THerwig6    *fHerwig;        // Herwig
  79     Int_t       fProcess;        // Process number
  80     Double_t    fPtHardMin;      // lower pT-hard cut
  81     Double_t    fPtRMS;          // intrinsic pt of incoming hadrons
  82     Int_t       fMaxPr;          // maximum number of events to print out
  83     Int_t       fMaxErrors;      // maximum number of errors allowed
  84     Double_t    fEnSoft;          // change on soft energy distribution
  85
  86  private:
  87     // check if particle is selected as parent particle
  88     Bool_t ParentSelected(Int_t ip);
  89     // check if particle is selected as child particle
  90     Bool_t ChildSelected(Int_t ip);
  91     // adjust the weight from kinematic cuts
  92     void   AdjustWeights();
  93     // check seleted daughters
  94     Bool_t DaughtersSelection(TParticle* iparticle, TClonesArray* particles);
  95     // check if stable
  96     Bool_t Stable(TParticle*  particle);
  97
  98     void InitPDF();
  99
 100     ClassDef(AliGenHerwig,1) // AliGenerator interface to Herwig
 101 };
 102 #endif
 103
 104
 105
 106
 107