pygorithm/imgs/test_geometry/test_extrapolated_intersection/utils.py at 9db4ca4e3b85c3b9757c7bd0d28644e8a9b35529 · OmkarPathak/pygorithm · 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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
"""
Collection of functions that make making graphs easier.
"""

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as patches
import sys

def prepare_figure(fig, ax, title, xlim, ylim):
    """
    xlim and ylim must start at a negative number and end
    at a positive number. they must both be completely integer
    values
    """

    # set small title
    fig.suptitle(title)

    # force limits (defaults are always too thin)
    ax.set_xlim(xlim[0], xlim[1])
    ax.set_ylim(ylim[0], ylim[1])

    # force reasonable tick sizes (default is always bad except for values between 4 and 7)
    # note this is not the same as the defaults (removes the outer lines) which are clutter
    # (the edges of the graph aren't used anyway... dont put grid lines there!)
    ax.xaxis.set_ticks(range(xlim[0]+1, xlim[1]))
    ax.yaxis.set_ticks(range(ylim[0]+1, ylim[1]))

    # force reasonable aspect ratio (default is scaled weird)
    ax.set_aspect('equal')

    # remove outer spines (clutter) and move left and bottom spines to 0 (instead of xmin and ymin)
    ax.spines['left'].set_position('zero')
    ax.spines['right'].set_color('none')
    ax.spines['top'].set_color('none')
    ax.spines['bottom'].set_position('zero')

    # add dashed grid
    ax.grid(True, linestyle='dashed')

    # remove unnecessary tick marks and labels (why would you label 0, 0 by default?)
    ax.xaxis.get_major_ticks()[(-xlim[0]) - 1].label1.set_visible(False)
    ax.yaxis.get_major_ticks()[(-ylim[0]) - 1].label1.set_visible(False)

def annotate_point(fig, ax, renderer, ptx, pty, dir, family="sans-serif", size="x-small", spacing=5, **kwargs):
    if dir == 'none':
        return

    anstr = "({}, {})".format(ptx, pty)

    an = ax.annotate(s=anstr, xy=(ptx, pty), family=family, size=size, **kwargs)
    an_extents = an.get_window_extent(renderer)
    an.remove()

    offsetx = 0
    offsety = 0
    if dir == 'left':
        offsetx = -an_extents.width - spacing*2
        offsety = -an_extents.height / 2
    elif dir == 'topleft':
        offsetx = -an_extents.width - spacing
        offsety = spacing
    elif dir == 'top':
        offsetx = -an_extents.width / 2
        offsety = spacing*2
    elif dir == 'topright':
        offsetx = spacing
        offsety = spacing
    elif dir == 'right':
        offsetx = spacing*2
        offsety = -an_extents.height / 2
    elif dir == 'botright':
        offsetx = spacing
        offsety = -an_extents.height - spacing
    elif dir == 'bot':
        offsetx = -an_extents.width / 2
        offsety = -an_extents.height - spacing*2
    elif dir == 'botleft':
        offsetx = -an_extents.width - spacing
        offsety = -an_extents.height - spacing

    return ax.annotate(s=anstr, xy=(ptx, pty), xytext=(offsetx, offsety), textcoords='offset pixels', family=family, size=size, **kwargs)

def create_moving_point(fig, ax, renderer, ptx, pty, arendx, arendy, dir='botleft'):
    pt = ax.scatter([ptx], [pty], zorder=5)

    an = annotate_point(fig, ax, renderer, ptx, pty, dir, zorder=5)

    ar = ax.annotate("", xy=(ptx, pty), xytext=(arendx, arendy), arrowprops=dict(arrowstyle="<-", shrinkA=0, shrinkB=0, zorder=4, mutation_scale=15))
    return pt, an, ar

def create_moving_line(fig, ax, renderer, pt1tup, pt2tup, movetup, dir='botleft', dir2=None):
    dir2 = dir if dir2 is None else dir2
    pts = ax.scatter([pt1tup[0], pt2tup[0]], [pt1tup[1], pt2tup[1]], zorder=5)
    anpt1 = annotate_point(fig, ax, renderer, pt1tup[0], pt1tup[1], dir, zorder=5)
    anpt2 = annotate_point(fig, ax, renderer, pt2tup[0], pt2tup[1], dir2, zorder=5)

    conn_arrow = ax.annotate("", xy=pt1tup, xytext=pt2tup, arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='g', zorder=4))

    move_pt1_arrow = ax.annotate("", xy=pt1tup, xytext=(pt1tup[0] + movetup[0], pt1tup[1] + movetup[1]), arrowprops=dict(arrowstyle="<-", linestyle=":", shrinkA=0, shrinkB=0, zorder=4, mutation_scale=15))
    move_pt2_arrow = ax.annotate("", xy=pt2tup, xytext=(pt2tup[0] + movetup[0], pt2tup[1] + movetup[1]), arrowprops=dict(arrowstyle="<-", linestyle=":", shrinkA=0, shrinkB=0, zorder=4, mutation_scale=15))

    #end_pts = ax.scatter([pt1tup[0] + movetup[0], pt2tup[0] + movetup[0]], [pt1tup[1] + movetup[1], pt2tup[1] + movetup[1]], zorder=5)
    end_pts_conn = ax.annotate("", xy=(pt1tup[0] + movetup[0], pt1tup[1] + movetup[1]), xytext=(pt2tup[0] + movetup[0], pt2tup[1] + movetup[1]), arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='k', zorder=4, linestyle="dashed"))
    return pts, anpt1, anpt2, move_pt1_arrow, move_pt2_arrow, end_pts_conn

def create_moving_polygon(fig, ax, renderer, points, move, dir='botleft'):
    pointsx = list(p[0] for p in points)
    pointsy = list(p[1] for p in points)
    ax.scatter(pointsx, pointsy, zorder=5)

    last = points[-1]
    for p in points:
        pdir = p[2] if len(p) > 2 else dir
        annotate_point(fig, ax, renderer, p[0], p[1], pdir, zorder=5)

        ax.annotate("", xy=(last[0], last[1]), xytext=(p[0], p[1]), arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='g', zorder=4))
        ax.annotate("", xy=(p[0], p[1]), xytext = (p[0] + move[0], p[1] + move[1]), arrowprops=dict(arrowstyle="<-", linestyle=":", shrinkA=0, shrinkB=0, zorder=4, mutation_scale=15))
        ax.annotate("", xy=(last[0] + move[0], last[1] + move[1]), xytext=(p[0] + move[0], p[1] + move[1]), arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='k', zorder=4, linestyle="dashed"))

        last = p

def create_still_segment(fig, ax, renderer, pt1tup, pt2tup, dir='botleft', dir2=None):
    dir2 = dir if dir2 is None else dir2

    segment_pts = ax.scatter([ pt1tup[0], pt2tup[0] ], [ pt1tup[1], pt2tup[1] ], c='r', zorder=5)
    segment_arrow = ax.annotate("", xy=pt1tup, xytext=pt2tup, arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='b', zorder=4))
    an1 = annotate_point(fig, ax, renderer, pt1tup[0], pt1tup[1], dir, zorder=5)
    an2 = annotate_point(fig, ax, renderer, pt2tup[0], pt2tup[1], dir2, zorder=5)
    return segment_pts, segment_arrow, an1, an2

def create_still_polygon(fig, ax, renderer, points, dir='botleft'):
    pointsx = list(p[0] for p in points)
    pointsy = list(p[1] for p in points)
    ax.scatter(pointsx, pointsy, zorder=5)

    last = points[-1]
    for p in points:
        pdir = p[2] if len(p) > 2 else dir
        annotate_point(fig, ax, renderer, p[0], p[1], pdir, zorder=5)

        ax.annotate("", xy=(last[0], last[1]), xytext=(p[0], p[1]), arrowprops=dict(arrowstyle="-", shrinkA=0, shrinkB=0, color='b', zorder=4))
        last = p

def create_newfig(title, xlim=(-1, 7), ylim=(-1, 5)):
    fig, ax = plt.subplots()
    renderer = fig.canvas.get_renderer()
    prepare_figure(fig, ax, title, xlim, ylim)
    return fig, ax, renderer

def run_or_export(*args):
    fns = args

    found_export_command = False
    skip_next = False
    found_an_only = False
    just_found_only = False
    indexes = []
    for i in range(1, len(sys.argv)):
        if just_found_only:
            indexes.append(int(sys.argv[i]) - 1)
            just_found_only = False
        elif sys.argv[i] == '--export':
            found_export_command = True
        elif sys.argv[i] == '--only':
            found_an_only = True
            just_found_only = True
        else:
            print('Unknown Command: {}'.format(sys.argv[i]))


    figaxtitletups = []
    for i in range(len(fns)):
        if not found_an_only or i in indexes:
            figaxtitletups.append(fns[i]())

    if found_export_command:
        for fig, ax, longtitle in figaxtitletups:
            fig.savefig('out/{}.png'.format(longtitle))
    else:
        plt.show()