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

#include "TFile.h"
#include "TTree.h"

#include <filesystem>
#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 (!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;
}