#!/usr/local/bin/python -i

import aa
from ROOT import *

f = EventFile("/sps/km3net/users/heijboer/example_files/km3_v4_nueCC_8.JTE_r2356.aareco.root")



for evt in f :

    rec = evt.trks[2]
    nu  = evt.mc_trks[0]
    print " angular error  : ", angle_between( rec.dir, nu.dir )
    print " position error : ", (rec.pos - nu.pos ).len()
    print " energy ratio   : ", rec.E / nu.E
    print 

    if f.index > 10 : break



# note E exits because root has opened the file, so the next line
# can be omitted
 
E = f.rootfile().Get("E")

E.SetAlias("angdeg","57.2*acos(mc_trks[0].dir.x*trks[2].dir.x+mc_trks[0].dir.y*trks[2].dir.y+mc_trks[0].dir.z*trks[2].dir.z)");
E.SetAlias("Enu","mc_trks[0].E");

E.Draw("angdeg:log10(Enu)","angdeg<20","colz")