#include "hto2taus.h"
#include "TFile.h"
#include "TROOT.h"
#include "TTree.h"
#include <iostream>
#include <string>
#include <vector>
#include <TKey.h>
#include "TCanvas.h"

using namespace std;

int main ( int argc, char * argv[]) {

    // root file
    vector<string> vinput {
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    "../flattuple/.root",
    };  

    // directories
    vector<string>vtree{"hto2taus"};

    // loop over input files
    for (auto &input : vinput){ 	

      cout << "Processing file: " << input << endl;
      TFile* ifile = new TFile(input.c_str(),"READ");
      if (!ifile->IsOpen()) { cout<<"Can't open "<<input<<endl; return 1; }

      string output {input};
      //output.replace(0, output.find_first_of("_"), "hist");
      output.replace(0, output.find_last_of("/")+1, "hist_");
      cout << "output file: " << output << endl;

      TFile* ofile = new TFile(output.c_str(),"RECREATE");
      if (!ofile->IsOpen()) { cout<<"Can't create "<<output<<endl; return 1; }

      // fill histogram
      cout<<"filling hist...\n";
      for(auto &dir : vtree ){

	string treeName="demo/"+dir;
	cout<<"tree name: "<<treeName<<endl;

	TTree *tree = (TTree*) ifile->GetObjectChecked(treeName.c_str(), "TTree");

	// analysis
	string sample{output};
	sample.replace(0, sample.find_first_of("_")+1, "");
	cout<<"sample: "<<sample<<endl;

	//string smass{sample};
	//smass.replace(0, smass.find_first_of("M")+1, "");
	//smass.replace(smass.find_first_of("_"), smass.find_last_of("t"),"");
	//cout<<"string mass: "<<smass<<endl;

	//double mass{stod(smass)};
	//cout<<"double mass: "<<mass<<endl;

	//svfitmem an(tree, dir, sample, mass);
	svfitmem an(tree, sample);

	an.Loop();
	cout<<"end loop!!"<<endl;

	delete tree;
      } // dir
      ifile->Close();
      ofile->Close();

    } // input file

    return 0;
}