JZebraMantis.cc File Reference

Program to compare histograms in root files that have same directory structure,
and where the histograms have the same names. More...

#include <iostream>
#include <fstream>
#include "TString.h"
#include "TRegexp.h"
#include "TObjArray.h"
#include "TObjString.h"
#include "TFile.h"
#include "TKey.h"
#include "Jeep/JParser.hh"
#include "JSupport/JMeta.hh"
#include "JCompareHistograms/JTest_t.hh"
#include "JCompareHistograms/JTestDictionary.hh"

Go to the source code of this file.

Functions
void	readDir (TDirectory *dir, vector< TString > &v)
int	main (int argc, char **argv)

Detailed Description

Program to compare histograms in root files that have same directory structure,
and where the histograms have the same names.

The input histograms and the test to be applied for each histogram are specified
in an ASCII formatted steering file which is passed by the command line.

Each row of the steering file should have multiple columns.

Column 1 is the name of the histogram to be compared (including the full path inside the root file)
Column 2 is an integer value that indicates the test to be performed. See JTestDictionary()
Columns 3..n are reserved for the different parameters of the test.

Author

rgruiz

Definition in file JZebraMantis.cc.

Function Documentation

void readDir ( TDirectory *  dir, vector< TString > &  v  )

inline

Definition at line 26 of file JZebraMantis.cc.

                                                       {

  TIter iter(dir->GetListOfKeys());
  
  for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {

    if (key->IsFolder()){

      dir->cd(key->GetName());
      TDirectory *subdir = gDirectory;
      readDir(subdir,v);
      dir->cd();
    }else{

      const TString fullPath(dir->GetPath());
      TObjArray* t = fullPath.Tokenize(":");
      v.push_back(TString(((TObjString *)(t->At(1)))->String() + "/" + (TString)key->GetName()));
    }
  }  
}

int main	(	int	argc,
		char **	argv
	)

Definition at line 63 of file JZebraMantis.cc.

                                { 

  string steeringFile;
  string file1;
  string file2;
  string output;
  string ascii;
  bool   onlyFailures;

  
  try {
    JParser<> zap("\nProgram to compare histograms in root files that have the same directory structure. See the link below this usage for further details.\n"); 
    zap['s'] = make_field(steeringFile , "ASCII steering file with list of histograms and tests");
    zap['a'] = make_field(file1        , "input file 1");
    zap['b'] = make_field(file2        , "input file 2");
    zap['o'] = make_field(output       , "output file root") = "zebramantis.root";
    zap['t'] = make_field(ascii        , "output file txt" ) = "zebramantis.txt";
    zap['w'] = make_field(onlyFailures , "write only failed tests" );
    zap(argc,argv);
  }
  catch(const exception &error) {
    ERROR(error.what() << endl);
  }

  TFile* f1 = TFile::Open(file1.c_str());
  TFile* f2 = TFile::Open(file2.c_str());
  
  TFile out(output.c_str(),"recreate");
  out.cd();

  ofstream results;
  results.open (ascii);

  vector<TString> keys;
  readDir(f1,keys);

  std::ifstream infile(steeringFile);

  JTestDictionary d;

  TString name;
  int     testID;
  string  rest;

  int npassed = 0;
  int nfailed = 0;

  for (vector<TString>::const_iterator key = keys.begin() ; key != keys.end() ; ++key) {
    
    string line;
    while (getline(infile, line)) {
      while (line.length()==0) 
            getline(infile, line);
      
      istringstream iss(line);
      iss >> name >> testID;
      
      const TRegexp regexp(name); 
     
      if (((*key).Index(regexp) != -1) && (f2->Get(*key))) {

        TObject* obj1 = (TObject*)f1->Get(*key);
        TObject* obj2 = (TObject*)f2->Get(*key);

        d[testID]->read(iss);
        d[testID]->test(obj1,obj2);
        
        for (vector<JTestResult>::const_iterator r = d[testID]->results.begin() ; r != d[testID]->results.end() ; ++r) {
          (r->passed ? npassed++ : nfailed++);
        }

        string Key  = MAKE_STRING(*key);        
        string path = MAKE_STRING(Key.substr (Key.find ('/') + 1 , Key.rfind ('/')));
        
        d[testID]->write(cout);
        d[testID]->write(results, ";", onlyFailures); 
        d[testID]->save (&out, path, onlyFailures);
        d[testID]->clear();
      }
    }
    infile.clear();
    infile.seekg(0, ios::beg);
  }

  results << WHITE << "PASSED: " << npassed << " " << " FAILED: " << nfailed << " FAILURE FRACTION: " << float (nfailed)/(nfailed+npassed) << endl;

  putObject(&out, JMeta(argc, argv));
  JMeta::copy(file1.c_str(), out);
  JMeta::copy(file2.c_str(), out);

  results.close();
  out    .Close();  
  return 0; 
}