convertXMLToGBRForestROOT.cc File Reference

#include "CommonTools/MVAUtils/interface/GBRForestTools.h"
#include "TFile.h"
#include "TTree.h"
#include <filesystem>
#include <iostream>

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 9 of file convertXMLToGBRForestROOT.cc.

References dir2webdir::argc, GCPpyPlots::argv, gather_cfg::cout, createGBRForest(), f, InefficientDoubleROC::inputFileName, and makeListRunsInFiles::outputFileName.

                                {
  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 (!std::filesystem::exists(inputFileName)) {
    std::cout << "Input file " << inputFileName << " does not exists." << std::endl;
    return 1;
  }

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

  std::vector<std::string> variableNames;
  auto gbrForest = createGBRForest(inputFileName, variableNames);
  std::cout << "Read GBRForest " << inputFileName << " successfully." << std::endl;

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

  return 0;
}