int filter(TString inputDir, long n_evt=100,bool is11=false,bool isFR=false)
{
long TotalEvents=n_evt;
//options
bool is2011=is11;
int N_ALL=4;
int N_ELMU=-3;
if(isFR){
N_ALL=-4;
N_ELMU=3;
}
double cut_mu_pt=9.0;
double cut_mu_eta=2.4;
double cut_el_pt=9.0;
double cut_el_eta=2.5;
double cut_tau_pt=19.0;
double cut_tau_eta=2.3;
if(isFR) cut_tau_pt=0.0;
double cut_dR=-0.09;
double lepton_mass_min=50.0;
double lepton_mass_max=150.0;
std::string doubEle="HLT_Ele17_CaloIdT_CaloIsoVL_TrkIdVL_TrkIsoVL_Ele8_CaloIdT_CaloIsoVL_TrkIdVL_TrkIsoVL_v";
std::string doubEle2="HLT_Ele17_CaloIdL_CaloIsoVL_Ele8_CaloIdL_CaloIsoVL_v";
std::string doubMu="HLT_Mu17_Mu8_v";
std::string doubMu2="HLT_Mu17_TkMu8_v"; // "HLT_Mu13_Mu8_v" in 2011
if(is2011) doubMu2="HLT_Mu13_Mu8_v";
std::string doubMu3="HLT_DoubleMu7_v";
bool debug = false;
TChain *chain = new TChain("t");
chain->Add(inputDir+"*root*");
//////////////////////// end of modular part ////////////////////////////////////////
/////////////////////////COMMON part //////////////////////////////////////////
myevent* in = new myevent();
chain->SetBranchAddress("myevent",&in);
long MaxEvents = chain->GetEntries();
if(TotalEvents > MaxEvents || TotalEvents < 0) TotalEvents=MaxEvents;
std::cout << "There are " << MaxEvents << " entries." << std::endl;
//bookkeeping
ofstream lumi;
long current_run=-1;
long current_lumi=-1;
long m_allEvents=0;
lumi.open("lumi.csv");
//output definition
TFile *newFile = new TFile("output.root","recreate");
TChain *newchain = (TChain*)chain->CloneTree(0);
TTree *tree = newchain->GetTree();
for(uint i =0; i < TotalEvents; i++)
{
chain->GetEntry(i);
m_allEvents++;
if(debug) std::cout << "Processing entry " << i << " event: " << in->eventNumber << std::endl;
if(i%1000==0)std::cout << "Analyzing entry no. " << i << std::endl;
// storing lumis
if(in->runNumber!=current_run || in->lumiNumber!=current_lumi){
lumi << in->runNumber << " " << in->lumiNumber << std::endl;
current_run=in->runNumber;
current_lumi=in->lumiNumber;
}
if(PassFilter(in, debug, doubEle, doubEle2, doubMu, doubMu2, doubMu3,
is2011, N_ALL, N_ELMU,
cut_mu_pt, cut_mu_eta, cut_el_pt, cut_el_eta, cut_tau_pt, cut_tau_eta,
cut_dR, lepton_mass_min, lepton_mass_max)
) tree->Fill();
}
newFile->cd();
newFile->Write();
newFile->Close();
ofstream log1;
log1.open("total.txt");
log1 << m_allEvents << std::endl;
log1.close();
return 0;
}
int main ()
{
TFile * outputfile = TFile::Open ("outputTMVA.root","RECREATE");
TMVA::Factory * TMVAtest = new TMVA::Factory ("TMVAtest", outputfile, "S") ;
//PG get the signal and deliver it to the TMVA factory
TChain signalTree ("sample") ;
signalTree.Add ("data/sig_0.root") ;
std::cout << "READ " << signalTree.GetEntries () << " signal events\n" ;
TMVAtest->AddSignalTree (&signalTree, 1) ;
//PG get the bkg and deliver it to the TMVA factory
TChain bkgTree ("sample") ;
bkgTree.Add ("data/bkg_0.root") ;
std::cout << "READ " << bkgTree.GetEntries () << " bkg events\n" ;
TMVAtest->AddBackgroundTree (&bkgTree, 1) ;
//PG get the training and test samples and deliver them to the TMVA factory
TChain signalTrainTree ("sample") ;
signalTrainTree.Add ("data/sig_1.root") ;
std::cout << "READ " << signalTrainTree.GetEntries () << " signal train events\n" ;
TChain bkgTrainTree ("sample") ;
bkgTrainTree.Add ("data/bkg_1.root") ;
std::cout << "READ " << bkgTrainTree.GetEntries () << " bkg train events\n" ;
TMVAtest->SetInputTrees (signalTrainTree.GetTree (), bkgTrainTree.GetTree (), 1., 1.) ;
//PG variables to be used for the selection
//PG must be defined in the TTrees
TMVAtest->AddVariable ("vars.x", 'F') ;
TMVAtest->AddVariable ("vars.y" , 'F') ;
int signalNumTrain = signalTrainTree.GetEntries () * 4 / 5 ;
int bkgNumTrain = bkgTrainTree.GetEntries () * 4 / 5 ;
int signalNumTest = signalTrainTree.GetEntries () - signalNumTrain ;
int bkgNumTest = bkgTrainTree.GetEntries () - bkgNumTrain ;
char trainOptions[120] ;
sprintf (trainOptions,"NSigTrain=%d:NBkgTrain=%d:NSigTest=%d:NBkgTest=%d",
signalNumTrain, bkgNumTrain,
signalNumTest, bkgNumTest) ;
sprintf (trainOptions,"NSigTrain=%d:NBkgTrain=%d:NSigTest=%d:NBkgTest=%d",
0,0,0,0) ;
std::cout << "TRAINING CONFIGURATION : " << trainOptions << "\n" ;
TMVAtest->PrepareTrainingAndTestTree ("",trainOptions) ;
//PG prepare the classifier
//PG cut-based, default params
TMVAtest->BookMethod (TMVA::Types::kCuts, "Cuts") ;
TMVAtest->TrainAllMethods () ;
TMVAtest->TestAllMethods () ;
TMVAtest->EvaluateAllMethods () ;
delete TMVAtest ;
delete outputfile ;
}
void combinefiles(const TString input="temp.txt",
const TString outputfile="test.root") {
// read input file
TChain *eventChain = new TChain("Events");
TChain *countChain = new TChain("Count");
ifstream ifs;
ifs.open(input.Data());
assert(ifs.is_open());
string line;
getline(ifs,line);
while(getline(ifs,line)) {
eventChain->Add(TString(line));
countChain->Add(TString(line));
}
ifs.close();
Int_t iHmatch1=-1, iHmatch2=-1, iHmatch3=-1, iHmatch4=-1;
Int_t eventType;
Int_t genInfo;
Float_t eventWeight;
Int_t sampleNo;
Int_t isBBTT;
Int_t isBBGG;
Int_t isBBBB;
Int_t isVBFTT;
Int_t isVBF4B;
Float_t met, metPhi;
Float_t ppMet, ppMetPhi;
Float_t pileupMet, pileupMetPhi;
Int_t nCentral=0, nBtag=0;
Int_t centB=0;
Int_t nJets=0;
Int_t tauCat1=0, tauCat2=0;
Int_t bTag1=0, bTag2=0, bTag3=0, bTag4=0;
Int_t jbTag_tt1=0, jbTag_tt2=0, jbTag_6j1=0, jbTag_6j2=0;
Double_t mt2=0;
Double_t mt2puppi=0;
Double_t mt2pileup=0;
Double_t m_sv=0;
Double_t m_svpileup=0;
Double_t m_svpuppi=0;
Float_t ptTau1, ptTau2, ptTrk1, ptTrk2, ptB1, ptB2, ptB3, ptB4, ptG1, ptG2, ptJet_tt1, ptJet_tt2, ptJet_6j1, ptJet_6j2, tauIso1, tauIso2;
Float_t etaTau1, etaTau2, etaTrk1, etaTrk2, etaB1, etaB2, etaB3, etaB4, etaG1, etaG2, etaJet_tt1, etaJet_tt2, etaJet_6j1, etaJet_6j2;
Float_t phiTau1, phiTau2, phiTrk1, phiTrk2, phiB1, phiB2, phiB3, phiB4, phiG1, phiG2, phiJet_tt1, phiJet_tt2, phiJet_6j1, phiJet_6j2;
Float_t mTau1, mTau2, mTrk1, mTrk2, mB1, mB2, mB3, mB4, eG1, eG2, mJet_tt1, mJet_tt2, mJet_6j1, mJet_6j2;
Float_t ptTau1_gen, ptTau2_gen, ptB1_gen, ptB2_gen, ptB3_gen, ptB4_gen, ptG1_gen, ptG2_gen, ptJet_tt1_gen, ptJet_tt2_gen, ptJet_6j1_gen, ptJet_6j2_gen;
Float_t etaTau1_gen, etaTau2_gen, etaB1_gen, etaB2_gen, etaB3_gen, etaB4_gen, etaG1_gen, etaG2_gen, etaJet_tt1_gen, etaJet_tt2_gen, etaJet_6j1_gen, etaJet_6j2_gen;
Float_t phiTau1_gen, phiTau2_gen, phiB1_gen, phiB2_gen, phiB3_gen, phiB4_gen, phiG1_gen, phiG2_gen, phiJet_tt1_gen, phiJet_tt2_gen, phiJet_6j1_gen, phiJet_6j2_gen;
Float_t mTau1_gen, mTau2_gen, mB1_gen, mB2_gen, mB3_gen, mB4_gen, eG1_gen, eG2_gen, mJet_tt1_gen, mJet_tt2_gen, mJet_6j1_gen, mJet_6j2_gen;
Float_t gamIso1, gamIso2;
Float_t ptH1_gen, etaH1_gen, phiH1_gen, mH1_gen;
Float_t ptH2_gen, etaH2_gen, phiH2_gen, mH2_gen;
Float_t ptTau1_genJet, ptTau2_genJet, etaTau1_genJet, etaTau2_genJet, phiTau1_genJet, phiTau2_genJet, mTau1_genJet, mTau2_genJet;
Float_t ptTT, ptBB1, ptBB2, ptGG, ptJJ_tt, ptJJ_6j, ptHH;
Float_t etaTT, etaBB1, etaBB2, etaGG, etaJJ_tt, etaJJ_6j, etaHH;
Float_t phiTT, phiBB1, phiBB2, phiGG, phiJJ_tt, phiJJ_6j, phiHH;
Float_t mTT, mBB1, mBB2, mGG, mJJ_tt, mJJ_6j, mHH;
Float_t dEta_tt=0, dEta_6j=0;
Int_t n;
Float_t rho_0=0, rho_1=0, rho_2=0;
Float_t dEtaBB1=0, dEtaTT=0, dEtaHH=0;
Float_t dPhiBB1=0, dPhiTT=0, dPhiHH=0;
Float_t dRBB1=0, dRTT=0, dRHH=0;
Float_t bdtVal=0;
Float_t chi2=0, corrMet=0, corrMetPhi=0;
Int_t conv=-999;
TFile *outFile = new TFile(outputfile, "RECREATE");
countChain->SetBranchAddress("n", &n);
Int_t nevents=0;
for (Int_t i=0; i<countChain->GetEntries(); i++) {
countChain->GetEntry(i);
nevents+=n;
}
cout << nevents << endl;
eventChain->SetBranchAddress("eventWeight", &eventWeight);
eventChain->SetBranchAddress("sampleNo", &sampleNo);
eventChain->SetBranchAddress("genInfo", &genInfo);
eventChain->SetBranchAddress("isBBTT", &isBBTT);
eventChain->SetBranchAddress("isBBGG", &isBBGG);
eventChain->SetBranchAddress("isBBBB", &isBBBB);
eventChain->SetBranchAddress("isVBFTT", &isVBFTT);
eventChain->SetBranchAddress("isVBF4B", &isVBF4B);
eventChain->SetBranchAddress("eventType", &eventType);
eventChain->SetBranchAddress("met", &met);
eventChain->SetBranchAddress("metPhi", &metPhi);
eventChain->SetBranchAddress("pileupmet", &pileupMet);
eventChain->SetBranchAddress("pileupmetPhi", &pileupMetPhi);
eventChain->SetBranchAddress("puppiMet", &ppMet);
eventChain->SetBranchAddress("puppiMetPhi", &ppMetPhi);
eventChain->SetBranchAddress("ptTau1", &ptTau1);
eventChain->SetBranchAddress("etaTau1", &etaTau1);
eventChain->SetBranchAddress("phiTau1", &phiTau1);
eventChain->SetBranchAddress("mTau1", &mTau1);
eventChain->SetBranchAddress("tauCat1", &tauCat1);
eventChain->SetBranchAddress("tauIso1", &tauIso1);
eventChain->SetBranchAddress("ptTau2", &ptTau2);
eventChain->SetBranchAddress("etaTau2", &etaTau2);
eventChain->SetBranchAddress("phiTau2", &phiTau2);
eventChain->SetBranchAddress("mTau2", &mTau2);
eventChain->SetBranchAddress("tauCat2", &tauCat2);
eventChain->SetBranchAddress("tauIso2", &tauIso2);
eventChain->SetBranchAddress("ptTrk1", &ptTrk1);
eventChain->SetBranchAddress("etaTrk1", &etaTrk1);
eventChain->SetBranchAddress("phiTrk1", &phiTrk1);
eventChain->SetBranchAddress("mTrk1", &mTrk1);
eventChain->SetBranchAddress("ptTrk2", &ptTrk2);
eventChain->SetBranchAddress("etaTrk2", &etaTrk2);
eventChain->SetBranchAddress("phiTrk2", &phiTrk2);
eventChain->SetBranchAddress("mTrk2", &mTrk2);
eventChain->SetBranchAddress("ptG1", &ptG1);
eventChain->SetBranchAddress("etaG1", &etaG1);
eventChain->SetBranchAddress("phiG1", &phiG1);
eventChain->SetBranchAddress("eG1", &eG1);
//eventChain->SetBranchAddress("gamIso1", &gamIso1);
eventChain->SetBranchAddress("ptG2", &ptG2);
eventChain->SetBranchAddress("etaG2", &etaG2);
eventChain->SetBranchAddress("phiG2", &phiG2);
eventChain->SetBranchAddress("eG2", &eG2);
//eventChain->SetBranchAddress("gamIso2", &gamIso2);
eventChain->SetBranchAddress("ptB1", &ptB1);
eventChain->SetBranchAddress("etaB1", &etaB1);
eventChain->SetBranchAddress("phiB1", &phiB1);
eventChain->SetBranchAddress("mB1", &mB1);
eventChain->SetBranchAddress("bTag1", &bTag1);
eventChain->SetBranchAddress("ptB2", &ptB2);
eventChain->SetBranchAddress("etaB2", &etaB2);
eventChain->SetBranchAddress("phiB2", &phiB2);
eventChain->SetBranchAddress("mB2", &mB2);
eventChain->SetBranchAddress("bTag2", &bTag2);
eventChain->SetBranchAddress("ptB3", &ptB3);
eventChain->SetBranchAddress("etaB3", &etaB3);
eventChain->SetBranchAddress("phiB3", &phiB3);
eventChain->SetBranchAddress("mB3", &mB3);
eventChain->SetBranchAddress("bTag3", &bTag3);
eventChain->SetBranchAddress("ptB4", &ptB4);
eventChain->SetBranchAddress("etaB4", &etaB4);
eventChain->SetBranchAddress("phiB4", &phiB4);
eventChain->SetBranchAddress("mB4", &mB4);
eventChain->SetBranchAddress("bTag4", &bTag4);
eventChain->SetBranchAddress("ptJet_tt1", &ptJet_tt1);
eventChain->SetBranchAddress("etaJet_tt1", &etaJet_tt1);
eventChain->SetBranchAddress("phiJet_tt1", &phiJet_tt1);
eventChain->SetBranchAddress("mJet_tt1", &mJet_tt1);
eventChain->SetBranchAddress("jbTag_tt1", &jbTag_tt1);
eventChain->SetBranchAddress("ptJet_tt2", &ptJet_tt2);
eventChain->SetBranchAddress("etaJet_tt2", &etaJet_tt2);
eventChain->SetBranchAddress("phiJet_tt2", &phiJet_tt2);
eventChain->SetBranchAddress("mJet_tt2", &mJet_tt2);
eventChain->SetBranchAddress("jbTag_tt2", &jbTag_tt2);
eventChain->SetBranchAddress("ptJet_6j1", &ptJet_6j1);
eventChain->SetBranchAddress("etaJet_6j1", &etaJet_6j1);
eventChain->SetBranchAddress("phiJet_6j1", &phiJet_6j1);
eventChain->SetBranchAddress("mJet_6j1", &mJet_6j1);
eventChain->SetBranchAddress("jbTag_6j1", &jbTag_6j1);
eventChain->SetBranchAddress("ptJet_6j2", &ptJet_6j2);
eventChain->SetBranchAddress("etaJet_6j2", &etaJet_6j2);
eventChain->SetBranchAddress("phiJet_6j2", &phiJet_6j2);
eventChain->SetBranchAddress("mJet_6j2", &mJet_6j2);
eventChain->SetBranchAddress("jbTag_6j2", &jbTag_6j2);
eventChain->SetBranchAddress("ptTau1_gen", &ptTau1_gen);
eventChain->SetBranchAddress("etaTau1_gen", &etaTau1_gen);
eventChain->SetBranchAddress("phiTau1_gen", &phiTau1_gen);
eventChain->SetBranchAddress("mTau1_gen", &mTau1_gen);
eventChain->SetBranchAddress("ptTau2_gen", &ptTau2_gen);
eventChain->SetBranchAddress("etaTau2_gen", &etaTau2_gen);
eventChain->SetBranchAddress("phiTau2_gen", &phiTau2_gen);
eventChain->SetBranchAddress("mTau2_gen", &mTau2_gen);
eventChain->SetBranchAddress("ptTau1_genJet", &ptTau1_genJet);
eventChain->SetBranchAddress("etaTau1_genJet", &etaTau1_genJet);
eventChain->SetBranchAddress("phiTau1_genJet", &phiTau1_genJet);
eventChain->SetBranchAddress("mTau1_genJet", &mTau1_genJet);
eventChain->SetBranchAddress("ptTau2_genJet", &ptTau2_genJet);
eventChain->SetBranchAddress("etaTau2_genJet", &etaTau2_genJet);
eventChain->SetBranchAddress("phiTau2_genJet", &phiTau2_genJet);
eventChain->SetBranchAddress("mTau2_genJet", &mTau2_genJet);
eventChain->SetBranchAddress("ptG1_gen", &ptG1_gen);
eventChain->SetBranchAddress("etaG1_gen", &etaG1_gen);
eventChain->SetBranchAddress("phiG1_gen", &phiG1_gen);
eventChain->SetBranchAddress("eG1_gen", &eG1_gen);
eventChain->SetBranchAddress("ptG2_gen", &ptG2_gen);
eventChain->SetBranchAddress("etaG2_gen", &etaG2_gen);
eventChain->SetBranchAddress("phiG2_gen", &phiG2_gen);
eventChain->SetBranchAddress("eG2_gen", &eG2_gen);
eventChain->SetBranchAddress("ptB1_gen", &ptB1_gen);
eventChain->SetBranchAddress("etaB1_gen", &etaB1_gen);
eventChain->SetBranchAddress("phiB1_gen", &phiB1_gen);
eventChain->SetBranchAddress("mB1_gen", &mB1_gen);
eventChain->SetBranchAddress("iHmatch1", &iHmatch1);
eventChain->SetBranchAddress("ptB2_gen", &ptB2_gen);
eventChain->SetBranchAddress("etaB2_gen", &etaB2_gen);
eventChain->SetBranchAddress("phiB2_gen", &phiB2_gen);
eventChain->SetBranchAddress("mB2_gen", &mB2_gen);
eventChain->SetBranchAddress("iHmatch2", &iHmatch2);
eventChain->SetBranchAddress("ptB3_gen", &ptB3_gen);
eventChain->SetBranchAddress("etaB3_gen", &etaB3_gen);
eventChain->SetBranchAddress("phiB3_gen", &phiB3_gen);
eventChain->SetBranchAddress("mB3_gen", &mB3_gen);
eventChain->SetBranchAddress("iHmatch3", &iHmatch3);
eventChain->SetBranchAddress("ptB4_gen", &ptB4_gen);
eventChain->SetBranchAddress("etaB4_gen", &etaB4_gen);
eventChain->SetBranchAddress("phiB4_gen", &phiB4_gen);
eventChain->SetBranchAddress("mB4_gen", &mB4_gen);
eventChain->SetBranchAddress("iHmatch4", &iHmatch4);
eventChain->SetBranchAddress("ptH1_gen", &ptH1_gen);
eventChain->SetBranchAddress("etaH1_gen", &etaH1_gen);
eventChain->SetBranchAddress("phiH1_gen", &phiH1_gen);
eventChain->SetBranchAddress("mH1_gen", &mH1_gen);
eventChain->SetBranchAddress("ptH2_gen", &ptH2_gen);
eventChain->SetBranchAddress("etaH2_gen", &etaH2_gen);
eventChain->SetBranchAddress("phiH2_gen", &phiH2_gen);
eventChain->SetBranchAddress("mH2_gen", &mH2_gen);
eventChain->SetBranchAddress("ptJet_tt1_gen", &ptJet_tt1_gen);
eventChain->SetBranchAddress("etaJet_tt1_gen", &etaJet_tt1_gen);
eventChain->SetBranchAddress("phiJet_tt1_gen", &phiJet_tt1_gen);
eventChain->SetBranchAddress("mJet_tt1_gen", &mJet_tt1_gen);
eventChain->SetBranchAddress("ptJet_tt2_gen", &ptJet_tt2_gen);
eventChain->SetBranchAddress("etaJet_tt2_gen", &etaJet_tt2_gen);
eventChain->SetBranchAddress("phiJet_tt2_gen", &phiJet_tt2_gen);
eventChain->SetBranchAddress("mJet_tt2_gen", &mJet_tt2_gen);
eventChain->SetBranchAddress("ptJet_6j1_gen", &ptJet_6j1_gen);
eventChain->SetBranchAddress("etaJet_6j1_gen", &etaJet_6j1_gen);
eventChain->SetBranchAddress("phiJet_6j1_gen", &phiJet_6j1_gen);
eventChain->SetBranchAddress("mJet_6j1_gen", &mJet_6j1_gen);
eventChain->SetBranchAddress("ptJet_6j2_gen", &ptJet_6j2_gen);
eventChain->SetBranchAddress("etaJet_6j2_gen", &etaJet_6j2_gen);
eventChain->SetBranchAddress("phiJet_6j2_gen", &phiJet_6j2_gen);
eventChain->SetBranchAddress("mJet_6j2_gen", &mJet_6j2_gen);
eventChain->SetBranchAddress("ptTT", &ptTT);
eventChain->SetBranchAddress("etaTT", &etaTT);
eventChain->SetBranchAddress("phiTT", &phiTT);
eventChain->SetBranchAddress("mTT", &mTT);
eventChain->SetBranchAddress("ptBB1", &ptBB1);
eventChain->SetBranchAddress("etaBB1", &etaBB1);
eventChain->SetBranchAddress("phiBB1", &phiBB1);
eventChain->SetBranchAddress("mBB1", &mBB1);
eventChain->SetBranchAddress("ptBB2", &ptBB2);
eventChain->SetBranchAddress("etaBB2", &etaBB2);
eventChain->SetBranchAddress("phiBB2", &phiBB2);
eventChain->SetBranchAddress("mBB2", &mBB2);
eventChain->SetBranchAddress("ptGG", &ptGG);
eventChain->SetBranchAddress("etaGG", &etaGG);
eventChain->SetBranchAddress("phiGG", &phiGG);
eventChain->SetBranchAddress("mGG", &mGG);
eventChain->SetBranchAddress("ptJJ_tt", &ptJJ_tt);
eventChain->SetBranchAddress("etaJJ_tt", &etaJJ_tt);
eventChain->SetBranchAddress("phiJJ_tt", &phiJJ_tt);
eventChain->SetBranchAddress("mJJ_tt", &mJJ_tt);
eventChain->SetBranchAddress("ptJJ_6j", &ptJJ_6j);
eventChain->SetBranchAddress("etaJJ_6j", &etaJJ_6j);
eventChain->SetBranchAddress("phiJJ_6j", &phiJJ_6j);
eventChain->SetBranchAddress("mJJ_6j", &mJJ_6j);
eventChain->SetBranchAddress("ptHH", &ptHH);
eventChain->SetBranchAddress("etaHH", &etaHH);
eventChain->SetBranchAddress("phiHH", &phiHH);
eventChain->SetBranchAddress("mHH", &mHH);
eventChain->SetBranchAddress("mt2", &mt2);
eventChain->SetBranchAddress("mt2puppi", &mt2puppi);
eventChain->SetBranchAddress("mt2pileup", &mt2pileup);
eventChain->SetBranchAddress("m_sv", &m_sv);
eventChain->SetBranchAddress("m_svpileup", &m_svpileup);
eventChain->SetBranchAddress("m_svpuppi", &m_svpuppi);
eventChain->SetBranchAddress("nBtag", &nBtag);
eventChain->SetBranchAddress("nCentral", &nCentral);
eventChain->SetBranchAddress("centB", ¢B);
eventChain->SetBranchAddress("nJets", &nJets);
eventChain->SetBranchAddress("dEta_tt", &dEta_tt);
eventChain->SetBranchAddress("dEta_6j", &dEta_6j);
eventChain->SetBranchAddress("rho_0", &rho_0);
eventChain->SetBranchAddress("rho_1", &rho_1);
eventChain->SetBranchAddress("rho_2", &rho_2);
eventChain->SetBranchAddress("dEtaBB1", &dEtaBB1);
eventChain->SetBranchAddress("dPhiBB1", &dPhiBB1);
eventChain->SetBranchAddress("dRBB1", &dRBB1);
eventChain->SetBranchAddress("dEtaTT", &dEtaTT);
eventChain->SetBranchAddress("dPhiTT", &dPhiTT);
eventChain->SetBranchAddress("dRTT", &dRTT);
eventChain->SetBranchAddress("dEtaHH", &dEtaHH);
eventChain->SetBranchAddress("dPhiHH", &dPhiHH);
eventChain->SetBranchAddress("dRHH", &dRHH);
eventChain->GetEntry(0);
TTree *outTree=(TTree*)eventChain->GetTree()->CloneTree(0);
outTree->Branch("bdtVal", &bdtVal, "bdtVal/f");
outTree->Branch("chi2", &chi2, "chi2/f");
outTree->Branch("corrMet", &corrMet, "corrMet/f");
outTree->Branch("corrMetPhi", &corrMetPhi, "corrMetPhi/f");
outTree->Branch("conv", &conv, "conv/i");
//hhMVA::MVAType why1=hhMVA::kTauTau;
hhMVA::MVAType why2=hhMVA::kMuTau;
hhMVA::MVAType why3=hhMVA::kElTau;
hhMVA::MVAType why4=hhMVA::kElMu;
//hhMVA *ttMVA = new hhMVA();
hhMVA *mtMVA = new hhMVA();
hhMVA *etMVA = new hhMVA();
hhMVA *emMVA = new hhMVA();
//ttMVA->Intialize(why1);
mtMVA->Intialize(why2);
etMVA->Intialize(why3);
emMVA->Intialize(why4);
for (Int_t i=0; i<eventChain->GetEntries(); i++) {
eventChain->GetEntry(i);
eventWeight/=float(nevents);
bdtVal=999;
// hack to fix modified SM samples
//if (sampleNo>95 && sampleNo<100) eventWeight=2.92/float(nevents);
if (isBBTT!=1) continue;
if (ptB1<30 || ptB2<30) continue;
//if (TMath::Sqrt((etaTau1-etaTau2)*(etaTau1-etaTau2)+(phiTau1-phiTau2)*(phiTau1-phiTau2))<0.4) continue;
/*
TLorentzVector v1;
TLorentzVector v2;
v1.SetPtEtaPhiM(ptB1, etaB1, phiB1, mB1);
v2.SetPtEtaPhiM(ptB2, etaB2, phiB2, mB2);
TMatrixD m1(3,3);
TMatrixD m2(3,3);
m1.Zero();
m2.Zero();
m1(0,0) = ErrEt (v1.Et(), v1.Eta()); // et
m1(1,1) = ErrEta(v1.Et(), v1.Eta()); // eta
m1(2,2) = ErrPhi(v1.Et(), v1.Eta()); // phi
m2(0,0) = ErrEt (v2.Et(), v2.Eta()); // et
m2(1,1) = ErrEta(v2.Et(), v2.Eta()); // eta
m2(2,2) = ErrPhi(v2.Et(), v2.Eta()); // phi
TFitParticleEtEtaPhi *jet1 = new TFitParticleEtEtaPhi( "Jet1", "Jet1", &v1, &m1 );
TFitParticleEtEtaPhi *jet2 = new TFitParticleEtEtaPhi( "Jet2", "Jet2", &v2, &m2 );
TFitConstraintM *mCons1 = new TFitConstraintM( "HMassConstraint", "HMass-Constraint", 0, 0 , 125.);
mCons1->addParticles1( jet1, jet2 );
TKinFitter* fitter = new TKinFitter("fitter", "fitter");
fitter->addMeasParticle( jet1 );
fitter->addMeasParticle( jet2 );
fitter->addConstraint( mCons1 );
//Set convergence criteria
fitter->setMaxNbIter( 30 );
fitter->setMaxDeltaS( 1e-2 );
fitter->setMaxF( 1e-1 );
fitter->setVerbosity(1);
TVector2 b1_i; b1_i.SetMagPhi(fitter->get4Vec(0)->Pt(), fitter->get4Vec(0)->Phi());
TVector2 b2_i; b2_i.SetMagPhi(fitter->get4Vec(1)->Pt(), fitter->get4Vec(1)->Phi());
//Perform the fit
conv=fitter->fit();
chi2=fitter->getS();
TVector2 metTemp; metTemp.SetMagPhi(pileupMet,pileupMetPhi);
TVector2 b1_f; b1_f.SetMagPhi(fitter->get4Vec(0)->Pt(), fitter->get4Vec(0)->Phi());
TVector2 b2_f; b2_f.SetMagPhi(fitter->get4Vec(1)->Pt(), fitter->get4Vec(1)->Phi());
metTemp=metTemp+b1_i+b2_i-b1_f-b2_f;
corrMet=metTemp.Mod();
corrMetPhi=metTemp.Phi();
*/
if (tauCat1==1 && tauCat2==1 && ptTau1>45 && ptTau2>45) {
bdtVal=999;//ttMVA->GetBDTValue(mTT, ptTT, mBB1, ptBB1, dRBB1, dRTT);
}
else if (tauCat1==1 && tauCat2==3 && ptTau1>30 && ptTau2>20) {
bdtVal=mtMVA->GetBDTValue(mTT, ptTT, mBB1, ptBB1, mHH, ptHH, mt2pileup, dRBB1, dRTT, dRHH);
}
else if (tauCat1==1 && tauCat2==2 && ptTau1>30 && ptTau2>20) {
bdtVal=etMVA->GetBDTValue(mTT, ptTT, mBB1, ptBB1, mHH, ptHH, mt2pileup, dRBB1, dRTT, dRHH);
}
else if (tauCat1==3 && tauCat2==2 && ptTau1>20 && ptTau2>20) {
bdtVal=emMVA->GetBDTValue(mTT, ptTT, mBB1, ptBB1, mHH, ptHH, mt2pileup, dRBB1, dRTT, dRHH);
}
else continue;
outTree->Fill();
}
outFile->Write();
outFile->Save();
}
//
// Fucntions
//
void skimThisBaby(TString inPath, TString inFileName, TString inTreeName, TString outPath){
// Talk to user
cout << " Skimming: "<< endl;
cout << " " << inPath+inFileName <<endl;
// Load input TChain
TChain *ch = new TChain(inTreeName);
TString inFile = inPath;
inFile += inFileName;
ch->Add(inFile);
Long64_t nentries = ch->GetEntries();
TTreeCache::SetLearnEntries(10);
ch->SetCacheSize(128*1024*1024);
// Initialize Branches
babyAnalyzer.Init(ch->GetTree());
babyAnalyzer.LoadAllBranches();
// Setup output file name and path
TString outFileName = outPath;
outFileName += inFileName;
outFileName.ReplaceAll("*", "");
outFileName.ReplaceAll(".root", "_skimmed.root");
cout << " Output will be written to: " << endl;
cout << " " << outFileName << endl << endl;
// Open outputfile and Clone input TTree
TFile *newfile = new TFile(outFileName, "recreate");
TTree *newtree = (TTree*)ch->GetTree()->CloneTree(0);
if(!newtree) cout << "BAD TTREE CLONE" << endl;
TH1D *newCounter=NULL;
// Get nEntries
unsigned int nEventsTotal = 0;
unsigned int nEventsChain = ch->GetEntries();
// Grab list of files
TObjArray *listOfFiles = ch->GetListOfFiles();
TIter fileIter(listOfFiles);
TFile *currentFile = 0;
// File Loop
int iFile=0;
while ( (currentFile = (TFile*)fileIter.Next()) ) {
// Get File Content
TFile *file = new TFile( currentFile->GetTitle() );
TTree *tree = (TTree*)file->Get("t");
TTreeCache::SetLearnEntries(10);
tree->SetCacheSize(128*1024*1024);
babyAnalyzer.Init(tree);
if(iFile==0){
TH1D *temp = (TH1D*)file->Get("h_counter");
newCounter = (TH1D*)temp->Clone("h_counter");
newCounter->SetDirectory(newfile);
}
//else{
//TH1D *temp = (TH1D*)file->Get("h_counter");
//newCounter->Add(temp);
//}
// Loop over Events in current file
if( nEventsTotal >= nEventsChain ) continue;
unsigned int nEventsTree = tree->GetEntriesFast();
for( unsigned int event = 0; event < nEventsTree; ++event) {
// Progress
stop_1l_babyAnalyzer::progress( nEventsTotal, nEventsChain );
// Load Tree
tree->LoadTree(event);
babyAnalyzer.GetEntry(event);
++nEventsTotal;
// Selection
if(nvetoleps()<1) continue;
if(nvetoleps()<2 && PassTrackVeto_v3() && PassTauVeto()) continue;
if(ngoodjets()<2) continue;
if(mt_met_lep()<150.0) continue;
if(pfmet()<200.0) continue;
// Turn on all branches on input
babyAnalyzer.LoadAllBranches();
// Fill output tree
newtree->Fill();
} // end loop over entries
iFile++;
} // end loop over files in TChain
// Clean up
//newtree->Print();
//newtree->AutoSave();
newfile->Write();
newfile->Close();
delete ch;
delete newfile;
}
int main ()
{
TFile * outputfile = TFile::Open ("outputTMVA.root","RECREATE");
TMVA::Factory * TMVAtest = new TMVA::Factory ("TMVAtest", outputfile, "S") ;
//PG get the signal and deliver it to the TMVA factory
TChain signalTree ("shrinked") ;
signalTree.Add ("data/H160_VBF.root") ;
std::cout << "READ " << signalTree.GetEntries () << " signal events\n" ;
TMVAtest->AddSignalTree (&signalTree, 1) ;
//PG get the bkg and deliver it to the TMVA factory
TChain bkgTree ("shrinked") ;
bkgTree.Add ("data/H160_NOVBF.root") ;
std::cout << "READ " << bkgTree.GetEntries () << " bkg events\n" ;
TMVAtest->AddBackgroundTree (&bkgTree, 1) ;
//PG get the training and test samples and deliver them to the TMVA factory
TChain signalTrainTree ("shrinked") ;
signalTrainTree.Add ("data/H150_VBF.root") ;
std::cout << "READ " << signalTrainTree.GetEntries () << " signal train events\n" ;
TChain bkgTrainTree ("shrinked") ;
bkgTrainTree.Add ("data/H150_NOVBF.root") ;
std::cout << "READ " << bkgTrainTree.GetEntries () << " bkg train events\n" ;
// PG this is useless!!
TMVAtest->SetInputTrees (signalTrainTree.GetTree (), bkgTrainTree.GetTree (), 1., 1.) ;
//PG variables to be used for the selection
//PG must be defined in the TTrees
TMVAtest->AddVariable ("vars.Deta", 'F') ;
TMVAtest->AddVariable ("vars.Dphi" , 'F') ;
TMVAtest->AddVariable ("vars.Minv" , 'F') ;
int signalNumTrain = signalTrainTree.GetEntries () * 4 / 5 ;
int bkgNumTrain = bkgTrainTree.GetEntries () * 4 / 5 ;
int signalNumTest = signalTrainTree.GetEntries () - signalNumTrain ;
int bkgNumTest = bkgTrainTree.GetEntries () - bkgNumTrain ;
char trainOptions[120] ;
sprintf (trainOptions,"NSigTrain=%d:NBkgTrain=%d:NSigTest=%d:NBkgTest=%d",
signalNumTrain, bkgNumTrain,
signalNumTest, bkgNumTest) ;
sprintf (trainOptions,"NSigTrain=%d:NBkgTrain=%d:NSigTest=%d:NBkgTest=%d",
0,0,0,0) ;
std::cout << "TRAINING CONFIGURATION : " << trainOptions << "\n" ;
TMVAtest->PrepareTrainingAndTestTree ("",trainOptions) ;
//PG prepare the classifiers
//PG ------------------------
//PG cut-based, default params
TMVAtest->BookMethod (TMVA::Types::kCuts, "Cuts") ;
//PG multi layer perceptron
TMVAtest->BookMethod (TMVA::Types::kMLP, "MLP",
"H:!V:!Normalise:NeuronType=tanh:NCycles=200:HiddenLayers=N+1,N:TestRate=5");
TMVAtest->TrainAllMethods () ;
TMVAtest->TestAllMethods () ;
TMVAtest->EvaluateAllMethods () ;
delete TMVAtest ;
delete outputfile ;
}