| 1 |
//==================================================================================================
|
| 2 |
// CPlot class:
|
| 3 |
// ------------
|
| 4 |
// Class to manage a plot. Member methods are provided to facilitate properties of the plot
|
| 5 |
// (e.g. styles, colors, combining histograms or graphs, etc.)
|
| 6 |
//
|
| 7 |
// USAGE: Need to be compiled on ROOT startup. Add the following line to rootlogon.C,
|
| 8 |
//
|
| 9 |
// gROOT->Macro("CPlot.cc+");
|
| 10 |
//
|
| 11 |
// (Modify path argument if rootlogon.C and CPlot.cc are in different directories)
|
| 12 |
//==================================================================================================
|
| 13 |
|
| 14 |
#ifndef CPLOT_HH
|
| 15 |
#define CPLOT_HH
|
| 16 |
|
| 17 |
#include <TROOT.h>
|
| 18 |
#include <TSystem.h>
|
| 19 |
#include <TCanvas.h>
|
| 20 |
#include <TFile.h>
|
| 21 |
#include <TLegend.h>
|
| 22 |
#include <TLegendEntry.h>
|
| 23 |
#include <TPaveStats.h>
|
| 24 |
#include <TF1.h>
|
| 25 |
#include <TH1F.h>
|
| 26 |
#include <TH2D.h>
|
| 27 |
#include <TGraph.h>
|
| 28 |
#include <TProfile.h>
|
| 29 |
#include <TLine.h>
|
| 30 |
#include <THStack.h>
|
| 31 |
#include <TBox.h>
|
| 32 |
#include <vector>
|
| 33 |
|
| 34 |
#include "RooGlobalFunc.h"
|
| 35 |
#include "RooPlot.h"
|
| 36 |
|
| 37 |
using namespace RooFit;
|
| 38 |
|
| 39 |
class CPlotItem {
|
| 40 |
public:
|
| 41 |
CPlotItem():hist1D(0),hist2D(0),graph(0),prof(0),drawopt(""){}
|
| 42 |
~CPlotItem(){}
|
| 43 |
|
| 44 |
TH1F* hist1D;
|
| 45 |
TH2D* hist2D;
|
| 46 |
TGraph* graph;
|
| 47 |
TProfile* prof;
|
| 48 |
TString drawopt;
|
| 49 |
};
|
| 50 |
|
| 51 |
class CPlot {
|
| 52 |
public:
|
| 53 |
CPlot();
|
| 54 |
CPlot(TString name, TString title, TString xtitle, TString ytitle);
|
| 55 |
CPlot(TString name, RooPlot* frame, TString title, TString xtitle, TString ytitle);
|
| 56 |
~CPlot(){}
|
| 57 |
|
| 58 |
static TString sOutDir; // output directory
|
| 59 |
|
| 60 |
// // Clear the plot by resetting all object properties
|
| 61 |
// void Clear(){}
|
| 62 |
|
| 63 |
// Draw the plot to a given canvas
|
| 64 |
void Draw(TCanvas *c, bool doSave=false, TString format="png");
|
| 65 |
|
| 66 |
// Adding a 1D histogram to the plot
|
| 67 |
void AddHist1D(TH1F *h, TString drawopt="", int color=kBlack, int linesty=1, int fillsty=0, int linecolor=0);
|
| 68 |
void AddHist1D(TH1F *h, TString label, TString drawopt, int color=kBlack, int linesty=1, int fillsty=0);
|
| 69 |
void AddHist1D(TFile *f, TString histName, TString drawopt="", int color=kBlack, int linesty=1, int fillsty=0);
|
| 70 |
void AddHist1D(TFile *f, TString histName, TString label, TString drawopt, int color=kBlack, int linesty=1, int fillsty=0);
|
| 71 |
|
| 72 |
// Adding a 1D histogram to a histogram stack
|
| 73 |
void AddToStack(TH1F *h, int color);
|
| 74 |
void AddToStack(TH1F *h, TString label, int color, int linecolor=1,
|
| 75 |
int linestyle=1, int linewidth=0, int fillstyle=1001);
|
| 76 |
void AddToStack(TFile *f, TString histName, int color);
|
| 77 |
void AddToStack(TFile *f, TString histName, TString label, int color);
|
| 78 |
|
| 79 |
// Adding a 2D histogram to the plot
|
| 80 |
void AddHist2D(TH2D *h, TString drawopt="", int fillcolor=kWhite, int linecolor=kBlack);
|
| 81 |
void AddHist2D(TFile *f, TString histName, TString drawopt="", int fillcolor=kWhite, int linecolor=kBlack);
|
| 82 |
|
| 83 |
// Adding a graph (with error bars) to the plot
|
| 84 |
void AddGraph(TGraph *gr, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 85 |
void AddGraph(TGraph *gr, TString label, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 86 |
void AddGraph(TFile *f, TString grName, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 87 |
void AddGraph(TFile *f, TString grName, TString label, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 88 |
|
| 89 |
// Adding a profile histogram to the plot
|
| 90 |
void AddProfile(TProfile *gr, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 91 |
void AddProfile(TProfile *gr, TString label, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 92 |
void AddProfile(TFile *f, TString prName, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 93 |
void AddProfile(TFile *f, TString prName, TString label, TString drawopt, int color=kBlack, int marksty=kFullDotLarge, int linesty=1);
|
| 94 |
|
| 95 |
// Adding a text box to the plot
|
| 96 |
void AddTextBox(TString text, double x1, double y1, double x2, double y2,
|
| 97 |
int bordersize=1, int textcolor=kBlack, int fillcolor=kWhite);
|
| 98 |
void AddTextBox(double x1, double y1, double x2, double y2,
|
| 99 |
int bordersize, int textcolor, int fillcolor, int nlines, ...);
|
| 100 |
|
| 101 |
// Add a line between two points (x1,y1) and (x2,y2)
|
| 102 |
void AddLine(double x1, double y1, double x2, double y2, int color=kBlack, int style=1);
|
| 103 |
void AddLine(double x1, double y1, double x2, double y2, int color, int style, TString label);
|
| 104 |
|
| 105 |
// Add a box with coordinates: bottom left (x1,y1), top right (x2,y2)
|
| 106 |
void AddBox(double x1, double y1, double x2, double y2, int linecolor=kBlack, int linesty=1, int fillcolor=kWhite);
|
| 107 |
void AddBox(double x1, double y1, double x2, double y2, int linecolor, int linesty, int fillcolor, TString label);
|
| 108 |
|
| 109 |
// Add a 1D function
|
| 110 |
void AddFcn(TF1* fcn, int color=kBlack, int linesty=1);
|
| 111 |
void AddFcn(TF1* fcn, TString label, int color=kBlack, int linesty=1);
|
| 112 |
|
| 113 |
// Set legend position
|
| 114 |
void SetLegend(double x1, double y1, double x2, double y2) {
|
| 115 |
assert(fLeg);
|
| 116 |
fLeg->SetX1(x1); fLeg->SetY1(y1); fLeg->SetX2(x2); fLeg->SetY2(y2);
|
| 117 |
}
|
| 118 |
// Translate legend box
|
| 119 |
void TransLegend(double dx, double dy) {
|
| 120 |
assert(fLeg);
|
| 121 |
fLeg->SetX1(fLeg->GetX1()+dx); fLeg->SetY1(fLeg->GetY1()+dy);
|
| 122 |
fLeg->SetX2(fLeg->GetX2()+dx); fLeg->SetY2(fLeg->GetY2()+dy);
|
| 123 |
}
|
| 124 |
|
| 125 |
// Set stats box position
|
| 126 |
void SetStats(double x, double y) { fStatsX = x; fStatsY = y; }
|
| 127 |
|
| 128 |
// Translate stats box
|
| 129 |
void TransStats(double dx, double dy) { fStatsX += dx; fStatsY += dy; }
|
| 130 |
|
| 131 |
//
|
| 132 |
// Set general properties of the plot
|
| 133 |
//
|
| 134 |
void SetName(TString str) { fName = str; } // plot name (for output)
|
| 135 |
void SetTitle(TString str) { fTitle = str; } // plot title
|
| 136 |
void SetXTitle(TString str) { fXTitle = str; } // x-axis title
|
| 137 |
void SetYTitle(TString str) { fYTitle = str; } // y-axis title
|
| 138 |
void SetXRange(double xmin, double xmax) { fXmin = xmin; fXmax = xmax; } // x-axis range
|
| 139 |
void SetYRange(double ymin, double ymax) { fYmin = ymin; fYmax = ymax; } // y-axis range
|
| 140 |
void SetLogx(int value=1) { fLogx = value; } // toggle logscale x-axis
|
| 141 |
void SetLogy(int value=1) { fLogy = value; } // toggle logscale y-axis
|
| 142 |
void SetGridx(bool value=1) { fGridx = value; } // toggle grid lines from x-axis ticks
|
| 143 |
void SetGridy(bool value=1) { fGridy = value; } // toggle grid lines from y-axis ticks
|
| 144 |
void Rebin(int ngroup) { fRebin = ngroup; } // 1D histogram re-bin
|
| 145 |
void ShowStats(int show=111) { fShowStats = show; } // display statistics
|
| 146 |
|
| 147 |
//
|
| 148 |
// Accessors
|
| 149 |
//
|
| 150 |
TLegend* GetLegend() { return fLeg; }
|
| 151 |
THStack* GetStack() { return fStack; }
|
| 152 |
|
| 153 |
protected:
|
| 154 |
vector<CPlotItem> fItems; // list of items to be plotted
|
| 155 |
vector<TPaveText*> fTextBoxes; // list of text boxes
|
| 156 |
vector<TLine*> fLines; // list of lines
|
| 157 |
vector<TBox*> fBoxes; // list of boxes
|
| 158 |
vector<TF1*> fFcns; // list of 1D functions
|
| 159 |
THStack *fStack; // histogram stack
|
| 160 |
TString fName; // plot name
|
| 161 |
TString fTitle; // plot title
|
| 162 |
TString fXTitle; // x-axis title
|
| 163 |
TString fYTitle; // y-axis title
|
| 164 |
double fXmin, fXmax; // x-axis range
|
| 165 |
double fYmin, fYmax; // y-axis range
|
| 166 |
int fLogx, fLogy; // logscale axes
|
| 167 |
int fGridx, fGridy; // grid lines
|
| 168 |
int fRebin; // grouping for histogram re-bin
|
| 169 |
TLegend *fLeg; // legend object
|
| 170 |
int fShowStats; // whether to display statistics
|
| 171 |
double fStatsX, fStatsY; // x,y coordinates of top left corner of stats box
|
| 172 |
|
| 173 |
vector<TLegendEntry*> fStackEntries; // pointer to legend entry objects for histograms in a stack
|
| 174 |
|
| 175 |
RooPlot *fRooPlot;
|
| 176 |
|
| 177 |
static int sCount; // number of CPlot instances
|
| 178 |
};
|
| 179 |
|
| 180 |
int CPlot::sCount = 0;
|
| 181 |
TString CPlot::sOutDir = ".";
|
| 182 |
|
| 183 |
#endif
|
