astroimsim-data 0.1.4

Data package for astroimsim ecosystem.
Documentation 
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
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233

#### >>>> file generated by plotpy <<<< ####

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.ticker as tck
import matplotlib.patches as pat
import matplotlib.path as pth
import matplotlib.patheffects as pff
import matplotlib.lines as lns
import matplotlib.transforms as tra
import mpl_toolkits.mplot3d
import matplotlib.tri as plt_tri
from mpl_toolkits.axes_grid1 import make_axes_locatable

# Variable to handle NaN values coming from Rust
NaN = np.nan

# List of additional objects to calculate bounding boxes
EXTRA_ARTISTS = []

# Adds an entity to the EXTRA_ARTISTS list to calculate bounding boxes
def add_to_ea(obj):
    global EXTRA_ARTISTS
    if obj!=None: EXTRA_ARTISTS.append(obj)

# Is a dictionary of mplot3d objects (one for each subplot_3d)
THREE_D = dict()

# Is a tuple holding the key to the current THREE_D object (defines the subplot_3d)
THREE_D_ACTIVE = (1,1,1)

# Creates or returns the mplot3d object with the current subplot_3d definition specified by THREE_D_ACTIVE
def ax3d():
    global THREE_D
    global THREE_D_ACTIVE
    if not THREE_D_ACTIVE in THREE_D:
        a, b, c = THREE_D_ACTIVE
        THREE_D[THREE_D_ACTIVE] = plt.gcf().add_subplot(a,b,c,projection='3d')
        THREE_D[THREE_D_ACTIVE].set_xlabel('x')
        THREE_D[THREE_D_ACTIVE].set_ylabel('y')
        THREE_D[THREE_D_ACTIVE].set_zlabel('z')
        add_to_ea(THREE_D[THREE_D_ACTIVE])
    return THREE_D[THREE_D_ACTIVE]

# Specifies the THREE_D_ACTIVE parameters to define a subplot_3d
def subplot_3d(a,b,c):
    global THREE_D_ACTIVE
    THREE_D_ACTIVE = (a,b,c)
    ax3d()

# Transforms data limits to axis limits
def data_to_axis(coords):
    plt.axis() # must call this first
    return plt.gca().transLimits.transform(coords)

# Transforms axis limits to data limits
def axis_to_data(coords):
    plt.axis() # must call this first
    return plt.gca().transLimits.inverted().transform(coords)

# Configures the aspect of axes with a same scaling from data to plot units for x, y and z.
def set_equal_axes():
    global THREE_D
    if len(THREE_D) == 0:
        ax = plt.gca()
        ax.axes.set_aspect('equal')
        return
    try:
        ax = ax3d()
        ax.set_box_aspect([1,1,1])
        limits = np.array([ax.get_xlim3d(), ax.get_ylim3d(), ax.get_zlim3d()])
        origin = np.mean(limits, axis=1)
        radius = 0.5 * np.max(np.abs(limits[:, 1] - limits[:, 0]))
        x, y, z = origin
        ax.set_xlim3d([x - radius, x + radius])
        ax.set_ylim3d([y - radius, y + radius])
        ax.set_zlim3d([z - radius, z + radius])
    except:
        import matplotlib
        print('VERSION of MATPLOTLIB = {}'.format(matplotlib.__version__))
        print('ERROR: set_box_aspect is missing in this version of Matplotlib')

# Function to ignore calls to plt such as the colorbar in an inset Axes
def ignore_this(*args, **kwargs):
    pass

################## plotting commands follow after this line ############################

xy=np.array([[0,0],[1,0],])
plt.plot(xy[:,0],xy[:,1],label=r'x axis for "test"',color='red',linewidth=2)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
xy=np.array([[0,0],[0,1],])
plt.plot(xy[:,0],xy[:,1],label=r'y axis for "test"',color='red',linewidth=1)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)
xy=np.array([[-1.25,-1.25],[1.25,-1.25],[1.25,1.25],[-1.25,1.25],[-1.25,-1.25],])
plt.plot(xy[:,0],xy[:,1],label=r'Frame of ""',color='yellow',linestyle='solid',linewidth=1,markerfacecolor='yellow',markevery=1,markersize=10,marker='.')
xy=np.array([[0,0],[1,0],])
plt.plot(xy[:,0],xy[:,1],label=r'x axis for "test"',color='red',linewidth=2)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
xy=np.array([[0,0],[0,1],])
plt.plot(xy[:,0],xy[:,1],label=r'y axis for "test"',color='red',linewidth=1)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)
xy=np.array([[-1.5,-1.5],[1.5,-1.5],[1.5,1.5],[-1.5,1.5],[-1.5,-1.5],])
plt.plot(xy[:,0],xy[:,1],label=r'Frame of ""',color='red',linestyle='solid',linewidth=1,markerfacecolor='red',markevery=1,markersize=10,marker='.')
xy=np.array([[0,0],[1,0],])
plt.plot(xy[:,0],xy[:,1],label=r'x axis for "test"',color='red',linewidth=2)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
xy=np.array([[0,0],[0,1],])
plt.plot(xy[:,0],xy[:,1],label=r'y axis for "test"',color='red',linewidth=1)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)
xy=np.array([[-1,0],[0,0],[0,-1],[-1,-1],[-1,0],])
plt.plot(xy[:,0],xy[:,1],label=r'Frame',linestyle='solid',linewidth=1,markerfacecolor='purple',markevery=1,markersize=10,marker='.')
t=plt.text(-1,-1,r'0',color='purple',fontsize=5)
t=plt.text(0,-1,r'1',color='purple',fontsize=5)
t=plt.text(1,-1,r'2',color='purple',fontsize=5)
t=plt.text(-1,0,r'3',color='purple',fontsize=5)
t=plt.text(0,0,r'4',color='purple',fontsize=5)
t=plt.text(1,0,r'5',color='purple',fontsize=5)
t=plt.text(-1,1,r'6',color='purple',fontsize=5)
t=plt.text(0,1,r'7',color='purple',fontsize=5)
t=plt.text(1,1,r'8',color='purple',fontsize=5)
xy=np.array([[-1,-1],[0,-1],[1,-1],[-1,0],[0,0],[1,0],[-1,1],[0,1],[1,1],])
plt.plot(xy[:,0],xy[:,1],label=r'Grid points: ""',linestyle='none',markerfacecolor='blue',markevery=1,markersize=7,marker='.')
xy=np.array([[-0.9044611903823121,-0.6890808322504309],])
plt.plot(xy[:,0],xy[:,1],label=r'x, y residuals: 0.096, 0.311',linestyle='none',markerfacecolor='#eeea83',markevery=1,markersize=10,marker='*')
xy=np.array([[-1,-1],])
plt.plot(xy[:,0],xy[:,1],label=r'Corner: (-1.000,-1.000)',linestyle='none',markerfacecolor='#eeea83',markevery=1,markersize=10,marker='.')
plt.gcf().set_size_inches(10,10)
plt.gca().get_yaxis().set_major_locator(tck.MaxNLocator(4))
plt.gca().get_xaxis().set_major_locator(tck.MaxNLocator(4))
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)
xy=np.array([[0,0],[1,0],])
plt.plot(xy[:,0],xy[:,1],label=r'x axis for "test"',color='red',linewidth=2)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
xy=np.array([[0,0],[0,1],])
plt.plot(xy[:,0],xy[:,1],label=r'y axis for "test"',color='red',linewidth=1)
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)
xy=np.array([[0,0.5],[0.5,0.5],[0.5,0],[0,0],[0,0.5],])
plt.plot(xy[:,0],xy[:,1],label=r'Frame',linestyle='solid',linewidth=1,markerfacecolor='purple',markevery=1,markersize=10,marker='.')
t=plt.text(-1,-1,r'0',color='purple',fontsize=5)
t=plt.text(-0.5,-1,r'1',color='purple',fontsize=5)
t=plt.text(0,-1,r'2',color='purple',fontsize=5)
t=plt.text(0.5,-1,r'3',color='purple',fontsize=5)
t=plt.text(1,-1,r'4',color='purple',fontsize=5)
t=plt.text(-1,-0.5,r'5',color='purple',fontsize=5)
t=plt.text(-0.5,-0.5,r'6',color='purple',fontsize=5)
t=plt.text(0,-0.5,r'7',color='purple',fontsize=5)
t=plt.text(0.5,-0.5,r'8',color='purple',fontsize=5)
t=plt.text(1,-0.5,r'9',color='purple',fontsize=5)
t=plt.text(-1,0,r'10',color='purple',fontsize=5)
t=plt.text(-0.5,0,r'11',color='purple',fontsize=5)
t=plt.text(0,0,r'12',color='purple',fontsize=5)
t=plt.text(0.5,0,r'13',color='purple',fontsize=5)
t=plt.text(1,0,r'14',color='purple',fontsize=5)
t=plt.text(-1,0.5,r'15',color='purple',fontsize=5)
t=plt.text(-0.5,0.5,r'16',color='purple',fontsize=5)
t=plt.text(0,0.5,r'17',color='purple',fontsize=5)
t=plt.text(0.5,0.5,r'18',color='purple',fontsize=5)
t=plt.text(1,0.5,r'19',color='purple',fontsize=5)
t=plt.text(-1,1,r'20',color='purple',fontsize=5)
t=plt.text(-0.5,1,r'21',color='purple',fontsize=5)
t=plt.text(0,1,r'22',color='purple',fontsize=5)
t=plt.text(0.5,1,r'23',color='purple',fontsize=5)
t=plt.text(1,1,r'24',color='purple',fontsize=5)
xy=np.array([[-1,-1],[-0.5,-1],[0,-1],[0.5,-1],[1,-1],[-1,-0.5],[-0.5,-0.5],[0,-0.5],[0.5,-0.5],[1,-0.5],[-1,0],[-0.5,0],[0,0],[0.5,0],[1,0],[-1,0.5],[-0.5,0.5],[0,0.5],[0.5,0.5],[1,0.5],[-1,1],[-0.5,1],[0,1],[0.5,1],[1,1],])
plt.plot(xy[:,0],xy[:,1],label=r'Grid points: ""',linestyle='none',markerfacecolor='blue',markevery=1,markersize=7,marker='.')
xy=np.array([[0.23491095388729488,0.432296952277462],])
plt.plot(xy[:,0],xy[:,1],label=r'x, y residuals: 0.235, -0.068',linestyle='none',markerfacecolor='#eeea83',markevery=1,markersize=10,marker='*')
xy=np.array([[-1,-1],])
plt.plot(xy[:,0],xy[:,1],label=r'Corner: (-1.000,-1.000)',linestyle='none',markerfacecolor='#eeea83',markevery=1,markersize=10,marker='.')
plt.gcf().set_size_inches(10,10)
plt.gca().get_yaxis().set_major_locator(tck.MaxNLocator(6))
plt.gca().get_xaxis().set_major_locator(tck.MaxNLocator(6))
plt.gca().set_axisbelow(True)
plt.grid(linestyle='--',color='grey',zorder=-1000)
plt.gca().set_xlabel(r'x')
plt.gca().set_ylabel(r'y')
h,l=plt.gca().get_legend_handles_labels()
if len(h)>0 and len(l)>0:
    leg=plt.legend(handlelength=3,ncol=1,loc='best')
    add_to_ea(leg)

fn=r'test'
plt.savefig(fn,bbox_inches='tight',bbox_extra_artists=EXTRA_ARTISTS)

plt.show()