JBinsMap.cc

Go to the documentation of this file.

 
#include <iostream>
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
#include "JROOT/JBinsMap.hh"
 
 
/**
 * \file
 *
 * Program to test methods of `JROOT::JBinsMap`.
 * \author bjung
 */
int main(int argc, char** argv)
{
  using namespace std;
  using namespace JPP;
 
  JBinsMap binsMap;
 
  string   outputFile;
 
  int      debug;
  
  try {
 
    JParser<> zap("Program to test methods of `JROOT::JBinsMap`");
 
    zap['@'] = make_field(binsMap);
    zap['o'] = make_field(outputFile);
    zap['d'] = make_field(debug);
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
 
  for (JBinsMap::const_iterator i = binsMap.cbegin(); i != binsMap.cend(); ++i) {
 
    unique_ptr<TH1> p;
    
    if        (i->second.find('z') != i->second.cend()) {
      p = binsMap.getHistogram<TH3D>(i->first);
    } else if (i->second.find('y') != i->second.cend()) {
      p = binsMap.getHistogram<TH2D>(i->first);
    } else if (i->second.find('x') != i->second.cend()) {
      p = binsMap.getHistogram<TH1D>(i->first);      
    }
 
    ASSERT(p.get() != NULL);
 
    const Int_t Ndim = p->GetDimension();
 
    if (Ndim > 0 && Ndim < 4) {
 
      const vector<double>& Xbins = i->second.at('x');
 
      ASSERT(p->GetXaxis()->GetNbins() == (Int_t) Xbins.size()-1);
 
      for (Int_t n = 1; n <= p->GetXaxis()->GetNbins(); ++n) {
        DEBUG (p->GetXaxis()->GetBinLowEdge(n) << " =?= " << Xbins[n-1] << endl);       
        ASSERT(p->GetXaxis()->GetBinLowEdge(n) == Xbins[n-1]);
      }
 
      if (Ndim > 1) {
 
        const vector<double>& Ybins = i->second.at('y');
 
        ASSERT(p->GetYaxis()->GetNbins() == (Int_t) Ybins.size()-1);
 
        for (Int_t n = 1; n <= p->GetYaxis()->GetNbins(); ++n) {
          DEBUG (p->GetYaxis()->GetBinLowEdge(n) << " =?= " << Ybins[n-1] << endl);
          ASSERT(p->GetYaxis()->GetBinLowEdge(n) == Ybins[n-1]);
        }
 
        if (Ndim > 2) {
 
          const vector<double>& Zbins = i->second.at('z');
 
          ASSERT(p->GetZaxis()->GetNbins() == (Int_t) Zbins.size()-1);
 
          for (Int_t n = 1; n <= p->GetZaxis()->GetNbins(); ++n) {
            DEBUG (p->GetZaxis()->GetBinLowEdge(n) << " =?= " << Zbins[n-1] << endl);
            ASSERT(p->GetZaxis()->GetBinLowEdge(n) == Zbins[n-1]);
          }
        }
      }
      
    } else {
      FATAL("Unexpected dimensionality for histogram " << p->GetName());      
    }
  }
 
  ofstream ofs(outputFile.c_str());
  
  ofs << binsMap;
 
  ofs.close();
  
  return 0;
}