#include <iostream>
#include <fstream>
#include <cstring>

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

#include "Event.h"
#include "treeGenFunctions.h"

using namespace std;

int main(){

 printBanner();

 char fileName[]="bmData.root", treeName[]="BlackMaxData", branchName[]="Event_", dataName[]="output.txt";

  //Generating File
  TFile * file = TFile::Open(fileName, "RECREATE", "A BlackMax simulation File"); //chopo look at TFile Open
  cout << " fileGen REPORT: File \"" << fileName << "\" created" << endl;

  TTree * tree= new TTree(treeName,"A BlackMax simulation TTree");
  cout << " fileGen REPORT: Tree \"" << treeName << "\" created" << endl;


  Event * event=0;
  Int_t bsize = 64000;

  tree->Branch(branchName, "Event", &event, bsize, 2);
  cout << " fileGen REPORT: Branch \"" << branchName << "\" created" << endl;

  //This data is includes and estracted from "output.txt" in the full version
  Int_t numberSimulations = 5;
  Int_t EPArraySize  = 5;       // 1 Energy + 3 Momentum (Px, Py, Pz) + 1 Extra dimention 
  
  //Accesing BM Data
  cout << " fileGen REPORT: Accessing \"" << dataName << "\" data" << endl;
  ifstream bmInput;
  bmInput.open(dataName);

    for(Int_t i=0; i<numberSimulations && bmInput.good(); i++){ 

      event = new Event();
      event->Build(EPArraySize, tree, bmInput);
      tree->Fill();
    }

    cout << "\n******************************************************************************\n";
    cout << " fileGen REPORT: Finished populating \"" << branchName << "\" branch" << endl;
    cout << " fileGen REPORT: Finished populating \"" << treeName    << "\" tree" << endl;
    
  bmInput.close();

  cout << "\n******************************************************************************\n\n";
  file->Write();
  cout << " fileGen REPORT: Finished writing to \"" << fileName << "\" file" << endl;
  file->Close();
  cout << " fileGen REPORT: \"" << fileName << "\" Closed" << endl;
  cout << " fileGen REPORT: job Done!\n" << endl;
  return 0;
}