#!/usr/bin/env python

# Original author: Nikiforos Nikiforou (CERN, May 2014)

from pyLCIO import EVENT, UTIL, IOIMPL
import ROOT

import sys, math
print "sys.argv = ", sys.argv
if not len(sys.argv)>=2:  raise(Exception, "Must specify inputFiles as argument!")

# Create a reader and open an LCIO file
reader = IOIMPL.LCFactory.getInstance().createLCReader()

    
vec = ROOT.std.vector("string")()

for fn in sys.argv[1:]:
    vec.push_back(fn)
    
reader.open( vec)

# Loop over the events
for event in reader:
    # get a hit collection
    
    hitId=0;
  
    for collname in ['BarrelEcalHits','EndcapEcalHits']:
        
        coll=0
        try:
            coll = event.getCollection( collname )
        except:
            continue
        
        print "Printing hits in",collname
        
        for hit in coll:
            hit_x = hit.getPosition()[0]
            hit_y = hit.getPosition()[1]
            hit_z = hit.getPosition()[2]
            print "%d. x=%f, y=%f, z=%f, R=%f, phi=%f" % (hitId,hit_x,hit_y,hit_z, math.sqrt(hit_x*hit_x+hit_y*hit_y), math.atan2(hit_y,hit_x))
            hitId = hitId + 1 

#if __name__ == '__main__':
   #rep = ''
   #while not rep in [ 'q', 'Q' ]:
      #rep = raw_input( 'enter "q" to quit: ' )
      #if 1 < len(rep):
         #rep = rep[0]