Better estimate of production point

[u/mrichter/AliRoot.git] / TDPMjet / AliGenDPMjet.h

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

// Generator using DPMJET as an external generator
// The main DPMJET options are accessable for the user through this interface.

#include "AliGenMC.h"
#include "TDPMjet.h"
#include <TString.h>
#include <TArrayI.h>

class TDPMjet;
class TArrayI;
class TParticle;
class TClonesArray;
class TGraph;
class AliGenEventHeader;
class AliStack;
class AliRunLoader;
class AliGenDPMjetEventHeader;



class AliGenDPMjet : public AliGenMC
{

 public:
    AliGenDPMjet();
    AliGenDPMjet(Int_t npart);
    AliGenDPMjet(const AliGenDPMjet &Dpmjet);
    virtual ~AliGenDPMjet(); 
    virtual void    Generate();
    virtual void    Init();
    virtual void    FinishRun();
    virtual void    SetEnergyCMS(Float_t energy = 14000.) {fEnergyCMS = energy; fBeamEn = energy / 2.;}
    virtual void    SetImpactParameterRange(Float_t bmin=0., Float_t bmax=1.)
			{fMinImpactParam=bmin; fMaxImpactParam=bmax;}
    virtual void    SetProcess(DpmProcess_t iproc) {fProcess = iproc;}
    virtual void    SetCentral(Int_t icentr=-2) {fICentr = icentr;}
    virtual void    SetFlavor(Int_t flag=0)           {fFlavor     = flag;}
    virtual void    SetSelectAll(Int_t flag=0)        {fSelectAll  = flag;}
    virtual void    SetSpectators(Int_t spects=1)     {fSpectators = spects;}
    virtual void    SetBoostLHC(Int_t flag=0)         {fLHC        = flag;}
    virtual void    SetPi0Decay(Int_t iPi0)  {fPi0Decay = iPi0;}
    virtual Float_t GetEnergyCMS() {return fEnergyCMS;}
    virtual void    GetProjectile(Int_t& a, Int_t& z)
			{a = fAProjectile; z = fZProjectile;}    
    virtual void    GetTarget(Int_t& a, Int_t& z)
			{a = fATarget; z = fZTarget;}    
    virtual void    GetImpactParameterRange(Float_t& bmin, Float_t& bmax)
			{bmin = fMinImpactParam; bmax = fMaxImpactParam;}
    virtual Int_t   GetSpectators()        {return fSpectators;}
    virtual Int_t   GetFlavor()            {return fFlavor;}

    virtual void    SetGenImpPar(Float_t bValue) {fGenImpPar=bValue;}
    virtual Float_t GetGenImpPar() {return fGenImpPar;}
    
    AliGenDPMjet &  operator=(const AliGenDPMjet & rhs);
    void     AddHeader(AliGenEventHeader* header);

 protected:
    Bool_t SelectFlavor(Int_t pid);
    void   MakeHeader();

 protected:
    Float_t       fBeamEn; 	   // beam energy
    Float_t       fEnergyCMS;      // Centre of mass energy
    Float_t       fMinImpactParam; // minimum impact parameter
    Float_t       fMaxImpactParam; // maximum impact parameter	
    Int_t	  fICentr;	   // Flag to force central production
    Int_t         fSelectAll;      // Flag to write the full event
    Int_t         fFlavor;         // Selected particle flavor 4: charm+beauty 5: beauty
    Int_t         fTrials;         // Number of trials
    Int_t         fSpectators;     // put spectators on stack
    Int_t 	  fSpecn;	   // Num. of spectator neutrons
    Int_t 	  fSpecp;	   // Num. of spectator protons
    TDPMjet      *fDPMjet;         // DPMjet
    Int_t         fNoGammas;       // Don't write gammas if flag "on"
    Int_t         fLHC;            // Assume LHC as lab frame
    Int_t         fPi0Decay;       // Flag for pi0 decays
    Float_t	  fGenImpPar;	   // GeneratedImpactParameter
    DpmProcess_t  fProcess;        // Process type
    
 private:
    // adjust the weight from kinematic cuts
    void   AdjustWeights();
    // check seleted daughters
    Bool_t DaughtersSelection(TParticle* iparticle);
    // check if stable
    Bool_t Stable(TParticle*  particle);
    
    ClassDef(AliGenDPMjet,1) // AliGenerator interface to DPMJET
};
#endif