StRoot: StEmbeddingUtilities/StEmbeddingQADraw.h Source File

00001 //----------------------------------------------------------------------------------------------------
00002 //   Draw QA histograms
00003 //     - Comparison of the results between embedding and real data
00004 //     - Print only one PDF file by default. 
00005 //
00006 //----------------------------------------------------------------------------------------------------
00007 //  NOTE: These functions are not used for the latest QA codes.
00008 //       file           function
00009 //       png            setPNGOn()
00010 //       gif            setGIFOn()
00011 //       jpg            setJPGOn()
00012 //       eps            setEPSOn()
00013 //       ps             setPSOn()
00014 //----------------------------------------------------------------------------------------------------
00015 //
00016 //  You can make the QA plots by drawEmbeddingQA.C under StRoot/macros/embedding
00017 //  Suppose you have output ROOT files: qa_embedding_2007_P08ic.root,
00018 //  and qa_real_2007_P08ic.root (the output file format is automatically 
00019 //  determined if you leave the output filename blank in StEmbeddingQAMaker)
00020 //  Suppose you have embedded single pion+ (geantid=8) in the output files.
00021 //
00022 //    > root4star -b -q drawEmbeddingQA.C'("./", "qa_embedding_2007_P08ic.root", "qa_real_2007_P08ic.root", 8)'
00023 //
00024 //  First argument is the directory where the output figures are printed.
00025 //  The default output directory is the current directory.
00026 //  The 6th argument is the input geantid
00027 //
00028 //  If you name the output ROOT files by hand, you need to put the year,
00029 //  production and particle name by yourself since those are used 
00030 //  for the output figure name and to print those informations in 
00031 //  the legend for each QA plot.
00032 //  Suppose you have output ROOT files: qa_embedding.root and qa_real.root
00033 //
00034 //    > root4star -b -q drawEmbeddingQA.C'("./", "qa_embedding.root", "qa_real.root", 2005, "P08ic", 8)'
00035 //
00036 //
00037 //  If you just want ot check the embedding output only, you can do 
00038 //
00039 //    > root4star -b -q drawEmbeddingQA.C'("./", "qa_embedding_2007_P08ic.root", "qa_real_2007_P08ic.root", 8, kTRUE)'
00040 //
00041 //  where the last argument is the switch to skip drawing the real data if it's true (default is false)
00042 //
00043 //----------------------------------------------------------------------------------------------------
00044 /****************************************************************************************************
00045  * $Id: StEmbeddingQADraw.h,v 1.16 2011/04/12 03:01:07 hmasui Exp $
00046  * $Log: StEmbeddingQADraw.h,v $
00047  * Revision 1.16  2011/04/12 03:01:07  hmasui
00048  * Fix isMatchedPairOk() to properly process particles with decay daughters
00049  *
00050  * Revision 1.15  2011/02/11 03:44:57  hmasui
00051  * Draw error messages in pdf if histogram is missing. Add error check for Ncommon histogram
00052  *
00053  * Revision 1.14  2011/01/31 21:33:51  hmasui
00054  * Add setParentGeantId() function to allow the multiple decays
00055  *
00056  * Revision 1.13  2010/06/22 16:31:17  hmasui
00057  * Separate 2D and 1D QA for MC tracks. Add pol0 fit for MC eta, y and phi distributions.
00058  *
00059  * Revision 1.12  2010/06/10 14:51:01  hmasui
00060  * Added particle name functions
00061  *
00062  * Revision 1.11  2010/04/07 19:45:11  hmasui
00063  * Use box option for dE/dx vs p to reduce the pdf file size
00064  *
00065  * Revision 1.10  2010/03/15 21:05:23  hmasui
00066  * Separate MC vertices QA into 2 pages. Added constraint on z-vertex cut for vx(vy) vs vz histograms.
00067  *
00068  * Revision 1.9  2010/02/24 18:11:47  hmasui
00069  * Added isMC flag in drawProjection2D() to switch real or MC tracks
00070  *
00071  * Revision 1.8  2010/02/23 16:56:37  hmasui
00072  * Add phi distributions QA (MC vs reconstructed)
00073  *
00074  * Revision 1.7  2010/02/16 02:14:03  hmasui
00075  * Print PDF file only for all QA plots
00076  *
00077  * Revision 1.6  2009/12/22 21:40:11  hmasui
00078  * Add comments for functions and members
00079  *
00080  ****************************************************************************************************/
00081 
00082 
00083 #ifndef __StEmbeddingQADraw_h__
00084 #define __StEmbeddingQADraw_h__
00085 
00086 #include <vector>
00087 
00088 class TCanvas ;
00089 class TH1 ;
00090 class TFile ;
00091 class TObject ;
00092 class TPad ;
00093 class TPaveText ;
00094 class TPDF ;
00095 #include "TString.h"
00096 
00097 class StEmbeddingQADraw {
00098   public:
00103     StEmbeddingQADraw(const TString embeddingFile, const TString realDataFile, const Int_t geantid,
00104         const Bool_t isEmbeddingOnly = kFALSE);
00105 
00107     StEmbeddingQADraw(const TString embeddingFile, const TString realDataFile,
00108         const Int_t year, const TString production, const Int_t geantid, const Bool_t isEmbeddingOnly = kFALSE);
00109     virtual ~StEmbeddingQADraw();
00110 
00112     void init() ;
00113 
00115     void setParentGeantId(const Int_t parentgeantid) ;
00116 
00118     void setOutputDirectory(const TString name = "./") ;
00119 
00120     Bool_t draw() const ;       
00121 
00122     // Event-wise
00123     Bool_t drawEvent() const;     
00124 
00125     // Track-wise
00126     Bool_t drawMcTrack() const;    
00127     Bool_t drawTrack() const ;     
00128 
00129     Bool_t drawGeantId() const;    
00130     Bool_t drawPhi() const ;       
00131     Bool_t drawRapidity() const;   
00132     Bool_t drawPt() const;         
00133     Bool_t drawMomentum() const;   
00134     Bool_t drawdEdx() const;       
00135     Bool_t drawDca() const;        
00136     Bool_t drawNHit() const;       
00137 
00138     Bool_t finish() ;               
00139 
00140     void setPNGOn() ;        
00141     void setGIFOn() ;        
00142     void setJPGOn() ;        
00143     void setEPSOn() ;        
00144     void setPSOn() ;         
00145 
00146     Int_t getYear() const ;               
00147     const Char_t* getProduction() const ; 
00148     Int_t getGeantId() const ;            
00149 
00151     //   Default id = -1, returns particle name from mGeantId
00152     const Char_t* getParticleName(const Int_t geantid = -1) const ;
00153 
00154     void setPtMax(const Double_t ptmax) ; 
00155 
00156   private:
00158     //  return true only if both embedding and real data files are opened properly
00159     Bool_t open(const TString embeddingFile, const TString realDataFile) ;
00160 
00162     void print(const TCanvas& canvas, const TString name) const;
00163 
00165     //   if we don't find geantid in the MC histogram, replace mGeantId to that found in the MC histogram
00166     void checkInputGeantId() ;
00167 
00169     Bool_t isOpen() const ;
00170 
00172     Bool_t isMatchedPairOk() const ;
00173 
00175     void setDaughterGeantId() ;
00176 
00178     Int_t getEntries() const ;
00179 
00181     Bool_t isDecay() const ;
00182 
00184     // if input geantid (or daughter geantid) is not e/pi/K/p, use pi+(=8)
00185     Int_t getGeantIdReal(const Int_t daughter) const ;
00186 
00188     //    For embedding
00189     //      stable particles (# of daughters = 0) --> matched pairs
00190     //      unstable particles (# of daughters > 0) --> contaminated pairs
00191     //    For real data  --> primary tracks
00192     Int_t  getCategoryId(const Bool_t isEmbedding = kTRUE) const ;
00193 
00195     TObject* getHistogram(const TString name, const Bool_t isEmbedding = kTRUE) const ;
00196 
00199     TObject* getHistogram(const TString name, const UInt_t daughter, const Bool_t isEmbedding,
00200         const UInt_t parentparentid = 0) const ;
00201 
00203     const Char_t* getHistogramName(const TString name, const UInt_t daughter, const Bool_t isEmbedding,
00204         const UInt_t parentparentid = 0) const ;
00205 
00206 
00208     Double_t getNormalization(const TH1& h) const ;
00209 
00211     const Char_t* getBaseName() const ;
00212 
00214     Bool_t drawStatistics(const Double_t x1=0.1, const Double_t y1=0.2, const Double_t x2=0.9, const Double_t y2=0.8,
00215         const Double_t textSize = 0.05) const;
00216 
00218     TPaveText* drawHeader(const TString description,
00219         const Double_t x1=0.0, const Double_t y1=0.9, const Double_t x2=1.0, const Double_t y2=0.95, 
00220         const Double_t textSize = 0.032) const;
00221 
00223     void drawLegend(const UInt_t id, TH1* hembed, TH1* hreal, const Option_t* option="L",
00224         const Bool_t doSplit=kFALSE) const ;
00225 
00227     void drawErrorMessages(const TString histogramName) const ;
00228 
00230     Bool_t drawVertices() const;          
00231     Bool_t drawRunEventId() const;        
00232     Bool_t drawNumberOfParticles() const; 
00233 
00235     Bool_t drawdEdxVsMomentum(const Bool_t isMcMomentum=kTRUE) const ;
00236 
00237     // Added isMC flag
00238     //  isMC = kTRUE            --> Compare reconstructed embedding tracks with MC tracks
00239     //  isMC = kFALSE (default) --> Compare reconstructed embedding tracks with real data
00240     Bool_t drawProjection2D(const TString name, const Bool_t isMC = kFALSE) const; 
00241     Bool_t drawProjection3D(const TString name) const; 
00242 
00244     // Header for current QA, division of (x,y), default is (0,0), no division
00245     TPaveText* initCanvas(const TString header, const Int_t nx=0, const Int_t ny=0) const;
00246 
00248     Double_t getVzAcceptedMinimum() const ; 
00249     Double_t getVzAcceptedMaximum() const ; 
00250 
00252     const Char_t* getMcParticleName() const ; 
00253 
00255     const Char_t* getEmbeddingParticleName(const UInt_t id, const Bool_t doSplit=kFALSE) const ; 
00256 
00258     const Char_t* getRealParticleName(const UInt_t id, const Bool_t doSplit=kFALSE) const ; 
00259 
00260     // Data members
00261     static UInt_t mCanvasId ; 
00262 
00263     TCanvas* mCanvas ;        
00264     TPad* mMainPad ;          
00265     TPDF* mPDF ;              
00266 
00267     const Bool_t mIsEmbeddingOnly ;  
00268 
00269     Bool_t mIsOpen ;          
00270 
00271     Bool_t mIsPNGOn ;         
00272     Bool_t mIsGIFOn ;         
00273     Bool_t mIsJPGOn ;         
00274     Bool_t mIsEPSOn ;         
00275     Bool_t mIsPSOn ;          
00276 
00277     Int_t mYear ;             
00278     TString mProduction ;     
00279     Int_t mGeantId ;          
00280     Double_t mPtMax ;         
00281 
00282     TFile* mInputEmbedding ; 
00283     TFile* mInputRealData ;  
00284 
00285     TString mOutputFigureDirectory ; 
00286 
00287     std::vector<Int_t> mDaughterGeantId ; 
00288     std::vector<Int_t> mParentGeantId ;   
00289     std::vector<Int_t> mParentParentGeantId ; 
00290     std::vector<Int_t> mMcGeantId ;       
00291     Int_t mInputParentGeantId ; 
00292 
00293     ClassDef(StEmbeddingQADraw, 1)
00294 };
00295 
00296 inline Int_t StEmbeddingQADraw::getYear()               const { return mYear ; }
00297 inline const Char_t* StEmbeddingQADraw::getProduction() const { return mProduction.Data() ; }
00298 inline Int_t StEmbeddingQADraw::getGeantId()            const { return mGeantId ; }
00299 
00300 #endif
00301