source: sasview/testbatch/guiframetest/local_perspectives/plotting/Edge.py @ 83a0940

import math
#import wx  
#from copy import deepcopy
from BaseInteractor import _BaseInteractor
#from sans.guiframe.events import NewPlotEvent
#from sans.guiframe.events import StatusEvent
#from sans.guiframe.events import SlicerParameterEvent
#from sans.guiframe.events import EVT_SLICER_PARS
   

class RadiusInteractor(_BaseInteractor):
    """
    Select an annulus through a 2D plot
    """
    def __init__(self, base, axes, color='black', zorder=5, arc1=None,
                 arc2=None, theta=math.pi/8):
        """
        """
        _BaseInteractor.__init__(self, base, axes, color=color)
        self.markers = []
        self.axes = axes
        self.r1 = arc1.get_radius()
        self.r2 = arc2.get_radius()
        #print "radius init", self.r1, self.r2
        self.theta = theta
        self.save_theta = theta
        #self.scale = 10.0
        self.move_stop = False
        self.theta_left = None
        self.theta_right = None
        self.arc1 = arc1
        self.arc2 = arc2
        #raise "Version error", message
        x1 = self.r1 * math.cos(self.theta)
        y1 = self.r1 * math.sin(self.theta)
        x2 = self.r2 * math.cos(self.theta)
        y2 = self.r2 * math.sin(self.theta)
        self.line = self.axes.plot([x1, x2], [y1, y2],
                                      linestyle='-', marker='',
                                      color=self.color,
                                      visible=True)[0]
        self.phi = theta
        self.npts = 20
        self.has_move = False
        self.connect_markers([self.line])
        self.update()
        
    def set_layer(self, n):
        """
        """
        self.layernum = n
        self.update()
        
    def clear(self):
        """
        """
        self.clear_markers()
        try:
            self.line.remove()
        except:
            # Old version of matplotlib
            for item in range(len(self.axes.lines)):
                del self.axes.lines[0]
    
    def get_angle(self):
        """
        """
        return self.theta
        
    def update(self, r1=None, r2=None, theta=None):
        """
        Draw the new roughness on the graph.
        """
        if r1 != None:
            self.r1 = r1
        if r2 != None:
            self.r2 = r2
        if theta != None:
            self.theta = theta
        #print "in the edge r1, r2",self.r1,self.r2,math.degrees(self.theta)
        x1 = self.r1 * math.cos(self.theta)
        y1 = self.r1 * math.sin(self.theta)
        x2 = self.r2 * math.cos(self.theta)
        y2 = self.r2 * math.sin(self.theta)  
        self.line.set(xdata=[x1, x2], ydata=[y1, y2])
        
    def save(self, ev):
        """
        Remember the roughness for this layer and the next so that we
        can restore on Esc.
        """
        self.save_theta = math.atan2(ev.y,ev.x)
        #self.save_theta= self.theta
        self.base.freeze_axes()
   
    def moveend(self, ev):
        """
        """
        self.has_move = False
        self.base.moveend(ev)
            
    def restore(self, ev):
        """
        Restore the roughness for this layer.
        """
        self.theta = self.save_theta

    def move(self, x, y, ev):
        """
        Process move to a new position, making sure that the move is allowed.
        """
        self.theta = math.atan2(y, x)
        self.has_move = True
        self.base.base.update()
        
    def set_cursor(self, r_min, r_max, theta):
        """
        """
        self.theta = theta
        self.r1 = r_min
        self.r2 = r_max
        self.update()
        
    def get_params(self):
        """
        """
        params = {}
        params["radius1"] = self.r1
        params["radius2"] = self.r2
        params["theta"] = self.theta
        return params
    
    def set_params(self, params):
        """
        """
        #print "when here set curcor arc"
        x1 = params["radius1"] 
        x2 = params["radius2"] 
        theta= params["theta"]
        self.set_cursor(x1, x2, theta)