UserCode/YiChenHelperCode/PlotHelper.h

#ifndef PlotHelper6125
#define PlotHelper6125

// Plot helper, version 0 - 6125 version
// Intended to help organizing plots and to avoid cluttering
// The main idea is to write a plot both to a pdf and to a TFile at the same time,
//    so that we can check quickly using pdf and come back to TFile later if needed
//    the helper functions here will be mostly on *.ps files
// Developer: Yi Chen, 6125

#include <iostream>
#include <string>
#include <vector>
#include <ctime>
using namespace std;

#include "TCanvas.h"
#include "TLatex.h"
#include "TROOT.h"
#include "TFile.h"
#include "TH1.h"
#include "TGraph.h"
#include "TVirtualPad.h"

class PsFileHelper
{
private:
   string FileName;
   string Option;
   bool Status;   // false - file not opened; true - file good and ongoing
public:
   PsFileHelper();
   PsFileHelper(string filename);
   PsFileHelper(string filename, string option);
   ~PsFileHelper();
   void Open(string filename);
   void Close();
   string GetFileName();
   void SetOption(string option);
   string GetOption();
   template <class PlotType> void AddPlot(PlotType *Histogram, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void Add4PanelPlot(PlotType *Histogram1, PlotType *Histogram2,
      PlotType *Histogram3, PlotType *Histogram4, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void AddPlot(PlotType &Histogram, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void Add4PanelPlot(PlotType &Histogram1, PlotType &Histogram2,
      PlotType &Histogram3, PlotType &Histogram4, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void AddNormalizedPlot(PlotType *Histogram, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void AddNormalizedPlot(PlotType &Histogram, string PlotOption = "",
      bool LogY = false, bool LogZ = false, bool Grid = false);
   template <class PlotType> void AddPlotWithText(PlotType *Histogram, string Text,
      string PlotOption = "", double X = 0.1, double Y = 0.9, double TextSize = 0.03);
   template <class PlotType> void AddPlotWithText(PlotType &Histogram, string Text,
      string PlotOption = "", double X = 0.1, double Y = 0.9, double TextSize = 0.03);
   void AddHistogramFromFile(TFile &File, string HistogramName,
      string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
   void Add4PanelHistogramFromFile(TFile &File, string HistogramName1, string HistogramName2,
      string HistogramName3, string HistogramName4,
      string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
   void AddGraphFromFile(TFile &File, string GraphName,
      string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
   void AddCanvas(TCanvas *Canvas);
   void AddCanvas(TCanvas &Canvas);
   void AddCanvasWithText(TCanvas *Canvas, string Text, double X = 0.1, double Y = 0.9, double TextSize = 0.03);
   void AddCanvasWithText(TCanvas &Canvas, string Text, double X = 0.1, double Y = 0.9, double TextSize = 0.03);
   void AddTextPage(string Text, double X = 0.15, double Y = 0.5, double TextSize = 0.05);
   void AddTextPage(vector<string> Text, double X = 0.1, double Y = 0.9, double TextSize = 0.04);
   void AddTimeStampPage();
};

PsFileHelper::PsFileHelper()
{
   Status = false;
   FileName = "";
   Option = "Landscape";
}

PsFileHelper::PsFileHelper(string filename)
{
   Option = "Landscape";

   Open(filename);
}

PsFileHelper::PsFileHelper(string filename, string option)
{
   Option = option;

   Open(filename);
}

PsFileHelper::~PsFileHelper()
{
   if(Status == true)
      Close();
}

void PsFileHelper::Open(string filename)
{
   FileName = filename;

   Status = true;

   TCanvas canvas;
   canvas.Print((FileName + "[").c_str(), Option.c_str());
}

void PsFileHelper::Close()
{
   TCanvas canvas;
   canvas.Print((FileName + "]").c_str(), Option.c_str());

   gROOT->ProcessLine((".! ps2pdf " + FileName).c_str());
   gROOT->ProcessLine((".! rm " + FileName).c_str());

   Status = false;
   FileName = "";
}

string PsFileHelper::GetFileName()
{
   return FileName;
}

void PsFileHelper::SetOption(string option)
{
   Option = option;
}

string PsFileHelper::GetOption()
{
   return Option;
}

template <class PlotType> void PsFileHelper::AddPlot(PlotType *Histogram, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   if(Histogram == NULL)
      return;

   TCanvas canvas;

   Histogram->Draw(PlotOption.c_str());

   if(LogY == true)
      canvas.SetLogy();
   if(LogZ == true)
      canvas.SetLogz();

   if(Grid == true)
   {
      canvas.SetGridx();
      canvas.SetGridy();
   }

   AddCanvas(canvas);
}

template <class PlotType> void PsFileHelper::Add4PanelPlot(PlotType *Histogram1, PlotType *Histogram2,
   PlotType *Histogram3, PlotType *Histogram4, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   TCanvas canvas;

   canvas.Divide(2, 2);

   PlotType *Histograms[4] = {Histogram1, Histogram2, Histogram3, Histogram4};

   for(int i = 0; i < 4; i++)
   {
      if(Histograms[i] == NULL)
         continue;

      TVirtualPad *Pad = canvas.cd(i + 1);

      Histograms[i]->Draw(PlotOption.c_str());

      if(LogY == true)
         Pad->SetLogy();
      if(LogZ == true)
         Pad->SetLogz();

      if(Grid == true)
      {
         Pad->SetGridx();
         Pad->SetGridy();
      }
   }

   AddCanvas(canvas);
}

template <class PlotType> void PsFileHelper::AddPlot(PlotType &Histogram, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   AddPlot(&Histogram, PlotOption, LogY, LogZ, Grid);
}

template <class PlotType> void Add4PanelPlot(PlotType &Histogram1, PlotType &Histogram2,
   PlotType &Histogram3, PlotType &Histogram4, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   Add4PanelPlot(&Histogram1, &Histogram2, &Histogram3, &Histogram4, PlotOption, LogY, LogZ, Grid);
}

template <class PlotType> void PsFileHelper::AddNormalizedPlot(PlotType *Histogram, string PlotOption,
   bool LogY, bool LogZ, bool Grid)
{
   if(Histogram == NULL)
      return;

   TCanvas canvas;

   Histogram->DrawNormalized(PlotOption.c_str());

   if(Histogram->GetMaximum() > 1e-10)
   {
      if(LogY == true)
         canvas.SetLogy();
      if(LogZ == true)
         canvas.SetLogz();
   }
   if(Grid == true)
   {
      canvas.SetGridx();
      canvas.SetGridy();
   }

   AddCanvas(canvas);
}

template <class PlotType> void PsFileHelper::AddNormalizedPlot(PlotType &Histogram, string PlotOption,
   bool LogY, bool LogZ, bool Grid)
{
   AddNormalizedPlot(&Histogram, PlotOption, LogY, LogZ, Grid);
}

template <class PlotType> void PsFileHelper::AddPlotWithText(PlotType *Histogram, string Text,
   string PlotOption, double X, double Y, double TextSize)
{
   TCanvas canvas;

   Histogram->Draw(PlotOption.c_str());

   AddCanvasWithText(canvas, Text, X, Y, TextSize);
}

template <class PlotType> void PsFileHelper::AddPlotWithText(PlotType &Histogram, string Text,
   string PlotOption, double X, double Y, double TextSize)
{
   AddPlotWithText(&Histogram, Text, PlotOption, X, Y, TextSize);
}

void PsFileHelper::AddHistogramFromFile(TFile &File, string HistogramName, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   TH1 *Histogram = (TH1 *)File.Get(HistogramName.c_str());
   if(Histogram == NULL)
      return;

   AddPlot(Histogram, PlotOption, LogY, LogZ, Grid);
}

void PsFileHelper::Add4PanelHistogramFromFile(TFile &File, string HistogramName1, string HistogramName2,
   string HistogramName3, string HistogramName4, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   TH1 *Histograms[4];

   Histograms[0] = (TH1 *)File.Get(HistogramName1.c_str());
   Histograms[1] = (TH1 *)File.Get(HistogramName2.c_str());
   Histograms[2] = (TH1 *)File.Get(HistogramName3.c_str());
   Histograms[3] = (TH1 *)File.Get(HistogramName4.c_str());

   Add4PanelPlot(Histograms[0], Histograms[1], Histograms[2], Histograms[3], PlotOption, LogY, LogZ, Grid);
}

void PsFileHelper::AddGraphFromFile(TFile &File, string GraphName, string PlotOption, bool LogY, bool LogZ, bool Grid)
{
   TGraph *Graph = (TGraph *)File.Get(GraphName.c_str());
   if(Graph == NULL)
      return;

   AddPlot(Graph, PlotOption, LogY, LogZ, Grid);
}

void PsFileHelper::AddCanvas(TCanvas *Canvas)
{
   if(Canvas == NULL)
      return;

   Canvas->Print(FileName.c_str(), Option.c_str());
}

void PsFileHelper::AddCanvas(TCanvas &Canvas)
{
   AddCanvas(&Canvas);
}

void PsFileHelper::AddCanvasWithText(TCanvas *Canvas, string Text, double X, double Y, double TextSize)
{
   Canvas->cd();

   TLatex text(X, Y, Text.c_str());
   text.SetTextSize(TextSize);
   text.Draw();

   Canvas->Print(FileName.c_str(), Option.c_str());
}

void PsFileHelper::AddCanvasWithText(TCanvas &Canvas, string Text, double X, double Y, double TextSize)
{
   AddCanvasWithText(&Canvas, Text, X, Y, TextSize);
}

void PsFileHelper::AddTextPage(string Text, double X, double Y, double TextSize)
{
   TCanvas canvas;

   TLatex text(X, Y, Text.c_str());
   text.SetTextSize(TextSize);
   text.Draw();

   canvas.Print(FileName.c_str(), Option.c_str());
}

void PsFileHelper::AddTextPage(vector<string> Text, double X, double Y, double TextSize)
{
   TCanvas canvas;

   vector<TLatex> texts;

   for(unsigned int i = 0; i < Text.size(); i++)
   {
      texts.push_back(TLatex(X, Y - i * 0.07, Text[i].c_str()));
      texts[i].SetName(Form("TextLine%d", i));
      texts[i].SetTextSize(TextSize);
   }

   for(unsigned int i = 0; i < Text.size(); i++)
      texts[i].Draw();

   canvas.Print(FileName.c_str(), Option.c_str());
}

void PsFileHelper::AddTimeStampPage()
{
   time_t CurrentTime = time(NULL);

   string str = "Generated at ";
   str = str + ctime(&CurrentTime);

   AddTextPage(str, 0.03);
}

#endif


Revision:	1.1
Committed:	Sun Aug 22 12:10:22 2010 UTC (14 years, 8 months ago) by chenyi
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Log Message:	First version of various helper files
#	User	Rev	Content
1	chenyi	1.1	#ifndef PlotHelper6125
2			#define PlotHelper6125
3
4			// Plot helper, version 0 - 6125 version
5			// Intended to help organizing plots and to avoid cluttering
6			// The main idea is to write a plot both to a pdf and to a TFile at the same time,
7			// so that we can check quickly using pdf and come back to TFile later if needed
8			// the helper functions here will be mostly on *.ps files
9			// Developer: Yi Chen, 6125
10
11			#include <iostream>
12			#include <string>
13			#include <vector>
14			#include <ctime>
15			using namespace std;
16
17			#include "TCanvas.h"
18			#include "TLatex.h"
19			#include "TROOT.h"
20			#include "TFile.h"
21			#include "TH1.h"
22			#include "TGraph.h"
23			#include "TVirtualPad.h"
24
25			class PsFileHelper
26			{
27			private:
28			string FileName;
29			string Option;
30			bool Status; // false - file not opened; true - file good and ongoing
31			public:
32			PsFileHelper();
33			PsFileHelper(string filename);
34			PsFileHelper(string filename, string option);
35			~PsFileHelper();
36			void Open(string filename);
37			void Close();
38			string GetFileName();
39			void SetOption(string option);
40			string GetOption();
41			template <class PlotType> void AddPlot(PlotType *Histogram, string PlotOption = "",
42			bool LogY = false, bool LogZ = false, bool Grid = false);
43			template <class PlotType> void Add4PanelPlot(PlotType Histogram1, PlotType Histogram2,
44			PlotType Histogram3, PlotType Histogram4, string PlotOption = "",
45			bool LogY = false, bool LogZ = false, bool Grid = false);
46			template <class PlotType> void AddPlot(PlotType &Histogram, string PlotOption = "",
47			bool LogY = false, bool LogZ = false, bool Grid = false);
48			template <class PlotType> void Add4PanelPlot(PlotType &Histogram1, PlotType &Histogram2,
49			PlotType &Histogram3, PlotType &Histogram4, string PlotOption = "",
50			bool LogY = false, bool LogZ = false, bool Grid = false);
51			template <class PlotType> void AddNormalizedPlot(PlotType *Histogram, string PlotOption = "",
52			bool LogY = false, bool LogZ = false, bool Grid = false);
53			template <class PlotType> void AddNormalizedPlot(PlotType &Histogram, string PlotOption = "",
54			bool LogY = false, bool LogZ = false, bool Grid = false);
55			template <class PlotType> void AddPlotWithText(PlotType *Histogram, string Text,
56			string PlotOption = "", double X = 0.1, double Y = 0.9, double TextSize = 0.03);
57			template <class PlotType> void AddPlotWithText(PlotType &Histogram, string Text,
58			string PlotOption = "", double X = 0.1, double Y = 0.9, double TextSize = 0.03);
59			void AddHistogramFromFile(TFile &File, string HistogramName,
60			string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
61			void Add4PanelHistogramFromFile(TFile &File, string HistogramName1, string HistogramName2,
62			string HistogramName3, string HistogramName4,
63			string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
64			void AddGraphFromFile(TFile &File, string GraphName,
65			string PlotOption = "", bool LogY = false, bool LogZ = false, bool Grid = false);
66			void AddCanvas(TCanvas *Canvas);
67			void AddCanvas(TCanvas &Canvas);
68			void AddCanvasWithText(TCanvas *Canvas, string Text, double X = 0.1, double Y = 0.9, double TextSize = 0.03);
69			void AddCanvasWithText(TCanvas &Canvas, string Text, double X = 0.1, double Y = 0.9, double TextSize = 0.03);
70			void AddTextPage(string Text, double X = 0.15, double Y = 0.5, double TextSize = 0.05);
71			void AddTextPage(vector<string> Text, double X = 0.1, double Y = 0.9, double TextSize = 0.04);
72			void AddTimeStampPage();
73			};
74
75			PsFileHelper::PsFileHelper()
76			{
77			Status = false;
78			FileName = "";
79			Option = "Landscape";
80			}
81
82			PsFileHelper::PsFileHelper(string filename)
83			{
84			Option = "Landscape";
85
86			Open(filename);
87			}
88
89			PsFileHelper::PsFileHelper(string filename, string option)
90			{
91			Option = option;
92
93			Open(filename);
94			}
95
96			PsFileHelper::~PsFileHelper()
97			{
98			if(Status == true)
99			Close();
100			}
101
102			void PsFileHelper::Open(string filename)
103			{
104			FileName = filename;
105
106			Status = true;
107
108			TCanvas canvas;
109			canvas.Print((FileName + "[").c_str(), Option.c_str());
110			}
111
112			void PsFileHelper::Close()
113			{
114			TCanvas canvas;
115			canvas.Print((FileName + "]").c_str(), Option.c_str());
116
117			gROOT->ProcessLine((".! ps2pdf " + FileName).c_str());
118			gROOT->ProcessLine((".! rm " + FileName).c_str());
119
120			Status = false;
121			FileName = "";
122			}
123
124			string PsFileHelper::GetFileName()
125			{
126			return FileName;
127			}
128
129			void PsFileHelper::SetOption(string option)
130			{
131			Option = option;
132			}
133
134			string PsFileHelper::GetOption()
135			{
136			return Option;
137			}
138
139			template <class PlotType> void PsFileHelper::AddPlot(PlotType *Histogram, string PlotOption, bool LogY, bool LogZ, bool Grid)
140			{
141			if(Histogram == NULL)
142			return;
143
144			TCanvas canvas;
145
146			Histogram->Draw(PlotOption.c_str());
147
148			if(LogY == true)
149			canvas.SetLogy();
150			if(LogZ == true)
151			canvas.SetLogz();
152
153			if(Grid == true)
154			{
155			canvas.SetGridx();
156			canvas.SetGridy();
157			}
158
159			AddCanvas(canvas);
160			}
161
162			template <class PlotType> void PsFileHelper::Add4PanelPlot(PlotType Histogram1, PlotType Histogram2,
163			PlotType Histogram3, PlotType Histogram4, string PlotOption, bool LogY, bool LogZ, bool Grid)
164			{
165			TCanvas canvas;
166
167			canvas.Divide(2, 2);
168
169			PlotType *Histograms[4] = {Histogram1, Histogram2, Histogram3, Histogram4};
170
171			for(int i = 0; i < 4; i++)
172			{
173			if(Histograms[i] == NULL)
174			continue;
175
176			TVirtualPad *Pad = canvas.cd(i + 1);
177
178			Histograms[i]->Draw(PlotOption.c_str());
179
180			if(LogY == true)
181			Pad->SetLogy();
182			if(LogZ == true)
183			Pad->SetLogz();
184
185			if(Grid == true)
186			{
187			Pad->SetGridx();
188			Pad->SetGridy();
189			}
190			}
191
192			AddCanvas(canvas);
193			}
194
195			template <class PlotType> void PsFileHelper::AddPlot(PlotType &Histogram, string PlotOption, bool LogY, bool LogZ, bool Grid)
196			{
197			AddPlot(&Histogram, PlotOption, LogY, LogZ, Grid);
198			}
199
200			template <class PlotType> void Add4PanelPlot(PlotType &Histogram1, PlotType &Histogram2,
201			PlotType &Histogram3, PlotType &Histogram4, string PlotOption, bool LogY, bool LogZ, bool Grid)
202			{
203			Add4PanelPlot(&Histogram1, &Histogram2, &Histogram3, &Histogram4, PlotOption, LogY, LogZ, Grid);
204			}
205
206			template <class PlotType> void PsFileHelper::AddNormalizedPlot(PlotType *Histogram, string PlotOption,
207			bool LogY, bool LogZ, bool Grid)
208			{
209			if(Histogram == NULL)
210			return;
211
212			TCanvas canvas;
213
214			Histogram->DrawNormalized(PlotOption.c_str());
215
216			if(Histogram->GetMaximum() > 1e-10)
217			{
218			if(LogY == true)
219			canvas.SetLogy();
220			if(LogZ == true)
221			canvas.SetLogz();
222			}
223			if(Grid == true)
224			{
225			canvas.SetGridx();
226			canvas.SetGridy();
227			}
228
229			AddCanvas(canvas);
230			}
231
232			template <class PlotType> void PsFileHelper::AddNormalizedPlot(PlotType &Histogram, string PlotOption,
233			bool LogY, bool LogZ, bool Grid)
234			{
235			AddNormalizedPlot(&Histogram, PlotOption, LogY, LogZ, Grid);
236			}
237
238			template <class PlotType> void PsFileHelper::AddPlotWithText(PlotType *Histogram, string Text,
239			string PlotOption, double X, double Y, double TextSize)
240			{
241			TCanvas canvas;
242
243			Histogram->Draw(PlotOption.c_str());
244
245			AddCanvasWithText(canvas, Text, X, Y, TextSize);
246			}
247
248			template <class PlotType> void PsFileHelper::AddPlotWithText(PlotType &Histogram, string Text,
249			string PlotOption, double X, double Y, double TextSize)
250			{
251			AddPlotWithText(&Histogram, Text, PlotOption, X, Y, TextSize);
252			}
253
254			void PsFileHelper::AddHistogramFromFile(TFile &File, string HistogramName, string PlotOption, bool LogY, bool LogZ, bool Grid)
255			{
256			TH1 Histogram = (TH1 )File.Get(HistogramName.c_str());
257			if(Histogram == NULL)
258			return;
259
260			AddPlot(Histogram, PlotOption, LogY, LogZ, Grid);
261			}
262
263			void PsFileHelper::Add4PanelHistogramFromFile(TFile &File, string HistogramName1, string HistogramName2,
264			string HistogramName3, string HistogramName4, string PlotOption, bool LogY, bool LogZ, bool Grid)
265			{
266			TH1 *Histograms[4];
267
268			Histograms[0] = (TH1 *)File.Get(HistogramName1.c_str());
269			Histograms[1] = (TH1 *)File.Get(HistogramName2.c_str());
270			Histograms[2] = (TH1 *)File.Get(HistogramName3.c_str());
271			Histograms[3] = (TH1 *)File.Get(HistogramName4.c_str());
272
273			Add4PanelPlot(Histograms[0], Histograms[1], Histograms[2], Histograms[3], PlotOption, LogY, LogZ, Grid);
274			}
275
276			void PsFileHelper::AddGraphFromFile(TFile &File, string GraphName, string PlotOption, bool LogY, bool LogZ, bool Grid)
277			{
278			TGraph Graph = (TGraph )File.Get(GraphName.c_str());
279			if(Graph == NULL)
280			return;
281
282			AddPlot(Graph, PlotOption, LogY, LogZ, Grid);
283			}
284
285			void PsFileHelper::AddCanvas(TCanvas *Canvas)
286			{
287			if(Canvas == NULL)
288			return;
289
290			Canvas->Print(FileName.c_str(), Option.c_str());
291			}
292
293			void PsFileHelper::AddCanvas(TCanvas &Canvas)
294			{
295			AddCanvas(&Canvas);
296			}
297
298			void PsFileHelper::AddCanvasWithText(TCanvas *Canvas, string Text, double X, double Y, double TextSize)
299			{
300			Canvas->cd();
301
302			TLatex text(X, Y, Text.c_str());
303			text.SetTextSize(TextSize);
304			text.Draw();
305
306			Canvas->Print(FileName.c_str(), Option.c_str());
307			}
308
309			void PsFileHelper::AddCanvasWithText(TCanvas &Canvas, string Text, double X, double Y, double TextSize)
310			{
311			AddCanvasWithText(&Canvas, Text, X, Y, TextSize);
312			}
313
314			void PsFileHelper::AddTextPage(string Text, double X, double Y, double TextSize)
315			{
316			TCanvas canvas;
317
318			TLatex text(X, Y, Text.c_str());
319			text.SetTextSize(TextSize);
320			text.Draw();
321
322			canvas.Print(FileName.c_str(), Option.c_str());
323			}
324
325			void PsFileHelper::AddTextPage(vector<string> Text, double X, double Y, double TextSize)
326			{
327			TCanvas canvas;
328
329			vector<TLatex> texts;
330
331			for(unsigned int i = 0; i < Text.size(); i++)
332			{
333			texts.push_back(TLatex(X, Y - i * 0.07, Text[i].c_str()));
334			texts[i].SetName(Form("TextLine%d", i));
335			texts[i].SetTextSize(TextSize);
336			}
337
338			for(unsigned int i = 0; i < Text.size(); i++)
339			texts[i].Draw();
340
341			canvas.Print(FileName.c_str(), Option.c_str());
342			}
343
344			void PsFileHelper::AddTimeStampPage()
345			{
346			time_t CurrentTime = time(NULL);
347
348			string str = "Generated at ";
349			str = str + ctime(&CurrentTime);
350
351			AddTextPage(str, 0.03);
352			}
353
354			#endif
355
356