StEmbeddingQA.h

//****************************************************************************************************
//  Class StEmbeddingQA
//
//  See http://drupal.star.bnl.gov/STAR/comp/embedding/embedding-procedures/qa-documentation
//  for instructions
//****************************************************************************************************
/****************************************************************************************************
 * $Id: StEmbeddingQA.h,v 1.12 2016/10/27 15:50:06 zhux Exp $
 * $Log: StEmbeddingQA.h,v $
 * Revision 1.12  2016/10/27 15:50:06  zhux
 * added an option to set the maximum pT cut, by Zachariah Miller
 *
 * Revision 1.11  2012/03/05 10:32:29  cpowell
 * Functions added to cut on refMult
 *
 * Revision 1.10  2011/04/01 05:05:47  hmasui
 * Track selections by StEmbeddingQAUtilities. Added 1/pt(RC)-1/pt(MC) vs pt, and pt dependent Ncommon vs NhitFit histograms
 *
 * Revision 1.9  2011/02/11 03:55:44  hmasui
 * Change geantid type to integer
 *
 * Revision 1.8  2011/01/31 21:32:10  hmasui
 * Modify histogram keys to TString to take into account parent geantid
 *
 * Revision 1.7  2010/07/12 21:29:40  hmasui
 * Move isGeantIdOk() function into StEmbeddingQAUtilities
 *
 * Revision 1.6  2010/05/14 19:50:11  hmasui
 * Add rapidity and trigger cuts.
 *
 * Revision 1.5  2010/04/24 20:21:18  hmasui
 * Add geant process check for contaminated pairs
 *
 * Revision 1.4  2010/02/16 02:13:34  hmasui
 * Add parent-parent geant id in the histogram name
 *
 * Revision 1.3  2010/01/28 21:50:37  hmasui
 * Add Vx vs Vz and Vy vs Vz histograms.
 *
 * Revision 1.2  2010/01/26 17:46:31  hmasui
 * Add histograms for eventid, runnumber, and number of particles
 *
 * Revision 1.1  2009/12/22 21:41:17  hmasui
 * Change class name from StEmbeddingQAMaker to StEmbeddingQA
 *
 ****************************************************************************************************/

#ifndef __StEmbeddingQA_h__
#define __StEmbeddingQA_h__

#include <map>
#include <vector>

#include "TMath.h"
#include "TString.h"

#include "StEmbeddingQAUtilities.h"

class TFile ;
class TH1 ;
class TH2 ;
class TH3 ;
class TObject ;
//class TTree ;

class StContamPair ;
class StEmbeddingQAPair ;
class StEmbeddingQATrack ;
class StMiniMcEvent ;
class StMuDstMaker ;
class StMuEvent ;
class StMuTrack ;
class StTinyMcTrack ;

//____________________________________________________________________________________________________
//  Analyze either minimc trees for embedding QA or microDST for real data QA
class StEmbeddingQA {
  public:
    StEmbeddingQA() ;

    StEmbeddingQA(const Int_t year, const TString production, const Bool_t isSimulation = kTRUE);

    virtual ~StEmbeddingQA();

    void init() ;  

    Bool_t book(const TString outputFileName = "");

    Bool_t make(const TString inputFileName, const Bool_t isSimulation = kTRUE);

    Bool_t run(const TString inputFileList) ;

    Bool_t runRealData(const TString inputFileList) ;

    Bool_t runEmbedding(const TString inputFileList) ;

    Bool_t end() const;

    void setZVertexCut(const Float_t vz) ;
    void setRefMultMinCut(const Int_t refMultMin) ;
    void setRefMultMaxCut(const Int_t refMultMax) ;

    void addTriggerIdCut(const UInt_t id) ;

    void setRapidityCut(const Float_t ycut) ;

    void setPtMax(Float_t ptmax) ; 

  private:
    const Int_t mYear ;               
    const TString mProduction ;       
    const Bool_t mIsSimulation ;      
    Float_t mPtMax ;                  

    void clear() ; 

    // Fill histograms
    Bool_t fillEmbedding(const TString inputFileName) ; // Fill embedding histograms
    Bool_t fillRealData(const TString inputFileName) ;  // Fill real data histograms

    void fillEmbeddingTracks(const StMiniMcEvent& mcevent, const Int_t categoryid, const Int_t itrk) ;

    StEmbeddingQATrack* getEmbeddingQATrack(const StMiniMcEvent& mcevent, const Int_t categoryid, const Int_t itrk) ;

    void fillRealTracks(const StMuTrack& track, const Int_t categoryid, const Int_t itrk);

    void fillHistograms(const StEmbeddingQATrack& track, const Int_t categoryid);

    void expandHistograms(const Int_t categoryid, const Int_t geantid, const Int_t parentid,
        const Int_t parentparentid, const Int_t geantprocess);

    Bool_t pushBackGeantId(const Int_t categoryid, const Int_t geantid, const Int_t parentid,
        const Int_t parentparentid, const Int_t geantprocess) ;

    Bool_t isZVertexOk(const StMiniMcEvent& mcevent) const ;

    Bool_t isRefMultOk(const StMiniMcEvent& mcevent) const ;

    Bool_t isTriggerOk(StMuEvent* event) const ;

    Int_t getNtrack(const Int_t categoryid, const StMiniMcEvent& mcevent) const ;

    TString getIdCollection(const Int_t geantid, const Int_t parentid, const Int_t parentparentid) const ;

    StMuDstMaker* mMuDstMaker ; 

    TFile* mOutput ; 

    // Event-wise histograms
    Double_t mVz ;      
    TH1* mhVz ;         
    TH1* mhVzAccepted ; 
    TH1* mhRef ;         
    TH1* mhRefAccepted ; 

    TH2* mhVyVx ; 
    TH2* mhVxVz ; 
    TH2* mhVyVz ; 
    TH1* mhdVx ; 
    TH1* mhdVy ; 
    TH1* mhdVz ; 

    TH1* mhEventId ; 
    TH1* mhRunNumber ; 
    TH1* mhNParticles[StEmbeddingQAConst::mNCategory] ; 

    //
    //  - Use MC momentum instead of reconstructed momentum (Update on Nov/13/2009)
    //  - Add p (reco) vs p (MC) (Update on Nov/13/2009)
    std::vector<Int_t> mGeantId[StEmbeddingQAConst::mNCategory] ; 
    std::vector<TString> mGeantIdCollection ;                       

    TH1* mhGeantId[StEmbeddingQAConst::mNCategory];                               
    std::map<TString, TH3*> mhNHit[StEmbeddingQAConst::mNCategory] ;                
    std::map<TString, TH3*> mhDca[StEmbeddingQAConst::mNCategory] ;                 
    std::map<TString, TH2*> mhPtVsEta[StEmbeddingQAConst::mNCategory] ;             
    std::map<TString, TH2*> mhPtVsY[StEmbeddingQAConst::mNCategory] ;               
    std::map<TString, TH2*> mhPtVsPhi[StEmbeddingQAConst::mNCategory] ;             
    std::map<TString, TH2*> mhPtVsMom[StEmbeddingQAConst::mNCategory] ;             
    std::map<TString, TH2*> mhdPtVsPt[StEmbeddingQAConst::mNCategory] ;             
    std::map<TString, TH2*> mhdInvPtVsPt[StEmbeddingQAConst::mNCategory] ;          
    std::map<TString, TH2*> mhMomVsEta[StEmbeddingQAConst::mNCategory] ;            
    std::map<TString, TH2*> mhdEdxVsMomMc[StEmbeddingQAConst::mNCategory] ;         
    std::map<TString, TH2*> mhdEdxVsMomMcPidCut[StEmbeddingQAConst::mNCategory] ;   
    std::map<TString, TH2*> mhdEdxVsMomReco[StEmbeddingQAConst::mNCategory] ;       
    std::map<TString, TH2*> mhdEdxVsMomRecoPidCut[StEmbeddingQAConst::mNCategory] ; 
    std::map<TString, TH2*> mhRecoPVsMcP[StEmbeddingQAConst::mNCategory] ;          
    std::map<TString, TH3*> mhNCommonHitVsNHit[StEmbeddingQAConst::mNCategory] ;      

    std::map<TString, TH2*> mhEtaVsPhi[StEmbeddingQAConst::mNCategory] ;  
    std::map<TString, TH2*> mhEtaVsVz[StEmbeddingQAConst::mNCategory] ;   
    std::map<TString, TH2*> mhYVsVz[StEmbeddingQAConst::mNCategory] ;     

    ClassDef(StEmbeddingQA, 1);
};

#endif