#include "test_profile_fit.hh"
#include "EventFile.hh"
#include "Det.hh"
#include "Trk.hh"

int main(int argc, char *argv[] ){

  EventFile f( "/sps/km3net/users/jseneca/aanet/data/km3_v5/mcv5.1.genhen_nueCC.km3_AAv1.jte.jchain.aashower.94.root" );
  f.set_index( 2147 );
  Det det( "/sps/km3net/users/jseneca/aanet/data/km3_v5/KM3NeT_-00000001_20171212.detx" );
  
  Trk trk = f.evt.mc_trks[0];
  
  vector<Hit> pre_hits = f.evt.hits;
  det.apply( pre_hits );
  vector<Hit> hits;
  foreach( h, pre_hits ){
    h.t = h.t - ( f.evt.mc_t - f.evt.t.AsDouble()*1e9 );
    double D = ( f.evt.mc_trks[0].pos - h.pos ).len();
    double dt = abs( h.t - D/v_light );
    if( D < 500 && dt < 100){
      hits.push_back( h );
    };
  };
    
  print( "len hits: ", hits.size() );

  assert( hits.size() );

  JppShower* pdf = new JppShower( );

  cout << "Made pdf here" << endl;
  
  pdf->init_event( det, hits, trk.pos, 300, true );

  cout << "Init event here" << endl;

  JppShowerFit jfit( "5d_pdf",
		     pdf,
		     "Minuit2",
		     "MIGRAD",
		     1);
  
  cout << "Made jfit here" << endl;

  jfit.fix_vars( 0, 1, 2, 5 );
  
  Trk fit_5d = jfit.fit( trk, det, hits, 1, true );

  print( fit_5d );
  
  return 1;

};