Jpp  19.0.0
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/JEquals.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/JProperties.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 221 of file JPlot2D.cc.

222 {
223  using namespace std;
224  using namespace JPP;
225 
226  typedef JRange<double> JRange_t;
227 
228  vector<JRootObjectID> inputFile;
229  string outputFile;
230  JCanvas canvas;
231  JStyle::JParameters parameters;
232  int stats;
233  JRange_t X;
234  JRange_t Y;
235  JRange_t Z;
236  JCounter logx;
237  JCounter logy;
238  bool logz;
239  string project;
240  string xLabel;
241  string yLabel;
242  string zLabel;
243  double markerSize;
244  string option;
245  set<char> grid;
246  bool batch;
247  string title;
248  map<string, int> Ndivisions;
249  int palette;
250  int debug;
251 
252  try {
253 
254  JProperties properties = parameters.getProperties();
255 
256  JParser<> zap("General purpose plot program for 2D ROOT objects.");
257 
258  zap['f'] = make_field(inputFile, "<input file>:<object name>");
259  zap['o'] = make_field(outputFile, "graphics output") = "";
260  zap['w'] = make_field(canvas, "size of canvas <nx>x<ny> [pixels]") = JCanvas(500, 500);
261  zap['@'] = make_field(properties) = JPARSER::initialised();
262  zap['s'] = make_field(stats) = -1;
263  zap['x'] = make_field(X, "x-abscissa range") = JRange_t::DEFAULT_RANGE();
264  zap['y'] = make_field(Y, "y-abscissa range") = JRange_t::DEFAULT_RANGE();
265  zap['z'] = make_field(Z, "ordinate range") = JRange_t::DEFAULT_RANGE();
266  zap['X'] = make_field(logx, "logarithmic x-axis (-XX log10 axis)");
267  zap['Y'] = make_field(logy, "logarithmic y-axis (-YY log10 axis)");
268  zap['Z'] = make_field(logz, "logarithmic z-axis");
269  zap['P'] = make_field(project, "projection") = "", "xy", "yx", "xz", "zx", "yz", "zy";
270  zap['>'] = make_field(xLabel, "x-axis label") = "";
271  zap['<'] = make_field(yLabel, "y-axis label") = "";
272  zap['^'] = make_field(zLabel, "z-axis label") = "";
273  zap['S'] = make_field(markerSize, "marker size") = 1.0;
274  zap['O'] = make_field(option, "plotting option") = "";
275  zap['G'] = make_field(grid, "grid lines [X][Y]") = JPARSER::initialised();
276  zap['B'] = make_field(batch, "batch processing");
277  zap['T'] = make_field(title, "graphics title ("
278  << "\"" << JName_t << "\" -> ROOT name; "
279  << "\"" << JTitle_t << "\" -> ROOT title)") = "KM3NeT preliminary";
280  zap['N'] = make_field(Ndivisions, "axis divisioning (e.g. \"X 505\")") = JPARSER::initialised();
281  zap['p'] = make_field(palette, "palette") = -1;
282  zap['d'] = make_field(debug) = 0;
283 
284  zap(argc, argv);
285  }
286  catch(const exception &error) {
287  FATAL(error.what() << endl);
288  }
289 
290 
291  gROOT->SetBatch(batch);
292 
293  TApplication* tp = new TApplication("user", NULL, NULL);
294  TCanvas* cv = new TCanvas("c1", "c1", canvas.x, canvas.y);
295 
296  unique_ptr<TStyle> gStyle(new JStyle("gplot", cv->GetWw(), cv->GetWh(), parameters));
297 
298  if (palette != -1) {
299  gStyle->SetPalette(palette);
300  }
301 
302  gROOT->SetStyle("gplot");
303  gROOT->ForceStyle();
304 
305 
306  cv->SetFillStyle(4000);
307  cv->SetFillColor(kWhite);
308  cv->Divide(1,1);
309  cv->cd(1);
310 
311  JMarkerAttributes::getInstance().setMarkerSize(markerSize);
312 
313  Double_t xmin = numeric_limits<double>::max();
314  Double_t xmax = numeric_limits<double>::lowest();
315  Double_t ymin = numeric_limits<double>::max();
316  Double_t ymax = numeric_limits<double>::lowest();
317  Double_t zmin = numeric_limits<double>::max();
318  Double_t zmax = numeric_limits<double>::lowest();
319 
320  vector<JRootObject> listOfObjects;
321 
322  //TH2* master = NULL;
323  JMaster master;
324 
325  for (vector<JRootObjectID>::const_iterator input = inputFile.begin(); input != inputFile.end(); ++input) {
326 
327  DEBUG("Input: " << *input << endl);
328 
329  TDirectory* dir = getDirectory(*input);
330 
331  if (dir == NULL) {
332  ERROR("File: " << input->getFullFilename() << " not opened." << endl);
333  continue;
334  }
335 
336  const TRegexp regexp(input->getObjectName());
337 
338  TIter iter(dir->GetListOfKeys());
339 
340  for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {
341 
342  const TString tag(key->GetName());
343 
344  DEBUG("Key: " << tag << " match = " << tag.Contains(regexp) << endl);
345 
346  // option match
347 
348  if (tag.Contains(regexp) && isTObject(key)) {
349 
350  if (title == JName_t) {
351  title = key->GetName();
352  } else if (title == JTitle_t) {
353  title = key->GetTitle();
354  }
355 
356  JRootObject object(key->ReadObj());
357 
358  try {
359 
360  TH3& h3 = dynamic_cast<TH3&>(*object);
361 
362  object = h3.Project3D(project.c_str());
363  }
364  catch(exception&) {}
365 
366  try {
367 
368  TH2& h2 = dynamic_cast<TH2&>(*object);
369 
370  h2.SetStats(stats != -1);
371 
372  xmin = min(xmin, h2.GetXaxis()->GetXmin());
373  xmax = max(xmax, h2.GetXaxis()->GetXmax());
374  ymin = min(ymin, h2.GetYaxis()->GetXmin());
375  ymax = max(ymax, h2.GetYaxis()->GetXmax());
376  zmin = min(zmin, logz ? h2.GetMinimum(0.0) : h2.GetMinimum());
377  zmax = max(zmax, h2.GetMaximum());
378  }
379  catch(exception&) {}
380 
381  try {
382 
383  TGraph& g1 = dynamic_cast<TGraph&>(*object);
384 
385  for (Int_t i = 0; i != g1.GetN(); ++i) {
386  xmin = min(xmin, g1.GetX()[i] - numeric_limits<float>::epsilon());
387  xmax = max(xmax, g1.GetX()[i] + numeric_limits<float>::epsilon());
388  ymin = min(ymin, g1.GetY()[i] - numeric_limits<float>::epsilon());
389  ymax = max(ymax, g1.GetY()[i] + numeric_limits<float>::epsilon());
390  }
391 
392  int ng = 0;
393 
394  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
395  if (dynamic_cast<TGraph*>(i->get()) != NULL) {
396  ++ng;
397  }
398  }
399 
400  static_cast<TAttMarker&>(g1) = JMarkerAttributes::getInstance().get(ng);
401  }
402  catch(exception&) {}
403 
404  try {
405 
406  TGraph2D& g2 = dynamic_cast<TGraph2D&>(*object);
407 
408  for (Int_t i = 0; i != g2.GetN(); ++i) {
409  if (!logz || g2.GetZ()[i] > 0.0) {
410  xmin = min(xmin, g2.GetX()[i]);
411  xmax = max(xmax, g2.GetX()[i]);
412  ymin = min(ymin, g2.GetY()[i]);
413  ymax = max(ymax, g2.GetY()[i]);
414  zmin = min(zmin, g2.GetZ()[i]);
415  zmax = max(zmax, g2.GetZ()[i]);
416  }
417  }
418 
419  if (Z.is_valid()) {
420  g2.SetMinimum(Z.getLowerLimit());
421  g2.SetMaximum(Z.getUpperLimit());
422  }
423  }
424  catch(exception&) {}
425 
426  try {
427 
428  TF2& f2 = dynamic_cast<TF2&>(*object);
429 
430  f2.SetLineColor(kRed);
431  f2.SetTitle((title + ";" + xLabel + ";" + yLabel + ";" + zLabel).c_str());
432 
433  double __xmin;
434  double __xmax;
435  double __ymin;
436  double __ymax;
437 
438  f2.GetRange(__xmin, __ymin, __xmax, __ymax);
439 
440  xmin = min(xmin, __xmin);
441  xmax = max(xmax, __xmax);
442  ymin = min(ymin, __ymin);
443  ymax = max(ymax, __ymax);
444  zmin = min(zmin, f2.GetMinimum());
445  zmax = max(zmax, f2.GetMaximum());
446  }
447  catch(exception&) {}
448 
449  // label
450 
451  for (TString buffer[] = { object.getLabel(), input->getFilename().c_str(), "" }, *i = buffer; *i != ""; ++i) {
452 
453  *i = (*i)(TRegexp("\\[.*\\]"));
454 
455  if ((*i).Length() > 2) {
456  object.setLabel((*i)(1, (*i).Length() - 2));
457  }
458  }
459 
460  if (dynamic_cast<TH2*> (object.get()) != NULL ||
461  dynamic_cast<TGraph*> (object.get()) != NULL ||
462  dynamic_cast<TGraph2D*>(object.get()) != NULL ||
463  dynamic_cast<TEllipse*>(object.get()) != NULL ||
464  dynamic_cast<TLine*> (object.get()) != NULL ||
465  dynamic_cast<TText*> (object.get()) != NULL ||
466  dynamic_cast<TF2*> (object.get()) != NULL) {
467 
468  DEBUG("Add object: " << tag << " with label <" << object.getLabel() << ">" << endl);
469 
470  if (master == NULL) {
471  //master = dynamic_cast<TH2*>(object.get());
472  master = object.get();
473  }
474 
475  listOfObjects.push_back(object);
476 
477  } else {
478  ERROR("For other objects than 2D histograms, use JPlot1D" << endl);
479  }
480  }
481  }
482  }
483 
484  if (listOfObjects.empty()) {
485  ERROR("Nothing to draw." << endl);
486  }
487 
488  // plot frame
489 
490  cv->cd(1);
491 
492  if (option.find("COLZ") != string::npos ||
493  option.find("colz") != string::npos) {
494  gPad->SetRightMargin(0.20);
495  }
496 
497  if (X.is_valid()) {
498  xmin = X.getLowerLimit();
499  xmax = X.getUpperLimit();
500  }
501 
502  if (Y.is_valid()) {
503  ymin = Y.getLowerLimit();
504  ymax = Y.getUpperLimit();
505  }
506 
507  if (Z.is_valid()) {
508  zmin = Z.getLowerLimit();
509  zmax = Z.getUpperLimit();
510  } else if (zmax > zmin) {
511  setRange(zmin, zmax, logz);
512  }
513 
514  if (!listOfObjects.empty()) {
515 
516  if (master == NULL) {
517 
518  master = new TH2D(MASTER.c_str(), NULL,
519  100, xmin, xmax,
520  100, ymin, ymax);
521 
522  master->SetStats(kFALSE);
523  }
524  }
525 
526  if (master == NULL) {
527 
528  TText* p = new TText(0.5, 0.5, "No data");
529 
530  p->SetTextAlign(21);
531  p->SetTextAngle(45);
532  p->Draw();
533 
534  } else {
535 
536  if (logx) { gPad->SetLogx(); }
537  if (logy) { gPad->SetLogy(); }
538  if (logz) { gPad->SetLogz(); }
539 
540  master->GetXaxis()->SetRangeUser(xmin, xmax);
541  master->GetYaxis()->SetRangeUser(ymin, ymax);
542 
543  if (logx > 1) { setLogarithmicX(master); }
544  if (logy > 1) { setLogarithmicY(master); }
545 
546  master->SetTitle(title.c_str());
547 
548  master->SetMinimum(zmin);
549  master->SetMaximum(zmax);
550 
551  if (xLabel != "") { master->GetXaxis()->SetTitle(xLabel.c_str()); master->GetXaxis()->CenterTitle(true); }
552  if (yLabel != "") { master->GetYaxis()->SetTitle(yLabel.c_str()); master->GetYaxis()->CenterTitle(true); }
553  if (zLabel != "") { master->GetZaxis()->SetTitle(zLabel.c_str()); master->GetZaxis()->CenterTitle(true); }
554 
555  master->GetXaxis()->SetMoreLogLabels((logx == 1 && log10(xmax/xmin) < 2) ||
556  (logx > 1 && xmax-xmin < 2));
557  master->GetYaxis()->SetMoreLogLabels((logy == 1 && log10(ymax/ymin) < 2) ||
558  (logy > 1 && ymax-ymin < 2));
559 
560  for (map<string, int>::const_iterator i = Ndivisions.begin(); i != Ndivisions.end(); ++i) {
561  master->SetNdivisions(i->second, i->first.c_str());
562  }
563 
564  DEBUG("Draw " << master->GetName() << ' ' << option << endl);
565 
566  master->Draw(option.c_str());
567  }
568 
569  if (logx > 1 || logy > 1) {
570  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
571  if (dynamic_cast<TH2*>(i->get()) != master) {
572  if (logx > 1) {
573  if (setLogX<TH2> (i->get())) {}
574  else if (setLogX<TF2> (i->get())) {}
575  else if (setLogX<TGraph2DErrors>(i->get())) {}
576  else if (setLogX<TGraph2D> (i->get())) {}
577  else if (setLogX<TGraphErrors> (i->get())) {}
578  else if (setLogX<TGraph> (i->get())) {}
579  else if (setLogX<TLine> (i->get())) {}
580  else if (setLogX<TEllipse> (i->get())) {}
581  }
582  if (logy > 1) {
583  if (setLogY<TH2> (i->get())) {}
584  else if (setLogY<TF2> (i->get())) {}
585  else if (setLogY<TGraph2DErrors>(i->get())) {}
586  else if (setLogY<TGraph2D> (i->get())) {}
587  else if (setLogY<TGraphErrors> (i->get())) {}
588  else if (setLogY<TGraph> (i->get())) {}
589  else if (setLogY<TLine> (i->get())) {}
590  else if (setLogY<TEllipse> (i->get())) {}
591  }
592  }
593  }
594  }
595 
596  if (grid.count('x') || grid.count('X')) { gPad->SetGridx(); }
597  if (grid.count('y') || grid.count('Y')) { gPad->SetGridy(); }
598 
599  if (stats != -1)
600  gStyle->SetOptStat(stats);
601  else
602  gStyle->SetOptFit(kFALSE);
603 
604 
605  for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
606 
607  if (i->get() != master) {
608 
609  DEBUG("Draw " << (*i)->GetName() << ' ' << (*i)->GetTitle() << endl);
610 
611  string buffer(option);
612 
613  buffer += "SAME";
614 
615  try {
616 
617  TH2& h2 = dynamic_cast<TH2&>(*(*i));
618 
619  h2.SetMinimum(zmin);
620  h2.SetMaximum(zmax);
621  }
622  catch(exception&) {}
623 
624  try {
625 
626  TGraph& g1 = dynamic_cast<TGraph&>(*(*i));
627 
628  buffer = "P";
629  }
630  catch(exception&) {}
631 
632  try {
633 
634  TGraph2D& g2 = dynamic_cast<TGraph2D&>(*(*i));
635 
636  g2.SetMinimum(zmin);
637  g2.SetMaximum(zmax);
638  }
639  catch(exception&) {}
640 
641  try {
642 
643  TF2& f2 = dynamic_cast<TF2&>(*(*i));
644 
645  f2.SetNpx(1000);
646  f2.SetNpy(1000);
647  /*
648  f2.SetRange(xmin, ymin, zmin,
649  xmax, ymax, zmax);
650  */
651  }
652  catch(exception&) {}
653 
654  (*i)->Draw(buffer.c_str());
655  }
656  }
657 
658  //gPad->RedrawAxis();
659 
660  cv->Update();
661 
662  if (outputFile != "") {
663  cv->SaveAs(outputFile.c_str());
664  }
665 
666  if (!batch) {
667  tp->Run();
668  }
669 
670  return (master != NULL ? 0 : 1);
671 }
GAUSS_LEGENDRE::xmax
const double xmax
Definition: JQuadrature.cc:24
JPARSER::JParser
Utility class to parse command line options.
Definition: JParser.hh:1711
master
then echo Test string reversed by master(hit< return > to continue)." $DIR/JProcess -c "$DIR/JEcho" -rC fi if (( 1 ))
create_update_issues.project
tuple project
Definition: create_update_issues.py:21
JDETECTOR::getLabel
std::string getLabel(const JLocation &location)
Get module label for monitoring and other applications.
Definition: JLocation.hh:246
JEEP::JProperties
Utility class to parse parameter values.
Definition: JProperties.hh:497
parameters
*fatal Wrong number of arguments esac JCookie sh typeset Z DETECTOR typeset Z SOURCE_RUN typeset Z TARGET_RUN set_variable PARAMETERS_FILE $WORKDIR parameters
Definition: diff-Tuna.sh:38
JPARSER::initialised
Empty structure for specification of parser element that is initialised (i.e. does not require input)...
Definition: JParser.hh:84
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:927
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:2158
log10
set_variable E_E log10(E_{fit}/E_{#mu})"
ERROR
#define ERROR(A)
Definition: JMessage.hh:66
JGIZMO::setLogarithmicY
void setLogarithmicY(TList *list)
Make y-axis of objects in list logarithmic (e.g. after using log10()).
Definition: JGizmoToolkit.hh:941
FATAL
#define FATAL(A)
Definition: JMessage.hh:67
GAUSS_LEGENDRE::xmin
const double xmin
Definition: JQuadrature.cc:23
JGIZMO::setRange
void setRange(double &xmin, double &xmax, const bool logx)
Set axis range.
Definition: JGizmoToolkit.hh:1417
JPARSER::JCounter
Auxiliary class to handle multiple boolean-like I/O.
Definition: JParser.hh:435
JGIZMO::isTObject
bool isTObject(const TKey *key)
Check if given key corresponds to a TObject.
Definition: JGizmoToolkit.hh:1526
X
no fit printf nominal n $STRING awk v X
Definition: detector-XY:fit.sh:102
JGIZMO::getDirectory
TDirectory * getDirectory(const JRootObjectID &id)
Get TDirectory pointer.
Definition: JGizmoToolkit.hh:141
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
GAUSS_LEGENDRE::epsilon
const double epsilon
Definition: JQuadrature.cc:21
debug
int debug
debug level
Definition: archive-put-wiki-detectors.sh:92
DEBUG
#define DEBUG(A)
Message macros.
Definition: JMessage.hh:62
g1
Double_t g1(const Double_t x)
Function.
Definition: JQuantiles.cc:25
Generated by   doxygen 1.8.5