StarGenerator.h

#ifndef __StarGenerator_h__
#define __StarGenerator_h__

#include "StMaker.h"
#include "TString.h"
#include "TDatabasePDG.h"
#include "StarCallf77.h"
#include "TLorentzVector.h"

#include "StarGenerator/EVENT/StarGenEvent.h"
#include "StarGenerator/EVENT/StarGenStats.h"
#include "StarGenerator/UTIL/StarParticleData.h"


class StarGenerator : public StMaker
{
 public:

  StarGenerator( const Char_t *name="" );   
  ~StarGenerator(){ /* nada */ };

 protected:
  const Double_t mm;
  const Double_t cm;

 public:
  virtual Int_t Init() = 0;

  Int_t InitRun( Int_t runnumber ) { return kStOK; };

  virtual Int_t PreGenerate(){ return kStOK; }
  virtual Int_t PostGenerate(){ return kStOK;  };

  virtual Int_t Generate() = 0;

  virtual void FillPP( StarGenEvent *event ){ };
  virtual void FillAA( StarGenEvent *event ){ };
  virtual void FillEP( StarGenEvent *event ){ };
  virtual void FillEA( StarGenEvent *event ){ };
  virtual void FillUser( StarGenEvent *event ){ };

  Int_t PreGenerateHook();
  Int_t PostGenerateHook();

  Int_t Make() { return Generate(); }

  void Clear( const Option_t *opts="" );
 
  Int_t Finalize();


  /********************************************************************************
   *
   * Configuration Options
   *
   *******************************************************************************/


  void SetBlue( const Char_t *b );// { mBlue = b; }
  void SetYell( const Char_t *y ); //{ mYell = y; }
  void SetFrame( const Char_t *frame, const Double_t val );
  void SetFrame( const Char_t *frame, const Double_t *pblue, const Double_t *pyell );

  void SetImpact( Double_t bmin, Double_t bmax )
  {
    mImpactMin = bmin;
    mImpactMax = bmax;
  }

  Int_t GetNumberOfParticles(){ return mNumberOfParticles; }

  void   SetPileup( Bool_t p, Double_t prob=0.10 ){ mIsPileup = p; if ( p ) mPileup=prob; else mPileup=1.0; }


  Bool_t IsPileup(){ return mIsPileup; }

  Double_t GetPileup(){ return mPileup; }


  void SetOutputTree ( TTree *tree );

  void SetInputTree ( TTree *tree, const Char_t *bname=0 );

  void SetInputFile ( const Char_t *name, const Char_t *treename="genevents", const Char_t *bname=0 );



  Int_t IOmode(){ return mIOMode; }

  StarGenEvent *Event(){ return mEvent; }

  virtual StarGenStats Stats() { return StarGenStats(); }


  virtual const char *GetCVS() const
  {static const char cvs[]="Tag $Name:  $ $Id: StarGenerator.h,v 1.6 2015/04/24 18:21:11 jwebb Exp $ built " __DATE__ " " __TIME__ ; return cvs;}


 private:

  // Generator ID
  Int_t mId;
  // Set generator ID
  void SetId( Int_t id ) { mId = id; }

 protected:
  // Number of particles in this event
  Int_t mNumberOfParticles;

  StarGenEvent *mEvent;

  TString  mBlue;
  TString  mYell;
  TString  mFrame;
  Double_t mRootS;      
  Double_t mDirect;

  Double_t mImpactMin;
  Double_t mImpactMax;

  TLorentzVector mBlueMomentum;
  TLorentzVector mYellMomentum;
  Double_t mBlueMass;
  Double_t mYellMass;

  Double_t mVertex[3]; 
  Double_t mSpread[3]; 

  Int_t  mIOMode;         
  TTree *mInputTree;      
  TTree *mOutputTree;     

  Bool_t   mIsPileup;
  Double_t mPileup;

  StarParticleData *mParticleDb;

  TString mLibrary;

  friend class StarPrimaryMaker;

  ClassDef(StarGenerator,1);

};
#endif