#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <map>
#include <string>
#include <random>
#include <climits>
#include <ctime>

#include "TChain.h"
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TObjString.h"
#include "TSystem.h"
#include "TROOT.h"
#include "TKey.h"

#include "TMVA/Types.h"
#include "TMVA/TMVAGui.h"
//#include "TMVA/Factory.h"
//#include "TMVA/DataLoader.h"
//#include "TMVA/Tools.h"
//#include "TMVA/TMVAGui.h"
#include "TMVA/Config.h"
#include "TMVA/correlations.h"
#include "TMVA/mvas.h"
#include "TMVA/efficiencies.h"

using namespace std;
/*
int rename_mass_mode(char massName[7], char modeName[5], string defaultName, string suffix="eps") 
{
  // Output plot path prefix (for renaming)
  char i_Plot_prefix[100];
  sprintf(i_Plot_prefix, "/home/psh/localwork/tmva-qq/dataset/plots");
  char o_Plot_prefix[100];
  sprintf(o_Plot_prefix, "/home/psh/localwork/tmva-qq/plots/%s", modeName);

  char i_Plot[128];
  sprintf(i_Plot, "%s/%s.%s", i_Plot_prefix, defaultName.c_str(), suffix.c_str());
  char o_Plot[128];
  sprintf(o_Plot, "%s/%s-%s-%s.%s", o_Plot_prefix, defaultName.c_str(), massName, modeName, suffix.c_str());
  return rename(i_Plot, o_Plot); 
}
*/
void draw()
{
  TString treeName = "variables";

  TString fName = "tmva-test.root";
  TString dataset = "dataset";
  
  // Input file path for correlation plot drawing
  TString fName_corr = "tmva-test.root";

  // Draw correlation plots
  TMVA::correlations(dataset.Data(), fName_corr);
  //rename_mass_mode(massName, modeName, "CorrelationMatrixS");
  //rename_mass_mode(massName, modeName, "CorrelationMatrixB");

  // Draw Fisher output (test and training samples)
  TMVA::mvas(dataset.Data(), fName, TMVA::kCompareType);
  //rename_mass_mode(massName, modeName, "overtrain_Fisher");

  // Draw Fisher probability output (test samples)
  TMVA::mvas(dataset.Data(), fName, TMVA::kProbaType);
  //rename_mass_mode(massName, modeName, "proba_Fisher");

  // Draw ROC curve
  TMVA::efficiencies(dataset.Data(), fName);
  //rename_mass_mode(massName, modeName, "rejBvsS");

  return;
}

int main( int argc, char** argv )
{
  /*
  int mass, mode;
  if (argc != 3) return 1;
  mass = atoi(argv[1]);
  mode = atoi(argv[2]);

  draw(mass, mode);
  */

  if (argc != 1) return 1;
  draw();
  /*  draw( 400, 10);
  draw(1100, 10);
  draw(1200, 10);
  draw(1300, 10);
  draw(1400, 10);
  draw(1500, 10);
  draw(1600, 10);
  draw(1700, 10);
  draw(1800, 10);
  draw(1900, 10);
  draw(2000, 10);
  draw(2100, 10);
  draw(2200, 10);
  draw(2300, 10);
  draw(2400, 10);
  draw(2500, 10);
  draw(2600, 10);
  draw( 300, 11);
  draw( 400, 11);
  draw(1100, 11);
  draw(1200, 11);
  draw(1300, 11);
  draw(1400, 11);
  draw(1500, 11);
  draw(1600, 11);
  draw(1700, 11);
  draw(1800, 11);
  draw(1900, 11);
  draw(2000, 11);
  draw(2100, 11);
  draw(2200, 11);
  draw(2300, 11);
  draw(2400, 11);
  draw(2500, 11);
  draw(2600, 11);
  */
  return 0;
}