StPrepEmbedMaker.h

#ifndef StPrepEmbedMaker_hh     
#define StPrepEmbedMaker_hh

#include "StMaker.h"
#include "TGiant3.h"
#include "TString.h"

class StEvent;
class StEvtHddr ;
class StTrack;
class TFile;
class TGiant3;
class TTree;

class StPrepEmbedMaker : public StMaker {
 public:
  
  StPrepEmbedMaker(const Char_t *name="PrepEmbed");     // constructor
  ~StPrepEmbedMaker();                                 // destructor
  
  Int_t  Init();                      // called once at the beginning of your job
  Int_t  Make();                      // invoked for every event
  Int_t  Finish();
  Int_t  InitRun(const int runnum);
  virtual void   Do(const Char_t *option = "dcut cave x 0.1 10 10 0.03 0.03"); // *MENU 
  virtual const char *GetCVS() const {
    static const char cvs[]="Tag $Name:  $ $Id: StPrepEmbedMaker.h,v 1.9 2014/08/06 11:43:55 jeromel Exp $ built " __DATE__ " " __TIME__ ; 
    return cvs;
  }
  
  void SetPartOpt(const Int_t pid, const Double_t mult); 

  void SetOpt(const Double_t ptlow, const Double_t pthigh,
              const Double_t etalow, const Double_t etahigh, const Double_t philow,
              const Double_t phihigh, const TString type="FlatPt");
  void SetTemp(const double t);
  void SetTagFile(const Char_t *file) ;
  void SetSkipMode(const Bool_t flag=kTRUE) ;
  void SetSpreadMode(const Bool_t flag=kFALSE) ;
  void SetTrgOpt(const Int_t TrgId); // Set trigger id cut
  void SetZVertexCut(const Double_t vzlow, const Double_t vzhigh); // Set z-vertex cut
  void SetVrCut(const Double_t vr) ; // Set vr = sqrt{vx^2 + vy^2} cut
  void SetVpdVzCut(const Double_t vpdvz) ; // Set |vpdvz-vz| cut
  void SetPVRankCut(const Double_t pvrank) ; // Set cut on P.V. rank > pvrank
  void OpenFzFile() ; 
  void SetPrimeMode(const Bool_t flag=kFALSE) ; //Switch to prime mode for nucleus (with geantID > 10000) embedding
  void SetVpdVzCutMode(const Bool_t flag=kFALSE) ; //Switch to turn on cut for |VpdVz-Vz|
  void SetPVRankCutMode(const Bool_t flag=kFALSE) ; //Switch to turn on cut for P.V. rank

  void SetRapidityMode(const Bool_t flag=kTRUE) ; //Switch to assigin input kinematic range to rapidity (true) or pseudorapidity(false)

  void phasespace(const Int_t mult) ;

  void gkine(const Int_t mult, const Double_t vzmin, const Double_t vzmax) ;

 private:
  static const Double_t mRapidityMaximumCut ; 

  Int_t getMultiplicity(const StEvtHddr& EvtHddr, const Int_t nprimarytracks) const ;

  TGiant3 *mGeant3;
  TString mTagFile; 
  TString mMoreTagsFile; 
  TString mFzFile ; 
  Int_t mEventCounter; 
  TFile *mFile; 
  TFile *mMoreFile;
  TTree *mTree;
  TTree *mMoreTree;
  Bool_t mSkipMode;
  Bool_t mSpreadMode;
  Bool_t mOpenFzFile; 

  Bool_t mRapidityMode; 

  Bool_t mPrimeMode;  
  Int_t  mSavePid;
  Bool_t mPrimed;  // 

  Bool_t mVpdVzCutMode;  
  Bool_t mPVRankCutMode;  

  ClassDef(StPrepEmbedMaker,0)    
};
#endif