config = configp.start()

if args is None:

args = parser.start()

elif isinstance(args,basestring):

args = parser.start(args)

cedfile = args.ced

stdfile = args.std

plotFields = args.field

multi = args.multi

print_shapes = args.print_shapes

print_global_atts = args.print_global_atts

print_axis = args.print_axis

print_list_synth = args.print_list_synth

terrain = args.terrain

slope = args.slope

meteo = args.meteo

valid = args.valid

prof = args.prof

nearest = args.nearest

noplot = args.no_plot

""" print synhesis availables """

if print_list_synth:

synth_folder=fh.set_working_files(config=config)

out=os.listdir(synth_folder)

print "Synthesis directories available:"

for f in out:

print f

usr_input = raw_input('\nIndicate directory: ')

out=os.listdir(synth_folder+'/'+usr_input)

print "\nSyntheses available:"

out.sort()

for f in out:

if f[-3:]=='cdf': print f

print '\n'

sys.exit()

""" retrieves synthesis and flight instances

from AircraftAnalysis

"""

SYNTH, FLIGHT, TERRAIN = fh.set_working_files(cedfile=cedfile,

stdfile=stdfile,

config=config)

""" print shape of attribute arrays """

if print_shapes:

SYNTH.print_shapes()

if not print_global_atts:

sys.exit()

""" print global attirutes of cedric synthesis """

if print_global_atts:

SYNTH.print_global_atts()

sys.exit()

""" print axis values """

if print_axis:

for ax in print_axis:

if ax.isupper():

ax=ax.lower()

SYNTH.print_axis(ax)

sys.exit()

""" print synthesis time """

# print "Synthesis start time :%s" % SYNTH.start

# print "Synthesis end time :%s\n" % SYNTH.end

""" make synthesis plots """

if plotFields:

for f in plotFields:

P=Plotter.plot_synth(SYNTH,FLIGHT,TERRAIN,

var=f,

wind=args.wind,

panel=args.panel,

slicem = args.slicem,

slicez = args.slicez,

slice = args.slice,

# azimuth = args.azimuth,

# distance = args.distance,

zoomIn=args.zoomin,

mask = args.mask,

config=config)

""" make terrain plots """

if terrain or slope:

# P[0] might produce error if P is not a list,

# check in ploth_synth.cross_section

Plotter.plot_terrain(P[0],

terrain=terrain,

slope=slope,

terrain_file=config['filepath_dtm'])

""" make flight level meteo plot """

if meteo:

Plotter.plot_flight_meteo(SYNTH,FLIGHT)

""" compare synth and flight level """

if valid:

out = Plotter.compare_synth_flight(SYNTH,FLIGHT,

level=valid,

zoomin=args.zoomin,

noplot=noplot)

# if 'wind_profiler' in config:

# case=int(cedfile[1:3])

# Plotter.compare_with_windprof(SYNTH,

# location=config['wind_profiler'],

# case=case)

return out

""" make profile from synthesis """

if prof:

if nearest is None:

markers=['o','s','D','*']

out = Plotter.make_synth_profile(SYNTH,coords=prof,

markers=markers,

noplot=noplot)

else:

' (4.5km,n=12) or (7.0km,n=30) seem good choices '

out = Plotter.make_synth_profile_withnearest(SYNTH,

target_latlon=prof,

max_dist=nearest[0][0], # [km]

n_neigh =nearest[0][1])

try:

' if P exists '

for i,p in enumerate(prof):

lat,lon = p

P.haxis.scatter(lon,lat,

s=config['sounding_size'],

c=config['sounding_color'],

marker=config['sounding_marker'],

lw=3)

except UnboundLocalError:

' if P does not exist just pass '

pass

return out

# if turbulence:

# Plotter.print_covariance(SYNTH,FLIGHT)

# Plotter.print_correlation(SYNTH,FLIGHT)

# Plotter.plot_wind_comp_var(SYNTH,FLIGHT)

# Plotter.plot_tke(SYNTH,FLIGHT)

# Plotter.plot_vertical_heat_flux(SYNTH,FLIGHT)

# Plotter.plot_vertical_momentum_flux(SYNTH,FLIGHT,config['filepath_dtm'])

# Plotter.plot_turbulence_spectra(SYNTH,FLIGHT)

if multi:

plt.close('all')

return P

else:

''' use this one with ipython '''

plt.show(block=False)

''' use this one with the shell '''