BmnDrawHist.h

Go to the documentation of this file.

 
#ifndef BMNDRAWHIST_H_
#define BMNDRAWHIST_H_
 
#include "TH1.h"
#include "TH1D.h"
#include "TH2.h"
#include "TPad.h"
#include "TLegend.h"
#include "TGraph.h"
#include "TGraph2D.h"
 
#include <string>
#include <vector>
#include <fstream>
using std::string;
using std::vector;
 
class BmnDrawingOptions {
public:
 
    static Int_t Color(Int_t colorIndex) {
        static const Int_t nofColors = 6;
        static Int_t colors[nofColors] = {kRed, kBlue + 1, kGreen + 3, kMagenta + 4, kYellow + 2, kViolet};
        return (colorIndex < nofColors) ? colors[colorIndex] : colorIndex;
    }
 
    static Int_t LineWidth() {
        return 2;
    }
 
    static Int_t LineStyle(Int_t lineStyleIndex) {
        return lineStyleIndex + 1;
    }
 
    static Float_t MarkerSize() {
        return 0.8;
    }
 
    static Int_t MarkerStyle(Int_t markerIndex) {
        static const Int_t nofMarkers = 4;
        //        static Int_t markers[nofMarkers] = {kOpenCircle, kOpenSquare, kOpenTriangleUp, kOpenDiamond,
        //            kFullCircle, kFullSquare, kFullTriangleUp};
        static Int_t markers[nofMarkers] = {kFullCircle, kFullSquare, kFullTriangleUp, kFullDiamond};
        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 string& drawOpt = "",
        Int_t color = BmnDrawingOptions::Color(0),
        Int_t lineWidth = BmnDrawingOptions::LineWidth(),
        Int_t lineStyle = BmnDrawingOptions::LineStyle(0),
        Float_t markerSize = BmnDrawingOptions::MarkerSize(),
        Int_t markerStyle = BmnDrawingOptions::MarkerStyle(0),
        Int_t fillColor = -1);
 
 
void DrawH2(
        TH2* hist,
        HistScale logx = kLinear,
        HistScale logy = kLinear,
        HistScale logz = kLinear,
        const string& drawOpt = "COLZ");
 
 
void DrawH1(
        const vector<TH1*>& histos,
        const vector<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 string& drawOpt = "", Bool_t outputMeanValue01 = kFALSE);
 
void DrawGraph(
        TGraph* graph,
        HistScale logx = kLinear,
        HistScale logy = kLinear,
        const string& drawOpt = "AC",
        Int_t color = BmnDrawingOptions::Color(0),
        Int_t lineWidth = BmnDrawingOptions::LineWidth(),
        Int_t lineStyle = BmnDrawingOptions::LineStyle(0),
        Int_t markerSize = BmnDrawingOptions::MarkerSize(),
        Int_t markerStyle = BmnDrawingOptions::MarkerStyle(0));
 
void DrawGraph(
        const vector<TGraph*>& graphs,
        const vector<string>& graphLabels,
        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);
 
void DrawGraph2D(
        TGraph2D* graph,
        HistScale logx = kLinear,
        HistScale logy = kLinear,
        HistScale logz = kLinear,
        const string& drawOpt = "colz");
 
void DrawMeanLine(TH1* hist);
 
#endif