Coverage for /usr/local/lib/python3.7/dist-packages/pyrocko/plot/gmtpy.py: 77%

100

101

102

103

# http://pyrocko.org - GPLv3

# The Pyrocko Developers, 21st Century

# ---|P------/S----------~Lg----------

'''A Python interface to GMT.'''

# This file is part of GmtPy (http://emolch.github.io/gmtpy/)

# See there for copying and licensing information.

from __future__ import print_function, absolute_import

import subprocess

try:

from StringIO import StringIO as BytesIO

except ImportError:

from io import BytesIO

import re

import os

import sys

import shutil

from os.path import join as pjoin

import tempfile

import random

import logging

import math

import numpy as num

import copy

from select import select

from scipy.io import netcdf

from pyrocko import ExternalProgramMissing

try:

newstr = unicode

except NameError:

newstr = str

find_bb = re.compile(br'%%BoundingBox:((\s+[-0-9]+){4})')

find_hiresbb = re.compile(br'%%HiResBoundingBox:((\s+[-0-9.]+){4})')

encoding_gmt_to_python = {

'isolatin1+': 'iso-8859-1',

'standard+': 'ascii',

'isolatin1': 'iso-8859-1',

'standard': 'ascii'}

for i in range(1, 11):

encoding_gmt_to_python['iso-8859-%i' % i] = 'iso-8859-%i' % i

def have_gmt():

try:

get_gmt_installation('newest')

return True

except GMTInstallationProblem:

return False

def check_have_gmt():

if not have_gmt():

raise ExternalProgramMissing('GMT is not installed or cannot be found')

def have_pixmaptools():

for prog in [['pdftocairo'], ['convert'], ['gs', '-h']]:

try:

p = subprocess.Popen(

prog,

stdout=subprocess.PIPE, stderr=subprocess.PIPE)

(stdout, stderr) = p.communicate()

except OSError:

return False

return True

class GmtPyError(Exception):

pass

class GMTError(GmtPyError):

pass

class GMTInstallationProblem(GmtPyError):

pass

def convert_graph(in_filename, out_filename, resolution=75., oversample=2.,

width=None, height=None, size=None):

_, tmp_filename_base = tempfile.mkstemp()

try:

if out_filename.endswith('.svg'):

fmt_arg = '-svg'

tmp_filename = tmp_filename_base

oversample = 1.0

else:

fmt_arg = '-png'

tmp_filename = tmp_filename_base + '-1.png'

if size is not None:

scale_args = ['-scale-to', '%i' % int(round(size*oversample))]

elif width is not None:

scale_args = ['-scale-to-x', '%i' % int(round(width*oversample))]

elif height is not None:

scale_args = ['-scale-to-y', '%i' % int(round(height*oversample))]

else:

scale_args = ['-r', '%i' % int(round(resolution * oversample))]

try:

subprocess.check_call(

['pdftocairo'] + scale_args +

[fmt_arg, in_filename, tmp_filename_base])

except OSError as e:

raise GmtPyError(

'Cannot start `pdftocairo`, is it installed? (%s)' % str(e))

if oversample > 1.:

try:

subprocess.check_call([

'convert',

tmp_filename,

'-resize', '%i%%' % int(round(100.0/oversample)),

out_filename])

except OSError as e:

raise GmtPyError(

'Cannot start `convert`, is it installed? (%s)' % str(e))

else:

if out_filename.endswith('.png') or out_filename.endswith('.svg'):

shutil.move(tmp_filename, out_filename)

else:

try:

subprocess.check_call(

['convert', tmp_filename, out_filename])

except Exception as e:

raise GmtPyError(

'Cannot start `convert`, is it installed? (%s)'

% str(e))

except Exception:

raise

finally:

if os.path.exists(tmp_filename_base):

os.remove(tmp_filename_base)

if os.path.exists(tmp_filename):

os.remove(tmp_filename)

def get_bbox(s):

for pat in [find_hiresbb, find_bb]:

m = pat.search(s)

if m:

bb = [float(x) for x in m.group(1).split()]

return bb

raise GmtPyError('Cannot find bbox')

def replace_bbox(bbox, *args):

def repl(m):

if m.group(1):

return ('%%HiResBoundingBox: ' + ' '.join(

'%.3f' % float(x) for x in bbox)).encode('ascii')

else:

return ('%%%%BoundingBox: %i %i %i %i' % (

int(math.floor(bbox[0])),

int(math.floor(bbox[1])),

int(math.ceil(bbox[2])),

int(math.ceil(bbox[3])))).encode('ascii')

pat = re.compile(br'%%(HiRes)?BoundingBox:((\s+[0-9.]+){4})')

if len(args) == 1:

s = args[0]

return pat.sub(repl, s)

else:

fin, fout = args

nn = 0

for line in fin:

line, n = pat.subn(repl, line)

nn += n

fout.write(line)

if nn == 2:

break

if nn == 2:

for line in fin:

fout.write(line)

def escape_shell_arg(s):

'''This function should be used for debugging output only - it could be

insecure.'''

if re.search(r'[^a-zA-Z0-9._/=-]', s):

return "'" + s.replace("'", "'\\''") + "'"

else:

return s

def escape_shell_args(args):

'''This function should be used for debugging output only - it could be

insecure.'''

return ' '.join([escape_shell_arg(x) for x in args])

golden_ratio = 1.61803

# units in points

_units = {

'i': 72.,

'c': 72./2.54,

'm': 72.*100./2.54,

'p': 1.}

inch = _units['i']

cm = _units['c']

# some awsome colors

tango_colors = {

'butter1': (252, 233, 79),

'butter2': (237, 212, 0),

'butter3': (196, 160, 0),

'chameleon1': (138, 226, 52),

'chameleon2': (115, 210, 22),

'chameleon3': (78, 154, 6),

'orange1': (252, 175, 62),

'orange2': (245, 121, 0),

'orange3': (206, 92, 0),

'skyblue1': (114, 159, 207),

'skyblue2': (52, 101, 164),

'skyblue3': (32, 74, 135),

'plum1': (173, 127, 168),

'plum2': (117, 80, 123),

'plum3': (92, 53, 102),

'chocolate1': (233, 185, 110),

'chocolate2': (193, 125, 17),

'chocolate3': (143, 89, 2),

'scarletred1': (239, 41, 41),

'scarletred2': (204, 0, 0),

'scarletred3': (164, 0, 0),

'aluminium1': (238, 238, 236),

'aluminium2': (211, 215, 207),

'aluminium3': (186, 189, 182),

'aluminium4': (136, 138, 133),

'aluminium5': (85, 87, 83),

'aluminium6': (46, 52, 54)

}

graph_colors = [tango_colors[_x] for _x in (

'scarletred2', 'skyblue3', 'chameleon3', 'orange2', 'plum2', 'chocolate2',

'butter2')]

def color(x=None):

'''Generate a string for GMT option arguments expecting a color.

If ``x`` is None, a random color is returned. If it is an integer, the

corresponding ``gmtpy.graph_colors[x]`` or black returned. If it is a

string and the corresponding ``gmtpy.tango_colors[x]`` exists, this is

returned, or the string is passed through. If ``x`` is a tuple, it is

transformed into the string form which GMT expects.

'''

if x is None:

return '%i/%i/%i' % tuple(random.randint(0, 255) for _ in 'rgb')

if isinstance(x, int):

if 0 <= x < len(graph_colors):

return '%i/%i/%i' % graph_colors[x]

else:

return '0/0/0'

elif isinstance(x, str):

if x in tango_colors:

return '%i/%i/%i' % tango_colors[x]

else:

return x

return '%i/%i/%i' % x

def color_tup(x=None):

if x is None:

return tuple([random.randint(0, 255) for _x in 'rgb'])

if isinstance(x, int):

if 0 <= x < len(graph_colors):

return graph_colors[x]

else:

return (0, 0, 0)

elif isinstance(x, str):

if x in tango_colors:

return tango_colors[x]

return x

_gmt_installations = {}

# Set fixed installation(s) to use...

# (use this, if you want to use different GMT versions simultaneously.)

# _gmt_installations['4.2.1'] = {'home': '/sw/etch-ia32/gmt-4.2.1',

# 'bin': '/sw/etch-ia32/gmt-4.2.1/bin'}

# _gmt_installations['4.3.0'] = {'home': '/sw/etch-ia32/gmt-4.3.0',

# 'bin': '/sw/etch-ia32/gmt-4.3.0/bin'}

# _gmt_installations['4.3.1'] = {'home': '/sw/share/gmt',

# 'bin': '/sw/bin' }

# ... or let GmtPy autodetect GMT via $PATH and $GMTHOME

def key_version(a):

a = a.split('_')[0] # get rid of revision id

return [int(x) for x in a.split('.')]

def newest_installed_gmt_version():

return sorted(_gmt_installations.keys(), key=key_version)[-1]

def all_installed_gmt_versions():

return sorted(_gmt_installations.keys(), key=key_version)

# To have consistent defaults, they are hardcoded here and should not be

# changed.

_gmt_defaults_by_version = {}

_gmt_defaults_by_version['4.2.1'] = r'''

# GMT-SYSTEM 4.2.1 Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = 255/255/255

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE = 12p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = 0/0/0

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

TICK_LENGTH = 0.075i

POLAR_CAP = 85/90

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = -0.75i/-0.75i

#-------- Color System Parameters -----------

COLOR_BACKGROUND = 0/0/0

COLOR_FOREGROUND = 255/255/255

COLOR_NAN = 128/128/128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = none

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

#-------- I/O Format Parameters -------------

D_FORMAT = %lg

FIELD_DELIMITER = tab

GRIDFILE_SHORTHAND = FALSE

GRID_FORMAT = nf

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = +D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_SYSTEM = other

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE'''

_gmt_defaults_by_version['4.3.0'] = r'''

# GMT-SYSTEM 4.3.0 Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = 255/255/255

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE_PRIMARY = 12p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = 0/0/0

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

POLAR_CAP = 85/90

TICK_LENGTH = 0.075i

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = BL/-0.75i/-0.75i

UNIX_TIME_FORMAT = %Y %b %d %H:%M:%S

#-------- Color System Parameters -----------

COLOR_BACKGROUND = 0/0/0

COLOR_FOREGROUND = 255/255/255

COLOR_NAN = 128/128/128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = none

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

#-------- I/O Format Parameters -------------

D_FORMAT = %lg

FIELD_DELIMITER = tab

GRIDFILE_SHORTHAND = FALSE

GRID_FORMAT = nf

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = +D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE'''

_gmt_defaults_by_version['4.3.1'] = r'''

# GMT-SYSTEM 4.3.1 Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = 255/255/255

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE_PRIMARY = 12p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = 0/0/0

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

POLAR_CAP = 85/90

TICK_LENGTH = 0.075i

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = BL/-0.75i/-0.75i

UNIX_TIME_FORMAT = %Y %b %d %H:%M:%S

#-------- Color System Parameters -----------

COLOR_BACKGROUND = 0/0/0

COLOR_FOREGROUND = 255/255/255

COLOR_NAN = 128/128/128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = none

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

#-------- I/O Format Parameters -------------

D_FORMAT = %lg

FIELD_DELIMITER = tab

GRIDFILE_SHORTHAND = FALSE

GRID_FORMAT = nf

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = +D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE'''

_gmt_defaults_by_version['4.4.0'] = r'''

# GMT-SYSTEM 4.4.0 [64-bit] Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = 255/255/255

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE_PRIMARY = 14p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = 0/0/0

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

POLAR_CAP = 85/90

TICK_LENGTH = 0.075i

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = BL/-0.75i/-0.75i

UNIX_TIME_FORMAT = %Y %b %d %H:%M:%S

#-------- Color System Parameters -----------

COLOR_BACKGROUND = 0/0/0

COLOR_FOREGROUND = 255/255/255

COLOR_NAN = 128/128/128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = lzw

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

#-------- I/O Format Parameters -------------

D_FORMAT = %lg

FIELD_DELIMITER = tab

GRIDFILE_SHORTHAND = FALSE

GRID_FORMAT = nf

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = +D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE

'''

_gmt_defaults_by_version['4.5.2'] = r'''

# GMT-SYSTEM 4.5.2 [64-bit] Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = white

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE_PRIMARY = 14p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = black

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

POLAR_CAP = 85/90

TICK_LENGTH = 0.075i

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = BL/-0.75i/-0.75i

UNIX_TIME_FORMAT = %Y %b %d %H:%M:%S

#-------- Color System Parameters -----------

COLOR_BACKGROUND = black

COLOR_FOREGROUND = white

COLOR_NAN = 128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = lzw

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

TRANSPARENCY = 0

#-------- I/O Format Parameters -------------

D_FORMAT = %.12lg

FIELD_DELIMITER = tab

GRIDFILE_FORMAT = nf

GRIDFILE_SHORTHAND = FALSE

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

NAN_RECORDS = pass

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE

'''

_gmt_defaults_by_version['4.5.3'] = r'''

# GMT-SYSTEM 4.5.3 (CVS Jun 18 2010 10:56:07) [64-bit] Defaults file

#-------- Plot Media Parameters -------------

PAGE_COLOR = white

PAGE_ORIENTATION = portrait

PAPER_MEDIA = a4+

#-------- Basemap Annotation Parameters ------

ANNOT_MIN_ANGLE = 20

ANNOT_MIN_SPACING = 0

ANNOT_FONT_PRIMARY = Helvetica

ANNOT_FONT_SIZE_PRIMARY = 14p

ANNOT_OFFSET_PRIMARY = 0.075i

ANNOT_FONT_SECONDARY = Helvetica

ANNOT_FONT_SIZE_SECONDARY = 16p

ANNOT_OFFSET_SECONDARY = 0.075i

DEGREE_SYMBOL = ring

HEADER_FONT = Helvetica

HEADER_FONT_SIZE = 36p

HEADER_OFFSET = 0.1875i

LABEL_FONT = Helvetica

LABEL_FONT_SIZE = 14p

LABEL_OFFSET = 0.1125i

OBLIQUE_ANNOTATION = 1

PLOT_CLOCK_FORMAT = hh:mm:ss

PLOT_DATE_FORMAT = yyyy-mm-dd

PLOT_DEGREE_FORMAT = +ddd:mm:ss

Y_AXIS_TYPE = hor_text

#-------- Basemap Layout Parameters ---------

BASEMAP_AXES = WESN

BASEMAP_FRAME_RGB = black

BASEMAP_TYPE = plain

FRAME_PEN = 1.25p

FRAME_WIDTH = 0.075i

GRID_CROSS_SIZE_PRIMARY = 0i

GRID_PEN_PRIMARY = 0.25p

GRID_CROSS_SIZE_SECONDARY = 0i

GRID_PEN_SECONDARY = 0.5p

MAP_SCALE_HEIGHT = 0.075i

POLAR_CAP = 85/90

TICK_LENGTH = 0.075i

TICK_PEN = 0.5p

X_AXIS_LENGTH = 9i

Y_AXIS_LENGTH = 6i

X_ORIGIN = 1i

Y_ORIGIN = 1i

UNIX_TIME = FALSE

UNIX_TIME_POS = BL/-0.75i/-0.75i

UNIX_TIME_FORMAT = %Y %b %d %H:%M:%S

#-------- Color System Parameters -----------

COLOR_BACKGROUND = black

COLOR_FOREGROUND = white

COLOR_NAN = 128

COLOR_IMAGE = adobe

COLOR_MODEL = rgb

HSV_MIN_SATURATION = 1

HSV_MAX_SATURATION = 0.1

HSV_MIN_VALUE = 0.3

HSV_MAX_VALUE = 1

#-------- PostScript Parameters -------------

CHAR_ENCODING = ISOLatin1+

DOTS_PR_INCH = 300

GLOBAL_X_SCALE = 1

GLOBAL_Y_SCALE = 1

N_COPIES = 1

PS_COLOR = rgb

PS_IMAGE_COMPRESS = lzw

PS_IMAGE_FORMAT = ascii

PS_LINE_CAP = round

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_VERBOSE = FALSE

TRANSPARENCY = 0

#-------- I/O Format Parameters -------------

D_FORMAT = %.12lg

FIELD_DELIMITER = tab

GRIDFILE_FORMAT = nf

GRIDFILE_SHORTHAND = FALSE

INPUT_CLOCK_FORMAT = hh:mm:ss

INPUT_DATE_FORMAT = yyyy-mm-dd

IO_HEADER = FALSE

N_HEADER_RECS = 1

NAN_RECORDS = pass

OUTPUT_CLOCK_FORMAT = hh:mm:ss

OUTPUT_DATE_FORMAT = yyyy-mm-dd

OUTPUT_DEGREE_FORMAT = D

XY_TOGGLE = FALSE

#-------- Projection Parameters -------------

ELLIPSOID = WGS-84

MAP_SCALE_FACTOR = default

MEASURE_UNIT = inch

#-------- Calendar/Time Parameters ----------

TIME_FORMAT_PRIMARY = full

TIME_FORMAT_SECONDARY = full

TIME_EPOCH = 2000-01-01T00:00:00

TIME_IS_INTERVAL = OFF

TIME_INTERVAL_FRACTION = 0.5

TIME_LANGUAGE = us

TIME_UNIT = d

TIME_WEEK_START = Sunday

Y2K_OFFSET_YEAR = 1950

#-------- Miscellaneous Parameters ----------

HISTORY = TRUE

INTERPOLANT = akima

LINE_STEP = 0.01i

VECTOR_SHAPE = 0

VERBOSE = FALSE

'''

_gmt_defaults_by_version['5.1.2'] = r'''

# GMT 5.1.2 Defaults file

# vim:sw=8:ts=8:sts=8

# $Revision: 13836 $

# $LastChangedDate: 2014-12-20 03:45:42 -1000 (Sat, 20 Dec 2014) $

# COLOR Parameters

COLOR_BACKGROUND = black

COLOR_FOREGROUND = white

COLOR_NAN = 127.5

COLOR_MODEL = none

COLOR_HSV_MIN_S = 1

COLOR_HSV_MAX_S = 0.1

COLOR_HSV_MIN_V = 0.3

COLOR_HSV_MAX_V = 1

# DIR Parameters

DIR_DATA =

DIR_DCW =

DIR_GSHHG =

# FONT Parameters

FONT_ANNOT_PRIMARY = 14p,Helvetica,black

FONT_ANNOT_SECONDARY = 16p,Helvetica,black

FONT_LABEL = 14p,Helvetica,black

FONT_LOGO = 8p,Helvetica,black

FONT_TITLE = 24p,Helvetica,black

# FORMAT Parameters

FORMAT_CLOCK_IN = hh:mm:ss

FORMAT_CLOCK_OUT = hh:mm:ss

FORMAT_CLOCK_MAP = hh:mm:ss

FORMAT_DATE_IN = yyyy-mm-dd

FORMAT_DATE_OUT = yyyy-mm-dd

FORMAT_DATE_MAP = yyyy-mm-dd

FORMAT_GEO_OUT = D

FORMAT_GEO_MAP = ddd:mm:ss

FORMAT_FLOAT_OUT = %.12g

FORMAT_FLOAT_MAP = %.12g

FORMAT_TIME_PRIMARY_MAP = full

FORMAT_TIME_SECONDARY_MAP = full

FORMAT_TIME_STAMP = %Y %b %d %H:%M:%S

# GMT Miscellaneous Parameters

GMT_COMPATIBILITY = 4

GMT_CUSTOM_LIBS =

GMT_EXTRAPOLATE_VAL = NaN

GMT_FFT = auto

GMT_HISTORY = true

GMT_INTERPOLANT = akima

GMT_TRIANGULATE = Shewchuk

GMT_VERBOSE = compat

GMT_LANGUAGE = us

# I/O Parameters

IO_COL_SEPARATOR = tab

IO_GRIDFILE_FORMAT = nf

IO_GRIDFILE_SHORTHAND = false

IO_HEADER = false

IO_N_HEADER_RECS = 0

IO_NAN_RECORDS = pass

IO_NC4_CHUNK_SIZE = auto

IO_NC4_DEFLATION_LEVEL = 3

IO_LONLAT_TOGGLE = false

IO_SEGMENT_MARKER = >

# MAP Parameters

MAP_ANNOT_MIN_ANGLE = 20

MAP_ANNOT_MIN_SPACING = 0p

MAP_ANNOT_OBLIQUE = 1

MAP_ANNOT_OFFSET_PRIMARY = 0.075i

MAP_ANNOT_OFFSET_SECONDARY = 0.075i

MAP_ANNOT_ORTHO = we

MAP_DEFAULT_PEN = default,black

MAP_DEGREE_SYMBOL = ring

MAP_FRAME_AXES = WESNZ

MAP_FRAME_PEN = thicker,black

MAP_FRAME_TYPE = fancy

MAP_FRAME_WIDTH = 5p

MAP_GRID_CROSS_SIZE_PRIMARY = 0p

MAP_GRID_CROSS_SIZE_SECONDARY = 0p

MAP_GRID_PEN_PRIMARY = default,black

MAP_GRID_PEN_SECONDARY = thinner,black

MAP_LABEL_OFFSET = 0.1944i

MAP_LINE_STEP = 0.75p

MAP_LOGO = false

MAP_LOGO_POS = BL/-54p/-54p

MAP_ORIGIN_X = 1i

MAP_ORIGIN_Y = 1i

MAP_POLAR_CAP = 85/90

MAP_SCALE_HEIGHT = 5p

MAP_TICK_LENGTH_PRIMARY = 5p/2.5p

MAP_TICK_LENGTH_SECONDARY = 15p/3.75p

MAP_TICK_PEN_PRIMARY = thinner,black

MAP_TICK_PEN_SECONDARY = thinner,black

MAP_TITLE_OFFSET = 14p

MAP_VECTOR_SHAPE = 0

# Projection Parameters

PROJ_AUX_LATITUDE = authalic

PROJ_ELLIPSOID = WGS-84

PROJ_LENGTH_UNIT = cm

PROJ_MEAN_RADIUS = authalic

PROJ_SCALE_FACTOR = default

# PostScript Parameters

PS_CHAR_ENCODING = ISOLatin1+

PS_COLOR_MODEL = rgb

PS_COMMENTS = false

PS_IMAGE_COMPRESS = deflate,5

PS_LINE_CAP = butt

PS_LINE_JOIN = miter

PS_MITER_LIMIT = 35

PS_MEDIA = a4

PS_PAGE_COLOR = white

PS_PAGE_ORIENTATION = portrait

PS_SCALE_X = 1

PS_SCALE_Y = 1

PS_TRANSPARENCY = Normal

# Calendar/Time Parameters

TIME_EPOCH = 1970-01-01T00:00:00

TIME_IS_INTERVAL = off

TIME_INTERVAL_FRACTION = 0.5

TIME_UNIT = s

TIME_WEEK_START = Monday

TIME_Y2K_OFFSET_YEAR = 1950

'''

def get_gmt_version(gmtdefaultsbinary, gmthomedir=None):

args = [gmtdefaultsbinary]

environ = os.environ.copy()

environ['GMTHOME'] = gmthomedir or ''

p = subprocess.Popen(

args,

stdout=subprocess.PIPE,

stderr=subprocess.PIPE,

env=environ)

(stdout, stderr) = p.communicate()

m = re.search(br'(\d+(\.\d+)*)', stderr) \

or re.search(br'# GMT (\d+(\.\d+)*)', stdout)

if not m:

raise GMTInstallationProblem(

"Can't extract version number from output of %s"

% gmtdefaultsbinary)

return str(m.group(1).decode('ascii'))

def detect_gmt_installations():

installations = {}

errmesses = []

# GMT 4.x:

try:

p = subprocess.Popen(

['GMT'],

stdout=subprocess.PIPE,

stderr=subprocess.PIPE)

(stdout, stderr) = p.communicate()

m = re.search(br'Version\s+(\d+(\.\d+)*)', stderr, re.M)

if not m:

raise GMTInstallationProblem(

"Can't version number from output of GMT")

version = str(m.group(1).decode('ascii'))

if version[0] != '5':

m = re.search(br'^\s+executables\s+(.+)$', stderr, re.M)

if not m:

raise GMTInstallationProblem(

"Can't extract executables dir from output of GMT")

gmtbin = str(m.group(1).decode('ascii'))

m = re.search(br'^\s+shared data\s+(.+)$', stderr, re.M)

if not m:

raise GMTInstallationProblem(

"Can't extract shared dir from output of GMT")

gmtshare = str(m.group(1).decode('ascii'))

if not gmtshare.endswith('/share'):

raise GMTInstallationProblem(

"Can't determine GMTHOME from output of GMT")

gmthome = gmtshare[:-6]

installations[version] = {

'home': gmthome,

'bin': gmtbin}

except OSError as e:

errmesses.append(('GMT', str(e)))

try:

version = str(subprocess.check_output(

['gmt', '--version']).strip().decode('ascii')).split('_')[0]

gmtbin = str(subprocess.check_output(

['gmt', '--show-bindir']).strip().decode('ascii'))

installations[version] = {

'bin': gmtbin}

except (OSError, subprocess.CalledProcessError) as e:

errmesses.append(('gmt', str(e)))

if not installations:

s = []

for (progname, errmess) in errmesses:

s.append('Cannot start "%s" executable: %s' % (progname, errmess))

raise GMTInstallationProblem(', '.join(s))

return installations

def appropriate_defaults_version(version):

avails = sorted(_gmt_defaults_by_version.keys(), key=key_version)

for iavail, avail in enumerate(avails):

if key_version(version) == key_version(avail):

return version

elif key_version(version) < key_version(avail):

return avails[max(0, iavail-1)]

return avails[-1]

def gmt_default_config(version):

'''Get default GMT configuration dict for given version.'''

xversion = appropriate_defaults_version(version)

# if not version in _gmt_defaults_by_version:

# raise GMTError('No GMT defaults for version %s found' % version)

gmt_defaults = _gmt_defaults_by_version[xversion]

d = {}

for line in gmt_defaults.splitlines():

sline = line.strip()

if not sline or sline.startswith('#'):

continue

k, v = sline.split('=', 1)

d[k.strip()] = v.strip()

return d

def diff_defaults(v1, v2):

d1 = gmt_default_config(v1)

d2 = gmt_default_config(v2)

for k in d1:

if k not in d2:

print('%s not in %s' % (k, v2))

else:

if d1[k] != d2[k]:

print('%s %s = %s' % (v1, k, d1[k]))

print('%s %s = %s' % (v2, k, d2[k]))

for k in d2:

if k not in d1:

print('%s not in %s' % (k, v1))

# diff_defaults('4.5.2', '4.5.3')

def check_gmt_installation(installation):

home_dir = installation.get('home', None)

bin_dir = installation['bin']

version = installation['version']

for d in home_dir, bin_dir:

if d is not None:

if not os.path.exists(d):

logging.error(('Directory does not exist: %s\n'

'Check your GMT installation.') % d)

if version[0] != '5':

gmtdefaults = pjoin(bin_dir, 'gmtdefaults')

versionfound = get_gmt_version(gmtdefaults, home_dir)

if versionfound != version:

raise GMTInstallationProblem((

'Expected GMT version %s but found version %s.\n'

'(Looking at output of %s)') % (

version, versionfound, gmtdefaults))

def get_gmt_installation(version):

setup_gmt_installations()

if version != 'newest' and version not in _gmt_installations:

logging.warn('GMT version %s not installed, taking version %s instead'

% (version, newest_installed_gmt_version()))

version = 'newest'

if version == 'newest':

version = newest_installed_gmt_version()

installation = dict(_gmt_installations[version])

return installation

def setup_gmt_installations():

if not setup_gmt_installations.have_done:

if not _gmt_installations:

_gmt_installations.update(detect_gmt_installations())

# store defaults as dicts into the gmt installations dicts

for version, installation in _gmt_installations.items():

installation['defaults'] = gmt_default_config(version)

installation['version'] = version

for installation in _gmt_installations.values():

check_gmt_installation(installation)

setup_gmt_installations.have_done = True

setup_gmt_installations.have_done = False

_paper_sizes_a = '''A0 2380 3368

A1 1684 2380

A2 1190 1684

A3 842 1190

A4 595 842

A5 421 595

A6 297 421

A7 210 297

A8 148 210

A9 105 148

A10 74 105

B0 2836 4008

B1 2004 2836

B2 1418 2004

B3 1002 1418

B4 709 1002

B5 501 709

archA 648 864

archB 864 1296

archC 1296 1728

archD 1728 2592

archE 2592 3456

flsa 612 936

halfletter 396 612

note 540 720

letter 612 792

legal 612 1008

11x17 792 1224

ledger 1224 792'''

_paper_sizes = {}

def setup_paper_sizes():

if not _paper_sizes:

for line in _paper_sizes_a.splitlines():

k, w, h = line.split()

_paper_sizes[k.lower()] = float(w), float(h)

def get_paper_size(k):

setup_paper_sizes()

return _paper_sizes[k.lower().rstrip('+')]

def all_paper_sizes():

setup_paper_sizes()

return _paper_sizes

def measure_unit(gmt_config):

for k in ['MEASURE_UNIT', 'PROJ_LENGTH_UNIT']:

if k in gmt_config:

return gmt_config[k]

raise GmtPyError('cannot get measure unit / proj length unit from config')

def paper_media(gmt_config):

for k in ['PAPER_MEDIA', 'PS_MEDIA']:

if k in gmt_config:

return gmt_config[k]

raise GmtPyError('cannot get paper media from config')

def page_orientation(gmt_config):

for k in ['PAGE_ORIENTATION', 'PS_PAGE_ORIENTATION']:

if k in gmt_config:

return gmt_config[k]

raise GmtPyError('cannot get paper orientation from config')

def make_bbox(width, height, gmt_config, margins=(0.8, 0.8, 0.8, 0.8)):

leftmargin, topmargin, rightmargin, bottommargin = margins

portrait = page_orientation(gmt_config).lower() == 'portrait'

paper_size = get_paper_size(paper_media(gmt_config))

if not portrait:

paper_size = paper_size[1], paper_size[0]

xoffset = (paper_size[0] - (width + leftmargin + rightmargin)) / \

2.0 + leftmargin

yoffset = (paper_size[1] - (height + topmargin + bottommargin)) / \

2.0 + bottommargin

if portrait:

bb1 = int((xoffset - leftmargin))

bb2 = int((yoffset - bottommargin))

bb3 = bb1 + int((width+leftmargin+rightmargin))

bb4 = bb2 + int((height+topmargin+bottommargin))

else:

bb1 = int((yoffset - topmargin))

bb2 = int((xoffset - leftmargin))

bb3 = bb1 + int((height+topmargin+bottommargin))

bb4 = bb2 + int((width+leftmargin+rightmargin))

return xoffset, yoffset, (bb1, bb2, bb3, bb4)

def gmtdefaults_as_text(version='newest'):

'''Get the built-in gmtdefaults.'''

if version not in _gmt_installations:

logging.warn('GMT version %s not installed, taking version %s instead'

% (version, newest_installed_gmt_version()))

version = 'newest'

if version == 'newest':

version = newest_installed_gmt_version()

return _gmt_defaults_by_version[version]

def savegrd(x, y, z, filename, title=None, naming='xy'):

'''Write COARDS compliant netcdf (grd) file.'''

assert y.size, x.size == z.shape

ny, nx = z.shape

nc = netcdf.netcdf_file(filename, 'w')

assert naming in ('xy', 'lonlat')

if naming == 'xy':

kx, ky = 'x', 'y'

else:

kx, ky = 'lon', 'lat'

nc.node_offset = 0

if title is not None:

nc.title = title

nc.Conventions = 'COARDS/CF-1.0'

nc.createDimension(kx, nx)

nc.createDimension(ky, ny)

xvar = nc.createVariable(kx, 'd', (kx,))

yvar = nc.createVariable(ky, 'd', (ky,))

if naming == 'xy':

xvar.long_name = kx

yvar.long_name = ky

else:

xvar.long_name = 'longitude'

xvar.units = 'degrees_east'

yvar.long_name = 'latitude'

yvar.units = 'degrees_north'

zvar = nc.createVariable('z', 'd', (ky, kx))

xvar[:] = x.astype(num.float64)

yvar[:] = y.astype(num.float64)

zvar[:] = z.astype(num.float64)

nc.close()

def to_array(var):

arr = var[:].copy()

if hasattr(var, 'scale_factor'):

arr *= var.scale_factor

if hasattr(var, 'add_offset'):

arr += var.add_offset

return arr

def loadgrd(filename):

'''Read COARDS compliant netcdf (grd) file.'''

nc = netcdf.netcdf_file(filename, 'r')

vkeys = list(nc.variables.keys())

kx = 'x'

ky = 'y'

if 'lon' in vkeys:

kx = 'lon'

if 'lat' in vkeys:

ky = 'lat'

x = to_array(nc.variables[kx])

y = to_array(nc.variables[ky])

z = to_array(nc.variables['z'])

nc.close()

return x, y, z

def centers_to_edges(asorted):

return (asorted[1:] + asorted[:-1])/2.

def nvals(asorted):

eps = (asorted[-1]-asorted[0])/asorted.size

return num.sum(asorted[1:] - asorted[:-1] >= eps) + 1

def guess_vals(asorted):

eps = (asorted[-1]-asorted[0])/asorted.size

indis = num.nonzero(asorted[1:] - asorted[:-1] >= eps)[0]

indis = num.concatenate((num.array([0]), indis+1,

num.array([asorted.size])))

asum = num.zeros(asorted.size+1)

asum[1:] = num.cumsum(asorted)

return (asum[indis[1:]] - asum[indis[:-1]]) / (indis[1:]-indis[:-1])

def blockmean(asorted, b):

indis = num.nonzero(asorted[1:] - asorted[:-1])[0]

indis = num.concatenate((num.array([0]), indis+1,

num.array([asorted.size])))

bsum = num.zeros(b.size+1)

bsum[1:] = num.cumsum(b)

return (

asorted[indis[:-1]],

(bsum[indis[1:]] - bsum[indis[:-1]]) / (indis[1:]-indis[:-1]))

def griddata_regular(x, y, z, xvals, yvals):

nx, ny = xvals.size, yvals.size

xindi = num.digitize(x, centers_to_edges(xvals))

yindi = num.digitize(y, centers_to_edges(yvals))

zindi = yindi*nx+xindi

order = num.argsort(zindi)

z = z[order]

zindi = zindi[order]

zindi, z = blockmean(zindi, z)

znew = num.empty(nx*ny, dtype=num.float)

znew[:] = num.nan

znew[zindi] = z

return znew.reshape(ny, nx)

def guess_field_size(x_sorted, y_sorted, z=None, mode=None):

critical_fraction = 1./num.e - 0.014*3

xs = x_sorted

ys = y_sorted

nxs, nys = nvals(xs), nvals(ys)

if mode == 'nonrandom':

return nxs, nys, 0

elif xs.size == nxs*nys:

# exact match

return nxs, nys, 0

elif nxs >= xs.size*critical_fraction and nys >= xs.size*critical_fraction:

# possibly randomly sampled

nxs = int(math.sqrt(xs.size))

nys = nxs

return nxs, nys, 2

else:

return nxs, nys, 1

def griddata_auto(x, y, z, mode=None):

'''Grid tabular XYZ data by binning.

This function does some extra work to guess the size of the grid. This

should work fine if the input values are already defined on an rectilinear

grid, even if data points are missing or duplicated. This routine also

tries to detect a random distribution of input data and in that case

creates a grid of size sqrt(N) x sqrt(N).

The points do not have to be given in any particular order. Grid nodes

without data are assigned the NaN value. If multiple data points map to the

same grid node, their average is assigned to the grid node. '''

x, y, z = [num.asarray(X) for X in (x, y, z)]

assert x.size == y.size == z.size

xs, ys = num.sort(x), num.sort(y)

nx, ny, badness = guess_field_size(xs, ys, z, mode=mode)

if badness <= 1:

xf = guess_vals(xs)

yf = guess_vals(ys)

zf = griddata_regular(x, y, z, xf, yf)

else:

xf = num.linspace(xs[0], xs[-1], nx)

yf = num.linspace(ys[0], ys[-1], ny)

zf = griddata_regular(x, y, z, xf, yf)

return xf, yf, zf

def tabledata(xf, yf, zf):

assert yf.size, xf.size == zf.shape

x = num.tile(xf, yf.size)

y = num.repeat(yf, xf.size)

z = zf.flatten()

return x, y, z

def double1d(a):

a2 = num.empty(a.size*2-1)

a2[::2] = a

a2[1::2] = (a[:-1] + a[1:])/2.

return a2

def double2d(f):

f2 = num.empty((f.shape[0]*2-1, f.shape[1]*2-1))

f2[:, :] = num.nan

f2[::2, ::2] = f

f2[1::2, ::2] = (f[:-1, :] + f[1:, :])/2.

f2[::2, 1::2] = (f[:, :-1] + f[:, 1:])/2.

f2[1::2, 1::2] = (f[:-1, :-1] + f[1:, :-1] + f[:-1, 1:] + f[1:, 1:])/4.

diag = f2[1::2, 1::2]

diagA = (f[:-1, :-1] + f[1:, 1:]) / 2.

diagB = (f[1:, :-1] + f[:-1, 1:]) / 2.

f2[1::2, 1::2] = num.where(num.isnan(diag), diagA, diag)

f2[1::2, 1::2] = num.where(num.isnan(diag), diagB, diag)

return f2

def doublegrid(x, y, z):

x2 = double1d(x)

y2 = double1d(y)

z2 = double2d(z)

return x2, y2, z2

class Guru(object):

'''Abstract base class providing template interpolation, accessible as

attributes.

Classes deriving from this one, have to implement a :py:meth:`get_params`

method, which is called to get a dict to do ordinary

``"%(key)x"``-substitutions. The deriving class must also provide a dict

with the templates.'''

def __init__(self):

self.templates = {}

def fill(self, templates, **kwargs):

params = self.get_params(**kwargs)

strings = [t % params for t in templates]

return strings

# hand through templates dict

def __getitem__(self, template_name):

return self.templates[template_name]

def __setitem__(self, template_name, template):

self.templates[template_name] = template

def __contains__(self, template_name):

return template_name in self.templates

def __iter__(self):

return iter(self.templates)

def __len__(self):

return len(self.templates)

def __delitem__(self, template_name):

del(self.templates[template_name])

def _simple_fill(self, template_names, **kwargs):

templates = [self.templates[n] for n in template_names]

return self.fill(templates, **kwargs)

def __getattr__(self, template_names):

if [n for n in template_names if n not in self.templates]:

raise AttributeError(template_names)

def f(**kwargs):

return self._simple_fill(template_names, **kwargs)

return f

def nice_value(x):

'''Round ``x`` to nice value.'''

exp = 1.0

sign = 1

if x < 0.0:

x = -x

sign = -1

while x >= 1.0:

x /= 10.0

exp *= 10.0

while x < 0.1:

x *= 10.0

exp /= 10.0

if x >= 0.75:

return sign * 1.0 * exp

if x >= 0.375:

return sign * 0.5 * exp

if x >= 0.225:

return sign * 0.25 * exp

if x >= 0.15:

return sign * 0.2 * exp

return sign * 0.1 * exp

class AutoScaler(object):

'''Tunable 1D autoscaling based on data range.

Instances of this class may be used to determine nice minima, maxima and

increments for ax annotations, as well as suitable common exponents for

notation.

The autoscaling process is guided by the following public attributes:

.. py:attribute:: approx_ticks

Approximate number of increment steps (tickmarks) to generate.

.. py:attribute:: mode

Mode of operation: one of ``'auto'``, ``'min-max'``, ``'0-max'``,

``'min-0'``, ``'symmetric'`` or ``'off'``.

================ ==================================================

mode description

================ ==================================================

``'auto'``: Look at data range and choose one of the choices

below.

``'min-max'``: Output range is selected to include data range.

``'0-max'``: Output range shall start at zero and end at data

max.

``'min-0'``: Output range shall start at data min and end at

zero.

``'symmetric'``: Output range shall by symmetric by zero.

``'off'``: Similar to ``'min-max'``, but snap and space are

disabled, such that the output range always

exactly matches the data range.

================ ==================================================

.. py:attribute:: exp

If defined, override automatically determined exponent for notation

by the given value.

.. py:attribute:: snap

If set to True, snap output range to multiples of increment. This

parameter has no effect, if mode is set to ``'off'``.

.. py:attribute:: inc

If defined, override automatically determined tick increment by the

given value.

.. py:attribute:: space

Add some padding to the range. The value given, is the fraction by

which the output range is increased on each side. If mode is

``'0-max'`` or ``'min-0'``, the end at zero is kept fixed at zero.

This parameter has no effect if mode is set to ``'off'``.

.. py:attribute:: exp_factor

Exponent of notation is chosen to be a multiple of this value.

.. py:attribute:: no_exp_interval:

Range of exponent, for which no exponential notation is allowed.

'''

def __init__(

self,

approx_ticks=7.0,

mode='auto',

exp=None,

snap=False,

inc=None,

space=0.0,

exp_factor=3,

no_exp_interval=(-3, 5)):

'''Create new AutoScaler instance.

The parameters are described in the AutoScaler documentation.

'''

self.approx_ticks = approx_ticks

self.mode = mode

self.exp = exp

self.snap = snap

self.inc = inc

self.space = space

self.exp_factor = exp_factor

self.no_exp_interval = no_exp_interval

def make_scale(self, data_range, override_mode=None):

'''Get nice minimum, maximum and increment for given data range.

Returns ``(minimum, maximum, increment)`` or ``(maximum, minimum,

-increment)``, depending on whether data_range is ``(data_min,

data_max)`` or ``(data_max, data_min)``. If ``override_mode`` is

defined, the mode attribute is temporarily overridden by the given

value. '''

data_min = min(data_range)

data_max = max(data_range)

is_reverse = (data_range[0] > data_range[1])

a = self.mode

if self.mode == 'auto':

a = self.guess_autoscale_mode(data_min, data_max)

if override_mode is not None:

a = override_mode

mi, ma = 0, 0

if a == 'off':

mi, ma = data_min, data_max

elif a == '0-max':

mi = 0.0

if data_max > 0.0:

ma = data_max

else:

ma = 1.0

elif a == 'min-0':

ma = 0.0

if data_min < 0.0:

mi = data_min

else:

mi = -1.0

elif a == 'min-max':

mi, ma = data_min, data_max

elif a == 'symmetric':

m = max(abs(data_min), abs(data_max))

mi = -m

ma = m

nmi = mi

if (mi != 0. or a == 'min-max') and a != 'off':

nmi = mi - self.space*(ma-mi)

nma = ma

if (ma != 0. or a == 'min-max') and a != 'off':

nma = ma + self.space*(ma-mi)

mi, ma = nmi, nma

if mi == ma and a != 'off':

mi -= 1.0

ma += 1.0

# make nice tick increment

if self.inc is not None:

inc = self.inc

else:

if self.approx_ticks > 0.:

inc = nice_value((ma-mi) / self.approx_ticks)

else:

inc = nice_value((ma-mi)*10.)

if inc == 0.0:

inc = 1.0

# snap min and max to ticks if this is wanted

if self.snap and a != 'off':

ma = inc * math.ceil(ma/inc)

mi = inc * math.floor(mi/inc)

if is_reverse:

return ma, mi, -inc

else:

return mi, ma, inc

def make_exp(self, x):

'''Get nice exponent for notation of ``x``.

For ax annotations, give tick increment as ``x``.'''

if self.exp is not None:

return self.exp

x = abs(x)

if x == 0.0:

return 0

if 10**self.no_exp_interval[0] <= x <= 10**self.no_exp_interval[1]:

return 0

return math.floor(math.log10(x)/self.exp_factor)*self.exp_factor

def guess_autoscale_mode(self, data_min, data_max):

'''Guess mode of operation, based on data range.

Used to map ``'auto'`` mode to ``'0-max'``, ``'min-0'``, ``'min-max'``

or ``'symmetric'``.'''

a = 'min-max'

if data_min >= 0.0:

if data_min < data_max/2.:

a = '0-max'

else:

a = 'min-max'

if data_max <= 0.0:

if data_max > data_min/2.:

a = 'min-0'

else:

a = 'min-max'

if data_min < 0.0 and data_max > 0.0:

if abs((abs(data_max)-abs(data_min)) /

(abs(data_max)+abs(data_min))) < 0.5:

a = 'symmetric'

else:

a = 'min-max'

return a

class Ax(AutoScaler):

'''Ax description with autoscaling capabilities.

The ax is described by the :py:class:`AutoScaler` public attributes, plus

the following additional attributes (with default values given in

paranthesis):

.. py:attribute:: label

Ax label (without unit).

.. py:attribute:: unit

Physical unit of the data attached to this ax.

.. py:attribute:: scaled_unit

(see below)

.. py:attribute:: scaled_unit_factor

Scaled physical unit and factor between unit and scaled_unit so that

unit = scaled_unit_factor x scaled_unit.

(E.g. if unit is 'm' and data is in the range of nanometers, you may

want to set the scaled_unit to 'nm' and the scaled_unit_factor to

1e9.)

.. py:attribute:: limits

If defined, fix range of ax to limits=(min,max).

.. py:attribute:: masking

If true and if there is a limit on the ax, while calculating ranges,

the data points are masked such that data points outside of this axes

limits are not used to determine the range of another dependant ax.

'''

def __init__(self, label='', unit='', scaled_unit_factor=1.,

scaled_unit='', limits=None, masking=True, **kwargs):

AutoScaler.__init__(self, **kwargs)

self.label = label

self.unit = unit

self.scaled_unit_factor = scaled_unit_factor

self.scaled_unit = scaled_unit

self.limits = limits

self.masking = masking

def label_str(self, exp, unit):

'''Get label string including the unit and multiplier.'''

slabel, sunit, sexp = '', '', ''

if self.label:

slabel = self.label

if unit or exp != 0:

if exp != 0:

sexp = '\\327 10@+%i@+' % exp

sunit = '[ %s %s ]' % (sexp, unit)

else:

sunit = '[ %s ]' % unit

p = []

if slabel:

p.append(slabel)

if sunit:

p.append(sunit)

return ' '.join(p)

def make_params(self, data_range, ax_projection=False, override_mode=None,

override_scaled_unit_factor=None):

'''Get minimum, maximum, increment and label string for ax display.'

Returns minimum, maximum, increment and label string including unit and

multiplier for given data range.

If ``ax_projection`` is True, values suitable to be displayed on the ax

are returned, e.g. min, max and inc are returned in scaled units.

Otherwise the values are returned in the original units, without any

scaling applied. '''

sf = self.scaled_unit_factor

if override_scaled_unit_factor is not None:

sf = override_scaled_unit_factor

dr_scaled = [sf*x for x in data_range]

mi, ma, inc = self.make_scale(dr_scaled, override_mode=override_mode)

if self.inc is not None:

inc = self.inc*sf

if ax_projection:

exp = self.make_exp(inc)

if sf == 1. and override_scaled_unit_factor is None:

unit = self.unit

else:

unit = self.scaled_unit

label = self.label_str(exp, unit)

return mi/10**exp, ma/10**exp, inc/10**exp, label

else:

label = self.label_str(0, self.unit)

return mi/sf, ma/sf, inc/sf, label

class ScaleGuru(Guru):

'''2D/3D autoscaling and ax annotation facility.

Instances of this class provide automatic determination of plot ranges,

tick increments and scaled annotations, as well as label/unit handling. It

can in particular be used to automatically generate the -R and -B option

arguments, which are required for most GMT commands.

It extends the functionality of the :py:class:`Ax` and

:py:class:`AutoScaler` classes at the level, where it can not be handled

anymore by looking at a single dimension of the dataset's data, e.g.:

* The ability to impose a fixed aspect ratio between two axes.

* Recalculation of data range on non-limited axes, when there are

limits imposed on other axes.

'''

def __init__(self, data_tuples=None, axes=None, aspect=None,

percent_interval=None, copy_from=None):

Guru.__init__(self)

if copy_from:

self.templates = copy.deepcopy(copy_from.templates)

self.axes = copy.deepcopy(copy_from.axes)

self.data_ranges = copy.deepcopy(copy_from.data_ranges)

self.aspect = copy_from.aspect

if percent_interval is not None:

from scipy.stats import scoreatpercentile as scap

self.templates = dict(

R='-R%(xmin)g/%(xmax)g/%(ymin)g/%(ymax)g',

B='-B%(xinc)g:%(xlabel)s:/%(yinc)g:%(ylabel)s:WSen',

T='-T%(zmin)g/%(zmax)g/%(zinc)g')

maxdim = 2

if data_tuples:

maxdim = max(maxdim, max([len(dt) for dt in data_tuples]))

else:

if axes:

maxdim = len(axes)

data_tuples = [([],) * maxdim]

if axes is not None:

self.axes = axes

else:

self.axes = [Ax() for i in range(maxdim)]

# sophisticated data-range calculation

data_ranges = [None] * maxdim

for dt_ in data_tuples:

dt = num.asarray(dt_)

in_range = True

for ax, x in zip(self.axes, dt):

if ax.limits and ax.masking:

ax_limits = list(ax.limits)

if ax_limits[0] is None:

ax_limits[0] = -num.inf

if ax_limits[1] is None:

ax_limits[1] = num.inf

in_range = num.logical_and(

in_range,

num.logical_and(ax_limits[0] <= x, x <= ax_limits[1]))

for i, ax, x in zip(range(maxdim), self.axes, dt):

if not ax.limits or None in ax.limits:

if len(x) >= 1:

if in_range is not True:

xmasked = num.where(in_range, x, num.NaN)

if percent_interval is None:

range_this = (

num.nanmin(xmasked),

num.nanmax(xmasked))

else:

xmasked_finite = num.compress(

num.isfinite(xmasked), xmasked)

range_this = (

scap(xmasked_finite,

(100.-percent_interval)/2.),

scap(xmasked_finite,

100.-(100.-percent_interval)/2.))

else:

if percent_interval is None:

range_this = num.nanmin(x), num.nanmax(x)

else:

xmasked_finite = num.compress(

num.isfinite(xmasked), xmasked)

range_this = (

scap(xmasked_finite,

(100.-percent_interval)/2.),

scap(xmasked_finite,

100.-(100.-percent_interval)/2.))

else:

range_this = (0., 1.)

if ax.limits:

if ax.limits[0] is not None:

range_this = ax.limits[0], max(ax.limits[0],

range_this[1])

if ax.limits[1] is not None:

range_this = min(ax.limits[1],

range_this[0]), ax.limits[1]

else:

range_this = ax.limits

if data_ranges[i] is None and range_this[0] <= range_this[1]:

data_ranges[i] = range_this

else:

mi, ma = range_this

if data_ranges[i] is not None:

mi = min(data_ranges[i][0], mi)

ma = max(data_ranges[i][1], ma)

data_ranges[i] = (mi, ma)

for i in range(len(data_ranges)):

if data_ranges[i] is None or not (

num.isfinite(data_ranges[i][0])

and num.isfinite(data_ranges[i][1])):

data_ranges[i] = (0., 1.)

self.data_ranges = data_ranges

self.aspect = aspect

def copy(self):

return ScaleGuru(copy_from=self)

def get_params(self, ax_projection=False):

'''Get dict with output parameters.

For each data dimension, ax minimum, maximum, increment and a label

string (including unit and exponential factor) are determined. E.g. in

for the first dimension the output dict will contain the keys

``'xmin'``, ``'xmax'``, ``'xinc'``, and ``'xlabel'``.

Normally, values corresponding to the scaling of the raw data are

produced, but if ``ax_projection`` is ``True``, values which are

suitable to be printed on the axes are returned. This means that in the

latter case, the :py:attr:`Ax.scaled_unit` and

:py:attr:`Ax.scaled_unit_factor` attributes as set on the axes are

respected and that a common 10^x factor is factored out and put to the

label string. '''

xmi, xma, xinc, xlabel = self.axes[0].make_params(

self.data_ranges[0], ax_projection)

ymi, yma, yinc, ylabel = self.axes[1].make_params(

self.data_ranges[1], ax_projection)

if len(self.axes) > 2:

zmi, zma, zinc, zlabel = self.axes[2].make_params(

self.data_ranges[2], ax_projection)

# enforce certain aspect, if needed

if self.aspect is not None:

xwid = xma-xmi

ywid = yma-ymi

if ywid < xwid*self.aspect:

ymi -= (xwid*self.aspect - ywid)*0.5

yma += (xwid*self.aspect - ywid)*0.5

ymi, yma, yinc, ylabel = self.axes[1].make_params(

(ymi, yma), ax_projection, override_mode='off',

override_scaled_unit_factor=1.)

elif xwid < ywid/self.aspect:

xmi -= (ywid/self.aspect - xwid)*0.5

xma += (ywid/self.aspect - xwid)*0.5

xmi, xma, xinc, xlabel = self.axes[0].make_params(

(xmi, xma), ax_projection, override_mode='off',

override_scaled_unit_factor=1.)

params = dict(xmin=xmi, xmax=xma, xinc=xinc, xlabel=xlabel,

ymin=ymi, ymax=yma, yinc=yinc, ylabel=ylabel)

if len(self.axes) > 2:

params.update(dict(zmin=zmi, zmax=zma, zinc=zinc, zlabel=zlabel))

return params

class GumSpring(object):

'''Sizing policy implementing a minimal size, plus a desire to grow.'''

def __init__(self, minimal=None, grow=None):

self.minimal = minimal

if grow is None:

if minimal is None:

self.grow = 1.0

else:

self.grow = 0.0

else:

self.grow = grow

self.value = 1.0

def get_minimal(self):

if self.minimal is not None:

return self.minimal

else:

return 0.0

def get_grow(self):

return self.grow

def set_value(self, value):

self.value = value

def get_value(self):

return self.value

def distribute(sizes, grows, space):

sizes = list(sizes)

gsum = sum(grows)

if gsum > 0.0:

for i in range(len(sizes)):

sizes[i] += space*grows[i]/gsum

return sizes

class Widget(Guru):

'''Base class of the gmtpy layout system.

The Widget class provides the basic functionality for the nesting and

placing of elements on the output page, and maintains the sizing policies

of each element. Each of the layouts defined in gmtpy is itself a Widget.

Sizing of the widget is controlled by :py:meth:`get_min_size` and

:py:meth:`get_grow` which should be overloaded in derived classes. The

basic behaviour of a Widget instance is to have a vertical and a horizontal

minimum size which default to zero, as well as a vertical and a horizontal

desire to grow, represented by floats, which default to 1.0. Additionally

an aspect ratio constraint may be imposed on the Widget.

After layouting, the widget provides its width, height, x-offset and

y-offset in various ways. Via the Guru interface (see :py:class:`Guru`

class), templates for the -X, -Y and -J option arguments used by GMT

arguments are provided. The defaults are suitable for plotting of linear

(-JX) plots. Other projections can be selected by giving an appropriate 'J'

template, or by manual construction of the -J option, e.g. by utilizing the

:py:meth:`width` and :py:meth:`height` methods. The :py:meth:`bbox` method

can be used to create a PostScript bounding box from the widgets border,

e.g. for use in the :py:meth:`save` method of :py:class:`GMT` instances.

The convention is, that all sizes are given in PostScript points.

Conversion factors are provided as constants :py:const:`inch` and

:py:const:`cm` in the gmtpy module. '''

def __init__(self, horizontal=None, vertical=None, parent=None):

'''Create new widget.'''

Guru.__init__(self)

self.templates = dict(

X='-Xa%(xoffset)gp',

Y='-Ya%(yoffset)gp',

J='-JX%(width)gp/%(height)gp')

if horizontal is None:

self.horizontal = GumSpring()

else:

self.horizontal = horizontal

if vertical is None:

self.vertical = GumSpring()

else:

self.vertical = vertical

self.aspect = None

self.parent = parent

self.dirty = True

def set_parent(self, parent):

'''Set the parent widget.

This method should not be called directly. The :py:meth:`set_widget`

methods are responsible for calling this. '''

self.parent = parent

self.dirtyfy()

def get_parent(self):

'''Get the widgets parent widget.'''

return self.parent

def get_root(self):

'''Get the root widget in the layout hierarchy.'''

if self.parent is not None:

return self.get_parent()

else:

return self

def set_horizontal(self, minimal=None, grow=None):

'''Set the horizontal sizing policy of the Widget.

:param minimal: new minimal width of the widget

:param grow: new horizontal grow disire of the widget

'''

self.horizontal = GumSpring(minimal, grow)

self.dirtyfy()

def get_horizontal(self):

return self.horizontal.get_minimal(), self.horizontal.get_grow()

def set_vertical(self, minimal=None, grow=None):

'''Set the horizontal sizing policy of the Widget.

:param minimal: new minimal height of the widget

:param grow: new vertical grow disire of the widget

'''

self.vertical = GumSpring(minimal, grow)

self.dirtyfy()

def get_vertical(self):

return self.vertical.get_minimal(), self.vertical.get_grow()

def set_aspect(self, aspect=None):

'''Set aspect constraint on the widget.

The aspect is given as height divided by width.

'''

self.aspect = aspect

self.dirtyfy()

def set_policy(self, minimal=(None, None), grow=(None, None), aspect=None):

'''Shortcut to set sizing and aspect constraints in a single method

call.'''

self.set_horizontal(minimal[0], grow[0])

self.set_vertical(minimal[1], grow[1])

self.set_aspect(aspect)

def get_policy(self):

mh, gh = self.get_horizontal()

mv, gv = self.get_vertical()

return (mh, mv), (gh, gv), self.aspect

def legalize(self, size, offset):

'''Get legal size for widget.

Returns: (new_size, new_offset)

Given a box as ``size`` and ``offset``, return ``new_size`` and

``new_offset``, such that the widget's sizing and aspect constraints

are fullfilled. The returned box is centered on the given input box.

'''

sh, sv = size

oh, ov = offset

shs, svs = Widget.get_min_size(self)

ghs, gvs = Widget.get_grow(self)

if ghs == 0.0:

oh += (sh-shs)/2.

sh = shs

if gvs == 0.0:

ov += (sv-svs)/2.

sv = svs

if self.aspect is not None:

if sh > sv/self.aspect:

oh += (sh-sv/self.aspect)/2.

sh = sv/self.aspect

if sv > sh*self.aspect:

ov += (sv-sh*self.aspect)/2.

sv = sh*self.aspect

return (sh, sv), (oh, ov)

def get_min_size(self):

'''Get minimum size of widget.

Used by the layout managers. Should be overloaded in derived classes.

'''

mh, mv = self.horizontal.get_minimal(), self.vertical.get_minimal()

if self.aspect is not None:

if mv == 0.0:

return mh, mh*self.aspect

elif mh == 0.0:

return mv/self.aspect, mv

return mh, mv

def get_grow(self):

'''Get widget's desire to grow.

Used by the layout managers. Should be overloaded in derived classes.

'''

return self.horizontal.get_grow(), self.vertical.get_grow()

def set_size(self, size, offset):

'''Set the widget's current size.

Should not be called directly. It is the layout manager's

responsibility to call this. '''

(sh, sv), inner_offset = self.legalize(size, offset)

self.offset = inner_offset

self.horizontal.set_value(sh)

self.vertical.set_value(sv)

self.dirty = False

def __str__(self):

def indent(ind, str):

return ('\n'+ind).join(str.splitlines())

size, offset = self.get_size()

s = "%s (%g x %g) (%g, %g)\n" % ((self.__class__,) + size + offset)

children = self.get_children()

if children:

s += '\n'.join([' ' + indent(' ', str(c)) for c in children])

return s

def policies_debug_str(self):

def indent(ind, str):

return ('\n'+ind).join(str.splitlines())

mins, grows, aspect = self.get_policy()

s = "%s: minimum=(%s, %s), grow=(%s, %s), aspect=%s\n" % (

(self.__class__,) + mins+grows+(aspect,))

children = self.get_children()

if children:

s += '\n'.join([' ' + indent(

' ', c.policies_debug_str()) for c in children])

return s

def get_corners(self, descend=False):

'''Get coordinates of the corners of the widget.

Returns list with coordinate tuples.

If ``descend`` is True, the returned list will contain corner

coordinates of all sub-widgets.

'''

self.do_layout()

(sh, sv), (oh, ov) = self.get_size()

corners = [(oh, ov), (oh+sh, ov), (oh+sh, ov+sv), (oh, ov+sv)]

if descend:

for child in self.get_children():

corners.extend(child.get_corners(descend=True))

return corners

def get_sizes(self):

'''Get sizes of this widget and all it's children.

Returns a list with size tuples.

'''

self.do_layout()

sizes = [self.get_size()]

for child in self.get_children():

sizes.extend(child.get_sizes())

return sizes

def do_layout(self):

'''Triggers layouting of the widget hierarchy, if needed.'''

if self.parent is not None:

return self.parent.do_layout()

if not self.dirty:

return

sh, sv = self.get_min_size()

gh, gv = self.get_grow()

if sh == 0.0 and gh != 0.0:

sh = 15.*cm

if sv == 0.0 and gv != 0.0:

sv = 15.*cm*gv/gh * 1./golden_ratio

self.set_size((sh, sv), (0., 0.))

def get_children(self):

'''Get sub-widgets contained in this widget.

Returns a list of widgets.

'''

return []

def get_size(self):

'''Get current size and position of the widget.

Triggers layouting and returns

``((width, height), (xoffset, yoffset))``

'''

self.do_layout()

return (self.horizontal.get_value(),

self.vertical.get_value()), self.offset

def get_params(self):

'''Get current size and position of the widget.

Triggers layouting and returns dict with keys ``'xoffset'``,

``'yoffset'``, ``'width'`` and ``'height'``. '''

self.do_layout()

(w, h), (xo, yo) = self.get_size()

return dict(xoffset=xo, yoffset=yo, width=w, height=h,

width_m=w/_units['m'])

def width(self):

'''Get current width of the widget.

Triggers layouting and returns width.'''

self.do_layout()

return self.horizontal.get_value()

def height(self):

'''Get current height of the widget.

Triggers layouting and return height.'''

self.do_layout()

return self.vertical.get_value()

def bbox(self):

'''Get PostScript bounding box for this widget.

Triggers layouting and returns values suitable to create PS bounding

box, representing the widgets current size and position.

'''

self.do_layout()

return (self.offset[0], self.offset[1], self.offset[0]+self.width(),

self.offset[1]+self.height())

def dirtyfy(self):

'''Set dirty flag on top level widget in the hierarchy.

Called by various methods, to indicate, that the widget hierarchy needs

new layouting.

'''

if self.parent is not None:

self.parent.dirtyfy()

self.dirty = True

class CenterLayout(Widget):

'''A layout manager which centers its single child widget.

The child widget may be oversized.

'''

def __init__(self, horizontal=None, vertical=None):

Widget.__init__(self, horizontal, vertical)

self.content = Widget(horizontal=GumSpring(grow=1.),

vertical=GumSpring(grow=1.), parent=self)

def get_min_size(self):

shs, svs = Widget.get_min_size(self)

sh, sv = self.content.get_min_size()

return max(shs, sh), max(svs, sv)

def get_grow(self):

ghs, gvs = Widget.get_grow(self)

gh, gv = self.content.get_grow()

return gh*ghs, gv*gvs

def set_size(self, size, offset):

(sh, sv), (oh, ov) = self.legalize(size, offset)

shc, svc = self.content.get_min_size()

ghc, gvc = self.content.get_grow()

if ghc != 0.:

shc = sh

if gvc != 0.:

svc = sv

ohc = oh+(sh-shc)/2.

ovc = ov+(sv-svc)/2.

self.content.set_size((shc, svc), (ohc, ovc))

Widget.set_size(self, (sh, sv), (oh, ov))

def set_widget(self, widget=None):

'''Set the child widget, which shall be centered.'''

if widget is None:

widget = Widget()

self.content = widget

widget.set_parent(self)

def get_widget(self):

return self.content

def get_children(self):

return [self.content]

class FrameLayout(Widget):

'''A layout manager containing a center widget sorrounded by four margin

widgets.

+---------------------------+

| top |

+---------------------------+

| | | |

| left | center | right |

| | | |

+---------------------------+

| bottom |

+---------------------------+

This layout manager does a little bit of extra effort to maintain the

aspect constraint of the center widget, if this is set. It does so, by

allowing for a bit more flexibility in the sizing of the margins. Two

shortcut methods are provided to set the margin sizes in one shot:

:py:meth:`set_fixed_margins` and :py:meth:`set_min_margins`. The first sets

the margins to fixed sizes, while the second gives them a minimal size and

a (neglectably) small desire to grow. Using the latter may be useful when

setting an aspect constraint on the center widget, because this way the

maximum size of the center widget may be controlled without creating empty

spaces between the widgets.

'''

def __init__(self, horizontal=None, vertical=None):

Widget.__init__(self, horizontal, vertical)

mw = 3.*cm

self.left = Widget(

horizontal=GumSpring(grow=0.15, minimal=mw), parent=self)

self.right = Widget(

horizontal=GumSpring(grow=0.15, minimal=mw), parent=self)

self.top = Widget(

vertical=GumSpring(grow=0.15, minimal=mw/golden_ratio),

parent=self)

self.bottom = Widget(

vertical=GumSpring(grow=0.15, minimal=mw/golden_ratio),

parent=self)

self.center = Widget(

horizontal=GumSpring(grow=0.7), vertical=GumSpring(grow=0.7),

parent=self)

def set_fixed_margins(self, left, right, top, bottom):

'''Give margins fixed size constraints.'''

self.left.set_horizontal(left, 0)

self.right.set_horizontal(right, 0)

self.top.set_vertical(top, 0)

self.bottom.set_vertical(bottom, 0)

def set_min_margins(self, left, right, top, bottom, grow=0.0001):

'''Give margins a minimal size and the possibility to grow.

The desire to grow is set to a very small number.'''

self.left.set_horizontal(left, grow)

self.right.set_horizontal(right, grow)

self.top.set_vertical(top, grow)

self.bottom.set_vertical(bottom, grow)

def get_min_size(self):

shs, svs = Widget.get_min_size(self)

sl, sr, st, sb, sc = [x.get_min_size() for x in (

self.left, self.right, self.top, self.bottom, self.center)]

gl, gr, gt, gb, gc = [x.get_grow() for x in (

self.left, self.right, self.top, self.bottom, self.center)]

shsum = sl[0]+sr[0]+sc[0]

svsum = st[1]+sb[1]+sc[1]

# prevent widgets from collapsing

for s, g in ((sl, gl), (sr, gr), (sc, gc)):

if s[0] == 0.0 and g[0] != 0.0:

shsum += 0.1*cm

for s, g in ((st, gt), (sb, gb), (sc, gc)):

if s[1] == 0.0 and g[1] != 0.0:

svsum += 0.1*cm

sh = max(shs, shsum)

sv = max(svs, svsum)

return sh, sv

def get_grow(self):

ghs, gvs = Widget.get_grow(self)

gh = (self.left.get_grow()[0] +

self.right.get_grow()[0] +

self.center.get_grow()[0]) * ghs

gv = (self.top.get_grow()[1] +

self.bottom.get_grow()[1] +

self.center.get_grow()[1]) * gvs

return gh, gv

def set_size(self, size, offset):

(sh, sv), (oh, ov) = self.legalize(size, offset)

sl, sr, st, sb, sc = [x.get_min_size() for x in (

self.left, self.right, self.top, self.bottom, self.center)]

gl, gr, gt, gb, gc = [x.get_grow() for x in (

self.left, self.right, self.top, self.bottom, self.center)]

ah = sh - (sl[0]+sr[0]+sc[0])

av = sv - (st[1]+sb[1]+sc[1])

if ah < 0.0:

raise GmtPyError("Container not wide enough for contents "

"(FrameLayout, available: %g cm, needed: %g cm)"

% (sh/cm, (sl[0]+sr[0]+sc[0])/cm))

if av < 0.0:

raise GmtPyError("Container not high enough for contents "

"(FrameLayout, available: %g cm, needed: %g cm)"

% (sv/cm, (st[1]+sb[1]+sc[1])/cm))

slh, srh, sch = distribute((sl[0], sr[0], sc[0]),

(gl[0], gr[0], gc[0]), ah)

stv, sbv, scv = distribute((st[1], sb[1], sc[1]),

(gt[1], gb[1], gc[1]), av)

if self.center.aspect is not None:

ahm = sh - (sl[0]+sr[0] + scv/self.center.aspect)

avm = sv - (st[1]+sb[1] + sch*self.center.aspect)

if 0.0 < ahm < ah:

slh, srh, sch = distribute(

(sl[0], sr[0], scv/self.center.aspect),

(gl[0], gr[0], 0.0), ahm)

elif 0.0 < avm < av:

stv, sbv, scv = distribute((st[1], sb[1],

sch*self.center.aspect),

(gt[1], gb[1], 0.0), avm)

ah = sh - (slh+srh+sch)

av = sv - (stv+sbv+scv)

oh += ah/2.

ov += av/2.

sh -= ah

sv -= av

self.left.set_size((slh, scv), (oh, ov+sbv))

self.right.set_size((srh, scv), (oh+slh+sch, ov+sbv))

self.top.set_size((sh, stv), (oh, ov+sbv+scv))

self.bottom.set_size((sh, sbv), (oh, ov))

self.center.set_size((sch, scv), (oh+slh, ov+sbv))

Widget.set_size(self, (sh, sv), (oh, ov))

def set_widget(self, which='center', widget=None):

'''Set one of the sub-widgets.

``which`` should be one of ``'left'``, ``'right'``, ``'top'``,

``'bottom'`` or ``'center'``. '''

if widget is None:

widget = Widget()

if which in ('left', 'right', 'top', 'bottom', 'center'):

self.__dict__[which] = widget

else:

raise GmtPyError('No such sub-widget: %s' % which)

widget.set_parent(self)

def get_widget(self, which='center'):

'''Get one of the sub-widgets.

``which`` should be one of ``'left'``, ``'right'``, ``'top'``,

``'bottom'`` or ``'center'``. '''

if which in ('left', 'right', 'top', 'bottom', 'center'):

return self.__dict__[which]

else:

raise GmtPyError('No such sub-widget: %s' % which)

def get_children(self):

return [self.left, self.right, self.top, self.bottom, self.center]

class GridLayout(Widget):

'''A layout manager which arranges its sub-widgets in a grid.

The grid spacing is flexible and based on the sizing policies of the

contained sub-widgets. If an equidistant grid is needed, the sizing

policies of the sub-widgets have to be set equally.

The height of each row and the width of each column is derived from the

sizing policy of the largest sub-widget in the row or column in question.

The algorithm is not very sophisticated, so conflicting sizing policies

might not be resolved optimally.

'''

def __init__(self, nx=2, ny=2, horizontal=None, vertical=None):

'''Create new grid layout with ``nx`` columns and ``ny`` rows.'''

Widget.__init__(self, horizontal, vertical)

self.grid = []

for iy in range(ny):

row = []

for ix in range(nx):

w = Widget(parent=self)

row.append(w)

self.grid.append(row)

def sub_min_sizes_as_array(self):

esh = num.array(

[[w.get_min_size()[0] for w in row] for row in self.grid],

dtype=num.float)

esv = num.array(

[[w.get_min_size()[1] for w in row] for row in self.grid],

dtype=num.float)

return esh, esv

def sub_grows_as_array(self):

egh = num.array(

[[w.get_grow()[0] for w in row] for row in self.grid],

dtype=num.float)

egv = num.array(

[[w.get_grow()[1] for w in row] for row in self.grid],

dtype=num.float)

return egh, egv

def get_min_size(self):

sh, sv = Widget.get_min_size(self)

esh, esv = self.sub_min_sizes_as_array()

if esh.size != 0:

sh = max(sh, num.sum(esh.max(0)))

if esv.size != 0:

sv = max(sv, num.sum(esv.max(1)))

return sh, sv

def get_grow(self):

ghs, gvs = Widget.get_grow(self)

egh, egv = self.sub_grows_as_array()

if egh.size != 0:

gh = num.sum(egh.max(0))*ghs

else:

gh = 1.0

if egv.size != 0:

gv = num.sum(egv.max(1))*gvs

else:

gv = 1.0

return gh, gv

def set_size(self, size, offset):

(sh, sv), (oh, ov) = self.legalize(size, offset)

esh, esv = self.sub_min_sizes_as_array()

egh, egv = self.sub_grows_as_array()

# available additional space

empty = esh.size == 0

if not empty:

ah = sh - num.sum(esh.max(0))

av = sv - num.sum(esv.max(1))

else:

av = sv

ah = sh

if ah < 0.0:

raise GmtPyError("Container not wide enough for contents "

"(GridLayout, available: %g cm, needed: %g cm)"

% (sh/cm, (num.sum(esh.max(0)))/cm))

if av < 0.0:

raise GmtPyError("Container not high enough for contents "

"(GridLayout, available: %g cm, needed: %g cm)"

% (sv/cm, (num.sum(esv.max(1)))/cm))

nx, ny = esh.shape

if not empty:

# distribute additional space on rows and columns

# according to grow weights and minimal sizes

gsh = egh.sum(1)[:, num.newaxis].repeat(ny, axis=1)

nesh = esh.copy()

nesh += num.where(gsh > 0.0, ah*egh/gsh, 0.0)

nsh = num.maximum(nesh.max(0), esh.max(0))

gsv = egv.sum(0)[num.newaxis, :].repeat(nx, axis=0)

nesv = esv.copy()

nesv += num.where(gsv > 0.0, av*egv/gsv, 0.0)

nsv = num.maximum(nesv.max(1), esv.max(1))

ah = sh - sum(nsh)

av = sv - sum(nsv)

oh += ah/2.

ov += av/2.

sh -= ah

sv -= av

# resize child widgets

neov = ov + sum(nsv)

for row, nesv in zip(self.grid, nsv):

neov -= nesv

neoh = oh

for w, nesh in zip(row, nsh):

w.set_size((nesh, nesv), (neoh, neov))

neoh += nesh

Widget.set_size(self, (sh, sv), (oh, ov))

def set_widget(self, ix, iy, widget=None):

'''Set one of the sub-widgets.

Sets the sub-widget in column ``ix`` and row ``iy``. The indices are

counted from zero. '''

if widget is None:

widget = Widget()

self.grid[iy][ix] = widget

widget.set_parent(self)

def get_widget(self, ix, iy):

'''Get one of the sub-widgets.

Gets the sub-widget from column ``ix`` and row ``iy``. The indices are

counted from zero.'''

return self.grid[iy][ix]

def get_children(self):

children = []

for row in self.grid:

children.extend(row)

return children

def is_gmt5(version='newest'):

return get_gmt_installation(version)['version'][0] == '5'

def aspect_for_projection(gmtversion, *args, **kwargs):

gmt = GMT(version=gmtversion, eps_mode=True)

if gmt.is_gmt5():

gmt.psbasemap('-B+gblack', finish=True, *args, **kwargs)

fn = gmt.tempfilename('test.eps')

gmt.save(fn, crop_eps_mode=True)

with open(fn, 'rb') as f:

s = f.read()

l, b, r, t = get_bbox(s)

else:

gmt.psbasemap('-G0', finish=True, *args, **kwargs)

l, b, r, t = gmt.bbox()

return (t-b)/(r-l)

def text_box(

text, font=0, font_size=12., angle=0, gmtversion='newest', **kwargs):

gmt = GMT(version=gmtversion)

if gmt.is_gmt5():

row = [0, 0, text]

farg = ['-F+f%gp,%s,%s+j%s' % (font_size, font, 'black', 'BL')]

else:

row = [0, 0, font_size, 0, font, 'BL', text]

farg = []

gmt.pstext(

in_rows=[row],

finish=True,

R=(0, 1, 0, 1),

J='x10p',

N=True,

*farg,

**kwargs)

fn = gmt.tempfilename() + '.ps'

gmt.save(fn)

(_, stderr) = subprocess.Popen(

['gs', '-q', '-dNOPAUSE', '-dBATCH', '-r720', '-sDEVICE=bbox', fn],

stderr=subprocess.PIPE).communicate()

dx, dy = None, None

for line in stderr.splitlines():

if line.startswith(b'%%HiResBoundingBox:'):

l, b, r, t = [float(x) for x in line.split()[-4:]]

dx, dy = r-l, t-b

break

return dx, dy

class TableLiner(object):

'''Utility class to turn tables into lines.'''

def __init__(self, in_columns=None, in_rows=None, encoding='utf-8'):

self.in_columns = in_columns

self.in_rows = in_rows

self.encoding = encoding

def __iter__(self):

if self.in_columns is not None:

for row in zip(*self.in_columns):

yield (' '.join([newstr(x) for x in row])+'\n').encode(

self.encoding)

if self.in_rows is not None:

for row in self.in_rows:

yield (' '.join([newstr(x) for x in row])+'\n').encode(

self.encoding)

class LineStreamChopper(object):

'''File-like object to buffer data.'''

def __init__(self, liner):

self.chopsize = None

self.liner = liner

self.chop_iterator = None

self.closed = False

def _chopiter(self):

buf = BytesIO()

for line in self.liner:

buf.write(line)

buflen = buf.tell()

if self.chopsize is not None and buflen >= self.chopsize:

buf.seek(0)

while buf.tell() <= buflen-self.chopsize:

yield buf.read(self.chopsize)

newbuf = BytesIO()

newbuf.write(buf.read())

buf.close()

buf = newbuf

yield(buf.getvalue())

buf.close()

def read(self, size=None):

if self.closed:

raise ValueError('Cannot read from closed LineStreamChopper.')

if self.chop_iterator is None:

self.chopsize = size

self.chop_iterator = self._chopiter()

self.chopsize = size

try:

return next(self.chop_iterator)

except StopIteration:

return ''

def close(self):

self.chopsize = None

self.chop_iterator = None

self.closed = True

def flush(self):

pass

font_tab = {

0: 'Helvetica',

1: 'Helvetica-Bold',

}

font_tab_rev = dict((v, k) for (k, v) in font_tab.items())

class GMT(object):

'''A thin wrapper to GMT command execution.

A dict ``config`` may be given to override some of the default GMT

parameters. The ``version`` argument may be used to select a specific GMT

version, which should be used with this GMT instance. The selected

version of GMT has to be installed on the system, must be supported by

gmtpy and gmtpy must know where to find it.

Each instance of this class is used for the task of producing one PS or PDF

output file.

Output of a series of GMT commands is accumulated in memory and can then be

saved as PS or PDF file using the :py:meth:`save` method.

GMT commands are accessed as method calls to instances of this class. See

the :py:meth:`__getattr__` method for details on how the method's

arguments are translated into options and arguments for the GMT command.

Associated with each instance of this class, a temporary directory is

created, where temporary files may be created, and which is automatically

deleted, when the object is destroyed. The :py:meth:`tempfilename` method

may be used to get a random filename in the instance's temporary directory.

Any .gmtdefaults files are ignored. The GMT class uses a fixed

set of defaults, which may be altered via an argument to the constructor.

If possible, GMT is run in 'isolation mode', which was introduced with GMT

version 4.2.2, by setting `GMT_TMPDIR` to the instance's temporary

directory. With earlier versions of GMT, problems may arise with parallel

execution of more than one GMT instance.

Each instance of the GMT class may pick a specific version of GMT which

shall be used, so that, if multiple versions of GMT are installed on the

system, different versions of GMT can be used simultaneously such that

backward compatibility of the scripts can be maintained.

'''

def __init__(

self,

config=None,

kontinue=None,

version='newest',

config_papersize=None,

eps_mode=False):

self.installation = get_gmt_installation(version)

self.gmt_config = dict(self.installation['defaults'])

self.eps_mode = eps_mode

self._shutil = shutil

if config:

self.gmt_config.update(config)

if config_papersize:

if not isinstance(config_papersize, str):

config_papersize = 'Custom_%ix%i' % (

int(config_papersize[0]), int(config_papersize[1]))

if self.is_gmt5():

self.gmt_config['PS_MEDIA'] = config_papersize

else:

self.gmt_config['PAPER_MEDIA'] = config_papersize

self.tempdir = tempfile.mkdtemp("", "gmtpy-")

self.gmt_config_filename = pjoin(self.tempdir, 'gmt.conf')

self.gen_gmt_config_file(self.gmt_config_filename, self.gmt_config)

if kontinue is not None:

self.load_unfinished(kontinue)

self.needstart = False

else:

self.output = BytesIO()

self.needstart = True

self.finished = False

self.environ = os.environ.copy()

self.environ['GMTHOME'] = self.installation.get('home', '')

# GMT isolation mode: works only properly with GMT version >= 4.2.2

self.environ['GMT_TMPDIR'] = self.tempdir

self.layout = None

self.command_log = []

self.keep_temp_dir = False

def is_gmt5(self):

return self.installation['version'][0] == '5'

def get_version(self):

return self.installation['version']

def get_config(self, key):

return self.gmt_config[key]

def to_points(self, string):

if not string:

return 0

unit = string[-1]

if unit in _units:

return float(string[:-1])/_units[unit]

else:

default_unit = measure_unit(self.gmt_config).lower()[0]

return float(string)/_units[default_unit]

def label_font_size(self):

if self.is_gmt5():

return self.to_points(self.gmt_config['FONT_LABEL'].split(',')[0])

else:

return self.to_points(self.gmt_config['LABEL_FONT_SIZE'])

def label_font(self):

if self.is_gmt5():

return font_tab_rev(self.gmt_config['FONT_LABEL'].split(',')[1])

else:

return self.gmt_config['LABEL_FONT']

def gen_gmt_config_file(self, config_filename, config):

f = open(config_filename, 'wb')

f.write(

('#\n# GMT %s Defaults file\n'

% self.installation['version']).encode('ascii'))

for k, v in config.items():

f.write(('%s = %s\n' % (k, v)).encode('ascii'))

f.close()

def __del__(self):

if not self.keep_temp_dir:

self._shutil.rmtree(self.tempdir)

def _gmtcommand(self, command, *addargs, **kwargs):

'''Execute arbitrary GMT command.

See docstring in __getattr__ for details.

'''

in_stream = kwargs.pop('in_stream', None)

in_filename = kwargs.pop('in_filename', None)

in_string = kwargs.pop('in_string', None)

in_columns = kwargs.pop('in_columns', None)

in_rows = kwargs.pop('in_rows', None)

out_stream = kwargs.pop('out_stream', None)

out_filename = kwargs.pop('out_filename', None)

out_discard = kwargs.pop('out_discard', None)

finish = kwargs.pop('finish', False)

suppressdefaults = kwargs.pop('suppress_defaults', False)

config_override = kwargs.pop('config', None)

assert(not self.finished)

# check for mutual exclusiveness on input and output possibilities

assert(1 >= len(

[x for x in [

in_stream, in_filename, in_string, in_columns, in_rows]

if x is not None]))

assert(1 >= len([x for x in [out_stream, out_filename, out_discard]

if x is not None]))

options = []

gmt_config = self.gmt_config

if not self.is_gmt5():

gmt_config_filename = self.gmt_config_filename

if config_override:

gmt_config = self.gmt_config.copy()

gmt_config.update(config_override)

gmt_config_override_filename = pjoin(

self.tempdir, 'gmtdefaults_override')

self.gen_gmt_config_file(

gmt_config_override_filename, gmt_config)

gmt_config_filename = gmt_config_override_filename

else: # gmt5 needs override variables as --VAR=value

if config_override:

for k, v in config_override.items():

options.append('--%s=%s' % (k, v))

if out_discard:

out_filename = '/dev/null'

out_mustclose = False

if out_filename is not None:

out_mustclose = True

out_stream = open(out_filename, 'wb')

if in_filename is not None:

in_stream = open(in_filename, 'rb')

if in_string is not None:

in_stream = BytesIO(in_string)

encoding_gmt = gmt_config.get(

'PS_CHAR_ENCODING',

gmt_config.get('CHAR_ENCODING', 'ISOLatin1+'))

encoding = encoding_gmt_to_python[encoding_gmt.lower()]

if in_columns is not None or in_rows is not None:

in_stream = LineStreamChopper(TableLiner(in_columns=in_columns,

in_rows=in_rows,

encoding=encoding))

# convert option arguments to strings

for k, v in kwargs.items():

if len(k) > 1:

raise GmtPyError('Found illegal keyword argument "%s" '

'while preparing options for command "%s"'

% (k, command))

if type(v) is bool:

if v:

options.append('-%s' % k)

elif type(v) is tuple or type(v) is list:

options.append('-%s' % k + '/'.join([str(x) for x in v]))

else:

options.append('-%s%s' % (k, str(v)))

# if not redirecting to an external sink, handle -K -O

if out_stream is None:

if not finish:

options.append('-K')

else:

self.finished = True

if not self.needstart:

options.append('-O')

else:

self.needstart = False

out_stream = self.output

# run the command

if self.is_gmt5():

args = [pjoin(self.installation['bin'], 'gmt'), command]

else:

args = [pjoin(self.installation['bin'], command)]

if not os.path.isfile(args[0]):

raise OSError('No such file: %s' % args[0])

args.extend(options)

args.extend(addargs)

if not self.is_gmt5() and not suppressdefaults:

# does not seem to work with GMT 5 (and should not be necessary

args.append('+'+gmt_config_filename)

bs = 2048

p = subprocess.Popen(args, stdin=subprocess.PIPE,

stdout=subprocess.PIPE, bufsize=bs,

env=self.environ)

while True:

cr, cw, cx = select([p.stdout], [p.stdin], [])

if cr:

out_stream.write(p.stdout.read(bs))

if cw:

if in_stream is not None:

data = in_stream.read(bs)

if len(data) == 0:

break

p.stdin.write(data)

else:

break

if not cr and not cw:

break

p.stdin.close()

while True:

data = p.stdout.read(bs)

if len(data) == 0:

break

out_stream.write(data)

p.stdout.close()

retcode = p.wait()

if in_stream is not None:

in_stream.close()

if out_mustclose:

out_stream.close()

if retcode != 0:

self.keep_temp_dir = True

raise GMTError('Command %s returned an error. '

'While executing command:\n%s'

% (command, escape_shell_args(args)))

self.command_log.append(args)

def __getattr__(self, command):

'''Maps to call self._gmtcommand(command, \\*addargs, \\*\\*kwargs).

Execute arbitrary GMT command.

Run a GMT command and by default append its postscript output to the

output file maintained by the GMT instance on which this method is

called.

Except for a few keyword arguments listed below, any ``kwargs`` and

``addargs`` are converted into command line options and arguments and

passed to the GMT command. Numbers in keyword arguments are converted

into strings. E.g. ``S=10`` is translated into ``'-S10'``. Tuples of

numbers or strings are converted into strings where the elements of the

tuples are separated by slashes '/'. E.g. ``R=(10, 10, 20, 20)`` is

translated into ``'-R10/10/20/20'``. Options with a boolean argument

are only appended to the GMT command, if their values are True.

If no output redirection is in effect, the -K and -O options are

handled by gmtpy and thus should not be specified. Use

``out_discard=True`` if you don't want -K or -O beeing added, but are

not interested in the output.

The standard input of the GMT process is fed by data selected with one

of the following ``in_*`` keyword arguments:

=============== =======================================================

``in_stream`` Data is read from an open file like object.

``in_filename`` Data is read from the given file.

``in_string`` String content is dumped to the process.

``in_columns`` A 2D nested iterable whose elements can be accessed as

``in_columns[icolumn][irow]`` is converted into an

ascii

table, which is fed to the process.

``in_rows`` A 2D nested iterable whos elements can be accessed as

``in_rows[irow][icolumn]`` is converted into an ascii

table, which is fed to the process.

=============== =======================================================

The standard output of the GMT process may be redirected by one of the

following options:

================= =====================================================

``out_stream`` Output is fed to an open file like object.

``out_filename`` Output is dumped to the given file.

``out_discard`` If True, output is dumped to :file:`/dev/null`.

================= =====================================================

Additional keyword arguments:

===================== =================================================

``config`` Dict with GMT defaults which override the

currently active set of defaults exclusively

during this call.

``finish`` If True, the postscript file, which is maintained

by the GMT instance is finished, and no further

plotting is allowed.

``suppress_defaults`` Suppress appending of the ``'+gmtdefaults'``

option to the command.

===================== =================================================

'''

def f(*args, **kwargs):

return self._gmtcommand(command, *args, **kwargs)

return f

def tempfilename(self, name=None):

'''Get filename for temporary file in the private temp directory.

If no ``name`` argument is given, a random name is picked. If

``name`` is given, returns a path ending in that ``name``.'''

if not name:

name = ''.join(

[random.choice('abcdefghijklmnopqrstuvwxyz')

for i in range(10)])

fn = pjoin(self.tempdir, name)

return fn

def tempfile(self, name=None):

'''Create and open a file in the private temp directory.'''

fn = self.tempfilename(name)

f = open(fn, 'wb')

return f, fn

def save_unfinished(self, filename):

out = open(filename, 'wb')

out.write(self.output.getvalue())

out.close()

def load_unfinished(self, filename):

self.output = BytesIO()

self.finished = False

inp = open(filename, 'rb')

self.output.write(inp.read())

inp.close()

def dump(self, ident):

filename = self.tempfilename('breakpoint-%s' % ident)

self.save_unfinished(filename)

def load(self, ident):

filename = self.tempfilename('breakpoint-%s' % ident)

self.load_unfinished(filename)

def save(self, filename=None, bbox=None, resolution=150, oversample=2.,

width=None, height=None, size=None, crop_eps_mode=False,

psconvert=False):

'''

Finish and save figure as PDF, PS or PPM file.

If filename ends with ``'.pdf'`` a PDF file is created by piping the

GMT output through :program:`gmtpy-epstopdf`.

If filename ends with ``'.png'`` a PNG file is created by running

:program:`gmtpy-epstopdf`, :program:`pdftocairo` and

:program:`convert`. ``resolution`` specifies the resolution in DPI for

raster file formats. Rasterization is done at a higher resolution if

``oversample`` is set to a value higher than one. The output image size

can also be controlled by setting ``width``, ``height`` or ``size``

instead of ``resolution``. When ``size`` is given, the image is scaled

so that ``max(width, height) == size``.

The bounding box is set according to the values given in ``bbox``.

'''

if not self.finished:

self.psxy(R=True, J=True, finish=True)

if filename:

tempfn = pjoin(self.tempdir, 'incomplete')

out = open(tempfn, 'wb')

else:

out = sys.stdout

if bbox and not self.is_gmt5():

out.write(replace_bbox(bbox, self.output.getvalue()))

else:

out.write(self.output.getvalue())

if filename:

out.close()

if filename.endswith('.ps') or (

not self.is_gmt5() and filename.endswith('.eps')):

shutil.move(tempfn, filename)

return

if self.is_gmt5():

if crop_eps_mode:

addarg = ['-A0']

else:

addarg = []

subprocess.call(

[pjoin(self.installation['bin'], 'gmt'), 'psconvert',

'-Te', '-F%s' % tempfn + '.eps', tempfn, ] + addarg)

if bbox:

with open(tempfn + '.eps', 'rb') as fin:

with open(tempfn + '-fixbb.eps', 'wb') as fout:

replace_bbox(bbox, fin, fout)

shutil.move(tempfn + '-fixbb.eps', tempfn + '.eps')

else:

shutil.move(tempfn, tempfn + '.eps')

if filename.endswith('.pdf'):

if psconvert:

gmt_bin = pjoin(self.installation['bin'], 'gmt')

subprocess.call([gmt_bin, 'psconvert', tempfn, '-Tf',

'-F' + filename])

else:

subprocess.call(['gmtpy-epstopdf', '--res=%i' % resolution,

'--outfile=' + filename, tempfn + '.eps'])

else:

subprocess.call([

'gmtpy-epstopdf',

'--res=%i' % (resolution * oversample),

'--outfile=' + tempfn + '.pdf', tempfn + '.eps'])

convert_graph(

tempfn + '.pdf', filename,

resolution=resolution, oversample=oversample,

size=size, width=width, height=height)

def bbox(self):

return get_bbox(self.output.getvalue())

def get_command_log(self):

'''Get the command log.'''

return self.command_log

def __str__(self):

s = ''

for com in self.command_log:

s += com[0] + "\n " + "\n ".join(com[1:]) + "\n\n"

return s

def page_size_points(self):

'''Try to get paper size of output postscript file in points.'''

pm = paper_media(self.gmt_config).lower()

if pm.endswith('+') or pm.endswith('-'):

pm = pm[:-1]

orient = page_orientation(self.gmt_config).lower()

if pm in all_paper_sizes():

if orient == 'portrait':

return get_paper_size(pm)

else:

return get_paper_size(pm)[1], get_paper_size(pm)[0]

m = re.match(r'custom_([0-9.]+)([cimp]?)x([0-9.]+)([cimp]?)', pm)

if m:

w, uw, h, uh = m.groups()

w, h = float(w), float(h)

if uw:

w *= _units[uw]

if uh:

h *= _units[uh]

if orient == 'portrait':

return w, h

else:

return h, w

return None, None

def default_layout(self, with_palette=False):

'''Get a default layout for the output page.

One of three different layouts is choosen, depending on the

`PAPER_MEDIA` setting in the GMT configuration dict.

If `PAPER_MEDIA` ends with a ``'+'`` (EPS output is selected), a

:py:class:`FrameLayout` is centered on the page, whose size is

controlled by its center widget's size plus the margins of the

:py:class:`FrameLayout`.

If `PAPER_MEDIA` indicates, that a custom page size is wanted by

starting with ``'Custom_'``, a :py:class:`FrameLayout` is used to fill

the complete page. The center widget's size is then controlled by the

page's size minus the margins of the :py:class:`FrameLayout`.

In any other case, two FrameLayouts are nested, such that the outer

layout attaches a 1 cm (printer) margin around the complete page, and

the inner FrameLayout's center widget takes up as much space as

possible under the constraint, that an aspect ratio of 1/golden_ratio

is preserved.

In any case, a reference to the innermost :py:class:`FrameLayout`

instance is returned. The top-level layout can be accessed by calling

:py:meth:`Widget.get_parent` on the returned layout. '''

if self.layout is None:

w, h = self.page_size_points()

if w is None or h is None:

raise GmtPyError("Can't determine page size for layout")

pm = paper_media(self.gmt_config).lower()

if with_palette:

palette_layout = GridLayout(3, 1)

spacer = palette_layout.get_widget(1, 0)

palette_widget = palette_layout.get_widget(2, 0)

spacer.set_horizontal(0.5*cm)

palette_widget.set_horizontal(0.5*cm)

if pm.endswith('+') or self.eps_mode:

outer = CenterLayout()

outer.set_policy((w, h), (0., 0.))

inner = FrameLayout()

outer.set_widget(inner)

if with_palette:

inner.set_widget('center', palette_layout)

widget = palette_layout

else:

widget = inner.get_widget('center')

widget.set_policy((w/golden_ratio, 0.), (0., 0.),

aspect=1./golden_ratio)

mw = 3.0*cm

inner.set_fixed_margins(

mw, mw, mw/golden_ratio, mw/golden_ratio)

self.layout = inner

elif pm.startswith('custom_'):

layout = FrameLayout()

layout.set_policy((w, h), (0., 0.))

mw = 3.0*cm

layout.set_min_margins(

mw, mw, mw/golden_ratio, mw/golden_ratio)

if with_palette:

layout.set_widget('center', palette_layout)

self.layout = layout

else:

outer = FrameLayout()

outer.set_policy((w, h), (0., 0.))

outer.set_fixed_margins(1.*cm, 1.*cm, 1.*cm, 1.*cm)

inner = FrameLayout()

outer.set_widget('center', inner)

mw = 3.0*cm

inner.set_min_margins(mw, mw, mw/golden_ratio, mw/golden_ratio)

if with_palette:

inner.set_widget('center', palette_layout)

widget = palette_layout

else:

widget = inner.get_widget('center')

widget.set_aspect(1./golden_ratio)

self.layout = inner

return self.layout

def draw_layout(self, layout):

'''Use psxy to draw layout; for debugging'''

# corners = layout.get_corners(descend=True)

rects = num.array(layout.get_sizes(), dtype=num.float)

rects_wid = rects[:, 0, 0]

rects_hei = rects[:, 0, 1]

rects_center_x = rects[:, 1, 0] + rects_wid*0.5

rects_center_y = rects[:, 1, 1] + rects_hei*0.5

nrects = len(rects)

prects = (rects_center_x, rects_center_y, num.arange(nrects),

num.zeros(nrects), rects_hei, rects_wid)

# points = num.array(corners, dtype=num.float)

cptfile = self.tempfilename() + '.cpt'

self.makecpt(

C='ocean',

T='%g/%g/%g' % (-nrects, nrects, 1),

Z=True,

out_filename=cptfile, suppress_defaults=True)

bb = layout.bbox()

self.psxy(

in_columns=prects,

C=cptfile,

W='1p',

S='J',

R=(bb[0], bb[2], bb[1], bb[3]),

*layout.XYJ())

def simpleconf_to_ax(conf, axname):

c = {}

x = axname

for x in ('', axname):

for k in ('label', 'unit', 'scaled_unit', 'scaled_unit_factor',

'space', 'mode', 'approx_ticks', 'limits', 'masking', 'inc',

'snap'):

if x+k in conf:

c[k] = conf[x+k]

return Ax(**c)

class DensityPlotDef(object):

def __init__(self, data, cpt='ocean', tension=0.7, size=(640, 480),

contour=False, method='surface', zscaler=None, **extra):

self.data = data

self.cpt = cpt

self.tension = tension

self.size = size

self.contour = contour

self.method = method

self.zscaler = zscaler

self.extra = extra

class TextDef(object):

def __init__(

self,

data,

size=9,

justify='MC',

fontno=0,

offset=(0, 0),

color='black'):

self.data = data

self.size = size

self.justify = justify

self.fontno = fontno

self.offset = offset

self.color = color

class Simple(object):

def __init__(self, gmtconfig=None, gmtversion='newest', **simple_config):

self.data = []

self.symbols = []

self.config = copy.deepcopy(simple_config)

self.gmtconfig = gmtconfig

self.density_plot_defs = []

self.text_defs = []

self.gmtversion = gmtversion

self.data_x = []

self.symbols_x = []

self.data_y = []

self.symbols_y = []

self.default_config = {}

self.set_defaults(width=15.*cm,

height=15.*cm / golden_ratio,

margins=(2.*cm, 2.*cm, 2.*cm, 2.*cm),

with_palette=False,

palette_offset=0.5*cm,

palette_width=None,

palette_height=None,

zlabeloffset=2*cm,

draw_layout=False)

self.setup_defaults()

self.fixate_widget_aspect = False

def setup_defaults(self):

pass

def set_defaults(self, **kwargs):

self.default_config.update(kwargs)

def plot(self, data, symbol=''):

self.data.append(data)

self.symbols.append(symbol)

def density_plot(self, data, **kwargs):

dpd = DensityPlotDef(data, **kwargs)

self.density_plot_defs.append(dpd)

def text(self, data, **kwargs):

dpd = TextDef(data, **kwargs)

self.text_defs.append(dpd)

def plot_x(self, data, symbol=''):

self.data_x.append(data)

self.symbols_x.append(symbol)

def plot_y(self, data, symbol=''):

self.data_y.append(data)

self.symbols_y.append(symbol)

def set(self, **kwargs):

self.config.update(kwargs)

def setup_base(self, conf):

w = conf.pop('width')

h = conf.pop('height')

margins = conf.pop('margins')

gmtconfig = {}

if self.gmtconfig is not None:

gmtconfig.update(self.gmtconfig)

gmt = GMT(

version=self.gmtversion,

config=gmtconfig,

config_papersize='Custom_%ix%i' % (w, h))

layout = gmt.default_layout(with_palette=conf['with_palette'])

layout.set_min_margins(*margins)

if conf['with_palette']:

widget = layout.get_widget().get_widget(0, 0)

spacer = layout.get_widget().get_widget(1, 0)

spacer.set_horizontal(conf['palette_offset'])

palette_widget = layout.get_widget().get_widget(2, 0)

if conf['palette_width'] is not None:

palette_widget.set_horizontal(conf['palette_width'])

if conf['palette_height'] is not None:

palette_widget.set_vertical(conf['palette_height'])

widget.set_vertical(h-margins[2]-margins[3]-0.03*cm)

return gmt, layout, widget, palette_widget

else:

widget = layout.get_widget()

return gmt, layout, widget, None

def setup_projection(self, widget, scaler, conf):

pass

def setup_scaling(self, conf):

ndims = 2

if self.density_plot_defs:

ndims = 3

axes = [simpleconf_to_ax(conf, x) for x in 'xyz'[:ndims]]

data_all = []

data_all.extend(self.data)

for dsd in self.density_plot_defs:

if dsd.zscaler is None:

data_all.append(dsd.data)

else:

data_all.append(dsd.data[:2])

data_chopped = [ds[:ndims] for ds in data_all]

scaler = ScaleGuru(data_chopped, axes=axes[:ndims])

self.setup_scaling_plus(scaler, axes[:ndims])

return scaler

def setup_scaling_plus(self, scaler, axes):

pass

def setup_scaling_extra(self, scaler, conf):

scaler_x = scaler.copy()

scaler_x.data_ranges[1] = (0., 1.)

scaler_x.axes[1].mode = 'off'

scaler_y = scaler.copy()

scaler_y.data_ranges[0] = (0., 1.)

scaler_y.axes[0].mode = 'off'

return scaler_x, scaler_y

def draw_density(self, gmt, widget, scaler):

R = scaler.R()

# par = scaler.get_params()

rxyj = R + widget.XYJ()

innerticks = False

for dpd in self.density_plot_defs:

fn_cpt = gmt.tempfilename() + '.cpt'

if dpd.zscaler is not None:

s = dpd.zscaler

else:

s = scaler

gmt.makecpt(C=dpd.cpt, out_filename=fn_cpt, *s.T())

fn_grid = gmt.tempfilename()

fn_mean = gmt.tempfilename()

if dpd.method in ('surface', 'triangulate'):

gmt.blockmean(in_columns=dpd.data,

I='%i+/%i+' % dpd.size, # noqa

out_filename=fn_mean, *R)

if dpd.method == 'surface':

gmt.surface(

in_filename=fn_mean,

T=dpd.tension,

G=fn_grid,

I='%i+/%i+' % dpd.size, # noqa

out_discard=True,

*R)

if dpd.method == 'triangulate':

gmt.triangulate(

in_filename=fn_mean,

G=fn_grid,

I='%i+/%i+' % dpd.size, # noqa

out_discard=True,

V=True,

*R)

if gmt.is_gmt5():

gmt.grdimage(fn_grid, C=fn_cpt, E='i', n='l', *rxyj)

else:

gmt.grdimage(fn_grid, C=fn_cpt, E='i', S='l', *rxyj)

if dpd.contour:

gmt.grdcontour(fn_grid, C=fn_cpt, W='0.5p,black', *rxyj)

innerticks = '0.5p,black'

os.remove(fn_grid)

os.remove(fn_mean)

if dpd.method == 'fillcontour':

extra = dict(C=fn_cpt)

extra.update(dpd.extra)

gmt.pscontour(in_columns=dpd.data,

I=True, *rxyj, **extra) # noqa

if dpd.method == 'contour':

extra = dict(W='0.5p,black', C=fn_cpt)

extra.update(dpd.extra)

gmt.pscontour(in_columns=dpd.data, *rxyj, **extra)

return fn_cpt, innerticks

def draw_basemap(self, gmt, widget, scaler):

gmt.psbasemap(*(widget.JXY() + scaler.RB(ax_projection=True)))

def draw(self, gmt, widget, scaler):

rxyj = scaler.R() + widget.JXY()

for dat, sym in zip(self.data, self.symbols):

gmt.psxy(in_columns=dat, *(sym.split()+rxyj))

def post_draw(self, gmt, widget, scaler):

pass

def pre_draw(self, gmt, widget, scaler):

pass

def draw_extra(self, gmt, widget, scaler_x, scaler_y):

for dat, sym in zip(self.data_x, self.symbols_x):

gmt.psxy(in_columns=dat,

*(sym.split() + scaler_x.R() + widget.JXY()))

for dat, sym in zip(self.data_y, self.symbols_y):

gmt.psxy(in_columns=dat,

*(sym.split() + scaler_y.R() + widget.JXY()))

def draw_text(self, gmt, widget, scaler):

rxyj = scaler.R() + widget.JXY()

for td in self.text_defs:

x, y = td.data[0:2]

text = td.data[-1]

size = td.size

angle = 0

fontno = td.fontno

justify = td.justify

color = td.color

if gmt.is_gmt5():

gmt.pstext(

in_rows=[(x, y, text)],

F='+f%gp,%s,%s+a%g+j%s' % (

size, fontno, color, angle, justify),

D='%gp/%gp' % td.offset, *rxyj)

else:

gmt.pstext(

in_rows=[(x, y, size, angle, fontno, justify, text)],

D='%gp/%gp' % td.offset, *rxyj)

def save(self, filename, resolution=150):

conf = dict(self.default_config)

conf.update(self.config)

gmt, layout, widget, palette_widget = self.setup_base(conf)

scaler = self.setup_scaling(conf)

scaler_x, scaler_y = self.setup_scaling_extra(scaler, conf)

self.setup_projection(widget, scaler, conf)

if self.fixate_widget_aspect:

aspect = aspect_for_projection(

gmt.installation['version'], *(widget.J() + scaler.R()))

widget.set_aspect(aspect)

if conf['draw_layout']:

gmt.draw_layout(layout)

cptfile = None

if self.density_plot_defs:

cptfile, innerticks = self.draw_density(gmt, widget, scaler)

self.pre_draw(gmt, widget, scaler)

self.draw(gmt, widget, scaler)

self.post_draw(gmt, widget, scaler)

self.draw_extra(gmt, widget, scaler_x, scaler_y)

self.draw_text(gmt, widget, scaler)

self.draw_basemap(gmt, widget, scaler)

if palette_widget and cptfile:

nice_palette(gmt, palette_widget, scaler, cptfile,

innerticks=innerticks,

zlabeloffset=conf['zlabeloffset'])

gmt.save(filename, resolution=resolution)

class LinLinPlot(Simple):

pass

class LogLinPlot(Simple):

def setup_defaults(self):

self.set_defaults(xmode='min-max')

def setup_projection(self, widget, scaler, conf):

widget['J'] = '-JX%(width)gpl/%(height)gp'

scaler['B'] = '-B2:%(xlabel)s:/%(yinc)g:%(ylabel)s:WSen'

class LinLogPlot(Simple):

def setup_defaults(self):

self.set_defaults(ymode='min-max')

def setup_projection(self, widget, scaler, conf):

widget['J'] = '-JX%(width)gp/%(height)gpl'

scaler['B'] = '-B%(xinc)g:%(xlabel)s:/2:%(ylabel)s:WSen'

class LogLogPlot(Simple):

def setup_defaults(self):

self.set_defaults(mode='min-max')

def setup_projection(self, widget, scaler, conf):

widget['J'] = '-JX%(width)gpl/%(height)gpl'

scaler['B'] = '-B2:%(xlabel)s:/2:%(ylabel)s:WSen'

class AziDistPlot(Simple):

def __init__(self, *args, **kwargs):

Simple.__init__(self, *args, **kwargs)

self.fixate_widget_aspect = True

def setup_defaults(self):

self.set_defaults(

height=15.*cm,

width=15.*cm,

xmode='off',

xlimits=(0., 360.),

xinc=45.)

def setup_projection(self, widget, scaler, conf):

widget['J'] = '-JPa%(width)gp'

def setup_scaling_plus(self, scaler, axes):

scaler['B'] = '-B%(xinc)g:%(xlabel)s:/%(yinc)g:%(ylabel)s:N'

class MPlot(Simple):

def __init__(self, *args, **kwargs):

Simple.__init__(self, *args, **kwargs)

self.fixate_widget_aspect = True

def setup_defaults(self):

self.set_defaults(xmode='min-max', ymode='min-max')

def setup_projection(self, widget, scaler, conf):

par = scaler.get_params()

lon0 = (par['xmin'] + par['xmax'])/2.

lat0 = (par['ymin'] + par['ymax'])/2.

sll = '%g/%g' % (lon0, lat0)

widget['J'] = '-JM' + sll + '/%(width)gp'

scaler['B'] = \

'-B%(xinc)gg%(xinc)g:%(xlabel)s:/%(yinc)gg%(yinc)g:%(ylabel)s:WSen'

def nice_palette(gmt, widget, scaleguru, cptfile, zlabeloffset=0.8*inch,

innerticks=True):

par = scaleguru.get_params()

par_ax = scaleguru.get_params(ax_projection=True)

nz_palette = int(widget.height()/inch * 300)

px = num.zeros(nz_palette*2)

px[1::2] += 1

pz = num.linspace(par['zmin'], par['zmax'], nz_palette).repeat(2)

pdz = pz[2]-pz[0]

palgrdfile = gmt.tempfilename()

pal_r = (0, 1, par['zmin'], par['zmax'])

pal_ax_r = (0, 1, par_ax['zmin'], par_ax['zmax'])

gmt.xyz2grd(

G=palgrdfile, R=pal_r,

I=(1, pdz), in_columns=(px, pz, pz), # noqa

out_discard=True)

gmt.grdimage(palgrdfile, R=pal_r, C=cptfile, *widget.JXY())

if isinstance(innerticks, str):

tickpen = innerticks

gmt.grdcontour(palgrdfile, W=tickpen, R=pal_r, C=cptfile,

*widget.JXY())

negpalwid = '%gp' % -widget.width()

if not isinstance(innerticks, str) and innerticks:

ticklen = negpalwid

else:

ticklen = '0p'

TICK_LENGTH_PARAM = 'MAP_TICK_LENGTH' if gmt.is_gmt5() else 'TICK_LENGTH'

gmt.psbasemap(

R=pal_ax_r, B='4::/%(zinc)g::nsw' % par_ax,

config={TICK_LENGTH_PARAM: ticklen},

*widget.JXY())

if innerticks:

gmt.psbasemap(

R=pal_ax_r, B='4::/%(zinc)g::E' % par_ax,

config={TICK_LENGTH_PARAM: '0p'},

*widget.JXY())

else:

gmt.psbasemap(R=pal_ax_r, B='4::/%(zinc)g::E' % par_ax, *widget.JXY())

if par_ax['zlabel']:

label_font = gmt.label_font()

label_font_size = gmt.label_font_size()

label_offset = zlabeloffset

gmt.pstext(

R=(0, 1, 0, 2), D="%gp/0p" % label_offset,

N=True,

in_rows=[(1, 1, label_font_size, -90, label_font, 'CB',

par_ax['zlabel'])],

*widget.JXY())

Coverage for /usr/local/lib/python3.7/dist-packages/pyrocko/plot/gmtpy.py : 77%

1727 statements 1338 run 389 missing 43 excluded