#include "analysis.h"

#include <iostream>
#include <vector>

using namespace std;

void runAnalysis(vector<TString> files, TString InputTag, TString fileName, bool update);





int main() {

  cout << "In main" << endl;

  TString output = "test_histos.root";

  // -----------------------------------------------------------------
  //
  // Electron DataSets
  // 
  vector<TString> datafiles_e;
  datafiles_e.push_back("../ElisFiles_e/VisMassCycle.data.datG.root");
  datafiles_e.push_back("../ElisFiles_e/VisMassCycle.data.datH.root");
  datafiles_e.push_back("../ElisFiles_e/VisMassCycle.data.datI.root");
  //
  vector<TString> wenu_e;
  wenu_e.push_back("../ElisFiles_e/VisMassCycle.mc.wenu.root");
  //
  vector<TString> zee_e;
  zee_e.push_back("../ElisFiles_e/VisMassCycle.mc.zee.root");
  //
  vector<TString> wtaunu_e;
  wtaunu_e.push_back("../ElisFiles_e/VisMassCycle.mc.wtaunu.root");
  //
  vector<TString> ztt_e;
  ztt_e.push_back("../ElisFiles_e/VisMassCycle.mc.ztt.root");
  // -----------------------------------------------------------------
  //
  // Muon Datasets
  //
  vector<TString> datafiles_mu;
  datafiles_mu.push_back("../ElisFiles_mu/VisMassCycle.data.datG.root");
  datafiles_mu.push_back("../ElisFiles_mu/VisMassCycle.data.datH.root");
  datafiles_mu.push_back("../ElisFiles_mu/VisMassCycle.data.datI.root");
  //
  vector<TString> wmunu_mu;
  wmunu_mu.push_back("../ElisFiles_mu/VisMassCycle.mc.wmunu.root");
  //
  vector<TString> zmumu_mu;
  zmumu_mu.push_back("../ElisFiles_mu/VisMassCycle.mc.zmumu.root");
  //
  vector<TString> wtaunu_mu;
  wtaunu_mu.push_back("../ElisFiles_mu/VisMassCycle.mc.wtaunu.root");
  //
  vector<TString> ztt_mu;
  ztt_mu.push_back("../ElisFiles_mu/VisMassCycle.mc.ztt.root");

  // -----------------------------------------------------------------
  runAnalysis(datafiles_e, "dat_e", output, false);
  runAnalysis(wenu_e,      "wen_e", output, true);
  runAnalysis(zee_e,       "zee_e",  output, true);
  runAnalysis(wtaunu_e,    "wtn_e",  output, true);
  runAnalysis(ztt_e,       "ztt_e",  output, true);
  //
  runAnalysis(datafiles_mu, "dat_mu",  output, true);
  runAnalysis(wmunu_mu,     "wmn_mu",  output, true);
  runAnalysis(zmumu_mu,     "zmm_mu",  output, true);
  runAnalysis(wtaunu_mu,    "wtn_mu",  output, true);
  runAnalysis(ztt_mu,       "ztt_mu",  output, true);



}

void runAnalysis(vector<TString> files, TString InputTag, TString fileName, bool update) {

  TChain *tree = new TChain("VisMassCycleTree");
  for (int i = 0 ; i < files.size(); ++i) {
    cout << " >>>> Adding file " << files[i] << endl;
    tree->Add(files[i]);
  }
  analysis  myAna(tree, InputTag);
  myAna.runAnalysis(fileName, update);

  delete tree;

}