CbmDrawHist.h

Go to the documentation of this file.

/* Copyright (C) 2008-2020 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#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