"""
\namespace xboa::examples::Example_2

Example code to read a file and make some plots

\include xboa/examples/Example_2.py
"""

#############
# EXAMPLE 2 #
# PLOTTING  #
#############
#
# Load a data file; make some plots
#

#try to import the XBOA library - if this fails, review the installation procedure
from xboa  import *
from xboa.Hit   import Hit
from xboa.Bunch import Bunch
import sys

#some input data

if (len(sys.argv) < 2):
    print "No input data supplied. Fail. "
    raise SystemExit
#do nothing
    
filename = sys.argv[1]
filetype = "maus_root_virtual_hit"



print '========= XBOA example 2 ========='

#try to load an input file
#this will load the for009 file, and make a list of "bunches", one for each region
#a list is a python version of an array
print "This example shows how to make plots\nYou will need to have access to either ROOT library or the matplotlib library to make plots"
print "First loading the data... "
bunch_list = Bunch.new_list_from_read_builtin(filetype, filename)
print "Loaded"

#make some plots
#first try to make plots with ROOT if PyROOT exists
try:
  print 'Trying to make some plots using PyROOT plotting package'
  Common.has_root() #check for PyRoot library
  #momentum distribution at start and end
  bunch_list[0] .root_histogram('px', 'MeV/c')
  bunch_list[-1].root_histogram('px', 'MeV/c')

  #energy-time scatter plot at start
  bunch_list[0].root_scatter_graph('t', 'energy', 'ns', 'MeV/c')

  #histogram at start; note that in the first instance it *looks* like a scatter plot
  #but it is not; hence I draw the same histogram twice, once as a pseudo-scatter
  #and once as a contour plot
  bunch_list[0].root_histogram('t', 'ns', 'energy', 'MeV/c')
  (canvas, hist) = bunch_list[0] .root_histogram('t', 'ns', 'energy', 'MeV/c')
  hist.Draw('CONT')
  canvas.Update()

  #evolution of RMS emittance in x along the beamline
  Bunch.root_graph(bunch_list, 'mean', ['z'], 'mean', ['x'],     'm', 'mm')
  Bunch.root_graph(bunch_list, 'mean', ['z'], 'mean', ['pz'], 'm', 'mm')
*********************
## added by Tim **************	
*************************
  Bunch.root_graph(bunch_list, 'mean', ['z'], 'beta', ['x','y'],     'm', 'mm')
  Bunch.root_graph(bunch_list, 'mean', ['z'], 'emittance', ['x','y'], 'm', 'mm')
***********************************
except ImportError:
  print "PyROOT not detected - skipping PyROOT graphics"


#now wait for user to review plots before ending
print 'Press <return> key to finish'
raw_input()