JTestZero.hh

Go to the documentation of this file.

#ifndef __JTESTZERO__
#define __JTESTZERO__

#include <iostream>

#include <TROOT.h>
#include <TObject.h>
#include "TGraph.h"
#include "TCanvas.h"
#include "TLine.h"
#include "TMath.h"
#include "JTest_t.hh"
#include "JTools/JQuantile.hh"
#include "JTools/JRange.hh"
#include "JTestZero_t.hh"

using namespace std;

/*
 * Bin-by-Bin test for low content histograms.
 */
class JTestZero: 
  public JTest_t,
  public JTestZero_t
{ 
public:  

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

  /*
   * 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;
  };

  /*
   * Bin-by-bin comparison for ROOT TH2 histograms, of compatibility with a Poisson pdf of parameter 1. 
   * 
   * \param o1 First histogram
   * \param o2 Second histogram
   */
  void test(TObject* o1, TObject* o2) {    
    
    if (!(dynamic_cast<TH2*>(o1) == NULL) && !(dynamic_cast<TH2*>(o2) == NULL)) {
                  
      TH2D* h1 = dynamic_cast<TH2D*>(o1);
      TH2D* h2 = dynamic_cast<TH2D*>(o2);
      
      JTestResult r = JTestZero_2D(h1, h2, outliersThreshold, threshold);
      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 outliersThreshold; //!< Fraction of bins allowed to fail.
  double threshold;         //!< threshold p-value to decide if test is passed.
}; 
  
#endif