def onpress(self, event):
if event.inaxes != self.ax: return
if event.button!=1: return
# click location in screen coords
x, y = nx.array((event.x, event.y))
tx, ty = event.inaxes.transData.numerix_x_y(self.xs, self.ys)
d = nx.sqrt((x-tx)**2 + (y-ty)**2)
ind = nx.nonzero(d<5)
for i in ind:
self.pnts[i].plotraw()
def onpress(self, event):
if event.inaxes != self.ax: return
if event.button != 1: return
# click location in screen coords
x, y = nx.array((event.x, event.y))
tx, ty = event.inaxes.transData.numerix_x_y(self.xs, self.ys)
d = nx.sqrt((x - tx)**2 + (y - ty)**2)
ind = nx.nonzero(d < 5)
for i in ind:
self.pnts[i].plotraw()
def fill_below_intersection(x, S, Z):
"""
fill the region below the intersection of S and Z
"""
#find the intersection point
ind = nx.nonzero(nx.absolute(S - Z) == min(nx.absolute(S - Z)))[0]
# compute a new curve which we will fill below
Y = nx.zeros(S.shape, typecode=nx.Float)
Y[:ind] = S[:ind] # Y is S up to the intersection
Y[ind:] = Z[ind:] # and Z beyond it
p.fill(x, Y, facecolor='blue', alpha=0.5)
def fill_below_intersection(x, S, Z):
"""
fill the region below the intersection of S and Z
"""
#find the intersection point
ind = nx.nonzero( nx.absolute(S-Z)==min(nx.absolute(S-Z)))[0]
# compute a new curve which we will fill below
Y = nx.zeros(S.shape, typecode=nx.Float)
Y[:ind] = S[:ind] # Y is S up to the intersection
Y[ind:] = Z[ind:] # and Z beyond it
p.fill(x, Y, facecolor='blue', alpha=0.5)
def __draw_lines_hide(self, gc, x, y, transform=None):
"""
x and y are equal length arrays, draw lines connecting each
point in x, y
"""
if debugPS: self._pswriter.write('% draw_lines \n')
if transform:
if transform.need_nonlinear():
x, y, mask = transform.nonlinear_only_numerix(x,
y,
returnMask=1)
else:
mask = ones(x.shape)
vec6 = transform.as_vec6_val()
a, b, c, d, tx, ty = vec6
sx, sy = get_vec6_scales(vec6)
start = 0
end = 1000
points = zip(x, y)
write = self._pswriter.write
write('gsave\n')
self.push_gc(gc)
write('[%f %f %f %f %f %f] concat\n' % (a, b, c, d, tx, ty))
while start < len(x):
# put moveto on all the bad data and on the first good
# point after the bad data
codes = where(mask[start:end + 1], 'l', 'm')
ind = nonzero(mask[start:end + 1] == 0) + 1
if ind[-1] >= len(codes):
ind = ind[:-1]
put(codes, ind, 'm')
thisx = x[start:end + 1]
thisy = y[start:end + 1]
to_draw = izip(thisx, thisy, codes)
if not to_draw:
break
ps = ['%1.3f %1.3f m' % to_draw.next()[:2]]
ps.extend(["%1.3f %1.3f %c" % tup for tup in to_draw])
# we don't want to scale the line width, etc so invert the
# scale for the stroke
ps.append('\ngsave %f %f scale stroke grestore\n' %
(1. / sx, 1. / sy))
write('\n'.join(ps))
start = end
end += 1000
write("grestore\n")
def OnWhiz(self, evt):
self.x += numerix.pi / 15
self.y += numerix.pi / 20
z = numerix.sin(self.x) + numerix.cos(self.y)
self.im.set_array(z)
zmax = numerix.max(numerix.max(z)) - ERR_TOL
ymax_i, xmax_i = numerix.nonzero(numerix.greater_equal(z, zmax))
if self.im.origin == 'upper':
ymax_i = z.shape[0] - ymax_i
self.lines[0].set_data(xmax_i, ymax_i)
self.canvas.draw()
def __draw_lines_hide(self, gc, x, y, transform=None):
"""
x and y are equal length arrays, draw lines connecting each
point in x, y
"""
if debugPS: self._pswriter.write('% draw_lines \n')
if transform:
if transform.need_nonlinear():
x, y, mask = transform.nonlinear_only_numerix(x, y, returnMask=1)
else:
mask = ones(x.shape)
vec6 = transform.as_vec6_val()
a,b,c,d,tx,ty = vec6
sx, sy = get_vec6_scales(vec6)
start = 0
end = 1000
points = zip(x,y)
write = self._pswriter.write
write('gsave\n')
self.push_gc(gc)
write('[%f %f %f %f %f %f] concat\n'%(a,b,c,d,tx,ty))
while start < len(x):
# put moveto on all the bad data and on the first good
# point after the bad data
codes = where(mask[start:end+1], 'l', 'm')
ind = nonzero(mask[start:end+1]==0)+1
if ind[-1]>=len(codes):
ind = ind[:-1]
put(codes, ind, 'm')
thisx = x[start:end+1]
thisy = y[start:end+1]
to_draw = izip(thisx, thisy, codes)
if not to_draw:
break
ps = ['%1.3f %1.3f m' % to_draw.next()[:2]]
ps.extend(["%1.3f %1.3f %c" % tup for tup in to_draw])
# we don't want to scale the line width, etc so invert the
# scale for the stroke
ps.append('\ngsave %f %f scale stroke grestore\n'%(1./sx,1./sy))
write('\n'.join(ps))
start = end
end += 1000
write("grestore\n")
def OnWhiz(self,evt):
self.x += numerix.pi/15
self.y += numerix.pi/20
z = numerix.sin(self.x) + numerix.cos(self.y)
self.im.set_array(z)
zmax = mlab.max(mlab.max(z))-ERR_TOL
ymax_i, xmax_i = numerix.nonzero(
numerix.greater_equal(z, zmax))
if self.im.origin == 'upper':
ymax_i = z.shape[0]-ymax_i
self.lines[0].set_data(xmax_i,ymax_i)
self.canvas.draw()
def get_ind_under_point(self, event):
'get the index of the vertex under point if within epsilon tolerance'
x, y = zip(*self.poly.verts)
# display coords
xt, yt = self.poly.get_transform().numerix_x_y(x, y)
d = sqrt((xt-event.x)**2 + (yt-event.y)**2)
indseq = nonzero(equal(d, amin(d)))
ind = indseq[0]
if d[ind]>=self.epsilon:
ind = None
return ind
def get_ind_under_point(self, event):
'get the index of the vertex under point if within epsilon tolerance'
x, y = zip(*self.poly.verts)
# display coords
xt, yt = self.poly.get_transform().numerix_x_y(x, y)
d = sqrt((xt - event.x)**2 + (yt - event.y)**2)
indseq = nonzero(equal(d, amin(d)))
ind = indseq[0]
if d[ind] >= self.epsilon:
ind = None
return ind
def init_plot_data(self):
a = self.fig.add_subplot(111)
x = numerix.arange(120.0)*2*numerix.pi/60.0
y = numerix.arange(100.0)*2*numerix.pi/50.0
self.x, self.y = meshgrid(x, y)
z = numerix.sin(self.x) + numerix.cos(self.y)
self.im = a.imshow( z, cmap=cm.jet)#, interpolation='nearest')
zmax = numerix.max(numerix.max(z))-ERR_TOL
ymax_i, xmax_i = numerix.nonzero(
numerix.greater_equal(z, zmax))
if self.im.origin == 'upper':
ymax_i = z.shape[0]-ymax_i
self.lines = a.plot(xmax_i,ymax_i,'ko')
self.toolbar.update() # Not sure why this is needed - ADS
