test_TPadManipulator.cxx

Go to the documentation of this file.

// test_TPadManipulator.cxx

// David Adams
// September 2017
//
// This is a test and demonstration for TPadManipulator.

#undef NDEBUG

#include "../TPadManipulator.h"
#include <string>
#include <iostream>
#include <iomanip>
#include <cassert>
#include "TH1F.h"
#include "TGraph.h"
#include "TTimeStamp.h"
#include "TCanvas.h"
#include "TROOT.h"
#include "TFile.h"

using std::string;
using std::cout;
using std::endl;
using std::setw;

//**********************************************************************

int test_TPadManipulator() {
  const string myname = "test_TPadManipulator: ";
  cout << myname << "Starting test" << endl;
#ifdef NDEBUG
  cout << myname << "NDEBUG must be off." << endl;
  abort();
#endif
  string line = "-----------------------------";
  string scfg;

  cout << myname << line << endl;
  cout << myname << "Test data" << endl;

  cout << myname << line << endl;
  cout << myname << "Create histogram." << endl;
  double xmax = 10000;
  TH1* ph = new TH1F("h1", "My histo; X values; Y value [units]", 100, 0, xmax);
  ph->SetStats(0);
  ph->SetFillColor(2);
  for ( unsigned int ibin=1; ibin<=100; ++ ibin ) {
    ph->SetBinContent(ibin, ibin-1);
  }

  cout << myname << line << endl;
  cout << myname << "Create manipulator" << endl;
  TPadManipulator* pmantop = new TPadManipulator(700, 1000, 1, 2);

  TPadManipulator* pman = pmantop->man(0);
  TPadManipulator& man = *pman;
  assert( man.pad() == nullptr );
  assert( man.hist() == nullptr );

  cout << myname << line << endl;
  cout << myname << "Add histogram." << endl;
  man.add(ph, "H");
  {
    TLatex lin(0.15, 0.85, "Test of TPadManipulator");
    lin.SetNDC();
    man.add(&lin);
  }
  {
    TLatex lin(0.15, 0.80, "Histogram");
    lin.SetNDC();
    man.add(&lin);
  }
  assert( man.hist() != nullptr );

  cout << myname << line << endl;
  cout << myname << "Add axis." << endl;
  man.addAxis();

  cout << myname << line << endl;
  cout << myname << "Add vertical mod lines." << endl;
  man.addVerticalModLines(0.2*xmax, 0.1*xmax);

  cout << myname << line << endl;
  cout << myname << "Add sloped lines." << endl;
  man.addSlopedLine(0.01, 5, 2);
  man.addSlopedLine(0.01, -5, 3);

  cout << myname << line << endl;
  cout << myname << "Add label." << endl;
  assert( ! man.getLabel().size() );
  man.setLabel("This is my #beta data for E_{tot} = m#timesC^{2}.");
  assert( man.getLabel().size() );

  cout << myname << line << endl;
  cout << myname << "Copy the pad and draw copy with time format." << endl;
  *pmantop->man(1) = *pmantop->man(0);
  pmantop->man(1)->setRangeX(0, 8000);
  pmantop->man(1)->setTitle("Zoom of above");
  pmantop->man(1)->setRangeX(2000, 8000);
  pmantop->man(1)->setRangeY(10, 90);
  pmantop->man(1)->setTimeFormatX("%H:%M");
  {
    TLatex lin(0.15, 0.75, "Zoomed");
    lin.SetNDC();
    pmantop->man(1)->add(&lin);
  }
  {
    TLatex lin(0.15, 0.70, "Time format");
    lin.SetNDC();
    pmantop->man(1)->add(&lin);
  }

  cout << myname << line << endl;
  cout << myname << "Draw." << endl;
  assert( man.pad() == nullptr );
  pmantop->draw();
  assert( man.pad() != nullptr );

  cout << myname << line << endl;
  cout << myname << "Draw." << endl;
  pmantop->print("test_TPadManipulator.png");

  cout << myname << line << endl;
  cout << myname << "Copy the full plot." << endl;
  TPadManipulator man2(*pmantop);
  cout << myname << "Old subpad: " << pmantop->man(0) << endl;
  cout << myname << "Old canvas: " << pmantop << endl;
  cout << myname << "Old subpad parent: " << pmantop->man(0)->parent() << endl;
  cout << myname << "New subpad: " << man2.man(0) << endl;
  cout << myname << "New canvas: " << &man2 << endl;
  cout << myname << "New subpad parent: " << man2.man(0)->parent() << endl;

  cout << myname << line << endl;
  cout << myname << "Change title and y-label size on copy." << endl;
  cout << myname << "Old canvas label size: " << man2.getLabelSizeY() << endl;
  cout << myname << "Old pad label size: " << man2.man(0)->getLabelSizeY() << endl;
  man2.setLabelSizeY(0.02);
  man2.setTitleSize(0.025);
  cout << myname << "New canvas label size: " << man2.getLabelSizeY() << endl;
  // Next is zero before pad is drawn.
  cout << myname << "New pad label size: " << man2.man(0)->getLabelSizeY() << endl;

  cout << myname << line << endl;
  cout << myname << "Draw copy." << endl;
  man2.print("test_TPadManipulator-copy.{png,pdf,tpad}");
  cout << myname << "New pad label size: " << man2.man(0)->getLabelSizeY() << endl;


  cout << myname << line << endl;
  cout << myname << "Create time graph." << endl;
  TGraph* pg = new TGraph;
  pg->SetTitle("Clock");
  pg->GetXaxis()->SetTitle("March 6 daytime minutes");
  pg->GetYaxis()->SetTitle("Minute");
  pg->SetMarkerStyle(2);
  srand(12345);
  TTimeStamp ts0(2019, 3, 6, 6, 0, 0);
  for ( int i=0; i<12*60; ++i ) {
    double tsec = ts0 + 60*i;
    double min = fmod(i, 60);
    pg->SetPoint(i, tsec, min);
  }
  TPadManipulator mant(1400, 500);
  mant.add(pg, "P");
  mant.addAxis();
  double t0 = ts0.AsDouble();
  mant.setRangeX(t0 - 7200, t0+14*3600);
  mant.setRangeY(0, 61);
  mant.setTimeFormatX("%H:%M%F2019-06-06 00:00:00");
  mant.print("test_TPadManipulator-time.png");

  cout << myname << line << endl;
  string rfnam = "test_TPadManipulator.root";
  cout << myname << "Write pad to " << rfnam << endl;
  //assert( pmantop->write(rfnam) == 0 );
  assert( pmantop->print(rfnam) == 0 );

  cout << myname << line << endl;
  cout << myname << "Read pad from " << rfnam << endl;
  TPadManipulator* pmani = TPadManipulator::read(rfnam);
  assert( pmani != nullptr );
  pmani->man(1)->setTitle("My histo after read");
  {
    TLatex lin(0.01, 0.97, "Read from file");
    lin.SetNDC();
    pmani->add(&lin);
  }
  pmani->print("test_TPadManipulator-read.png");

  cout << myname << line << endl;
  cout << myname << "Root canvas count: " << gROOT->GetListOfCanvases()->GetEntries() << endl;
  cout << myname << "Deleting manipulator." << endl;
  delete pmantop;
  cout << myname << "Root canvas count: " << gROOT->GetListOfCanvases()->GetEntries() << endl;

  cout << myname << line << endl;
  cout << myname << "Done." << endl;
  return 0;
}

//**********************************************************************

int main() {
  return test_TPadManipulator();
}

//**********************************************************************