StRoot: St_geant_Maker/Embed/StPrepEmbedMaker.h Source File

00001 
00060 #ifndef StPrepEmbedMaker_hh     
00061 #define StPrepEmbedMaker_hh
00062 
00063 #include "StMaker.h"
00064 #include "TGiant3.h"
00065 #include "TString.h"
00066 
00067 class StEvent;
00068 class StEvtHddr ;
00069 class StTrack;
00070 class TFile;
00071 class TGiant3;
00072 class TTree;
00073 
00074 class StPrepEmbedMaker : public StMaker {
00075  public:
00076   
00077   StPrepEmbedMaker(const Char_t *name="PrepEmbed");     // constructor
00078   ~StPrepEmbedMaker();                                 // destructor
00079   
00080   Int_t  Init();                      // called once at the beginning of your job
00081   Int_t  Make();                      // invoked for every event
00082   Int_t  Finish();
00083   Int_t  InitRun(const int runnum);
00084   virtual void   Do(const Char_t *option = "dcut cave x 0.1 10 10 0.03 0.03"); // *MENU 
00085   virtual const char *GetCVS() const {
00086     static const char cvs[]="Tag $Name:  $ $Id: StPrepEmbedMaker.h,v 1.7 2012/05/13 06:36:59 zhux Exp $ built "__DATE__" "__TIME__ ; 
00087     return cvs;
00088   }
00089   
00090   void SetPartOpt(const Int_t pid, const Double_t mult); 
00091 
00099   void SetOpt(const Double_t ptlow, const Double_t pthigh,
00100               const Double_t etalow, const Double_t etahigh, const Double_t philow,
00101               const Double_t phihigh, const TString type="FlatPt");
00102   void SetTemp(const double t);
00103   void SetTagFile(const Char_t *file) ;
00104   void SetSkipMode(const Bool_t flag=kTRUE) ;
00105   void SetSpreadMode(const Bool_t flag=kFALSE) ;
00106   void SetTrgOpt(const Int_t TrgId); // Set trigger id cut
00107   void SetZVertexCut(const Double_t vzlow, const Double_t vzhigh); // Set z-vertex cut
00108   void SetVrCut(const Double_t vr) ; // Set vr = sqrt{vx^2 + vy^2} cut
00109   void SetVpdVzCut(const Double_t vpdvz) ; // Set |vpdvz-vz| cut
00110   void OpenFzFile() ; 
00111   void SetPrimeMode(const Bool_t flag=kFALSE) ; //Switch to prime mode for nucleus (with geantID > 10000) embedding
00112   void SetVpdVzCutMode(const Bool_t flag=kFALSE) ; //Switch to turn on cut for |VpdVz-Vz|
00113 
00114   void SetRapidityMode(const Bool_t flag=kTRUE) ; //Switch to assigin input kinematic range to rapidity (true) or pseudorapidity(false)
00115 
00118   void phasespace(const Int_t mult) ;
00119 
00122   void gkine(const Int_t mult, const Double_t vzmin, const Double_t vzmax) ;
00123 
00124  private:
00125   static const Double_t mRapidityMaximumCut ; 
00126 
00130   Int_t getMultiplicity(const StEvtHddr& EvtHddr, const Int_t nprimarytracks) const ;
00131 
00132   TGiant3 *mGeant3;
00133   TString mTagFile; 
00134   TString mMoreTagsFile; 
00135   TString mFzFile ; 
00136   Int_t mEventCounter; 
00137   TFile *mFile; 
00138   TFile *mMoreFile;
00139   TTree *mTree;
00140   TTree *mMoreTree;
00141   Bool_t mSkipMode;
00142   Bool_t mSpreadMode;
00143   Bool_t mOpenFzFile; 
00144 
00145   Bool_t mRapidityMode; 
00146 
00147   Bool_t mPrimeMode;  
00148   Int_t  mSavePid;
00149   Bool_t mPrimed;  // 
00150 
00151   Bool_t mVpdVzCutMode;  
00152 
00153   ClassDef(StPrepEmbedMaker,0)    
00154 };
00155 #endif