"""
Module that draws multiple arrows in 3D coordinates
"""
from matplotlib.patches import FancyArrowPatch
from mpl_toolkits.mplot3d import proj3d
import time
#from matplotlib.artist import allow_rasterization
[docs]class Arrow3D(FancyArrowPatch):
"""
Draw 3D arrow
"""
def __init__(self, base, xs, ys, zs, colors, *args, **kwargs):
"""
Init
:Params xs: [[x0, x0+dx0], [x1, x1+dx1], ...]
:Params ys: [[y0, y0+dy0], [y1, y1+dy1], ...]
:Params zs: [[z0, z0+dz0], [z1, z1+dz1], ...]
:Params colors: [[R0, G0, B0], [R1, G1, B1], ...]
where R, G, B ranges (0,1)
"""
FancyArrowPatch.__init__(self, (0,0), (0,0), *args, **kwargs)
self.leftdown = False
self.t_click = 0
self._verts3d = xs, ys, zs
self.colors = colors
self.base = base
if base != None:
# To turn the updating off during dragging
base.canvas.mpl_connect('button_press_event', self.onLeftDown)
base.canvas.mpl_connect('button_release_event', self.onLeftUp)
[docs] def onLeftDown(self, event):
"""
"""
self.leftdown = True
self.t_click = time.time()
[docs] def onLeftUp(self, event):
"""
"""
t_up = time.time() - self.t_click
# Avoid just clicking
if t_up > 0.1:
self.leftdown = False
self.base.canvas.draw()
[docs] def draw(self, renderer, rasterized=True):
"""
Drawing actually happens here
"""
# Draws only when the dragging finished
if self.leftdown:
return
xs3d, ys3d, zs3d = self._verts3d
for i in xrange(len(xs3d)):
xs, ys, _ = proj3d.proj_transform(xs3d[i], ys3d[i], zs3d[i],
renderer.M)
self.set_positions((xs[0],ys[0]),(xs[1],ys[1]))
self.set_color(self.colors[i])
FancyArrowPatch.draw(self, renderer)
self.leftdown = False