JTestKolmogorov_2D.hh

Go to the documentation of this file.

#ifndef __JTESTKOLMOGOROV_2D__
#define __JTESTKOLMOGOROV_2D__

#include <iostream>
#include "TMath.h"
#include "JTest_t.hh"
#include "JTestKolmogorov_t.hh"

using namespace std;

/*
 * Kolmogorov test.
 */
class JTestKolmogorov_2D: 
  public JTestKolmogorov_t, 
  public JTest_t 
{ 
public:  
  /*
   * Default constructor.
   */    
  JTestKolmogorov_2D(){}
  
  /*
   * Read test parameters from input.
   *
   * \param  in            input stream
   * \return               input stream
   */
  std::istream& read(std::istream& in) { 
    return in >> threshold >> slice;
  };

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

  /*
   * Applies Kolmogorov test for two ROOT TH2 histograms.
   *
   * \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);
      
      if(slice == 'x' || slice == 'X'){
            
        int nSlices1 = h1->GetNbinsX();
        int nSlices2 = h2->GetNbinsX();

        if(nSlices1 != nSlices2)
          ERROR("Histograms with different binning. The objects: " << o1->GetName() << " can not be compared." << endl);

        for (int i=1 ; i<=nSlices1 ; ++i){

          string name = MAKE_STRING(o1->GetName() + to_string("_") + to_string(i));
             
          TH1D* s1 = h1->ProjectionY (name.c_str(),i,i);
          TH1D* s2 = h2->ProjectionY (name.c_str(),i,i);
          
          JTestResult r = JKolmogorovTest(s1, s2, threshold);
          results.push_back(r);
        }
      }else if (slice == 'y' || slice == 'Y'){
           
        int nSlices1 = h1->GetNbinsY();
        int nSlices2 = h2->GetNbinsY();

        if(nSlices1 != nSlices2)
          ERROR("Histograms with different binning. The objects: " << o1->GetName() << " can not be compared." << endl);

        for (int i=1 ; i<=nSlices1 ; ++i){
             
          string name = MAKE_STRING(o1->GetName() + to_string("_") + to_string(i));

          TH1D* s1 = h1->ProjectionX (name.c_str(),i,i);
          TH1D* s2 = h2->ProjectionX (name.c_str(),i,i);
          
          JTestResult r = JKolmogorovTest(s1, s2, threshold);
          results.push_back(r);
        }
      }else if (slice == 'n' || slice == 'N'){  
        
        JTestResult r = JKolmogorovTest2D(h1,h2,threshold);
      }

    }else if (!(dynamic_cast<TH1*>(o1) == NULL) && !(dynamic_cast<TH1*>(o2) == NULL)) {
      ERROR("For 1D histograms call JTestKolmogorov_1D: " << o1->GetName() << endl);
   }
  };

  /*
   * 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; //!< threshold p-value to decide if test is passed.
  char   slice;     //!< Axis to slice.
}; 
  
#endif