JTestChi2R_2D.hh

Go to the documentation of this file.

#ifndef __JTESTCHI2R_2D__
#define __JTESTCHI2R_2D__

#include <istream>
#include <ostream>
#include "JCompareHistograms/JTest_t.hh"
#include "JCompareHistograms/JTestChi2_t.hh"

/**
 * \author rgruiz
 */

/**
 * Implementation of the Chi2 test for 2D histograms.\n
 * This class is derived from the abstract class JTest_t(). For a general description of the implementation of this and other tests derived from JTest_t(), see its documentation.\n
 * The parameter slice() can have the values x, X, y or Y. The histograms are sliced along the corresponding axis, and the JChi2Test() test is applied to each slice.\n 
 * The input parameter threshold(), is used to evaluate whether the test is passed or failed for each slice.\n
 * The evaluation is done by comparing the threshold() value with the result produced by JChi2Test(). The output of JChi2Test() is a p-value.\n 
 * The parameter threshold() should therefore be a real value between 0 and 1.
 */
class JTestChi2R_2D: 
  public JTestChi2_t, 
  public JTest_t 
{ 
public:
  
  /**
   * Default constructor.
   */  
  JTestChi2R_2D(){}
  
  /**
   * Read test parameters from input.
   *
   * \param  in            input stream
   * \return               input stream
   */
  std::istream& read(std::istream& in) override{ 
    return in >> threshold >> slice;
  };

  /**
   * Write test result to output.
   *
   * \param  out           output stream
   * \param  delimiter     field delimiter
   * \param  onlyFailures  If true, write only failures.
   * \return               output stream
   */
  std::ostream& write(std::ostream& out, std::string delimiter = " ", bool onlyFailures = false) const override{
    for (std::vector<JTestResult>::const_iterator r = results.begin() ; r != results.end() ; ++r) {
      if (onlyFailures){
        if (!r->passed)
          out << std::scientific << std::setprecision(2) << (r->passed ? JPP::GREEN : JPP::RED) << r->print(delimiter) << std::endl;
      }else{
        out << std::scientific << std::setprecision(2) << (r->passed ? JPP::GREEN : JPP::RED) << r->print(delimiter) << std::endl;
      }
    }
    return out;
  };

  /**
   * Applies Chi2 test for two ROOT TH2 histograms.
   *
   * \param o1 First histogram
   * \param o2 Second histogram
   */
  void test(TObject* o1, TObject* o2) override{    

    using namespace std;
    using namespace JPP;
    
    if (!(dynamic_cast<TH2*>(o1) == NULL) && !(dynamic_cast<TH2*>(o2) == NULL)) {
                  
      TH2D* h1 = dynamic_cast<TH2D*>(o1);
      TH2D* h2 = dynamic_cast<TH2D*>(o2);

      if(slice == 'x' || slice == 'X' || slice == 'y' || slice == 'Y'){
        
        JTestResult r = JChi2TestSlice(h1, h2, threshold, failuresThreshold, name, parameter, options, slice);

        results.push_back(r);
        
      } else {
        ERROR ("Slice option should be: x, X, y or Y, and not " + slice);
      }
      
    } else if (!(dynamic_cast<TH1*>(o1) == NULL) && !(dynamic_cast<TH1*>(o2) == NULL)) {
      ERROR("For 1D histograms call JChi2_1D: " << o1->GetName() << endl);
    }
  };

  /**
   * Writes the test result to root file
   * \param f A ROOT file 
   * \param path Path in root file.
   * \param onlyFailures  If true, write only failures.
   */
  virtual void save(TFile* f , std::string path, bool onlyFailures = false) override{
    
    using namespace std;
    
    for (vector<JTestResult>::const_iterator r = results.begin() ; r != results.end() ; ++r){
      
      if (onlyFailures){
        if (!r->passed){
          if (f -> GetDirectory(path.c_str())==0) f->mkdir(path.c_str());
          f->cd(path.c_str());
          r->obj->Write();
        }
      }else{
        if (f -> GetDirectory(path.c_str())==0) f->mkdir(path.c_str());
        f->cd(path.c_str());
        r->obj->Write();
      }
    }      
  }

  /**
   * Clear results
   */
  void clear() override{
    results.clear();
  }
  
  /**
   * Get test name.
   */
  std::string getName() override{
    return name;
  }

  /**
   * Get parameter name.
   */
  std::string getParameter() override{
    return parameter;
  }

private:
  char   slice;                             //!< axis to slice.
  double threshold;                         //!< threshold p-value to decide if test is passed.
  double failuresThreshold;                 //!< threshold p-value to decide if test is passed.
  const std::string name = "Chi2R_2D";      //!< test name.
  const std::string options = "CHI2/NDF";   //!< options for the ROOT chi2 test.
  const std::string parameter = "chi2/NDF"; //!< parameter name.
}; 
  
#endif