#include <iostream>
#include <iomanip>
using namespace std;

#include <TFile.h>
#include <TTree.h>

#include "cern_c.h"
#define MEMH 8000000
#define LREC 1024		/* record length of hbook direct access file in WORDS */
#define LUN 3			/* logical unit number of hbook file */
extern "C" {
	float pawc_[MEMH];
	int quest_[100];
};


#define MAX_PARTS 50

class event_t{
	public:
		int n;
		float E[MAX_PARTS];
		float px[MAX_PARTS];
		float py[MAX_PARTS];
		float pz[MAX_PARTS];
		float neuETot;
		int nGen;
		float genE[MAX_PARTS];
		float genPx[MAX_PARTS];
		float genPy[MAX_PARTS];
		float genPz[MAX_PARTS];
		int type[MAX_PARTS];
		float genNeuE;
};

int main(int narg, char *argv[])
{
	const char *fname ="primex_event.root";
	if(narg>1)fname = argv[1];

	// Open ROOT file for input
	TFile *f = new TFile(fname);
	if(!f || !f->IsOpen()){
		cerr<<"Unable to open file \""<<fname<<"\"!"<<endl;
		return -1;
	}	
	cout<<"Opened input file \""<<fname<<"\""<<endl;
	
	// Find TTree
	TTree *tree = (TTree*)f->Get("event");
	if(!tree){
		cerr<<"Unable to find TTree \"event\" in ROOT file!"<<endl;
		return -2;
	}
	
	// Set up branches
	event_t event;
	tree->SetBranchAddress("n", &event.n);
	tree->SetBranchAddress("E", &event.E);
	tree->SetBranchAddress("px", &event.px);
	tree->SetBranchAddress("py", &event.py);
	tree->SetBranchAddress("pz", &event.pz);
	tree->SetBranchAddress("neuETot", &event.neuETot);
	tree->SetBranchAddress("nGen", &event.nGen);
	tree->SetBranchAddress("genE", &event.genE);
	tree->SetBranchAddress("genPx", &event.genPx);
	tree->SetBranchAddress("genPy", &event.genPy);
	tree->SetBranchAddress("genPz", &event.genPz);
	tree->SetBranchAddress("type", &event.type);
	tree->SetBranchAddress("genNeuE", &event.genNeuE);

	// Initialize cernlib (monkey shines!)
	quest_[9] = 65000;
	int memh = MEMH;
	hlimit(memh);
	
	// Open HBOOK file for writing
	hropen(LUN, "lun", "primex_event.hbook" , "N", LREC, 0);
	
	// Create Ntuple
	hbnt(10,"event","");
	char ntp_str[256];
	sprintf(ntp_str, "n[0,%d]:I,E(n):R,px(n),py(n),pz(n),neuETot,nGen[0,%d]:I,genE(nGen):R,genPx(nGen),genPy(nGen),genPz(nGen),type(nGen):I,genNeuE:R", MAX_PARTS, MAX_PARTS);
	cout<<ntp_str<<endl;
	hbname(10,"EVNT", &event.n, ntp_str);
	
	// Loop over entries in ROOT file
	int Nevents = tree->GetEntries();
	for(int i=1; i<=Nevents; i++){
		// Read in event
		tree->GetEntry(i);

		// Write event to Ntuple
		hfnt(10);
		
		// Update ticker
		if(i%100 == 0)cout<<"  "<<i<<" events processed     \r";
	}
	
	cout<<endl;
	cout<<Nevents<<" events processed total."<<endl;
	
	// Close hbook file
	int icycle=0;
	hrout(0,icycle,"T");
	hrend("lun");
	
	// Close ROOT file
	f->Close();

	return 0;
}


#if 0
******************************************************************************
*Tree    :event     : Event Reconstruction                                   *
*Entries :    10000 : Total =         1882594 bytes  File  Size =    1363534 *
*        :          : Tree compression factor =   1.37                       *
******************************************************************************
*Br    0 :n         : n/I                                                    *
*Entries :    10000 : Total  Size=      40677 bytes  File Size  =       4586 *
*Baskets :        1 : Basket Size=      32000 bytes  Compression=   6.98     *
*............................................................................*
*Br    1 :E         : E[n]/F                                                 *
*Entries :    10000 : Total  Size=     228491 bytes  File Size  =     183329 *
*Baskets :        8 : Basket Size=      32000 bytes  Compression=   1.19     *
*............................................................................*
*Br    2 :px        : px[n]/F                                                *
*Entries :    10000 : Total  Size=     228505 bytes  File Size  =     187776 *
*Baskets :        8 : Basket Size=      32000 bytes  Compression=   1.16     *
*............................................................................*
*Br    3 :py        : py[n]/F                                                *
*Entries :    10000 : Total  Size=     228505 bytes  File Size  =     187842 *
*Baskets :        8 : Basket Size=      32000 bytes  Compression=   1.16     *
*............................................................................*
*Br    4 :pz        : pz[n]/F                                                *
*Entries :    10000 : Total  Size=     228505 bytes  File Size  =     184061 *
*Baskets :        8 : Basket Size=      32000 bytes  Compression=   1.18     *
*............................................................................*
*Br    5 :neuETot   : neuETot/F                                              *
*Entries :    10000 : Total  Size=      40719 bytes  File Size  =      26283 *
*Baskets :        1 : Basket Size=      32000 bytes  Compression=   1.22     *
*............................................................................*
*Br    6 :nGen      : nGen/I                                                 *
*Entries :    10000 : Total  Size=      40698 bytes  File Size  =        263 *
*Baskets :        1 : Basket Size=      32000 bytes  Compression= 121.66     *
*............................................................................*
*Br    7 :genE      : genE[nGen]/F                                           *
*Entries :    10000 : Total  Size=     161217 bytes  File Size  =     118567 *
*Baskets :        6 : Basket Size=      32000 bytes  Compression=   1.30     *
*............................................................................*
*Br    8 :genPx     : genPx[nGen]/F                                          *
*Entries :    10000 : Total  Size=     161229 bytes  File Size  =     122566 *
*Baskets :        6 : Basket Size=      32000 bytes  Compression=   1.25     *
*............................................................................*
*Br    9 :genPy     : genPy[nGen]/F                                          *
*Entries :    10000 : Total  Size=     161229 bytes  File Size  =     122616 *
*Baskets :        6 : Basket Size=      32000 bytes  Compression=   1.25     *
*............................................................................*
*Br   10 :genPz     : genPz[nGen]/F                                          *
*Entries :    10000 : Total  Size=     161229 bytes  File Size  =     118466 *
*Baskets :        6 : Basket Size=      32000 bytes  Compression=   1.30     *
*............................................................................*
*Br   11 :type      : type[nGen]/I                                           *
*Entries :    10000 : Total  Size=     161210 bytes  File Size  =      14124 *
*Baskets :        6 : Basket Size=      32000 bytes  Compression=  10.88     *
*............................................................................*
*Br   12 :genNeuE   : genNeuE/F                                              *
*Entries :    10000 : Total  Size=      40719 bytes  File Size  =      20990 *
*Baskets :        1 : Basket Size=      32000 bytes  Compression=   1.52     *
*............................................................................*
root [3] 
#endif