#include "ZtautauAnalysis.h"
#include "ZtautauTree.h"
#include "EtTree.h"
#include "TauTree.h"
#include "WeightTree.h"

#include <iostream>
#include <string>
#include <sstream>
#include "TString.h"

void addDijets(TChain* chain) {
  std::string dijetNames [] = {"J0", "J1", "J2", "J3", "J4", "J5", "J6"};
  
  for (int isample(0); isample < 7; isample++) {
     TString path = TString("/hepstore/store2/gwilliam/ZtautauNtuples/Ntuple.") + dijetNames[isample] +  ".All.root";
    chain->Add(path);
  }
}

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

  int nsamples = 8;
  std::string sampleNames[]  = {"Ztautau", "Zee", "Zmumu", "Wenu", "Wmunu", "Wtaulepnu", "Wtauhadnu", "Ttbar", "Dijet"};

  float       sampleXsects[] = {
    1.357           * 1000,              
    1.357           * 1000 * 0.96009632, 
    1.357           * 1000 * 0.96044032, 
    (8.071 + 5.743) * 1000 * 0.87661927, 
    (8.071 + 5.743) * 1000 * 0.87630332, 
    (8.071 + 5.743) * 1000 * 0.352 * 0.87424961,
    (8.071 + 5.743) * 1000 * (1-0.352),
    //(8.071 + 5.743) * 1000 * (/*0.352 * 0.87424961 +*/ (1-0.352)),
    //202.86 * 1.07 * 0.543                       
    401.60 * 0.543,
    1.170E+10 + 8.668E+08 + 5.601E+07 + 3.280E+06 + 1.516E+05 + 5.112E+03 + 1.119E+02             
  };

  float lumi = 100;

  for (int isample(0); isample < nsamples; isample++) {
    std::cout << "Starting " << sampleNames[isample] << std::endl;

    TChain* zchain = new TChain("ZtautauTree");
    TChain* etchain = new TChain("EtTree");
    TChain* tauchain = new TChain("TauTree");
    TChain* muonchain = new TChain("MuonTree");
    TChain* elecchain = new TChain("ElectronTree");
    TChain* wtchain = new TChain("WeightTree");

    //if (sampleNames[isample] == "Dijet") {
    //addDijets(zchain);
    //addDijets(etchain);
    //addDijets(tauchain);
    //addDijets(muonchain);
    //addDijets(elecchain);
    //addDijets(wtchain);
    //} else {
    TString path = TString("/hepstore/store2/gwilliam/ZtautauNtuples/Ntuple.") + sampleNames[isample] +  ".All.root";
    zchain->Add(path);
    etchain->Add(path);
    tauchain->Add(path);
    muonchain->Add(path);
    elecchain->Add(path);    
    wtchain->Add(path);
    //}

    ZtautauAnalysis* analysis = new ZtautauAnalysis(sampleNames[isample], sampleXsects[isample], lumi, zchain, etchain, tauchain, elecchain, muonchain, wtchain);
    //analysis->SetSafeTau();
    analysis->Initialise();
    analysis->EtMiss(etchain->GetEntries());
    analysis->Nprong(tauchain->GetEntries());
    analysis->Tau(tauchain->GetEntries());
    analysis->Ztautau(zchain->GetEntries());
    analysis->Finalise();
  }


//   for (int isample(0); isample < 1; isample++) {
//     std::cout << "Starting " << sampleNames[isample] << std::endl;
// 
//     TChain* zchain = new TChain("ZtautauTree");
//     zchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
// 
//     TChain* etchain = new TChain("EtTree");
//     etchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
// 
//     TChain* tauchain = new TChain("TauTree");
//     tauchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
// 
//     TChain* muonchain = new TChain("MuonTree");
//     muonchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
// 
//     TChain* elecchain = new TChain("ElectronTree");
//     elecchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
// 
//     TChain* wtchain = new TChain("WeightTree");
//     wtchain->Add("/user1/gwilliam/AtlasTest/14.4.0/groups/Liverpool/LiverpoolAnalysis/LivZAnalysis/run/Ntuple.Ztautau.root");
//     
//     ZtautauAnalysis* analysis = new ZtautauAnalysis(sampleNames[isample], sampleXsects[isample], lumi, zchain, etchain, tauchain, elecchain, muonchain, wtchain);
//     analysis->Initialise();
//     analysis->EtMiss(etchain->GetEntries());
//     analysis->Ztautau(zchain->GetEntries());
//     analysis->Finalise();
//   }

  return 0;
}