CbmDrawHist.h

Go to the documentation of this file.

 
#ifndef CBMDRAWHIST_H_
#define CBMDRAWHIST_H_
 
#include <Rtypes.h>      // for kBlue, kGreen, kOrange, kRed, kYellow, kBlack
#include <RtypesCore.h>  // for Int_t, Double_t, Bool_t
#include <TAttMarker.h>  // for kFullCircle, kFullSquare, kFullTriangleUp
 
#include <string>  // for string
#include <vector>  // for vector
 
class TGraph2D;
class TGraph;
class TH1;
class TH2;
class TH2D;
class TH3;
 
class CbmDrawingOptions {
public:
  static Int_t Color(Int_t colorIndex) {
    static const Int_t nofColors   = 15;
    static Int_t colors[nofColors] = {kRed,
                                      kBlue + 1,
                                      kGreen + 3,
                                      kMagenta + 4,
                                      kYellow + 2,
                                      kViolet,
                                      kOrange - 5,
                                      kOrange + 3,
                                      kRed - 8,
                                      kPink + 9,
                                      kBlue - 6,
                                      kCyan + 1,
                                      kGreen - 6,
                                      kYellow - 1,
                                      kGray + 1};
    return (colorIndex < nofColors) ? colors[colorIndex] : colorIndex;
  }
 
  static Int_t LineWidth() { return 2; }
 
  static Int_t LineStyle(Int_t lineStyleIndex) { return lineStyleIndex + 1; }
 
  static Int_t MarkerSize() { return 1; }
 
  static Int_t MarkerStyle(Int_t markerIndex) {
    static const Int_t nofMarkers    = 8;
    static Int_t markers[nofMarkers] = {kOpenCircle,
                                        kOpenSquare,
                                        kOpenTriangleUp,
                                        kOpenDiamond,
                                        kFullCircle,
                                        kFullSquare,
                                        kFullTriangleUp};
    return (markerIndex < nofMarkers) ? markers[markerIndex] : markerIndex;
  }
 
  static Double_t TextSize() { return 0.06; }
};
 
enum HistScale {
  kLog    = 0, 
  kLinear = 1  
};
 
 
void SetDefaultDrawStyle();
 
void DrawH1(TH1* hist,
            HistScale logx             = kLinear,
            HistScale logy             = kLinear,
            const std::string& drawOpt = "",
            Int_t color                = CbmDrawingOptions::Color(0),
            Int_t lineWidth            = CbmDrawingOptions::LineWidth(),
            Int_t lineStyle            = CbmDrawingOptions::LineStyle(0),
            Int_t markerSize           = CbmDrawingOptions::MarkerSize(),
            Int_t markerStyle          = CbmDrawingOptions::MarkerStyle(0));
 
 
void DrawH2(TH2* hist,
            HistScale logx             = kLinear,
            HistScale logy             = kLinear,
            HistScale logz             = kLinear,
            const std::string& drawOpt = "COLZ");
 
 
void DrawH1(const std::vector<TH1*>& histos,
            const std::vector<std::string>& histLabels,
            HistScale logx             = kLinear,
            HistScale logy             = kLinear,
            Bool_t drawLegend          = true,
            Double_t x1                = 0.25,
            Double_t y1                = 0.99,
            Double_t x2                = 0.55,
            Double_t y2                = 0.75,
            const std::string& drawOpt = "");
 
void DrawGraph(TGraph* graph,
               HistScale logx             = kLinear,
               HistScale logy             = kLinear,
               const std::string& drawOpt = "AC",
               Int_t color                = CbmDrawingOptions::Color(0),
               Int_t lineWidth            = CbmDrawingOptions::LineWidth(),
               Int_t lineStyle            = CbmDrawingOptions::LineStyle(0),
               Int_t markerSize           = CbmDrawingOptions::MarkerSize(),
               Int_t markerStyle          = CbmDrawingOptions::MarkerStyle(0));
 
void DrawGraph(const std::vector<TGraph*>& graphs,
               const std::vector<std::string>& graphLabels,
               HistScale logx    = kLinear,
               HistScale logy    = kLinear,
               Bool_t drawLegend = true,
               Double_t x1       = 0.50,
               Double_t y1       = 0.78,
               Double_t x2       = 0.99,
               Double_t y2       = 0.99);
 
void DrawGraph2D(TGraph2D* graph,
                 HistScale logx             = kLinear,
                 HistScale logy             = kLinear,
                 HistScale logz             = kLinear,
                 const std::string& drawOpt = "colz");
 
void DrawTextOnPad(const std::string& text,
                   Double_t x1,
                   Double_t y1,
                   Double_t x2,
                   Double_t y2);
 
 
void DrawH1andFitGauss(TH1* hist,
                       Bool_t drawResults    = true,
                       Bool_t doScale        = true,
                       Double_t userRangeMin = 0.,
                       Double_t userRangeMax = 0.);
 
void DrawH2WithProfile(TH2* hist,
                       Bool_t doGaussFit          = false,
                       Bool_t drawOnlyMean        = false,
                       const std::string& drawOpt = "COLZ",
                       Int_t profileColor         = kBlack,
                       Int_t profileLineWidth     = 4);
 
TH2D* DrawH3Profile(TH3* h,
                    Bool_t drawMean        = true,
                    Bool_t doGaussFit      = false,
                    Double_t zUserRangeMin = 0.,
                    Double_t zUserRangeMax = 0.);
 
#endif