Subversion Repositories lagranto.ecmwf

# =============================================================================

#  Imports

# Check whether the input file is available

if os.path.exists(inpfile_folder+inpfile_filename):

    print 'input file found: ' + inpfile_folder + inpfile_filename

else:

    sys.exit( 'input file is missing : ' + inpfile_folder + inpfile_filename )

# Check whether field is available on input file

cmd        = 'getvars ' + inpfile_folder + '/' + inpfile_filename

proc       = subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)

(out, err) = proc.communicate()  

ok         = any( (elem.strip() == field_fieldname) for elem in out.split('\n') ) 

# If not available, try to calculate it with p2s

if ( ok == False ):

    print 'Try to calculate '+field_fieldname+' with p2s'

    cmd = 'p2s ' + inpfile_folder + '/' + inpfile_filename

    print 'NOT YET IMPLEMTED'

if ok == False:

    sys.exit('variable '+field_fieldname+' missing on '+inpfile_filename);

# ----------------------------------------------------------------------------

# Set the color table

# ----------------------------------------------------------------------------

color_split = options.color.split(',')

if len(color_split) == 5:

    color_cmin = color_split[0]

    color_cmax = color_split[1]

    color_mode = color_split[2]

    color_ncol = color_split[3]

    color_name = color_split[4]

elif len(color_split) == 4:

    color_cmin = color_split[0]

    color_cmax = color_split[1]

    color_mode = color_split[2]

    color_ncol = color_split[3]

    color_name = 'adjust'

elif len(color_split) == 3:

    color_cmin = color_split[0]

    color_cmax = color_split[1]

    color_mode = color_split[2]

    color_ncol = 18

    color_name = 'hotcold_18lev'

elif len(color_split) == 2:

    color_cmin = color_split[0]

    color_cmax = color_split[1]

    color_mode = 'fill'

    color_ncol = '18'

    color_name = 'hotcold_18lev'

elif (color_split[0] == 'PV.default') | ( color_split[0] == 'NONE' ) & ( field_fieldname == 'PV' ):

    color_cmin = '-0.5'

    color_cmax = '10.'

    color_mode = 'fill'

    color_ncol = 'adjust'

    color_name = 'PV.default'

elif color_split[0] == 'NONE':

    color_cmin = 'p95'

    color_cmax = 'p05'

    color_mode = 'fill'

    color_ncol = '18'

    color_name = 'hotcold_18lev'

# ----------------------------------------------------------------------------

# Set the horizontal domain

# ----------------------------------------------------------------------------

# Set the geographical projection 

projection = options.projection

print projection      

domain_split = options.domain.split(',')

print domain_split

if len(domain_split) == 4:

    domain_xmin = domain_split[0]

    domain_xmax = domain_split[1]

    domain_ymin = domain_split[2]

    domain_ymax = domain_split[3]

elif ( len(domain_split) == 1 ) & (options.domain == 'europe' ):

    domain_xmin = -20

    domain_xmax = 40

    domain_ymin = 30

    domain_ymax = 75

elif ( len(domain_split) == 1 ) & (options.domain == 'natlantic' ):

    domain_xmin = -100

    domain_xmax = 10

    domain_ymin = 10

    domain_ymax = 75

elif ( len(domain_split) == 1 ) & (options.domain == 'global' ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = -90

    domain_ymax = 90

elif ( len(domain_split) == 1 ) & (options.domain == 'nhem' ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = 0

    domain_ymax = 90

elif ( len(domain_split) == 1 ) & (options.domain == 'shem' ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = -90

    domain_ymax = 0

elif ( len(domain_split) == 2 ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = domain_split[0]

    domain_ymax = domain_split[1]

elif ( projection == 'np' ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = 30

    domain_ymax = 90

elif ( projection == 'sp' ):

    domain_xmin = -180

    domain_xmax = 180

    domain_ymin = -90

    domain_ymax = -30

else:

    domain_xmin = 'adjust'

    domain_xmax = 'adjust'

    domain_ymin = 'adjust'

    domain_ymax = 'adjust'

# Check that the geographical domain is reasonable

if ( domain_xmin != 'adjust' ):

    if (domain_xmin >= domain_xmax) | (domain_ymin >= domain_ymax):

        sys.exit('domain problem:')

# ----------------------------------------------------------------------------

# Set the settings for vertical cross section

# ----------------------------------------------------------------------------

vertical_split = options.vertical.split(',')

if len(vertical_split) == 7:

    vertical_lon1 = vertical_split[0]

    vertical_lat1 = vertical_split[1]

    vertical_lon2 = vertical_split[2]

    vertical_lat2 = vertical_split[3]

    vertical_lev1 = vertical_split[4]

    vertical_lev2 = vertical_split[5]

    vertical_type = vertical_split[6]

if len(vertical_split) == 6:

    vertical_lon1 = vertical_split[0]

    vertical_lat1 = vertical_split[1]

    vertical_lon2 = vertical_split[2]

    vertical_lat2 = vertical_split[3]

    vertical_lev1 = vertical_split[4]

    vertical_lev2 = vertical_split[5]

    vertical_type = 'linear'

if len(vertical_split) == 4:

    vertical_lon1 = vertical_split[0]

    vertical_lat1 = vertical_split[1]

    vertical_lon2 = vertical_split[2]

    vertical_lat2 = vertical_split[3]

    vertical_lev1 = 1030

    vertical_lev2 = 100

    vertical_type = 'linear'

elif ( len(vertical_split) == 1 ) & (options.vertical == 'zurich.ns' ):

    vertical_lon1 = 8.538

    vertical_lat1 = 47.3690 - 7.5

    vertical_lon2 = 8.538

    vertical_lat2 = 47.3690 + 7.5

    vertical_lev1 = 1030

    vertical_lev2 = 100

    vertical_type = 'linear'

# ----------------------------------------------------------------------------

# Set the profile settings

# ----------------------------------------------------------------------------

profile_split = options.profile.split(',')

if len(profile_split) == 4:

    profile_lon  = profile_split[0]

    profile_lat  = profile_split[1]

    profile_lev1 = profile_split[2]

    profile_lev2 = profile_split[3]

elif len(profile_split) == 2:

    profile_lon  = profile_split[0]

    profile_lat  = profile_split[1]

    profile_lev1 = 1030

    profile_lev2 = 100

# ----------------------------------------------------------------------------

# Horizontal cross section 

# ----------------------------------------------------------------------------

if ( mode == 'horizontal' ):

    os.system('python ' + SCRIPT_PATH + '/ext_scripts/horizontal.py ' + 

        ' '.join([inpfile_filename, 

                  inpfile_folder, 

                  field_fieldname, 

                  str(field_level),

                  str(domain_xmin),str(domain_xmax),str(domain_ymin),str(domain_ymax), 

                  str(options.npixel), 

                  wsname, 

                  figpath,figname, 

                  str(color_cmin), str(color_ncol), str(color_cmax), color_name, 

                  field_unit,projection,color_mode]))

# ----------------------------------------------------------------------------

# Vertical cross section 

# ----------------------------------------------------------------------------

if ( mode == 'vertical' ) :

    os.system('python ' + SCRIPT_PATH + '/ext_scripts/vertical.py ' +

          ' '.join([inpfile_filename, 

                    inpfile_folder, 

                    field_fieldname,

                    str(vertical_lon1), str(vertical_lon2), 

                    str(vertical_lat1), str(vertical_lat2), 

                    str(vertical_lev1), str(vertical_lev2),

                    vertical_type, 

                    str(options.npixel), 

                    wsname, 

                    figpath, figname,

                    str(color_cmin), str(color_ncol), str(color_cmax), 

                    color_name,color_mode ]))

# ----------------------------------------------------------------------------

# Skew-T/log-P (skewtlogp)

# ----------------------------------------------------------------------------

    os.system('python ' + SCRIPT_PATH + '/ext_scripts/skewtlogp.py ' +

          ' '.join([inpfile_filename,inpfile_folder,

          str(profile_lon), str(profile_lat), str(profile_lev1), str(profile_lev2),

          str(options.npixel), wsname, figpath, figname ]))

# ----------------------------------------------------------------------------

# Show the figure if requested

# ----------------------------------------------------------------------------

if options.show:

    cmd = 'xv ' + figpath + '/' + figname + '.png'

    subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)

if options.showremove:

    cmd = 'xv ' + figpath + '/' + figname + '.png; rm ' + figpath + '/' + figname + '.png'

    subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)

if options.labelbar | ( mode == 'colorbar' ):

    figname = 'quickview.color'

    os.system('python ' + SCRIPT_PATH + '/ext_scripts/colorbar.py ' +

          ' '.join([str(options.npixel), wsname, figpath, figname])) 

        cmd = 'xv ' + figpath + '/quickview.color.png;' + 'rm ' + figpath + '/quickview.color.png'

        subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)

    else:

        cmd = 'xv ' + figpath + '/quickview.color.png;'

        subprocess.Popen(cmd,stdout=subprocess.PIPE,shell=True)