JTestRange1D.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <map>
 
#include "TROOT.h"
#include "TFile.h"
#include "TKey.h"
#include "TString.h"
#include "TRegexp.h"
#include "TH1.h"
#include "TProfile.h"
#include "TGraph.h"
 
#include "JTools/JRange.hh"
#include "JGizmo/JRootObjectID.hh"
#include "JGizmo/JGizmoToolkit.hh"
 
#include "Jeep/JColor.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 * Auxiliary program to test contents of of 1D histograms.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  typedef JRange<Double_t>        JRange_t;
  typedef map<TString, JRange_t>  map_type;
 
  vector<JRootObjectID>  inputFile;
  JRange_t               X;
  JRange_t               Y;
  bool                   invertX;
  bool                   invertY;
  int                    numberOfOutliers;
  map_type               zmap;
  int                    debug;
 
  try {
 
    JParser<> zap("Auxiliary program to test contents of 1D histograms.");
 
    zap['f'] = make_field(inputFile,  "measurement histogram, e.g: <file name>:<object name>");
    zap['x'] = make_field(X,          "accepted x-range values")   = JRange_t();
    zap['y'] = make_field(Y,          "accepted y-range values")   = JRange_t();
    zap['X'] = make_field(invertX);
    zap['Y'] = make_field(invertY);
    zap['N'] = make_field(numberOfOutliers)                        = 0;
    zap['H'] = make_field(zmap,       "global tests")              = JPARSER::initialised();
    zap['d'] = make_field(debug)                                   = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  int number_of_failures = 0;
 
  for (vector<JRootObjectID>::const_iterator input = inputFile.begin(); input != inputFile.end(); ++input) {
 
    DEBUG("Input: " << *input << endl);
 
    TDirectory* dir = getDirectory(*input);
 
    if (dir == NULL) {
      FATAL("File: " << input->getFullFilename() << " not opened." << endl);
    }
 
    const TRegexp regexp(input->getObjectName());
 
    TIter iter(dir->GetListOfKeys());
 
    for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {
 
      const TString tag(key->GetName());
 
      DEBUG("Key: " << tag << " match = " << tag.Contains(regexp) << endl);
        
      // option match
 
      if (tag.Contains(regexp) && isTObject(key)) {
 
        TObject* p = key->ReadObj();
 
        const TH1*    h1 = ( dynamic_cast<TProfile*>(p) != NULL ? dynamic_cast<TProfile*>(p) :
                            (dynamic_cast<TH1*>(p)      != NULL ? dynamic_cast<TH1*>(p)      : NULL) );
        const TGraph* g1 = ( dynamic_cast<TGraph*>(p)   != NULL ? dynamic_cast<TGraph*>(p)   : NULL );
 
        for (map_type::const_iterator i = zmap.begin(); i != zmap.end(); ++i) {
 
          const double    value = getResult(i->first, p);
          const JRange_t& range = i->second;
 
          DEBUG("Global test " << i->first << ' ' << (range(value) ? "passed" : "failed") << endl);
 
          ASSERT(range(value));
        }       
 
 
        int number_of_events   = 0;
        int number_of_outliers = 0;
 
        if        (h1 != NULL) {
 
          for (Int_t i = 1; i <= h1->GetNbinsX(); ++i) {
 
            const Double_t x = h1->GetBinCenter (i);
            const Double_t y = h1->GetBinContent(i);
          
            if (X(x) == !invertX) {
 
              ++number_of_events;
 
              const bool ok = (Y(y) == !invertY);
 
              DEBUG("Test outlier " << h1->GetName() << " bin (" << i << ") (" << h1->GetXaxis()->GetBinLabel(i) << ") " << y << ' ' << (ok ? "passed" : "failed") << endl);
            
              if (!ok) {
                ++number_of_outliers;
              }
            }
          }
 
        } else if (g1 != NULL) {
 
          for (Int_t i = 0; i != g1->GetN(); ++i) {
 
            const Double_t x = g1->GetX()[i];
            const Double_t y = g1->GetY()[i];
 
            if (X(x) == !invertX) {
 
              ++number_of_events;
 
              const bool ok = (Y(y) == !invertY);
 
              DEBUG("Test outlier " << g1->GetName() << " bin (" << i << ") " << y << ' ' << (ok ? "passed" : "failed") << endl);
            
              if (!ok) {
                ++number_of_outliers;
              }
            }
          }
 
        } else {
 
          FATAL("Object at " << *input << " is not TH1 nor TGraph." << endl);
        }
 
 
        cout << (number_of_outliers > numberOfOutliers ? RED : GREEN);
        NOTICE("Number of outliers \"" << p->GetName() << "\" = " << number_of_outliers << "/" << number_of_events << endl);
        cout << RESET;
 
        if (number_of_outliers > numberOfOutliers) {
          ++number_of_failures;
        }
      }
    }
  }
 
  ASSERT(number_of_failures == 0);
 
  return 0;
}