#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"
#include "TBrowser.h"
#include "TH2.h"
#include <iostream>
#include <fstream>
#include <vector>
#include <string>


void treeRead() {
   
    bool DEBUG = 1; 

	TFile *f = new TFile("Marjorie/DAOD_STDM2.05858442._000011.pool.root.1");
	TTree *t1 = (TTree*)f->Get("CollectionTree");

    vector<float>* pt_RAC = new vector<float>;
    vector<float>* pt_vec = new vector<float>;
    TBranch *RAC = 0;
    t1->SetBranchAddress("AntiKt4TruthJetsAux.pt",&pt_RAC,&RAC);
    TBranch *pt  = t1->GetBranch("AntiKt4TruthJetsAux.pt"); 
    Long64_t nentries = pt->GetEntries();

    if (DEBUG) {
        RAC->Print();
        cout << "NEntries: " << nentries << endl;
    }

    for (Long64_t i=0; i < nentries; i++) {
        
        //RAC Error Here 
        RAC->GetEntry(i);
        int ntrk = pt_RAC->size();
        
        //vector<float> q;
        
        cout << "ntrk: " << ntrk << endl; 
        
        //for (int k = 0; k < ntrk; ++k) {
        //vector<float> x = (*pt_RAC)[k];
        //cout << "x size: " << x.size();
        //}

        //cout << "q size: " << q.size() << endl;
        if (DEBUG) {
            //cout << x << endl;
        }
        
        //pt_vec->push_back(x);
	}
    f->Close();

    if (std::ifstream("RAC.root"))
    {
        std::cout << "Output file " << "RAC.root"  << " already exists" << std::endl;
        exit();
    }

    TFile m("RAC.root","NEW"); 
    h1f = new TH1F("pt","Test",10000,0,65000);

    vector<float> &v = *pt_vec;
    int pt_vec_size = pt_vec->size();
    cout << "Size: " << pt_vec_size << endl;
    for (int j=0; j< pt_vec_size; j++) {
        if (DEBUG) {
            //cout << v[j] << endl;
        }
        h1f->Fill(v[j]);
    }
    
    h1f->Write();

}

void tree() {
	treeRead();
}