convertXMLToGBRForestROOT.cc File Reference

#include "CommonTools/MVAUtils/interface/GBRForestTools.h"
#include "TFile.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 8 of file convertXMLToGBRForestROOT.cc.

References dir2webdir::argc, cmsBatch::argv, gather_cfg::cout, createGBRForest(), InefficientDoubleROC::inputFileName, and reco_skim_cfg_mod::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;
  }

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