source: sasview/src/sas/plottools/arrow3d.py @ e46cbc5

"""
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
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)
            
    def onLeftDown(self, event):
        """
        """
        self.leftdown = True
        self.t_click = time.time()
        
    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()
    
    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