Jpp  16.0.3
the software that should make you happy
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Functions
JPlot2D.cc File Reference

General purpose plot program for 2D ROOT objects. More...

#include <string>
#include <iostream>
#include <vector>
#include <set>
#include <cmath>
#include "TROOT.h"
#include "TFile.h"
#include "TClass.h"
#include "TApplication.h"
#include "TCanvas.h"
#include "TKey.h"
#include "TStyle.h"
#include "TAttMarker.h"
#include "TAttLine.h"
#include "TH2.h"
#include "TH3.h"
#include "TH2D.h"
#include "TGraph.h"
#include "TGraphErrors.h"
#include "TGraph2D.h"
#include "TGraph2DErrors.h"
#include "TEllipse.h"
#include "TLine.h"
#include "TF2.h"
#include "TString.h"
#include "TRegexp.h"
#include "TText.h"
#include "JTools/JRange.hh"
#include "JLang/JSinglePointer.hh"
#include "JROOT/JStyle.hh"
#include "JROOT/JCanvas.hh"
#include "JROOT/JMarkerAttributes.hh"
#include "JGizmo/JRootObjectID.hh"
#include "JGizmo/JRootObject.hh"
#include "JGizmo/JGizmoToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Functions

int main (int argc, char **argv)
 

Detailed Description

General purpose plot program for 2D ROOT objects.

The option -f corresponds to <file name>:<object name>.

Author
mdejong

Definition in file JPlot2D.cc.

Function Documentation

int main ( int  argc,
char **  argv 
)

Definition at line 103 of file JPlot2D.cc.

104 {
105  using namespace std;
106  using namespace JPP;
107 
108  typedef JRange<double> JRange_t;
109  typedef pair<string, int> division_type;
110 
111  vector<JRootObjectID> inputFile;
112  string outputFile;
113  JCanvas canvas;
114  int stats;
115  JRange_t X;
116  JRange_t Y;
117  JRange_t Z;
118  JCounter logx;
119  JCounter logy;
120  bool logz;
121  string project;
122  string xLabel;
123  string yLabel;
124  string zLabel;
125  double markerSize;
126  string option;
127  set<char> grid;
128  bool batch;
129  string title;
130  vector<division_type> Ndivisions;
131  int palette;
132  int debug;
133 
134  try {
135 
136  JParser<> zap("General purpose plot program for 2D ROOT objects.");
137 
138  zap['f'] = make_field(inputFile, "<input file>:<object name>");
139  zap['o'] = make_field(outputFile, "graphics output") = "";
140  zap['w'] = make_field(canvas, "size of canvas <nx>x<ny> [pixels]") = JCanvas(500, 500);
141  zap['s'] = make_field(stats) = -1;
142  zap['x'] = make_field(X, "x-abscissa range") = JRange_t();
143  zap['y'] = make_field(Y, "y-abscissa range") = JRange_t();
144  zap['z'] = make_field(Z, "ordinate range") = JRange_t();
145  zap['X'] = make_field(logx, "logarithmic x-axis (-XX log10 axis)");
146  zap['Y'] = make_field(logy, "logarithmic y-axis (-YY log10 axis)");
147  zap['Z'] = make_field(logz, "logarithmic z-axis");
148  zap['P'] = make_field(project, "projection") = "", "xy", "yx", "xz", "zx", "yz", "zy";
149  zap['>'] = make_field(xLabel, "x-axis label") = "";
150  zap['<'] = make_field(yLabel, "y-axis label") = "";
151  zap['^'] = make_field(zLabel, "z-axis label") = "";
152  zap['S'] = make_field(markerSize, "marker size") = 1.0;
153  zap['O'] = make_field(option, "plotting option") = "";
154  zap['G'] = make_field(grid, "grid lines [X][Y]") = JPARSER::initialised();
155  zap['B'] = make_field(batch, "batch processing");
156  zap['T'] = make_field(title, "title") = "KM3NeT preliminary";
157  zap['N'] = make_field(Ndivisions, "axis divisioning (e.g. \"X 505\")") = JPARSER::initialised();
158  zap['p'] = make_field(palette, "palette") = -1;
159  zap['d'] = make_field(debug) = 0;
160 
161  zap(argc, argv);
162  }
163  catch(const exception &error) {
164  FATAL(error.what() << endl);
165  }
166 
167 
168  gROOT->SetBatch(batch);
169 
170  TApplication* tp = new TApplication("user", NULL, NULL);
171  TCanvas* cv = new TCanvas("c1", "c1", canvas.x, canvas.y);
172 
173  JSinglePointer<TStyle> gStyle(new JStyle("gplot", cv->GetWw(), cv->GetWh()));
174 
175  if (palette != -1) {
176  gStyle->SetPalette(palette);
177  }
178 
179  gROOT->SetStyle("gplot");
180  gROOT->ForceStyle();
181 
182 
183  cv->SetFillStyle(4000);
184  cv->SetFillColor(kWhite);
185  cv->Divide(1,1);
186  cv->cd(1);
187 
188  JMarkerAttributes::getInstance().setMarkerSize(markerSize);
189 
190  Double_t xmin = numeric_limits<double>::max();
191  Double_t xmax = numeric_limits<double>::lowest();
192  Double_t ymin = numeric_limits<double>::max();
193  Double_t ymax = numeric_limits<double>::lowest();
194  Double_t zmin = numeric_limits<double>::max();
195  Double_t zmax = numeric_limits<double>::lowest();
196 
197  vector<JRootObject> listOfObjects;
198 
199  TH2* master = NULL;
200 
201  for (vector<JRootObjectID>::const_iterator input = inputFile.begin(); input != inputFile.end(); ++input) {
202 
203  DEBUG("Input: " << *input << endl);
204 
205  TDirectory* dir = getDirectory(*input);
206 
207  if (dir == NULL) {
208  ERROR("File: " << input->getFullFilename() << " not opened." << endl);
209  continue;
210  }
211 
212  const TRegexp regexp(input->getObjectName());
213 
214  TIter iter(dir->GetListOfKeys());
215 
216  for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {
217 
218  const TString tag(key->GetName());
219 
220  DEBUG("Key: " << tag << " match = " << tag.Contains(regexp) << endl);
221 
222  // option match
223 
224  if (tag.Contains(regexp) && isTObject(key)) {
225 
226  if (title == JName_t) {
227  title = key->GetName();
228  } else if (title == JTitle_t) {
229  title = key->GetTitle();
230  }
231 
232  JRootObject object(key->ReadObj());
233 
234  try {
235 
236  TH3& h3 = dynamic_cast<TH3&>(*object);
237 
238  object = h3.Project3D(project.c_str());
239  }
240  catch(exception&) {}
241 
242  try {
243 
244  TH2& h2 = dynamic_cast<TH2&>(*object);
245 
246  h2.SetStats(stats != -1);
247 
248  xmin = min(xmin, h2.GetXaxis()->GetXmin());
249  xmax = max(xmax, h2.GetXaxis()->GetXmax());
250  ymin = min(ymin, h2.GetYaxis()->GetXmin());
251  ymax = max(ymax, h2.GetYaxis()->GetXmax());
252  zmin = min(zmin, logz ? h2.GetMinimum(0.0) : h2.GetMinimum());
253  zmax = max(zmax, h2.GetMaximum());
254  }
255  catch(exception&) {}
256 
257  try {
258 
259  TGraph& g1 = dynamic_cast<TGraph&>(*object);
260 
261  for (Int_t i = 0; i != g1.GetN(); ++i) {
262  xmin = min(xmin, g1.GetX()[i] - numeric_limits<float>::epsilon());
263  xmax = max(xmax, g1.GetX()[i] + numeric_limits<float>::epsilon());
264  ymin = min(ymin, g1.GetY()[i] - numeric_limits<float>::epsilon());
265  ymax = max(ymax, g1.GetY()[i] + numeric_limits<float>::epsilon());
266  }
267 
268  int ng = 0;
269 
270  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
271  if (dynamic_cast<TGraph*>(i->get()) != NULL) {
272  ++ng;
273  }
274  }
275 
276  static_cast<TAttMarker&>(g1) = JMarkerAttributes::getInstance().get(ng);
277  }
278  catch(exception&) {}
279 
280  try {
281 
282  TGraph2D& g2 = dynamic_cast<TGraph2D&>(*object);
283 
284  for (Int_t i = 0; i != g2.GetN(); ++i) {
285  if (!logz || g2.GetZ()[i] > 0.0) {
286  xmin = min(xmin, g2.GetX()[i]);
287  xmax = max(xmax, g2.GetX()[i]);
288  ymin = min(ymin, g2.GetY()[i]);
289  ymax = max(ymax, g2.GetY()[i]);
290  zmin = min(zmin, g2.GetZ()[i]);
291  zmax = max(zmax, g2.GetZ()[i]);
292  }
293  }
294 
295  if (Z != JRange_t()) {
296  g2.SetMinimum(Z.getLowerLimit());
297  g2.SetMaximum(Z.getUpperLimit());
298  }
299  }
300  catch(exception&) {}
301 
302  try {
303 
304  TF2& f2 = dynamic_cast<TF2&>(*object);
305 
306  f2.SetLineColor(kRed);
307  f2.SetTitle((title + ";" + xLabel + ";" + yLabel + ";" + zLabel).c_str());
308 
309  double __xmin;
310  double __xmax;
311  double __ymin;
312  double __ymax;
313 
314  f2.GetRange(__xmin, __ymin, __xmax, __ymax);
315 
316  xmin = min(xmin, __xmin);
317  xmax = max(xmax, __xmax);
318  ymin = min(ymin, __ymin);
319  ymax = max(ymax, __ymax);
320  zmin = min(zmin, f2.GetMinimum());
321  zmax = max(zmax, f2.GetMaximum());
322  }
323  catch(exception&) {}
324 
325  // label
326 
327  for (TString buffer[] = { object.getLabel(), input->getFilename().c_str(), "" }, *i = buffer; *i != ""; ++i) {
328 
329  *i = (*i)(TRegexp("\\[.*\\]"));
330 
331  if ((*i).Length() > 2) {
332  object.setLabel((*i)(1, (*i).Length() - 2));
333  }
334  }
335 
336  if (dynamic_cast<TH2*> (object.get()) != NULL ||
337  dynamic_cast<TGraph*> (object.get()) != NULL ||
338  dynamic_cast<TGraph2D*>(object.get()) != NULL ||
339  dynamic_cast<TEllipse*>(object.get()) != NULL ||
340  dynamic_cast<TLine*> (object.get()) != NULL ||
341  dynamic_cast<TF2*> (object.get()) != NULL) {
342 
343  DEBUG("Add object: " << tag << " with label <" << object.getLabel() << ">" << endl);
344 
345  if (master == NULL) {
346  master = dynamic_cast<TH2*>(object.get());
347  }
348 
349  listOfObjects.push_back(object);
350 
351  } else {
352  ERROR("For other objects than 2D histograms, use JPlot1D" << endl);
353  }
354  }
355  }
356  }
357 
358  if (listOfObjects.empty()) {
359  ERROR("Nothing to draw." << endl);
360  }
361 
362  // plot frame
363 
364  cv->cd(1);
365 
366  if (option.find("COLZ") != string::npos ||
367  option.find("colz") != string::npos) {
368  gPad->SetRightMargin(0.20);
369  }
370 
371  if (X != JRange_t()) {
372  xmin = X.getLowerLimit();
373  xmax = X.getUpperLimit();
374  }
375 
376  if (Y != JRange_t()) {
377  ymin = Y.getLowerLimit();
378  ymax = Y.getUpperLimit();
379  }
380 
381  if (Z != JRange_t()) {
382  zmin = Z.getLowerLimit();
383  zmax = Z.getUpperLimit();
384  } else {
385  setRange(zmin, zmax, logz);
386  }
387 
388  if (!listOfObjects.empty()) {
389 
390  if (master == NULL) {
391 
392  master = new TH2D(MASTER.c_str(), NULL,
393  100, xmin, xmax,
394  100, ymin, ymax);
395 
396  master->SetStats(kFALSE);
397  }
398  }
399 
400  if (master == NULL) {
401 
402  TText* p = new TText(0.5, 0.5, MAKE_CSTRING("No data"));
403 
404  p->SetTextAlign(21);
405  p->SetTextAngle(45);
406  p->Draw();
407 
408  } else {
409 
410  if (logx) { gPad->SetLogx(); }
411  if (logy) { gPad->SetLogy(); }
412  if (logz) { gPad->SetLogz(); }
413 
414  master->GetXaxis()->SetRangeUser(xmin, xmax);
415  master->GetYaxis()->SetRangeUser(ymin, ymax);
416 
417  if (logx > 1) { setLogarithmicX(master); }
418  if (logy > 1) { setLogarithmicY(master); }
419 
420  master->SetTitle(title.c_str());
421 
422  master->SetMinimum(zmin);
423  master->SetMaximum(zmax);
424 
425  if (xLabel != "") { master->GetXaxis()->SetTitle(xLabel.c_str()); master->GetXaxis()->CenterTitle(true); }
426  if (yLabel != "") { master->GetYaxis()->SetTitle(yLabel.c_str()); master->GetYaxis()->CenterTitle(true); }
427  if (zLabel != "") { master->GetZaxis()->SetTitle(zLabel.c_str()); master->GetZaxis()->CenterTitle(true); }
428 
429  master->GetXaxis()->SetMoreLogLabels((logx == 1 && log10(xmax/xmin) < 2) ||
430  (logx > 1 && xmax-xmin < 2));
431  master->GetYaxis()->SetMoreLogLabels((logy == 1 && log10(ymax/ymin) < 2) ||
432  (logy > 1 && ymax-ymin < 2));
433 
434  for (vector<division_type>::const_iterator i = Ndivisions.begin(); i != Ndivisions.end(); ++i) {
435  master->SetNdivisions(i->second, i->first.c_str());
436  }
437 
438  DEBUG("Draw " << master->GetName() << ' ' << option << endl);
439 
440  master->Draw(option.c_str());
441  }
442 
443  if (logx > 1 || logy > 1) {
444  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
445  if (dynamic_cast<TH2*>(i->get()) != master) {
446  if (logx > 1) {
447  if (setLogX<TH2> (i->get())) {}
448  else if (setLogX<TF2> (i->get())) {}
449  else if (setLogX<TGraph2DErrors>(i->get())) {}
450  else if (setLogX<TGraph2D> (i->get())) {}
451  else if (setLogX<TGraphErrors> (i->get())) {}
452  else if (setLogX<TGraph> (i->get())) {}
453  else if (setLogX<TLine> (i->get())) {}
454  else if (setLogX<TEllipse> (i->get())) {}
455  }
456  if (logy > 1) {
457  if (setLogY<TH2> (i->get())) {}
458  else if (setLogY<TF2> (i->get())) {}
459  else if (setLogY<TGraph2DErrors>(i->get())) {}
460  else if (setLogY<TGraph2D> (i->get())) {}
461  else if (setLogY<TGraphErrors> (i->get())) {}
462  else if (setLogY<TGraph> (i->get())) {}
463  else if (setLogY<TLine> (i->get())) {}
464  else if (setLogY<TEllipse> (i->get())) {}
465  }
466  }
467  }
468  }
469 
470  if (grid.count('x') || grid.count('X')) { gPad->SetGridx(); }
471  if (grid.count('y') || grid.count('Y')) { gPad->SetGridy(); }
472 
473  if (stats != -1)
474  gStyle->SetOptStat(stats);
475  else
476  gStyle->SetOptFit(kFALSE);
477 
478  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
479 
480  DEBUG("Draw " << (*i)->GetName() << ' ' << (*i)->GetTitle() << endl);
481 
482  string buffer(option);
483 
484  buffer += "SAME";
485 
486  try {
487 
488  TH2& h2 = dynamic_cast<TH2&>(*(*i));
489 
490  h2.SetMinimum(zmin);
491  h2.SetMaximum(zmax);
492  }
493  catch(exception&) {}
494 
495  try {
496 
497  TGraph& g1 = dynamic_cast<TGraph&>(*(*i));
498 
499  buffer = "P";
500  }
501  catch(exception&) {}
502 
503  try {
504 
505  TGraph2D& g2 = dynamic_cast<TGraph2D&>(*(*i));
506 
507  g2.SetMinimum(zmin);
508  g2.SetMaximum(zmax);
509  }
510  catch(exception&) {}
511 
512  try {
513 
514  TF2& f2 = dynamic_cast<TF2&>(*(*i));
515 
516  f2.SetNpx(1000);
517  f2.SetNpy(1000);
518  /*
519  f2.SetRange(xmin, ymin, zmin,
520  xmax, ymax, zmax);
521  */
522  }
523  catch(exception&) {}
524 
525  (*i)->Draw(buffer.c_str());
526  }
527 
528  gPad->RedrawAxis();
529 
530  cv->Update();
531 
532  if (outputFile != "") {
533  cv->SaveAs(outputFile.c_str());
534  }
535 
536  if (!batch) {
537  tp->Run();
538  }
539 
540  return (master != NULL ? 0 : 1);
541 }
JPARSER::JParser
Utility class to parse command line options.
Definition: JParser.hh:1500
master
then echo Test string reversed by master(hit< return > to continue)." $DIR/JProcess -c "$DIR/JEcho" -rC fi if (( 1 ))
JDETECTOR::getLabel
std::string getLabel(const JLocation &location)
Get module label for monitoring and other applications.
Definition: JLocation.hh:246
MAKE_CSTRING
#define MAKE_CSTRING(A)
Make C-string.
Definition: JPrint.hh:151
JPARSER::initialised
Empty structure for specification of parser element that is initialised (i.e. does not require input)...
Definition: JParser.hh:66
Y
then fatal Wrong number of arguments fi set_variable STRING $argv[1] set_variable DETECTORXY_TXT $WORKDIR $DETECTORXY_TXT tail read X Y CHI2 RMS printf optimum n $X $Y $CHI2 $RMS awk v Y
Definition: detector-XY:fit1d.sh:33
outputFile
string outputFile
Definition: JDAQTimesliceSelector.cc:37
JGIZMO::setLogarithmicX
void setLogarithmicX(TList *list)
Make x-axis of objects in list logarithmic (e.g. after using log10()).
Definition: JGizmoToolkit.hh:917
JLANG::getInstance
T & getInstance(const T &object)
Get static instance from temporary object.
Definition: JObject.hh:75
make_field
#define make_field(A,...)
macro to convert parameter to JParserTemplateElement object
Definition: JParser.hh:1961
log10
set_variable E_E log10(E_{fit}/E_{#mu})"
ERROR
#define ERROR(A)
Definition: JMessage.hh:66
X
then break fi done getCenter read X Y Z let X
Definition: acoustics-fit.sh:523
debug
int debug
debug level
Definition: JSirene.cc:63
JGIZMO::setLogarithmicY
void setLogarithmicY(TList *list)
Make y-axis of objects in list logarithmic (e.g. after using log10()).
Definition: JGizmoToolkit.hh:931
FATAL
#define FATAL(A)
Definition: JMessage.hh:67
JGIZMO::setRange
void setRange(double &xmin, double &xmax, const bool logx)
Set axis range.
Definition: JGizmoToolkit.hh:1157
DEBUG
then usage $script< input_file >< detector_file > fi set_variable OUTPUT_DIR set_variable SELECTOR JDAQTimesliceL1 set_variable DEBUG case set_variable DEBUG
Definition: JCalibratePMT.sh:21
JPARSER::JCounter
Auxiliary class to handle multiple boolean-like I/O.
Definition: JParser.hh:221
JGIZMO::isTObject
bool isTObject(const TKey *key)
Check if given key corresponds to a TObject.
Definition: JGizmoToolkit.hh:1266
JGIZMO::getDirectory
TDirectory * getDirectory(const JRootObjectID &id)
Get TDirectory pointer.
Definition: JGizmoToolkit.hh:135
Z
do set_variable MODULE getModule a $WORKDIR detector_a datx L $STRING JEditDetector a $WORKDIR detector_a datx M $MODULE setz o $WORKDIR detector_a datx JEditDetector a $WORKDIR detector_b datx M $MODULE setz o $WORKDIR detector_b datx done echo Output stored at $WORKDIR detector_a datx and $WORKDIR tripod_a txt JDrawDetector2D a $WORKDIR detector_a datx a $WORKDIR detector_b datx L BL o detector $FORMAT $BATCH JDrawDetector2D T $WORKDIR tripod_a txt T $WORKDIR tripod_b txt L BL o tripod $FORMAT $BATCH JCompareDetector a $WORKDIR detector_a datx b $WORKDIR detector_b datx o $WORKDIR abc root &dev null for KEY in X Y Z
Definition: JAlignDetector.sh:81
g1
Double_t g1(const Double_t x)
Function.
Definition: JQuantiles.cc:25
Generated by   doxygen 1.8.5