comparator.cc

Go to the documentation of this file.

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "Compare.hh"
 
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
 
#include "TFile.h"
#include "TH1.h"
#include "TTree.h"
 
class Result;
 
void usage();
bool FileExists(const std::string& fileName);
 
enum EXIT_CODE {
  _EXIT_SUCCESS        = 0,
  _EXIT_FAILED         = 1,
  _EXIT_INCORRECT_ARGS = 2,
  _EXIT_FILE_NOT_FOUND = 3,
  _EXIT_BAD_FILE       = 4,
  _EXIT_SUCCESS_NONE   = 5 // all 0 tests passed i.e. there were no tests
};
 
int main(int argc, char* argv[])
{
  if (argc != 3)
    { 
      usage();
      return EXIT_CODE::_EXIT_INCORRECT_ARGS;    
    }
 
  std::string fname1 = std::string(argv[1]);
  std::string fname2 = std::string(argv[2]);
  // try to open files - check validity
  if (!FileExists(fname1))
    {
      std::cout << "No such file \"" << fname1 << "\"" << std::endl;
      return EXIT_CODE::_EXIT_BAD_FILE;
    }
  TFile* f1 = new TFile(fname1.c_str());
  if(f1->IsZombie())
    {
      std::cout << "error : could not open " << fname1 << std::endl;
      return EXIT_CODE::_EXIT_FILE_NOT_FOUND;
    }
 
  if (!FileExists(fname2))
    {
      std::cout << "No such file \"" << fname2 << "\"" << std::endl;
      return EXIT_CODE::_EXIT_BAD_FILE;
    }
  TFile* f2 = new TFile(fname2.c_str());
  if (f2->IsZombie())
    {
      std::cout << "error : could not open " << fname2 << std::endl;
      return EXIT_CODE::_EXIT_FILE_NOT_FOUND;
    }
 
  std::vector<Result*> results = Compare::Files(f1,f2);
 
  f1->Close();
  delete f1;
  f2->Close();
  delete f2;
  
  bool allPassed = Compare::Summarise(results);
  for (auto r : results)
    {delete r;}
  if (!allPassed)
    {
      std::cout << "TESTS_FAILED" << std::endl; // key to look for
      return EXIT_CODE::_EXIT_FAILED;
    }
  else
    {
      if (results.empty())
        {
          std::cout << "No tests" << std::endl;
          return EXIT_CODE::_EXIT_SUCCESS_NONE;
        }
      else
        {return EXIT_CODE::_EXIT_SUCCESS;}
    }
}
 
void usage()
{ 
  std::cout << "Usage: comparator <rootfile1> <rootfile2>" << std::endl;
  std::cout << "Compares <rootfile2> to <rootfile1> - ie <rootfile1> is the reference." << std::endl;
}
 
bool FileExists(const std::string& fileName)
{
  std::ifstream infile(fileName.c_str());
  return infile.good();
  // note the destructor of ifstream will close the stream
}