CbmDrawHist.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmDrawHist.h

#ifndef CBMDRAWHIST_H_
#define CBMDRAWHIST_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>
using std::string;
using std::vector;

class CbmDrawingOptions
{
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 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 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 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 = "");

void DrawGraph(
   TGraph* graph,
   HistScale logx = kLinear,
   HistScale logy = kLinear,
   const 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 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");

#endif