StRoot: StEmbeddingUtilities/StEmbeddingQA.h Source File

00001 //****************************************************************************************************
00002 //  Class StEmbeddingQA
00003 //
00004 //  See http://drupal.star.bnl.gov/STAR/comp/embedding/embedding-procedures/qa-documentation
00005 //  for instructions
00006 //****************************************************************************************************
00007 /****************************************************************************************************
00008  * $Id: StEmbeddingQA.h,v 1.11 2012/03/05 10:32:29 cpowell Exp $
00009  * $Log: StEmbeddingQA.h,v $
00010  * Revision 1.11  2012/03/05 10:32:29  cpowell
00011  * Functions added to cut on refMult
00012  *
00013  * Revision 1.10  2011/04/01 05:05:47  hmasui
00014  * Track selections by StEmbeddingQAUtilities. Added 1/pt(RC)-1/pt(MC) vs pt, and pt dependent Ncommon vs NhitFit histograms
00015  *
00016  * Revision 1.9  2011/02/11 03:55:44  hmasui
00017  * Change geantid type to integer
00018  *
00019  * Revision 1.8  2011/01/31 21:32:10  hmasui
00020  * Modify histogram keys to TString to take into account parent geantid
00021  *
00022  * Revision 1.7  2010/07/12 21:29:40  hmasui
00023  * Move isGeantIdOk() function into StEmbeddingQAUtilities
00024  *
00025  * Revision 1.6  2010/05/14 19:50:11  hmasui
00026  * Add rapidity and trigger cuts.
00027  *
00028  * Revision 1.5  2010/04/24 20:21:18  hmasui
00029  * Add geant process check for contaminated pairs
00030  *
00031  * Revision 1.4  2010/02/16 02:13:34  hmasui
00032  * Add parent-parent geant id in the histogram name
00033  *
00034  * Revision 1.3  2010/01/28 21:50:37  hmasui
00035  * Add Vx vs Vz and Vy vs Vz histograms.
00036  *
00037  * Revision 1.2  2010/01/26 17:46:31  hmasui
00038  * Add histograms for eventid, runnumber, and number of particles
00039  *
00040  * Revision 1.1  2009/12/22 21:41:17  hmasui
00041  * Change class name from StEmbeddingQAMaker to StEmbeddingQA
00042  *
00043  ****************************************************************************************************/
00044 
00045 #ifndef __StEmbeddingQA_h__
00046 #define __StEmbeddingQA_h__
00047 
00048 #include <map>
00049 #include <vector>
00050 
00051 #include "TMath.h"
00052 #include "TString.h"
00053 
00054 #include "StEmbeddingQAUtilities.h"
00055 
00056 class TFile ;
00057 class TH1 ;
00058 class TH2 ;
00059 class TH3 ;
00060 class TObject ;
00061 //class TTree ;
00062 
00063 class StContamPair ;
00064 class StEmbeddingQAPair ;
00065 class StEmbeddingQATrack ;
00066 class StMiniMcEvent ;
00067 class StMuDstMaker ;
00068 class StMuEvent ;
00069 class StMuTrack ;
00070 class StTinyMcTrack ;
00071 
00072 //____________________________________________________________________________________________________
00073 //  Analyze either minimc trees for embedding QA or microDST for real data QA
00074 class StEmbeddingQA {
00075   public:
00077     StEmbeddingQA() ;
00078 
00080     StEmbeddingQA(const Int_t year, const TString production, const Bool_t isSimulation = kTRUE);
00081 
00083     virtual ~StEmbeddingQA();
00084 
00085     void init() ;  
00086 
00091     Bool_t book(const TString outputFileName = "");
00092 
00094     Bool_t make(const TString inputFileName, const Bool_t isSimulation = kTRUE);
00095 
00097     Bool_t run(const TString inputFileList) ;
00098 
00100     Bool_t runRealData(const TString inputFileList) ;
00101 
00103     Bool_t runEmbedding(const TString inputFileList) ;
00104 
00106     Bool_t end() const;
00107 
00110     void setZVertexCut(const Float_t vz) ;
00111     void setRefMultMinCut(const Int_t refMultMin) ;
00112     void setRefMultMaxCut(const Int_t refMultMax) ;
00113 
00116     void addTriggerIdCut(const UInt_t id) ;
00117 
00120     void setRapidityCut(const Float_t ycut) ;
00121 
00122   private:
00123     const Int_t mYear ;               
00124     const TString mProduction ;       
00125     const Bool_t mIsSimulation ;      
00126 
00127     void clear() ; 
00128 
00129     // Fill histograms
00130     Bool_t fillEmbedding(const TString inputFileName) ; // Fill embedding histograms
00131     Bool_t fillRealData(const TString inputFileName) ;  // Fill real data histograms
00132 
00134     void fillEmbeddingTracks(const StMiniMcEvent& mcevent, const Int_t categoryid, const Int_t itrk) ;
00135 
00137     StEmbeddingQATrack* getEmbeddingQATrack(const StMiniMcEvent& mcevent, const Int_t categoryid, const Int_t itrk) ;
00138 
00140     void fillRealTracks(const StMuTrack& track, const Int_t categoryid, const Int_t itrk);
00141 
00143     void fillHistograms(const StEmbeddingQATrack& track, const Int_t categoryid);
00144 
00146     void expandHistograms(const Int_t categoryid, const Int_t geantid, const Int_t parentid,
00147         const Int_t parentparentid, const Int_t geantprocess);
00148 
00150     Bool_t pushBackGeantId(const Int_t categoryid, const Int_t geantid, const Int_t parentid,
00151         const Int_t parentparentid, const Int_t geantprocess) ;
00152 
00154     Bool_t isZVertexOk(const StMiniMcEvent& mcevent) const ;
00155 
00157     Bool_t isRefMultOk(const StMiniMcEvent& mcevent) const ;
00158 
00160     Bool_t isTriggerOk(StMuEvent* event) const ;
00161 
00163     Int_t getNtrack(const Int_t categoryid, const StMiniMcEvent& mcevent) const ;
00164 
00167     TString getIdCollection(const Int_t geantid, const Int_t parentid, const Int_t parentparentid) const ;
00168 
00169     StMuDstMaker* mMuDstMaker ; 
00170 
00171     TFile* mOutput ; 
00172 
00173     // Event-wise histograms
00174     Double_t mVz ;      
00175     TH1* mhVz ;         
00176     TH1* mhVzAccepted ; 
00177     TH1* mhRef ;         
00178     TH1* mhRefAccepted ; 
00179 
00180     TH2* mhVyVx ; 
00181     TH2* mhVxVz ; 
00182     TH2* mhVyVz ; 
00183     TH1* mhdVx ; 
00184     TH1* mhdVy ; 
00185     TH1* mhdVz ; 
00186 
00187     TH1* mhEventId ; 
00188     TH1* mhRunNumber ; 
00189     TH1* mhNParticles[StEmbeddingQAConst::mNCategory] ; 
00190 
00193     //
00194     //  - Use MC momentum instead of reconstructed momentum (Update on Nov/13/2009)
00195     //  - Add p (reco) vs p (MC) (Update on Nov/13/2009)
00196     std::vector<Int_t> mGeantId[StEmbeddingQAConst::mNCategory] ; 
00197     std::vector<TString> mGeantIdCollection ;                       
00198 
00199 
00200     TH1* mhGeantId[StEmbeddingQAConst::mNCategory];                               
00201     std::map<TString, TH3*> mhNHit[StEmbeddingQAConst::mNCategory] ;                
00202     std::map<TString, TH3*> mhDca[StEmbeddingQAConst::mNCategory] ;                 
00203     std::map<TString, TH2*> mhPtVsEta[StEmbeddingQAConst::mNCategory] ;             
00204     std::map<TString, TH2*> mhPtVsY[StEmbeddingQAConst::mNCategory] ;               
00205     std::map<TString, TH2*> mhPtVsPhi[StEmbeddingQAConst::mNCategory] ;             
00206     std::map<TString, TH2*> mhPtVsMom[StEmbeddingQAConst::mNCategory] ;             
00207     std::map<TString, TH2*> mhdPtVsPt[StEmbeddingQAConst::mNCategory] ;             
00208     std::map<TString, TH2*> mhdInvPtVsPt[StEmbeddingQAConst::mNCategory] ;          
00209     std::map<TString, TH2*> mhMomVsEta[StEmbeddingQAConst::mNCategory] ;            
00210     std::map<TString, TH2*> mhdEdxVsMomMc[StEmbeddingQAConst::mNCategory] ;         
00211     std::map<TString, TH2*> mhdEdxVsMomMcPidCut[StEmbeddingQAConst::mNCategory] ;   
00212     std::map<TString, TH2*> mhdEdxVsMomReco[StEmbeddingQAConst::mNCategory] ;       
00213     std::map<TString, TH2*> mhdEdxVsMomRecoPidCut[StEmbeddingQAConst::mNCategory] ; 
00214     std::map<TString, TH2*> mhRecoPVsMcP[StEmbeddingQAConst::mNCategory] ;          
00215     std::map<TString, TH3*> mhNCommonHitVsNHit[StEmbeddingQAConst::mNCategory] ;      
00216 
00217     std::map<TString, TH2*> mhEtaVsPhi[StEmbeddingQAConst::mNCategory] ;  
00218     std::map<TString, TH2*> mhEtaVsVz[StEmbeddingQAConst::mNCategory] ;   
00219     std::map<TString, TH2*> mhYVsVz[StEmbeddingQAConst::mNCategory] ;     
00220 
00221     ClassDef(StEmbeddingQA, 1);
00222 };
00223 
00224 #endif
00225 
00226 
00227