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
|
import os
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
import WrightTools as wt
from WrightTools import datasets
here = os.path.abspath(os.path.dirname(__file__))
cmap = wt.artists.colormaps['default']
fig, gs = wt.artists.create_figure(width='double', nrows=2, cols=[1, 1, 1, 1, 'cbar'])
# get data
p = datasets.COLORS.v0p2_d1_d2_diagonal
data = wt.data.from_COLORS(p, invert_d1=False)
data.level(0, 0, 3)
data.ai0.symmetric_root(0.5)
data.ai0.normalize()
def dot_pixel_centers(ax, xi, yi):
for x in xi:
ax.scatter([x] * len(xi), yi, edgecolor=None, s=5, color='k')
def decorate(ax):
ax.set_xlim(-150, 150)
ax.set_ylim(-150, 150)
# pcolor
ax = plt.subplot(gs[0, 0])
ax.pcolor(data, cmap=cmap)
ax.set_title('pcolor', fontsize=20)
decorate(ax)
ax = plt.subplot(gs[1, 0])
ax.pcolor(data, cmap=cmap, edgecolors='k')
dot_pixel_centers(ax, data.d1.points, data.d2.points)
decorate(ax)
# tripcolor
xi = data.d1.points
yi = data.d2.points
zi = data.channels[0][:].T
ax = plt.subplot(gs[0, 1])
points = [xi, yi]
x, y = tuple(np.meshgrid(*points, indexing='ij'))
ax.tripcolor(x.flatten(), y.flatten(), zi.T.flatten(), cmap=cmap, vmin=0, vmax=1)
decorate(ax)
ax.set_title('tripcolor', fontsize=20)
ax = plt.subplot(gs[1, 1])
ax.tripcolor(x.flatten(), y.flatten(), zi.T.flatten(), edgecolor='k', cmap=cmap, vmin=0, vmax=1)
decorate(ax)
dot_pixel_centers(ax, xi, yi)
def plot_delaunay_edges(ax, xi, yi, zi):
x, y = tuple(np.meshgrid(xi, yi, indexing='ij'))
tri = matplotlib.tri.Triangulation(x.flatten(), y.flatten())
for i, j in tri.edges:
plt.plot([x.flatten()[i], x.flatten()[j]],
[y.flatten()[i], y.flatten()[j]],
c='k', lw=0.25)
ax.set_xlim(-125, 125)
ax.set_ylim(-125, 125)
# contourf
ax = plt.subplot(gs[0, 2])
levels = np.linspace(0, 1, 200)
ax.contourf(data)
decorate(ax)
ax.set_title('contourf', fontsize=20)
ax = plt.subplot(gs[1, 2])
ax.contourf(data)
plot_delaunay_edges(ax, xi, yi, zi)
dot_pixel_centers(ax, xi, yi)
# contour
ax = plt.subplot(gs[0, 3])
levels = np.linspace(0, 1, 11)
ax.contour(data)
decorate(ax)
ax.set_title('contour', fontsize=20)
ax = plt.subplot(gs[1, 3])
ax.contour(data)
decorate(ax)
plot_delaunay_edges(ax, xi, yi, zi)
dot_pixel_centers(ax, xi, yi)
# label
ticks = [-100, 0, 100]
wt.artists.set_fig_labels(xlabel=data.d1.label, ylabel=data.d2.label, xticks=ticks, yticks=ticks)
# colorbar
cax = plt.subplot(gs[:, -1])
wt.artists.plot_colorbar(cax=cax, label='amplitude')
# save
p = os.path.join(here, 'fill_types.png')
wt.artists.savefig(p)
|
