Jpp master_rocky-44-g75b7c4f75
the software that should make you happy
Loading...
Searching...
No Matches
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

◆ main()

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->SetTitle(title.c_str());
541
542 master->GetXaxis()->SetRangeUser(xmin, xmax);
543 master->GetYaxis()->SetRangeUser(ymin, ymax);
544
545 if (logx > 1) { setLogarithmicX(master); }
546 if (logy > 1) { setLogarithmicY(master); }
547
548 if (logx > 2) { master->GetXaxis()->SetNoExponent(); }
549 if (logy > 2) { master->GetYaxis()->SetNoExponent(); }
550
551 master->SetMinimum(zmin);
552 master->SetMaximum(zmax);
553
554 if (xLabel != "") { master->GetXaxis()->SetTitle(xLabel.c_str()); master->GetXaxis()->CenterTitle(true); }
555 if (yLabel != "") { master->GetYaxis()->SetTitle(yLabel.c_str()); master->GetYaxis()->CenterTitle(true); }
556 if (zLabel != "") { master->GetZaxis()->SetTitle(zLabel.c_str()); master->GetZaxis()->CenterTitle(true); }
557
558 master->GetXaxis()->SetMoreLogLabels((logx == 1 && log10(xmax/xmin) < 2) ||
559 (logx > 1 && xmax-xmin < 2));
560 master->GetYaxis()->SetMoreLogLabels((logy == 1 && log10(ymax/ymin) < 2) ||
561 (logy > 1 && ymax-ymin < 2));
562
563 for (map<string, int>::const_iterator i = Ndivisions.begin(); i != Ndivisions.end(); ++i) {
564 master->SetNdivisions(i->second, i->first.c_str());
565 }
566
567 DEBUG("Draw " << master->GetName() << ' ' << option << endl);
568
569 master->Draw(option.c_str());
570 }
571
572 if (logx > 1 || logy > 1) {
573 for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
574 if (dynamic_cast<TH2*>(i->get()) != master) {
575 if (logx > 1) {
576 if (setLogX<TH2> (i->get())) {}
577 else if (setLogX<TF2> (i->get())) {}
578 else if (setLogX<TGraph2DErrors>(i->get())) {}
579 else if (setLogX<TGraph2D> (i->get())) {}
580 else if (setLogX<TGraphErrors> (i->get())) {}
581 else if (setLogX<TGraph> (i->get())) {}
582 else if (setLogX<TLine> (i->get())) {}
583 else if (setLogX<TEllipse> (i->get())) {}
584 }
585 if (logy > 1) {
586 if (setLogY<TH2> (i->get())) {}
587 else if (setLogY<TF2> (i->get())) {}
588 else if (setLogY<TGraph2DErrors>(i->get())) {}
589 else if (setLogY<TGraph2D> (i->get())) {}
590 else if (setLogY<TGraphErrors> (i->get())) {}
591 else if (setLogY<TGraph> (i->get())) {}
592 else if (setLogY<TLine> (i->get())) {}
593 else if (setLogY<TEllipse> (i->get())) {}
594 }
595 }
596 }
597 }
598
599 if (grid.count('x') || grid.count('X')) { gPad->SetGridx(); }
600 if (grid.count('y') || grid.count('Y')) { gPad->SetGridy(); }
601
602 if (stats != -1)
603 gStyle->SetOptStat(stats);
604 else
605 gStyle->SetOptFit(kFALSE);
606
607
608 for (vector<JRootObject>::iterator i = listOfObjects.begin(); i != listOfObjects.end(); ++i) {
609
610 if (i->get() != master) {
611
612 DEBUG("Draw " << (*i)->GetName() << ' ' << (*i)->GetTitle() << endl);
613
614 string buffer(option);
615
616 buffer += "SAME";
617
618 try {
619
620 TH2& h2 = dynamic_cast<TH2&>(*(*i));
621
622 h2.SetMinimum(zmin);
623 h2.SetMaximum(zmax);
624 }
625 catch(exception&) {}
626
627 try {
628
629 TGraph& g1 = dynamic_cast<TGraph&>(*(*i));
630
631 buffer = "P";
632 }
633 catch(exception&) {}
634
635 try {
636
637 TGraph2D& g2 = dynamic_cast<TGraph2D&>(*(*i));
638
639 g2.SetMinimum(zmin);
640 g2.SetMaximum(zmax);
641 }
642 catch(exception&) {}
643
644 try {
645
646 TF2& f2 = dynamic_cast<TF2&>(*(*i));
647
648 f2.SetNpx(1000);
649 f2.SetNpy(1000);
650 /*
651 f2.SetRange(xmin, ymin, zmin,
652 xmax, ymax, zmax);
653 */
654 }
655 catch(exception&) {}
656
657 (*i)->Draw(buffer.c_str());
658 }
659 }
660
661 //gPad->RedrawAxis();
662
663 cv->Update();
664
665 if (outputFile != "") {
666 cv->SaveAs(outputFile.c_str());
667 }
668
669 if (!batch) {
670 tp->Run();
671 }
672
673 return (master != NULL ? 0 : 1);
674}
outputFile
string outputFile
Definition JDAQTimesliceSelector.cc:37
DEBUG
#define DEBUG(A)
Message macros.
Definition JMessage.hh:62
ERROR
#define ERROR(A)
Definition JMessage.hh:66
FATAL
#define FATAL(A)
Definition JMessage.hh:67
debug
int debug
debug level
Definition JSirene.cc:69
make_field
#define make_field(A,...)
macro to convert parameter to JParserTemplateElement object
Definition JParser.hh:2142
g1
Double_t g1(const Double_t x)
Function.
Definition JQuantiles.cc:25
JEEP::JProperties
Utility class to parse parameter values.
Definition JProperties.hh:501
JGIZMO::JRootObject
Auxiliary data structure for TObject with a user defined label.
Definition JRootObject.hh:31
JPARSER::JCounter
Auxiliary class to handle multiple boolean-like I/O.
Definition JParser.hh:423
JPARSER::JParser
Utility class to parse command line options.
Definition JParser.hh:1698
JROOT::JCanvas
Data structure for size of TCanvas.
Definition JCanvas.hh:26
JROOT::JCanvas::y
int y
number of pixels in Y
Definition JCanvas.hh:99
JROOT::JCanvas::x
int x
number of pixels in X
Definition JCanvas.hh:98
JROOT::JStyle
Wrapper class around ROOT TStyle.
Definition JStyle.hh:24
JTOOLS::JRange
Range of values.
Definition JRange.hh:42
GAUSS_LEGENDRE::xmax
const double xmax
Definition JQuadrature.cc:24
GAUSS_LEGENDRE::xmin
const double xmin
Definition JQuadrature.cc:23
JDETECTOR::getLabel
std::string getLabel(const JLocation &location)
Get module label for monitoring and other applications.
Definition JLocation.hh:247
JGIZMO::setLogarithmicX
void setLogarithmicX(TList *list)
Make x-axis of objects in list logarithmic (e.g. after using log10()).
Definition JGizmoToolkit.hh:977
JGIZMO::setRange
void setRange(double &xmin, double &xmax, const bool logx)
Set axis range.
Definition JGizmoToolkit.hh:1467
JGIZMO::getDirectory
TDirectory * getDirectory(const JRootObjectID &id)
Get TDirectory pointer.
Definition JGizmoToolkit.hh:144
JGIZMO::setLogarithmicY
void setLogarithmicY(TList *list)
Make y-axis of objects in list logarithmic (e.g. after using log10()).
Definition JGizmoToolkit.hh:991
JGIZMO::isTObject
bool isTObject(const TKey *key)
Check if given key corresponds to a TObject.
Definition JGizmoToolkit.hh:1576
JPP
This name space includes all other name spaces (except KM3NETDAQ, KM3NET and ANTARES).
Definition JAAnetToolkit.hh:43
create_update_issues.project
project
Definition create_update_issues.py:21
JPARSER::initialised
Empty structure for specification of parser element that is initialised (i.e. does not require input)...
Definition JParser.hh:68
JROOT::JStyle::JParameters::getProperties
JProperties getProperties()
Get properties of this class.
Definition JStyle.hh:44
Generated by doxygen 1.12.0