#include <stdio.h>
#include "TChain.h"
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TObjString.h"
#include "TSystem.h"
#include "TROOT.h"
#include <string>       // std::string
#include <iostream>     // std::cout
#include <sstream> 


using namespace ROOT;

void dataReduce(){

EnableImplicitMT();

//trip = TChain("tupleStrip/DecayTree")
//#for t in tupleFiles:
//    strip.AddFile(t)


RDataFrame data = RDataFrame("tupleStrip/DecayTree","/output/*tuple.root");


std::ostringstream sel, sela1, seljpsi, selBu, selBkg;

sela1 <<  "(pi1_MC15TuneV1_ProbNNk<0.5)" << "&&"<<
              "(pi2_MC15TuneV1_ProbNNk<0.5)" << "&&"<<
                   "(pi3_MC15TuneV1_ProbNNk<0.5)"<< "&&"<<
                   "(pi1_MC15TuneV1_ProbNNpi>0.2)"<< "&&"<<
                   "(pi2_MC15TuneV1_ProbNNpi>0.2)"<< "&&"<< 
		   "(pi3_MC15TuneV1_ProbNNpi>0.2)"<< "&&"<<
                   "(a1_IPCHI2_OWNPV<1000)";

seljpsi << "(mu1_MC15TuneV1_ProbNNmu>0.75)"<<"&&"<<  
                     "(mu2_MC15TuneV1_ProbNNmu>0.75)"<<"&&"<<
		     "(jpsi_M>2600)"<<"&&"<<
                     "(jpsi_IPCHI2_OWNPV<1000.)";
selBu << "(Bu_DIRA_OWNPV>0.9999)";
selBkg << "Bu_MM>5500";   // #high mass cut for backgorund sample

sel << sela1 << "&&" <<seljpsi<< "&&" << selBu;

auto data_reduced = data.Filter(sel.str());

std::string newFileName = "/hepstore/vifranco/a1mumu/Bu2a1mumu_Job55_Data_HighMassReduced.root";
std::string newTreeName = "reducedTree";
data_reduced.Snapshot(newTreeName,newFileName);

}