OceanPy/plot.py at master · mayajakes/OceanPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from pylab import *

# http://stackoverflow.com/questions/24535848/drawing-log-linear-plot-on-a-square-plot-area-in-matplotlib
def linlogplot(fig):

    ax = gca()
    ax.set_xscale("log")

    # now get the figure size in real coordinates:
    fig  = gcf()
    fwidth = fig.get_figwidth()
    fheight = fig.get_figheight()

    # get the axis size and position in relative coordinates
    # this gives a BBox object
    bb = ax.get_position()

    # calculate them into real world coordinates
    axwidth = fwidth * (bb.x1 - bb.x0)
    axheight = fheight * (bb.y1 - bb.y0)

    # if the axis is wider than tall, then it has to be narrowe
    if axwidth > axheight:
        # calculate the narrowing relative to the figure
        narrow_by = (axwidth - axheight) / fwidth
        # move bounding box edges inwards the same amount to give the correct width
        bb.x0 += narrow_by / 2
        bb.x1 -= narrow_by / 2
    # else if the axis is taller than wide, make it vertically smaller
    # works the same as above
    elif axheight > axwidth:
        shrink_by = (axheight - axwidth) / fheight
        bb.y0 += shrink_by / 2
        bb.y1 -= shrink_by / 2

    ax.set_position(bb)
    show()

def loglinplot(fig):

    ax = gca()
    ax.set_yscale("log")

    # now get the figure size in real coordinates:
    fig  = gcf()
    fwidth = fig.get_figwidth()
    fheight = fig.get_figheight()

    # get the axis size and position in relative coordinates
    # this gives a BBox object
    bb = ax.get_position()

    # calculate them into real world coordinates
    axwidth = fwidth * (bb.x1 - bb.x0)
    axheight = fheight * (bb.y1 - bb.y0)

    # if the axis is wider than tall, then it has to be narrowe
    if axwidth > axheight:
        # calculate the narrowing relative to the figure
        narrow_by = (axwidth - axheight) / fwidth
        # move bounding box edges inwards the same amount to give the correct width
        bb.x0 += narrow_by / 2
        bb.x1 -= narrow_by / 2
    # else if the axis is taller than wide, make it vertically smaller
    # works the same as above
    elif axheight > axwidth:
        shrink_by = (axheight - axwidth) / fheight
        bb.y0 += shrink_by / 2
        bb.y1 -= shrink_by / 2

    ax.set_position(bb)
    show()

# # plot circle around point
# circ1000 = plt.Circle(x,y,1000, color='blue', fill=False)
# fig.gca().add_artist(circ1000)