StGlauberHistogramMaker.h

/******************************************************************************
 * $Id: StGlauberHistogramMaker.h,v 1.2 2012/04/25 05:03:06 hmasui Exp $
 * $Log: StGlauberHistogramMaker.h,v $
 * Revision 1.2  2012/04/25 05:03:06  hmasui
 * Use StCentralityMaker. Added weight for 2D fill. Use STAR logger instead of iostream
 *
******************************************************************************/

#ifndef __StGlauberHistogramMaker_h__
#define __StGlauberHistogramMaker_h__

class TH1 ;
class TH2 ;
class TProfile ;
class StGlauberTree ;
class StCentralityMaker;
#include "TString.h"
#include <vector>

//____________________________________________________________________________________________________
// Class StGlauberHistogramMaker: Histogram maker
class StGlauberHistogramMaker {
  public:
    StGlauberHistogramMaker() ; 

    // Fixed bin width
    StGlauberHistogramMaker(const TString name, 
        const TString title, const TString ytitle,
        const Int_t ybin, const Double_t ymin, const Double_t ymax,
        const Bool_t isUnitWeight=kTRUE);

    ~StGlauberHistogramMaker() ; 

    void SetTableDirectory(const TString directory) ;

    virtual void Init();   

    void SetXaxis(const StGlauberTree& tree, const StCentralityMaker& centralityMaker, const TString type); 
    virtual void Fill(const Double_t y, const Double_t weight); 

    virtual void Finish(const TString type) ;

    const TString GetName() const ; 
    void DebugOn() ; 

  private:
    // Data members

    const TString mName             ; 
    const TString mTitle            ; 
    const TString mYTitle           ; 
    const Int_t mYbin               ; 
    const Double_t mYmin            ; 
    const Double_t mYmax            ; 
    const Bool_t mIsUnitWeight      ; 

    enum {
      mNXaxis = 4 
    };

    static const TString mXAxisName[mNXaxis]  ; 
    static const TString mXAxisTitle[mNXaxis] ; 
    static const Int_t mXAxisBin[mNXaxis]     ; 
    static const Double_t mXAxisMax[mNXaxis]  ; 

    std::vector<Double_t> mXaxis    ; 

    std::vector<TH1*> mHistogram1D  ; 
    std::vector<TH2*> mHistogram2D  ; 
    std::vector<TProfile*> mProfile ; 
    std::vector<TProfile*> mWeight  ; 

    TString mTableDirectory ; 

    UInt_t mDebug ; 

  protected:
    // Functions
    virtual void Reset();  

    Bool_t IsXaxisOk() const  ; 
    void Fill2D(std::vector<TH2*> collection, const Double_t y, const Double_t weight) ; 
    void FillProfile(std::vector<TProfile*> collection, const Double_t y) ;
    void Fill(const TString type, const Double_t y, const Double_t weight) ;

    void DoWeightCorrection(std::vector<TH1*> collection1d, std::vector<TProfile*> collectionp) ;

    void WriteTable(std::vector<TH1*> collection, const TString name) ;

    void WriteGraphs(std::vector<TH1*> collection) ;

    const TString GetHistogramName(const TString name, const UInt_t ix) const ; // histogram name
    TH1* GetTH1D(const TString name, const UInt_t ix)                         ; // 1D histogram
    TH2* GetTH2D(const TString name, const UInt_t ix)                         ; // 2D histogram
    TProfile* GetTProfile(const TString name, const UInt_t ix, const TString title="Profile") ; // Profile histogram

    UInt_t GetNXaxis() const ;

    ClassDef(StGlauberHistogramMaker, 0)
};

inline UInt_t StGlauberHistogramMaker::GetNXaxis() const { return mNXaxis ; }

#endif