#include <RootPlotter.h>

Inheritance diagram for RootPlotter:

Public Member Functions
void	setXaxisName (TString name)

void	setYaxisName (TString name)

virtual double	getGlobalMax ()

virtual double	getGlobalMin ()

	RootPlotter (TH1 *histogram, double width=300, double height=200)

TString &	drawOptions (int i)

TString &	objDrawOptions (int i)

void	plotStacked (TPad *pad, double scaleFactor)

void	plotSame (TPad *pad, TString plotOptions, double scaleFactor=1.0)

virtual void	plot (TString plotDirectory, TString plotOptions="", TPad *pad=0, double scaleFactor=1.0)

void	addText (TString text, double x, double y, int alignh=1, int alignv=2, double size=0.06, int ndc=true, int color=kBlack)

void	drawLegend ()

virtual	~RootPlotter ()

Public Member Functions inherited from Plotter
int	getColor (int i)

void	setColor (int i, int color)

void	usePresetColours (bool val=true)

void	allImageFormats (bool val=true)

void	setHistogramOwnership (bool i=1)

void	setObjectOwnership (bool i=1)

	Plotter (TString canvasName, double width, double height)

	Plotter (const Plotter &other)

void	add (TObject *histogram)

void	addDot (double xpos, double ypos, double size, int colour=1, TString shape="circle", double sizeY=0.0)

void	logX (bool log=1)

void	logY (bool log=1)

void	logZ (bool log=1)

void	addObject (TObject *obj)

int	getNumObjects ()

void	setImageFormat (TString format)

TPad *	getCanvas ()

void	scaleTextSize (double scale)

void	scaleAxisTitleSize (double scale)

void	scaleAxisTitleOffset (double scale)

void	setXAxisLabelSize (double val)

void	setYAxisLabelSize (double val)

void	setXAxisTitleSize (double val)

void	setYAxisTitleSize (double val)

void	setXAxisLabelOffset (double val)

void	setYAxisLabelOffset (double val)

void	setXAxisTitleOffset (double val)

void	setYAxisTitleOffset (double val)

void	setXAxisTickLength (double val)

void	setYAxisTickLength (double val)

void	setPropertiesFromTH1 (TH1 *hist)

void	setMin (double min)

void	setMax (double max)

void	setBMargin (double val)

void	setLMargin (double val)

void	setRMargin (double val)

void	setTMargin (double val)

virtual	~Plotter ()

Protected Member Functions
virtual void	setHistogramStyle (TH1 *histogram, bool setMinMax=1)=0

TH1 *	getHistogram (int i)

Protected Member Functions inherited from Plotter
virtual void	setCanvasDefaults (TPad *pad)

Protected Attributes
TString	_xAxisName

TString	_yAxisName

std::vector< TString >	_drawOptions

std::vector< TString >	_objDrawOptions

Protected Attributes inherited from Plotter
TPad *	_canvas

TLegend *	_legend

double	_forcedMax

double	_forcedMin

std::vector< TObject * >	_objToPlot

std::vector< TObject * >	_histograms

std::vector< int >	_colours

double	_lMargin

double	_rMargin

double	_tMargin

double	_bMargin

double	_xAxisTitleOffset

double	_yAxisTitleOffset

double	_xAxisLabelOffset

double	_yAxisLabelOffset

double	_xAxisTickLength

double	_yAxisTickLength

double	_xAxisLabelSize

double	_yAxisLabelSize

double	_xAxisTitleSize

double	_yAxisTitleSize

bool	_histogramOwnership

bool	_objectOwnership

bool	_usePresetColours

bool	_allImageFormats

Additional Inherited Members
Static Public Attributes inherited from Plotter
static TString	s_imageformat = ".eps"

static TString	s_imageformat2 = ""

static TString	s_legend_position = "RightTop"

static int	s_plotterCount = 0

static double	s_forcedMax = -999.999

static double	s_forcedMin = -999.999

Detailed Description

HyperPlot, Author: Sam Harnew, sam.h.nosp@m.arne.nosp@m.w@gma.nosp@m.il.c.nosp@m.om , Date: Dec 2015

Class to make plotting root histograms a little easier.

Definition at line 23 of file RootPlotter.h.

Constructor & Destructor Documentation

◆ RootPlotter()

RootPlotter::RootPlotter	(	TH1 *	histogram,
		double	width = `300`,
		double	height = `200`
	)

Construct a RootPlotter with one histogram on a canvas with specified width and height

Definition at line 4 of file RootPlotter.cpp.

                                                                     :
   Plotter("canvas_" + (TString)histogram->GetName(), width, height),
   _xAxisName(histogram->GetXaxis()->GetTitle()),
   _yAxisName(histogram->GetYaxis()->GetTitle())
 {
   _histograms.push_back(histogram);
 
   WELCOME_LOG << "Hello from the Plotter() constructor!";
 }

◆ ~RootPlotter()

RootPlotter::~RootPlotter ( )

virtual

Destructor.

Definition at line 210 of file RootPlotter.cpp.

                          {
 
 }

Member Function Documentation

◆ addText()

void RootPlotter::addText	(	TString	text,
		double	x,
		double	y,
		int	alignh = `1`,
		int	alignv = `2`,
		double	size = `0.06`,
		int	ndc = `true`,
		int	color = `kBlack`
	)

Add a text string at a point (x,y) on the histogram

Definition at line 33 of file RootPlotter.cpp.

                                                                                                                   {
   
   //1left 2center 3right
   
   TLatex* textL  = new TLatex(x, y, text);
 
   if (ndc == true) textL->SetNDC();
   textL->SetTextAlign(alignh*10 + alignv);
   textL->SetTextSize(size);
   textL->SetTextColor(color);
   
   this->addObject(textL);
 
 }

◆ drawLegend()

void RootPlotter::drawLegend ( )

Add a legend to the Canvas - not really tested

Definition at line 49 of file RootPlotter.cpp.

                             {
 
 
   if (_legend != 0) {delete _legend; _legend = 0;}
   
   if      (s_legend_position == "TopRight"   ) _legend = new TLegend(0.55,0.6,0.85,0.78);
   else if (s_legend_position == "BottomRight") _legend = new TLegend(0.55,0.2,0.85,0.38);
   else if (s_legend_position == "TopLeft"    ) _legend = new TLegend(0.15,0.6,0.5,0.78);
   else if (s_legend_position == "BottomLeft" ) _legend = new TLegend(0.6,0.6,0.85,0.8);
   
   VERBOSE_LOG << "Legend position is " << s_legend_position;
 
   if (_legend != 0){
     _legend->SetFillColor(0);
   
     for (unsigned int i = 1; i < _histograms.size(); i++){
         _legend->AddEntry(getHistogram(i),getHistogram(i)->GetTitle(),"l");
     }
   
     _legend->Draw();
   }
 
 }

◆ drawOptions()

TString & RootPlotter::drawOptions ( int i )

Get a reference to the draw options of histogram i - used for getting and setting

Definition at line 21 of file RootPlotter.cpp.

                                       {
   while ((int)_drawOptions.size() <= i) _drawOptions.push_back("");
   return _drawOptions.at(i);
 }

◆ getGlobalMax()

virtual double RootPlotter::getGlobalMax ( )

inlinevirtual

Get the minimum of all histograms added (or the forced min if set)

Reimplemented in RootPlotter1D, and RootPlotter2D.

Definition at line 44 of file RootPlotter.h.

◆ getGlobalMin()

virtual double RootPlotter::getGlobalMin ( )

inlinevirtual

Get the maximum of all histograms added (or the forced max if set)

Reimplemented in RootPlotter1D, and RootPlotter2D.

Definition at line 45 of file RootPlotter.h.

◆ getHistogram()

TH1 * RootPlotter::getHistogram ( int i )

protected

Get one of the histograms that has been added to the plotter

Definition at line 16 of file RootPlotter.cpp.

                                    {
   return (TH1*) _histograms.at(i);
 }

◆ objDrawOptions()

TString & RootPlotter::objDrawOptions ( int i )

Get a reference to the draw options of object i - used for getting and setting

Definition at line 27 of file RootPlotter.cpp.

                                          {
   while ((int)_objDrawOptions.size() <= i) _objDrawOptions.push_back("");
   return _objDrawOptions.at(i);
 }

◆ plot()

void RootPlotter::plot	(	TString	plotDirectory,
		TString	plotOptions = `""`,
		TPad *	pad = `0`,
		double	scaleFactor = `1.0`
	)

virtual

Call this to save the canvas to a specific directory.

Implements Plotter.

Definition at line 184 of file RootPlotter.cpp.

                                                                                                {
   
   if (pad == 0) pad = _canvas;
 
   this->setCanvasDefaults(pad);
   pad->cd();
   //set the histogram options on the first histogram   
 
   if (plotOptions == "STACKED")   plotStacked(pad, scaleFactor);
   else                            plotSame   (pad, plotOptions, scaleFactor);
 
   pad->Update();
   if (plotDirectory != "") {
     pad->Print(plotDirectory + s_imageformat);
     if (s_imageformat2 != "") pad->Print(plotDirectory + s_imageformat2);
     if (_allImageFormats == true){
       pad->Print(plotDirectory + ".pdf");
       pad->Print(plotDirectory + ".eps");
       pad->Print(plotDirectory + ".png");
       pad->Print(plotDirectory + ".C"  );
     }
   }
 }

◆ plotSame()

void RootPlotter::plotSame	(	TPad *	pad,
		TString	plotOptions,
		double	scaleFactor = `1.0`
	)

plot with the option SAME

Definition at line 125 of file RootPlotter.cpp.

                                                                             {
   
   if (pad == 0) pad = _canvas;
   this->setCanvasDefaults(pad);
   pad->cd();
   
   if (scaleFactor != 1.0){ 
     //std::cout << "Scaling histogram in pad " << pad->GetName() << " by " << scaleFactor << std::endl;
     _yAxisLabelSize   = _yAxisLabelSize   * scaleFactor;
     _yAxisTitleSize   = _yAxisTitleSize   * scaleFactor;
     _xAxisLabelOffset = _xAxisLabelOffset * scaleFactor;
     _xAxisTickLength  = _xAxisTickLength  * scaleFactor;
     _yAxisTitleOffset = _yAxisTitleOffset * (1.0/scaleFactor);
     //getHistogram(0)->GetXaxis()->SetTickLength( getHistogram(0)->GetXaxis()->GetTickLength()*scaleFactor );
     //getHistogram(0)->GetYaxis()->SetTickLength( getHistogram(0)->GetYaxis()->GetTickLength()*scaleFactor );
     //getHistogram(0)->GetYaxis()->SetNdivisions(502);
   }
   
 
   
   VERBOSE_LOG << "Setting histogram style";
   this->setHistogramStyle( getHistogram(0) );
   if (_usePresetColours) getHistogram(0)->SetLineColor(getColor(0));
   if (_usePresetColours) getHistogram(0)->SetMarkerColor(getColor(0));
   VERBOSE_LOG << "Draw axis";
 
   getHistogram(0)->DrawCopy(plotOptions + "AXIS");
   this->setHistogramStyle( getHistogram(0) );
   getHistogram(0)->DrawCopy(plotOptions + "AXIS");
 
 
   for (unsigned int i = 0; i < _histograms.size(); i++){
     VERBOSE_LOG << "Set style" << i;
     this->setHistogramStyle( getHistogram(i) );   
     if (_usePresetColours)getHistogram(i)->SetLineColor(getColor(i));
     //getHistogram(i)->SetFillColor(getColor(i));
     if (_usePresetColours)getHistogram(i)->SetMarkerColor(getColor(i));
     VERBOSE_LOG << "Draw" << i;
     getHistogram(i)->DrawCopy(plotOptions + drawOptions(i) + "SAME");
   }
 
 
 
   for (unsigned int i = 0; i < _objToPlot.size(); i++){
     if ( TString(_objToPlot.at(i)->ClassName()).Contains("TGraph") ){
       if (_usePresetColours) ((TGraph*)_objToPlot.at(i))->SetLineColor  (getColor(i));
       if (_usePresetColours) ((TGraph*)_objToPlot.at(i))->SetMarkerColor(getColor(i));      
     }
     _objToPlot.at(i)->DrawClone(plotOptions + "SAME" + objDrawOptions(i) );
     //INFO_LOG << "Drawing object" << i << " with option " << objDrawOptions(i);
   }
   VERBOSE_LOG << "Draw axis";
   getHistogram(0)->DrawCopy(plotOptions + "SAME AXIS");
 
   drawLegend();
 
 }

◆ plotStacked()

void RootPlotter::plotStacked	(	TPad *	pad,
		double	scaleFactor
	)

plot stacked

Definition at line 74 of file RootPlotter.cpp.

                                                           {
 
 
   if (scaleFactor != 1.0){ 
     _tMargin       = 0.0001;
     _bMargin       = _bMargin       * scaleFactor;
     _xAxisLabelSize = _xAxisLabelSize * scaleFactor;
     _yAxisLabelSize = _yAxisLabelSize * scaleFactor;
     _xAxisTitleSize = _xAxisTitleSize * scaleFactor;
     _yAxisTitleSize = _yAxisTitleSize * scaleFactor;
   }
   this->setCanvasDefaults(pad);
   gStyle->SetHistTopMargin(0.0);
 
   THStack* histogramStack = new THStack("stackedHistograms","stackedHistograms");
 
 
   for (unsigned int i = 0; i < _histograms.size(); i++) {
     this->setHistogramStyle( getHistogram(i) , 0);
     getHistogram(i)->SetFillColor(getColor(i));
     histogramStack->Add(getHistogram(i));
   }
 
   _canvas->cd();
   histogramStack->Draw("hist");
   pad->Draw(); 
 
   histogramStack->GetYaxis()->SetNdivisions( getHistogram(0)->GetYaxis()->GetNdivisions () );
   histogramStack->GetYaxis()->CenterTitle  ( getHistogram(0)->GetYaxis()->GetCenterTitle() );
   histogramStack->GetXaxis()->SetNdivisions( getHistogram(0)->GetXaxis()->GetNdivisions () );
   histogramStack->GetXaxis()->CenterTitle  ( getHistogram(0)->GetXaxis()->GetCenterTitle() );
 
   //The histogram is not drawn until the pad is drawn on a canvas - we must therefore 
   // draw the pad before the GetHistogram() returns something. Might run into problems
   // if we start using pads within pads
 
   this->setHistogramStyle( histogramStack->GetHistogram(), 0 );
 
   if (scaleFactor != 1.0){   
     //histogramStack->GetHistogram()->GetXaxis()->SetTickLength(histogramStack->GetHistogram()->GetXaxis()->GetTickLength()*scaleFactor);
     histogramStack->GetHistogram()->GetYaxis()->SetNdivisions(502);
   }
 
   pad->cd();
   histogramStack->DrawClone("hist");
 
   delete histogramStack;
 }

◆ setHistogramStyle()

virtual void RootPlotter::setHistogramStyle	(	TH1 *	histogram,
		bool	setMinMax = `1`
	)

protectedpure virtual

purely virtual function that sets the correct histogram style (different for TH1 and TH2)

Implemented in RootPlotter1D, and RootPlotter2D.

◆ setXaxisName()

void RootPlotter::setXaxisName ( TString name )

inline

Set the x-axis title name

Definition at line 41 of file RootPlotter.h.

◆ setYaxisName()

void RootPlotter::setYaxisName ( TString name )

inline

Set the y-axis title name

Definition at line 42 of file RootPlotter.h.

Member Data Documentation

◆ _drawOptions

std::vector<TString> RootPlotter::_drawOptions

protected

options passed to Root on how to draw each histogram added

Definition at line 30 of file RootPlotter.h.

◆ _objDrawOptions

std::vector<TString> RootPlotter::_objDrawOptions

protected

options passed to Root on how to draw each object added

Definition at line 31 of file RootPlotter.h.

◆ _xAxisName

TString RootPlotter::_xAxisName

protected

xAxis name - by default taken from the first histogram added

Definition at line 27 of file RootPlotter.h.

◆ _yAxisName

TString RootPlotter::_yAxisName

protected

yAxis name - by default taken from the first histogram added

Definition at line 28 of file RootPlotter.h.

The documentation for this class was generated from the following files:

Mint/RootPlotter.h
src/Mojito/Chi2Test/RootPlotter.cpp

Public Member Functions

Protected Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ RootPlotter()

◆ ~RootPlotter()

Member Function Documentation

◆ addText()

◆ drawLegend()

◆ drawOptions()

◆ getGlobalMax()

◆ getGlobalMin()

◆ getHistogram()

◆ objDrawOptions()

◆ plot()

◆ plotSame()

◆ plotStacked()

◆ setHistogramStyle()

◆ setXaxisName()

◆ setYaxisName()

Member Data Documentation

◆ _drawOptions

◆ _objDrawOptions

◆ _xAxisName

◆ _yAxisName