#!/usr/bin/env python

# This example shows basic a basic aanet event loop and the many
# options there are for printing event information. 
#
#


"""
 Usage print_event.py -f input_file

 Prints event information for the first events in a
 file. The file can be any format readable by aanet.

 options:
 -f : input file default=/sps/km3net/users/heijboer/example_files/JGandalf_aanet_muon-CC_3-100GeV_1.root
 -n : process this many events,                      default=10
 -p : print 1-line event summaries,                  default=True
 -d : print all members of event structures (dump)   default=False
 -u : print the "usr" information                    default=False
 -m : print MC tracks showing parent-child relation  default=False
 -a : print ascii format,                            default=False
 -j : print json                                     default=False
 -t : print tree information (for root files)        default=False
 -q : print info on first 10 hits                    default=False
 -r : print reconstructed tracks                     default=False
 -H : print Header information (if available)        default=False
 -h : this help message and exit                     default=False       
 -i : Python interative mode (prompt when done)      default=False  
"""

import sys, aa
from ROOT import *




# aa.py contains helper to deal with cmd line options
options = aa.Options( __doc__, sys.argv[1:] ) 

# open the Event File ; can by any format understood by aanet
f = EventFile( options.f )

if options.t :
    if f.roottree() : f.roottree().Print()
    sys.exit()

if options.H :
    print f.header
    sys.exit()

    
for evt in f :
    
    if options.p : print evt # calls Evt::__str__()
    
    if options.j : print jsons( evt )

    if options.d : dump( evt )

    if options.u : evt.printusr()

    if options.m : print pretty_mc_tracks( evt )
    
    if options.q : 
        print "-------------------------------------------------"
        print " hits : ", evt.hits.size()
        for h in evt.hits[:10]:
            print h, h.pmt_id, h.dom_id, h.channel_id, h.pos

        print
        print "MC hits : ", evt.mc_hits.size()
        for h in evt.mc_hits[:10]:
            print h


    if options.a :
        s = ostringstream()
        write( evt, s)
        print s.str()

    if options.r :

        for t in evt.trks :
            print t, t.rec_type, list(t.rec_stages)
        print "----------------------------------------------"

        
    if f.index >= options.n :
        break