JTestRuns_1D.hh

Go to the documentation of this file.

#ifndef __JTESTRUNS_1D__
#define __JTESTRUNS_1D__

#include <iostream>
#include "JTest_t.hh"
#include "JTestRuns_t.hh"

using namespace std;

/*
 * Interface to read input and write output for TObject tests.
 */
class JTestRuns_1D: 
  public JTest_t, 
  public JTestRuns_t
{ 
public:  
  
  JTestRuns_1D(){}

  /*
   * Read test parameters from input.
   *
   * \param  in            input stream
   * \return               input stream
   */
  std::istream& read(std::istream& in) { 
    return in >> threshold >> logX >> logY;
  };

  /*
   * Write test result to output.
   *
   * \param  out           output stream
   * \return               output stream
   */
  std::ostream& write(std::ostream& out) {
    for (vector<JTestResult>::const_iterator r = results.begin() ; r != results.end() ; ++r) {
      out << scientific << setprecision(2) << (r->passed ? GREEN : RED) << r->message << endl;
    }
    return out;
  };

  /*
   * Tests the statistical compatibility of two ROOT TObjects
   * 
   * \param o1 First object
   * \param o2 Second object
   */
  void test(TObject* o1, TObject* o2) {    
    
    if (!(dynamic_cast<TH2*>(o1) == NULL) || !(dynamic_cast<TH2*>(o2) == NULL)) {
      ERROR("For 2D histograms call JTestRuns_2D: " << o1->GetName() << endl);
    
    } else if (!(dynamic_cast<TH1*>(o1) == NULL) && !(dynamic_cast<TH1*>(o2) == NULL)) {
      
      TH1D* h1 =  dynamic_cast<TH1D*>(o1);
      TH1D* h2 =  dynamic_cast<TH1D*>(o2);
  
      if(h1 -> GetNbinsX() != h2 -> GetNbinsX())
        ERROR("Histograms with different bining. The objects: " << h1 -> GetName() << " can not be compared." << endl);

      JTestResult r = JRunsTest(h1, h2, threshold, logX, logY);
     
      results.push_back(r);
    }
  };

  /*
   * Writes the result to root file
   * \param f The file 
   * \param path Path in root file.
   */
  virtual void save(TFile* f , string path){
    
    for (vector<JTestResult>::const_iterator r = results.begin() ; r != results.end() ; ++r){
      
      if (f -> GetDirectory(path.c_str())==0) f->mkdir(path.c_str());
      f->cd(path.c_str());
      r->obj->Write();
    }
  }

  /*
   * Clear results
   */
  void clear(){
    results.clear();
  }

private:
  double threshold;
  bool   logX;
  bool   logY;
}; 
  
#endif