JZebraMantis.cc

Go to the documentation of this file.

#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"

using namespace JPP;
using namespace std;


/*
 * Gets list of keys in a ROOT TDirectory and stores it on a vector.
 *
 * \param dir The ROOT directory
 * \param v Vector to store the list of keys.  
 */
inline void readDir(TDirectory* dir,vector<TString> &v){

  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()));
    }
  }  
}

/**
 * \file
 *
 * Program to compare histograms in root files that have same directory structure,\n
 * and where the histograms have the same names.
 * The input histograms and the test to be applied for each histogram are specified\n
 * in an ASCII formatted steering file which is passed by the command line.\n\n
 * 
 * Each row of the steering file should have multiple columns.\n\n
 * 
 * Column  1 is the name of the histogram to be compared (including the full path inside the root file)\n
 * Column  2 is an integer value that indicates the test to be performed. See JTestDictionary()\n
 * Columns 3..n are reserved for the different parameters of the test.\n\n
 *
 * \author rgruiz
 */
int main(int argc, char** argv) { 

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