#!/bin/python

from sys import argv
from numpy import shape
from Modules.Spectrum import drawSpectrum, drawSpectrum_normaLepton
from Modules.Angle import drawAngle, drawRadial
from Modules.Timing import drawTiming
from Modules.Generation import drawGeneration
from Modules.Map import drawMap

def error(error_type):
   print error_type
   print "Give at least 2 arguments"
   print "first argument could be:"
   print "  - spectrum (flux versus energy)"
   print "  - angle (angle versus energy)"
   print "  - radial (angular distribution)"
   print "  - timing (time distribution)"
   print "  - generation (generation histogram)"
   print "  - maphisto"
   print "  - map"
   print "others arguments are the folders files to plot"
   exit()

if shape(argv)[0] < 3:
   error("not enough argument")

elif argv[1] == "spectrum": 
   drawSpectrum(argv[2:],PlotAnalytic=True)

elif argv[1] == "spectrumLept": 
   drawSpectrum_normaLepton(argv[2:])

elif argv[1] == "angle":
   drawAngle(argv[2:])

elif argv[1] == "radial":
   drawRadial(argv[2:])

elif argv[1] == "timing":
   drawTiming(argv[2:])

elif argv[1] == "generation":
   drawGeneration(argv[2:])

elif argv[1] == "map":
   drawMap(argv[2:])

else:
   error("bad 1st argument")