#include <string>

#include <vector>

#include "TLorentzVector.h"

#include "TTree.h"

#include "TFile.h"

#include "TDirectory.h"

#include "Math/GenVector/LorentzVector.h"

#include "Math/PositionVector3D.h"

#include "Math/SMatrix.h"

#include "Math/GenVector/PtEtaPhiM4D.h"

#include "Math/Vector4D.h"

#include "TClonesArray.h"

#include "baconheaders/TEventInfo.hh"

#include "baconheaders/TGenEventInfo.hh"

#include "baconheaders/TTrigger.hh"

#include "baconheaders/TriggerRecord.hh"

#include "baconheaders/TVertex.hh"

#include "baconheaders/TJet.hh"

#include "baconheaders/TAddJet.hh"

using namespace std;

typedef ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiE4D<float> > PtEtaPhiE4Vector;

typedef ROOT::Math::PositionVector3D<ROOT::Math::Cartesian3D<double>,ROOT::Math::DefaultCoordinateSystemTag> PointVector;

typedef ROOT::Math::SMatrix<double,3,3,ROOT::Math::MatRepSym<double,3> > CovMatrix;

struct JMETree {

}

void read(long i) {

tevent->GetEntry(i);

thlt->GetEntry(i);

tjet->GetEntry(i);

tvtx->GetEntry(i);

tmet->GetEntry(i);

}

};

struct BaconTree {

}

void fill(const JMETree& jme) {

eventinfo.runNum = jme.run;

eventinfo.evtNum = jme.evt;

eventinfo.lumiSec = jme.lumi;

eventinfo.nPU = jme.npus->size() ? (*jme.npus)[0] : 0;

eventinfo.nPUmean = jme.tnpus->size() ? (*jme.tnpus)[0] : 0;

eventinfo.rhoIso = 0;

eventinfo.rhoJet = jme.rho;

eventinfo.triggerBits = 0;

for(auto i = jme.paths->cbegin() ; i != jme.paths->cend() ; ++i) {

//cout << "testing " << *i << endl;

for(auto i2 = triggernames->cbegin(); i2 != triggernames->cend() ; ++i2) {

if(*i2 > *i) break;

//cout << "trying " << *i2 << endl;

if(i->compare(0,min(i->size(),i2->size()),*i2) == 0) {

//std::cout << "setting" << *i << ", " << *i2 << " " << i2-triggernames->cbegin() << endl;

eventinfo.triggerBits[i2-triggernames->cbegin()] = true;

}

}

//std::cout << *i << endl;

}

//eventinfo.triggerBits = jme.prescales->size();

eventinfo.pfMET = (*jme.met_p4)[0].Pt();// not stored atm, to be done later

eventinfo.pfMETphi = (*jme.met_p4)[0].Phi();

geneventinfo.id_1 = jme.pdf_id->first;

geneventinfo.id_2 = jme.pdf_id->second;

geneventinfo.x_1 = jme.pdf_x->first;

geneventinfo.x_2 = jme.pdf_x->second;

geneventinfo.weight = jme.weight;

//geneventinfo.pthat = jme.pthat; // is missing in the header - but anyway not needed (at the moment)

vertices->Clear();

jets->Clear();

addjets->Clear();

for(unsigned int j = 0 ; j < jme.position->size() ; ++j) {

assert(vertices->GetEntries() < vertices->GetSize());

const int index = vertices->GetEntries();

new((*vertices)[index]) baconhep::TVertex();

baconhep::TVertex *pVertex = (baconhep::TVertex*)(*vertices)[index];

//pVertex->nTracksFit =

pVertex->ndof = (*jme.ndof)[j];

pVertex->chi2 = (*jme.normalizedChi2)[j];

pVertex->x = (*jme.position)[j].X();

pVertex->y = (*jme.position)[j].Y();

pVertex->z = (*jme.position)[j].Z();

}

for(unsigned int j = 0 ; j < jme.p4->size() ; ++j) {

assert(jets->GetEntries() < jets->GetSize());

const int index = jets->GetEntries();

new((*jets)[index]) baconhep::TJet();

baconhep::TJet *pJet = (baconhep::TJet*)(*jets)[index];

pJet->pt = (*jme.p4)[j].pt();

pJet->eta = (*jme.p4)[j].eta();

pJet->phi = (*jme.p4)[j].phi();

pJet->mass = (*jme.p4)[j].mass();

pJet->ptRaw = (*jme.p4)[j].pt() * (*jme.jec_toraw)[j];

//pJet->csv = (*jme.pfCombinedSecondaryVertexV2BJetTags)[j]; // only stored in AK4PFCHS, to be done later

pJet->area = (*jme.jtarea)[j];

pJet->genpt = (*jme.gen_p4)[j].pt();

pJet->geneta = (*jme.gen_p4)[j].eta();

pJet->genphi = (*jme.gen_p4)[j].phi();

pJet->genm = (*jme.gen_p4)[j].mass();

pJet->betaStar = (*jme.betaStar)[j];

//std::cout << pJet->pt << std::endl;

}

tree->Fill();

}

};

int convert(string infile, string outfile, string jetcoll) {

JMETree jmetree = { };

TFile* fin = TFile::Open(infile.c_str());

long nev = jmetree.init(fin,jetcoll);

TFile* fout = TFile::Open(outfile.c_str(),"RECREATE");

BaconTree bacontree;

bacontree.init(fout,jetcoll);

// std::cout << "input file " << infile << " with " << nev << " events.\n";

for(int i = 0 ; i < nev ; ++i) {

jmetree.read(i);

if(jmetree.p4->size() < 2) continue;

//std::cout << "event " << jmetree.evt << " vertices:" << jmetree.position->size() << " rhos " << jmetree.rhos->size() << " jets " << jmetree.p4->size() << '\n';

/*

for(unsigned int j = 0 ; j < jmetree.p4->size() ; ++j) {

std::cout << (*jmetree.p4)[j].pt() << '\n';

}

*/

bacontree.fill(jmetree);

}

fin->Close();

bacontree.tree->Write();

fout->Close();

return 0;

}