AutoHistoCompare.cpp

Go to the documentation of this file.

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// A class for automatically detect objects in root files and automatically match them between
// reference files and new sample root files,finally perform comparison by using Ks Test or
// Chi2-test.
 
// input : package name
//         reference histogram file
//         new sample histogram file
//
// output:Pv distribution
//        comparison plots
//
// Created by Xingtao Huang <huangxt@sdu.edu.cn>
// Date: 2007-07-27
// Modified By Xingtao Huang
// Date: 2007-08-20
// Re-Design: combination of root macros inside this class in order to  support different root
// files. Data: 2008-07-30
//
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
#include <iostream>
#include <sstream>
 
#include <TCanvas.h>
#include <TKey.h>
#include <TPaveStats.h>
 
#include "AutoHistoCompare.h"
 
AutoHistoCompare::AutoHistoCompare( TH1F* h1, const char* oldfn, const char* newfn, int method, bool norm )
    : method( method )
    , norm( norm )
    , PvHisto( h1 )
    , oldlistfolder( 0 )
    , newlistfolder( 0 )
    , olddir( 0 )
    , newdir( 0 ) {
 
  oldfile = new TFile( oldfn );
  oldlist = oldfile->GetListOfKeys();
 
  newfile = new TFile( newfn );
  newlist = newfile->GetListOfKeys();
 
  te = new TText();
  te->SetTextSize( 0.1 );
 
  leg = new TLegend( 0.6, 0.83, 0.75, 0.90 );
  // std::cout << "AutoHistoCompare initialized." << std::endl;
}
 
AutoHistoCompare::~AutoHistoCompare() {
  delete oldfile;
  delete newfile;
  delete te;
  delete leg;
}
 
void AutoHistoCompare::DoCompare() {
  TIter    key_iter( oldlist );
  TKey*    key;
  TObject* obj;
  while ( ( key = (TKey*)key_iter() ) )
  {
    obj = key->ReadObj();
    if ( obj->InheritsFrom( "TH1" ) ) { MatchH1( obj ); }
    if ( key->IsFolder() )
    {
      olddir        = oldfile->GetDirectory( obj->GetName() );
      oldlistfolder = olddir->GetListOfKeys();
      newdir        = newfile->GetDirectory( obj->GetName() );
      newlistfolder = newdir->GetListOfKeys();
      FolderMatch();
    }
  }
}
 
void AutoHistoCompare::FolderMatch() {
  TIter    next( oldlistfolder );
  TKey*    key;
  TObject* obj;
  while ( ( key = (TKey*)next() ) )
  {
    obj = key->ReadObj();
    if ( obj->InheritsFrom( "TH1" ) ) { FolderMatchH1( obj ); }
    if ( key->IsFolder() ) { std::cout << " It's a 3rd-level Folder!" << std::endl; }
  }
  return;
}
 
void AutoHistoCompare::MatchH1( TObject* obj ) {
  char hname[50];
  sprintf( hname, "%s%s", obj->GetName(), ";1" );
  oldfile->GetObject( hname, myoldHisto1 );
  if ( !myoldHisto1 )
  {
    std::cout << "==> Failed when converting to TH1." << std::endl;
    exit( 1 );
  }
  newfile->GetObject( hname, mynewHisto1 );
  if ( !mynewHisto1 )
  {
    std::cout << "==> Failed when converting to TH1." << std::endl;
    exit( 1 );
  }
  if ( mynewHisto1->GetEntries() != 0 && myoldHisto1->GetEntries() != 0 ) { ComparisonH1(); }
}
 
void AutoHistoCompare::FolderMatchH1( TObject* obj ) {
 
  olddir->GetObject( obj->GetName(), myoldHisto1 );
  if ( !myoldHisto1 )
  {
    std::cout << "==> Failed when converting to TH1." << std::endl;
    exit( 1 );
  }
  newdir->GetObject( obj->GetName(), mynewHisto1 );
  if ( !mynewHisto1 )
  {
    std::cout << "==> Failed when converting to TH1." << std::endl;
    exit( 1 );
  }
  if ( mynewHisto1->GetEntries() != 0 && myoldHisto1->GetEntries() != 0 ) { ComparisonH1(); }
  return;
}
 
void AutoHistoCompare::ComparisonH1() {
 
  char    title[50];
  TCanvas c1;
 
  myoldHisto1->SetLineColor( 2 );
  mynewHisto1->SetLineColor( 4 );
  myoldHisto1->SetLineStyle( 3 );
  mynewHisto1->SetLineStyle( 5 );
  myoldHisto1->Draw();
  if ( norm )
  {
    mynewHisto1->Sumw2();
    mynewHisto1->Scale( myoldHisto1->Integral() / mynewHisto1->Integral() );
  }
  mynewHisto1->Draw( "Sames" );
  c1.Update();
  // statistics display
  TPaveStats* p1 = (TPaveStats*)myoldHisto1->GetListOfFunctions()->FindObject( "stats" );
  myoldHisto1->GetListOfFunctions()->Remove( p1 );
  myoldHisto1->SetStats( 0 );
  p1->SetLineColor( 2 );
  p1->SetX1NDC( 0.76 );
  p1->SetX2NDC( 0.95 );
  p1->SetY1NDC( 0.8 );
  p1->SetY2NDC( 0.95 );
  p1->Draw();
 
  TPaveStats* p2 = (TPaveStats*)mynewHisto1->GetListOfFunctions()->FindObject( "stats" );
  mynewHisto1->GetListOfFunctions()->Remove( p2 );
  mynewHisto1->SetStats( 0 );
  p2->SetLineColor( 4 );
  p2->SetX1NDC( 0.76 );
  p2->SetX2NDC( 0.95 );
  p2->SetY1NDC( 0.65 );
  p2->SetY2NDC( 0.8 );
  p2->Draw();
 
  leg->Clear();
  leg->AddEntry( myoldHisto1, "Reference", "l" );
  leg->AddEntry( mynewHisto1, "New ", "l" );
  leg->Draw();
 
  Double_t res;
  mypv = 0.000001;
  if ( method == 1 ) { mypv = myoldHisto1->Chi2Test( mynewHisto1, "UU", &res ); }
  else { mypv = myoldHisto1->KolmogorovTest( mynewHisto1, "UO" ); }
  if ( mypv < 0.01 ) { sprintf( title, "%s%s%s", "LT_", myoldHisto1->GetName(), ".eps" ); }
  else
  {
    //     sprintf(title,"%s%s", olddir->GetName(), ".eps");
    sprintf( title, "%s%s", myoldHisto1->GetName(), ".eps" );
  }
  //    c1.Print(title);
  PvHisto->Fill( mypv );
 
  std::stringstream buf;
  std::string       value;
  buf << "PV=" << mypv << std::endl;
  buf >> value;
 
  te->DrawTextNDC( 0.2, 0.2, value.c_str() );
  c1.Print( title );
  std::cout << "[OVAL] " << myoldHisto1->GetName() << " PV = " << mypv << std::endl;
 
  return;
}
 
void AutoHistoCompare::ComparisonH2() {
 
  mypv = 0.000001;
  if ( method == 1 )
  {
    mypv = myoldHisto2->Chi2Test( mynewHisto2, "OU" ); // ,res);
  }
  else { mypv = myoldHisto2->KolmogorovTest( mynewHisto2, "UO" ); }
 
  PvHisto->Fill( mypv );
 
  std::stringstream buf;
  std::string       value;
  buf << "PV=" << mypv << std::endl;
  buf >> value;
 
  te->DrawTextNDC( 0.43, 0.82, value.c_str() );
 
  std::cout << "[OVAL] " << myoldHisto2->GetName() << " PV = " << mypv << std::endl;
  return;
}
 
void AutoHistoCompare::ComparisonProfile() {
 
  mypv = 0.000001;
  if ( method == 1 )
  {
    mypv = myoldProfile->Chi2Test( mynewProfile, "OU" ); //,res);
  }
  else { mypv = myoldProfile->KolmogorovTest( mynewProfile, "UO" ); }
 
  PvHisto->Fill( mypv );
 
  std::stringstream buf;
  std::string       value;
  buf << "PV=" << mypv << std::endl;
  buf >> value;
 
  te->DrawTextNDC( 0.43, 0.82, value.c_str() );
 
  std::cout << "[OVAL] " << myoldProfile->GetName() << " PV = " << mypv << std::endl;
  return;
}