//C++
#include <type_traits>
#include <vector>
#include <map>
#include <exception>
#include <string>
#include <sstream>
#include <functional>
#include <chrono>
#include <thread>
//root
#include <TROOT.h>
#include <ROOT/TDataFrame.hxx>
#include <TObject.h>
#include <TFile.h>
#include <TString.h>
#include <TTree.h>
#include <TLeaf.h>
#include <TBranch.h>
#include <TFriendElement.h>
#include <TObjArray.h>
#include <TObjString.h>
#include <TEntryList.h>
#include <TEnv.h>
//local
#include "../IOjuggler/ProgressBar.h"
#include "../IOjuggler/IOjuggler.h"

#ifndef NTUPLE_GIZMO_GIT_HASH
#define NTUPLE_GIZMO_GIT_HASH " "
#endif

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

  //parse command line options
  const auto options = IOjuggler::parse_options(argc, argv, "c:d:i:o:t:v:h","nonoptions: any number of <file:friendtree> combinations");
  const auto wd = options.get<std::string>("workdir");
  const auto ofn = options.get<std::string>("outfilename");
  const auto ifn = options.get<std::string>("infilename");
  const auto itn = options.get<std::string>("treename");

  //init MessageService for easy and nice printout
  MessageService msgsvc("selector_rootMT",static_cast<MSG_LVL>(options.get<int>("verbosity")));
  msgsvc.debugmsg("Current ntuple-gizmo git hash: " + static_cast<std::string>(NTUPLE_GIZMO_GIT_HASH));

  //get input as TChain
  auto chain = IOjuggler::get_chain(itn,ifn,wd,msgsvc);

  //get config-file
  auto configtree = IOjuggler::get_ptree(options.get<std::string>("config"));

  //manipulations of the config file
  IOjuggler::auto_replace_in_ptree(configtree);
  IOjuggler::auto_append_in_ptree(configtree);

  auto trim_vars = configtree.get_child_optional("variables");

  auto tmp_msg = "Trimming input tree(s) with cut " + configtree.get<std::string>("basiccuts","");
  if(trim_vars)
    msgsvc.infomsg(tmp_msg + " and writing out " + configtree.get_child("variables").size() + " variables");
  else
    msgsvc.infomsg(tmp_msg + " and writing out full tree");

  const bool noImplicitCuts= (const bool) (configtree.get_child_optional("noImplicitCuts"));
  if(noImplicitCuts) msgsvc.infomsg("Implicit cuts switched off");

  std::vector<std::string> branches_in = {};
  std::vector<std::string> default_branches_out = {};
  std::vector<std::pair<std::string,std::string>> transformations;
  if(trim_vars)
    for(const auto& var : *trim_vars){
      branches_in.push_back(var.first);
      if(var.first.find("[0]") != std::string::npos){
        default_branches_out.push_back(boost::algorithm::replace_all_copy(var.first, "[0]", "Z"));
        transformations.emplace_back(default_branches_out.back(),var.first);
      }
      else if(auto nn = var.second.get_optional<std::string>("new_name")){
        default_branches_out.push_back(*nn);
        if(auto tfm = var.second.get_optional<std::string>("transformation"))
          transformations.emplace_back(default_branches_out.back(),*tfm);
        else
          transformations.emplace_back(default_branches_out.back(),var.first);
      }
      else if(!var.second.get("to_output",1))
        msgsvc.infomsg("Using " + var.first + " only as input branch");
      else
        default_branches_out.push_back(var.first);
    }

  //first, we need to know how many events there are and how many threads we use:
  ROOT::EnableImplicitMT();
  ROOT::Experimental::TDataFrame d(*chain.get(),branches_in);
  std::vector<ROOT::Experimental::TDF::TInterface<ROOT::Detail::TDF::TLoopManager>::TInterfaceJittedDefine> df{d};
  if(!transformations.empty())
    for(const auto& tf : transformations){
      msgsvc.debugmsg("transforming " + tf.second + " to a new branch: " + tf.first);
      df.push_back(df.back().Define(tf.first,tf.second));
    }
  //ROOT::Experimental::TDF::TSnapshotOptions tsno(configtree.get("outfopt","recreate"));
  //.Filter(configtree.get<std::string>("basiccuts",""))
  df.back().Snapshot(configtree.get("outtreename",itn),wd + "/" + ofn,default_branches_out);//,tsno

  return 0;
}