StarPrimaryMaker.h

#ifndef __StarPrimaryMaker_h__
#define __StarPrimaryMaker_h__

#include "StMaker.h"
#include "TLorentzVector.h"
#include "TVector3.h"
#include "TClonesArray.h"
#include "TDatabasePDG.h"
#include "StarParticleStack.h"
#include "StMessMgr.h"

#include "StarGenerator/UTIL/StarParticleData.h"

#include "TEventList.h"

#ifndef __CINT__
#include <functional>
#include <map>
#endif

class StarGenerator;
class StarGenEvent;
class StarFilterMaker;

class TTree;
class TFile;

class StarPrimaryMaker : public StMaker
{
 public:
  StarPrimaryMaker();
  virtual ~StarPrimaryMaker();
  
  Int_t InitRun( Int_t runnumber );
  Int_t Init();
  Int_t Make();
  void  Clear( const Option_t *opts="" );
  Int_t Finish();

  void  SetFileName( const Char_t *name ){ mFileName = name; }

  static TParticlePDG *pdg( Int_t pdgid );

  void AddGenerator( StarGenerator *gener );

  void AddFilter( StarFilterMaker *filt );

  void SetVertex( Double_t x, Double_t y, Double_t z ){ mVx=x; mVy=y; mVz=z; }

  void SetSigma( Double_t sx, Double_t sy, Double_t sz, Double_t rho=0 ){ mSx=sx; mSy=sy; mSz=sz; mRho=rho; }

  void SetSlope( Double_t dxdz, Double_t dydz ){ mVdxdz=dxdz; mVdydz=dydz; }

  void SetBeamline( Int_t beamline=1 ){ assert(beamline>=0); SetAttr("beamline",beamline); }

  void SetCuts( Double_t ptmin,    Double_t ptmax=-1, 
                Double_t ymin=0,   Double_t ymax=-1,
                Double_t phimin=0, Double_t phimax=-1,
                Double_t zmin=-999,   Double_t zmax=+999 );

  void SetPtRange( Double_t ptmin, Double_t ptmax=-1 ){ LOG_INFO << " -- StarPrimaryMaker -- ptmin=" << (mPtMin = ptmin) << " ptmax=" << (mPtMax = ptmax) << endm; }
  void SetEtaRange( Double_t etamin, Double_t etamax ){ LOG_INFO << " -- StarPrimaryMaker --  ymin=" << (mRapidityMin = etamin) << "  ymax= " << (mRapidityMax = etamax) << endm; }
  void SetPhiRange( Double_t phimin, Double_t phimax ){ LOG_INFO << " -- StarPrimaryMaker -- phimn=" << (mPhiMin = phimin) << " phimx=" <<  (mPhiMax = phimax) << endm; }

  void SetZvertexRange( Double_t zmin, Double_t zmax ){ LOG_INFO << " -- StarPrimaryMaker --  zmin=" << (mZMin = zmin) << "  zmax=" << (mZMax = zmax) << endm; }

  StarGenEvent *event() { return mPrimaryEvent; }

  virtual const char *GetCVS() const
  {static const char cvs[]="Tag $Name:  $ $Id: StarPrimaryMaker.h,v 1.8 2015/06/15 13:23:00 jwebb Exp $ built " __DATE__ " " __TIME__ ; return cvs;}

 private:
 protected:


  Int_t     PreGenerate();
  Int_t        Generate();
  Int_t    PostGenerate();
  Int_t        Finalize();
  void      BuildTables();
  
  Int_t mNumParticles;

  TTree   *mTree;
  TFile   *mFile;
  TString  mTreeName;
  TString  mFileName;
  StarParticleStack *mStack;
  StarGenEvent      *mPrimaryEvent;

  // Vertex position, width and x,y correlation
  Double_t mVx, mVy, mVz, mSx, mSy, mSz, mRho;

  // Vertex slope for beamline constraint
  Double_t mVdxdz, mVdydz;

  // Handle boost onto the beamline
  Bool_t mDoBeamline;

  void RotateBeamline( Double_t &px, Double_t &py, Double_t &pz, Double_t &E, Double_t &M, Double_t &vx, Double_t &vy, Double_t &vz, Double_t &vt );

  Double_t mPtMin, mPtMax, mRapidityMin, mRapidityMax, mPhiMin, mPhiMax, mZMin, mZMax;

  Bool_t Simulate( StarGenParticle *p );

  Int_t mRunNumber;

  TLorentzVector mPrimaryVertex;

  StarFilterMaker *mFilter;
  TEventList      *mAccepted; //*< event list containing accepted events

#ifndef __CINT__
  std::function< TLorentzVector() > mVertexFunction;
  std::map< std::string, std::function< TLorentzVector() > > mVertexFunctionMap;
  std::function< TLorentzVector() > GetVertexFunction(const char* name);
#endif

  TLorentzVector vertexGaussXYZ(); // gaussian XYZ
  TLorentzVector vertexFlatZ();    // uniform distribution along z (point in XY)
  TLorentzVector vertexFlatXYZ();  // uniform distribution in rectangle xyz
  TLorentzVector vertexFlatRZ();   // uniform distribution in cylinder rz
  TLorentzVector vertexFlatABZ();  // uniform distribution in ellipical cylinder xyz

  ClassDef(StarPrimaryMaker,1);

};

#endif