StMiniMcMaker.h

Go to the documentation of this file.

#ifndef StMiniMcMaker_H
#define StMiniMcMaker_H

#include "StMaker.h"

#include <vector>
#include <utility>
#include <algorithm>
#include <map>

#include "TString.h"

#include "StAssociationMaker/StAssociationMaker.h"
#include "StAssociationMaker/StTrackPairInfo.hh"

#include "StMiniMcEvent/StMiniMcEvent.h"
#include "StMiniMcEvent/StMiniMcPair.h"
#include "StMiniMcEvent/StContamPair.h"

class TFile;
class TTree;
class StEvent;
class StMcEvent;
class StMcTrack;
class StTrack;
class StPrimaryTrack;
class StIOInterFace;
#include "StThreeVectorF.hh"
class StTpcDedxPidAlgorithm;
class StuProbabilityPidAlgorithm;
class StTpcHit;
class StDedxPidTraits;
class StEmcRawHit;
class StEmcPoint;

// typdef's
#ifndef __CINT__

typedef map<UInt_t,Int_t> RCFOUNDMAP;
typedef map<long,Int_t> MCFOUNDMAP;
typedef map<Long_t,const StMcTrack*> MCMAP;
typedef vector<StTrackPairInfo*> PAIRVEC;
typedef pair<const StTpcHit*,const StTpcHit*> PAIRHIT;
inline bool pairCmp(StTrackPairInfo* p1, StTrackPairInfo* p2){
  return p1->commonTpcHits() < p2->commonTpcHits();
}
inline bool sortCmp(StTrackPairInfo* p1, StTrackPairInfo* p2){
  return p1->commonTpcHits() > p2->commonTpcHits();
}

#endif
//bool hitCmp(StTpcHit* p1, StTpcHit* p2);


// check StMiniMcEvent for the track categories...

//____________________________________________

class StMiniMcMaker : public StMaker{
 public:
  StMiniMcMaker(const Char_t* name="StMiniMcMaker",
                const Char_t* title="event/StMiniMcMaker");
  virtual ~StMiniMcMaker();

  void  Clear(Option_t *option="");
  Int_t Init();
  Int_t InitRun(int runnumber);
  Int_t Make();
  Int_t Finish();
  virtual const char *GetCVS() const
  {static const char cvs[]="Tag $Name:  $ $Id: StMiniMcMaker.h,v 1.20 2014/08/06 11:43:27 jeromel Exp $ built " __DATE__ " " __TIME__ ; return cvs;}

  //---- SETS -------

  void  setGhost(Bool_t doit=kTRUE)             { mGhost = doit; }
  void  setOutDir(const char* dir= "./")        { mOutDir = dir; }  
  void  setPtCut(Float_t minPt=0, Float_t maxPt=9999) 
                                                { mMinPt=minPt; mMaxPt=maxPt; }
  void  setFileName(const TString& val)                 { mInFileName = val; }
  void  setFilePrefix(const TString& val)       { mInFilePrefix = val; }

 private:
  //static const Float_t mSharedHitsCut = .5;

  // methods
  
  Bool_t           initAssociation();  // sets all the assoc maps
  Bool_t           initVertex(); // finds the primary vertex if it exists
  Bool_t           acceptRaw(const StMcTrack*);
  Bool_t           accept(const StMcTrack*);
  Bool_t           accept(const StTrack*);
  Bool_t           acceptGood20(const StTrack*);
  Bool_t           acceptGood20(const StMcTrack*);
  Bool_t           acceptCentrality(const StTrack*);
  Bool_t           acceptUncorrected(const StTrack*);
  Bool_t           acceptGlobals(const StTrack* track);
  Bool_t           acceptFTPC(const StTrack*);
  Bool_t           ok(const StTrack*);
  Bool_t           isSameSign(const StTrack*,const StMcTrack*);
  Bool_t           acceptPt(const StTrack*);
  Bool_t           acceptPt(const StMcTrack*);
  Bool_t           acceptDebug(const StMcTrack*);
  
  StTrackPairInfo* findBestMatchedGlobal(const StMcTrack*);
  PAIRVEC          findMatchedRc(const StMcTrack*);
  PAIRHIT          findFirstLastHit(const StTrack*);
  PAIRHIT          findFirstLastFitHit(const StTrack*);

  Float_t          computeXY(const StThreeVectorF*, const StTrack*);
  StDedxPidTraits* findDedxPidTraits(const StTrack*);
  //pair<Float_t,Float_t>  computeProj(const StThreeVectorF*,const StTrack*);
  Float_t          computeZDca(const StThreeVectorF*,const StTrack*);

  const StPrimaryTrack*  isPrimaryTrack(const StTrack*);
  Bool_t           isPrimaryTrack(const StMcTrack*);
  Int_t            openFile();
  Int_t            closeFile();
  void             trackLoop();
  void             trackLoopIdT();
  void             buildEmcIndexArray();
  void             fillEventInfo(Int_t nGoodTrack, Int_t nRcGlobal, Int_t nRcGoodGlobal20,
                                 //Int_t nAcceptedRaw,
                                 Int_t nMcGlobal, Int_t nMcGoodGlobal20,
                                 Int_t nMcNch, Int_t nMcHminus,
                                 Int_t nMcFtpcENch, Int_t nMcFtpcWNch,
                                 Int_t nFtpcEUncorrected, Int_t nFtpcWUncorrected, 
                                 Int_t nUncorrectedGlobals);
  void             fillTrackPairInfo(StMiniMcPair*,
                                     const StMcTrack*,
                                     const StTrack* prTrack, 
                                     const StTrack* glTrack,
                                     Int_t commonHits, Int_t nAssMc,
                                     Int_t nAssGl, Int_t nAssPr,
                                     Bool_t isBestContam=kFALSE);
  
  void             fillRcTrackInfo(StTinyRcTrack*,
                                   const StTrack* prTrack,
                                   const StTrack* glTrack,
                                   Int_t nAssMc);
  
  void             fillMcTrackInfo(StTinyMcTrack*,
                                   const StMcTrack*,
                                   Int_t nAssGl, Int_t nAssPr);

  void             checkMerged(const StMcTrack* merged, Int_t mergedCommonHits,
                               const StTrack* prTrack);
  void             checkSplit (const StMcTrack*,const StTrack*,Int_t);
  void             checkContam(const StMcTrack*,const StGlobalTrack*,Int_t);

  size_t           getIndex(size_t mult);

 void             AppendMCDaughterTrack();
 static void      dominatTkInfo(const StTrack* recTrack,int &dominatrackKey ,int& dominatrackHits,float& avgQuality);
  // members
  StMiniMcEvent*   mMiniMcEvent; 
  StIOInterFace*   mIOMaker;      
  TTree*           mMiniMcTree;   
  TFile*           mMiniMcDST;    
  TString          mInFileName;   
  TString          mInFilePrefix;   
  TString          mOutFileName;  
  TString          mOutDir;       
  TString          mParameterFileName; 

  StEvent*         mRcEvent;      
  StMcEvent*       mMcEvent;      
  rcTpcHitMapType* mRcHitMap;     
  rcTrackMapType*  mRcTrackMap;   
  mcTrackMapType*  mMcTrackMap;   
  const StThreeVectorF*  mRcVertexPos;  
  const StThreeVectorF*  mMcVertexPos;  
  StTpcDedxPidAlgorithm* mTpcDedxAlgo; 
  StuProbabilityPidAlgorithm* mPidAlgo; 
  vector<StEmcRawHit*> mEmcIndex; 
  Bool_t           mGhost;        
  
  Float_t          mMinPt;        
  Float_t          mMaxPt;        

  Int_t            mNSplit;     
  Int_t            mNRc;        
  Int_t            mNGhost;     
  Int_t            mNContam;    
  Int_t            mNMatched;   
  Int_t            mNMatGlob;   
  int              mMainVtx;
  ClassDef(StMiniMcMaker,0)
};
  
  

  
#endif  
//
// $Log $
//