convertXMLToGBRForestROOT.cc

Go to the documentation of this file.

#include "CommonTools/MVAUtils/interface/GBRForestTools.h"

#include "TFile.h"

#include <boost/filesystem.hpp>
#include <iostream>

int main(int argc, char **argv) {
  if (argc != 3) {
    std::cout << "Please pass a (gzipped) BDT weight file and a name for the output ROOT file." << std::endl;
    return 1;
  }

  char *inputFileName = argv[1];
  char *outputFileName = argv[2];

  if (!boost::filesystem::exists(inputFileName)) {
    std::cout << "Input file " << inputFileName << " does not exists." << std::endl;
    return 1;
  }

  if (boost::filesystem::exists(outputFileName)) {
    std::cout << "Output file " << outputFileName << " already exists." << std::endl;
    return 1;
  }

  auto gbrForest = createGBRForest(inputFileName);
  std::cout << "Read GBRForest " << inputFileName << " successfully." << std::endl;

  TFile{outputFileName, "RECREATE"}.WriteObject(gbrForest.get(), "gbrForest");
  std::cout << "GBRForest written to " << outputFileName << " successfully." << std::endl;

  return 0;
}