Source code for sas.guiframe.local_perspectives.plotting.AnnulusSlicer

#TODO: the line slicer should listen to all 2DREFRESH events, get the data and slice it
#      before pushing a new 1D data update.

#
#TODO: NEED MAJOR REFACTOR
#

import math
import wx
#from copy import deepcopy 
# Debug printout
from sas.guiframe.events import NewPlotEvent
from sas.guiframe.events import StatusEvent
from sas.guiframe.events import SlicerParameterEvent
from sas.guiframe.events import EVT_SLICER_PARS
from BaseInteractor import _BaseInteractor
from sas.guiframe.dataFitting import Data1D

[docs]class AnnulusInteractor(_BaseInteractor): """ Select an annulus through a 2D plot. This interactor is used to average 2D data with the region defined by 2 radius. this class is defined by 2 Ringinterators. """ def __init__(self, base, axes, color='black', zorder=3): _BaseInteractor.__init__(self, base, axes, color=color) self.markers = [] self.axes = axes self.base = base self.qmax = min(math.fabs(self.base.data2D.xmax), math.fabs(self.base.data2D.xmin)) #must be positive self.connect = self.base.connect ## Number of points on the plot self.nbins = 20 #Cursor position of Rings (Left(-1) or Right(1)) self.xmaxd = self.base.data2D.xmax self.xmind = self.base.data2D.xmin if (self.xmaxd + self.xmind) > 0: self.sign = 1 else: self.sign = -1 # Inner circle self.inner_circle = RingInteractor(self, self.base.subplot, zorder=zorder, r=self.qmax/2.0, sign=self.sign) self.inner_circle.qmax = self.qmax self.outer_circle = RingInteractor(self, self.base.subplot, zorder=zorder+1, r=self.qmax/1.8, sign=self.sign) self.outer_circle.qmax = self.qmax * 1.2 self.update() self._post_data() # Bind to slice parameter events self.base.Bind(EVT_SLICER_PARS, self._onEVT_SLICER_PARS) def _onEVT_SLICER_PARS(self, event): """ receive an event containing parameters values to reset the slicer :param event: event of type SlicerParameterEvent with params as attribute """ wx.PostEvent(self.base, StatusEvent(status="AnnulusSlicer._onEVT_SLICER_PARS")) event.Skip() if event.type == self.__class__.__name__: self.set_params(event.params) self.base.update()
[docs] def set_layer(self, n): """ Allow adding plot to the same panel :param n: the number of layer """ self.layernum = n self.update()
[docs] def clear(self): """ Clear the slicer and all connected events related to this slicer """ self.clear_markers() self.outer_circle.clear() self.inner_circle.clear() self.base.connect.clearall() self.base.Unbind(EVT_SLICER_PARS)
[docs] def update(self): """ Respond to changes in the model by recalculating the profiles and resetting the widgets. """ # Update locations self.inner_circle.update() self.outer_circle.update()
[docs] def save(self, ev): """ Remember the roughness for this layer and the next so that we can restore on Esc. """ self.base.freeze_axes() self.inner_circle.save(ev) self.outer_circle.save(ev)
def _post_data(self, nbins=None): """ Uses annulus parameters to plot averaged data into 1D data. :param nbins: the number of points to plot """ #Data to average data = self.base.data2D # If we have no data, just return if data == None: return from sas.dataloader.manipulations import Ring rmin = min(math.fabs(self.inner_circle.get_radius()), math.fabs(self.outer_circle.get_radius())) rmax = max(math.fabs(self.inner_circle.get_radius()), math.fabs(self.outer_circle.get_radius())) #if the user does not specify the numbers of points to plot # the default number will be nbins= 20 if nbins == None: self.nbins = 20 else: self.nbins = nbins ## create the data1D Q average of data2D sect = Ring(r_min=rmin, r_max=rmax, nbins=self.nbins) sector = sect(self.base.data2D) if hasattr(sector, "dxl"): dxl = sector.dxl else: dxl = None if hasattr(sector, "dxw"): dxw = sector.dxw else: dxw = None new_plot = Data1D(x=(sector.x - math.pi) * 180/math.pi, y=sector.y, dy=sector.dy) new_plot.dxl = dxl new_plot.dxw = dxw new_plot.name = "AnnulusPhi" +"("+ self.base.data2D.name+")" new_plot.source = self.base.data2D.source #new_plot.info=self.base.data2D.info new_plot.interactive = True new_plot.detector = self.base.data2D.detector # If the data file does not tell us what the axes are, just assume... new_plot.xaxis("\\rm{\phi}", 'degrees') new_plot.yaxis("\\rm{Intensity} ", "cm^{-1}") if hasattr(data, "scale") and data.scale == 'linear' and \ self.base.data2D.name.count("Residuals") > 0: new_plot.ytransform = 'y' new_plot.yaxis("\\rm{Residuals} ", "/") new_plot.group_id = "AnnulusPhi" + self.base.data2D.name new_plot.id = "AnnulusPhi" + self.base.data2D.name new_plot.is_data= True new_plot.xtransform = "x" new_plot.ytransform = "y" self.base.parent.update_theory(data_id=data.id, theory=new_plot) wx.PostEvent(self.base.parent, NewPlotEvent(plot=new_plot, title="AnnulusPhi"))
[docs] def moveend(self, ev): """ Called when any dragging motion ends. Post an event (type =SlicerParameterEvent) to plotter 2D with a copy slicer parameters Call _post_data method """ self.base.thaw_axes() # Post parameters to plotter 2D event = SlicerParameterEvent() event.type = self.__class__.__name__ event.params = self.get_params() wx.PostEvent(self.base, event) # create a 1D data plot #self._post_data()
[docs] def restore(self): """ Restore the roughness for this layer. """ self.inner_circle.restore() self.outer_circle.restore()
[docs] def move(self, x, y, ev): """ Process move to a new position, making sure that the move is allowed. """ pass
[docs] def set_cursor(self, x, y): pass
[docs] def get_params(self): """ Store a copy of values of parameters of the slicer into a dictionary. :return params: the dictionary created """ params = {} params["inner_radius"] = math.fabs(self.inner_circle._inner_mouse_x) params["outer_radius"] = math.fabs(self.outer_circle._inner_mouse_x) params["nbins"] = self.nbins return params
[docs] def set_params(self, params): """ Receive a dictionary and reset the slicer with values contained in the values of the dictionary. :param params: a dictionary containing name of slicer parameters and values the user assigned to the slicer. """ inner = math.fabs(params["inner_radius"]) outer = math.fabs(params["outer_radius"]) self.nbins = int(params["nbins"]) ## Update the picture self.inner_circle.set_cursor(inner, self.inner_circle._inner_mouse_y) self.outer_circle.set_cursor(outer, self.outer_circle._inner_mouse_y) ## Post the data given the nbins entered by the user self._post_data(self.nbins)
[docs] def freeze_axes(self): """ """ self.base.freeze_axes()
[docs] def thaw_axes(self): """ """ self.base.thaw_axes()
[docs] def draw(self): """ """ self.base.draw()
[docs]class RingInteractor(_BaseInteractor): """ Draw a ring Given a radius """ def __init__(self, base, axes, color='black', zorder=5, r=1.0, sign=1): """ :param: the color of the line that defined the ring :param r: the radius of the ring :param sign: the direction of motion the the marker """ _BaseInteractor.__init__(self, base, axes, color=color) self.markers = [] self.axes = axes # Current radius of the ring self._inner_mouse_x = r #Value of the center of the ring self._inner_mouse_y = 0 # previous value of that radius self._inner_save_x = r #Save value of the center of the ring self._inner_save_y = 0 #Class instantiating RingIterator class self.base = base #the direction of the motion of the marker self.sign = sign ## Create a marker try: # Inner circle marker x_value = [self.sign * math.fabs(self._inner_mouse_x)] self.inner_marker = self.axes.plot(x_value, [0], linestyle='', marker='s', markersize=10, color=self.color, alpha=0.6, pickradius=5, label="pick", zorder=zorder, visible=True)[0] except: x_value = [self.sign * math.fabs(self._inner_mouse_x)] self.inner_marker = self.axes.plot(x_value, [0], linestyle='', marker='s', markersize=10, color=self.color, alpha=0.6, label="pick", visible=True)[0] message = "\nTHIS PROTOTYPE NEEDS THE LATEST" message += " VERSION OF MATPLOTLIB\n" message += "Get the SVN version that is at " message += " least as recent as June 1, 2007" owner = self.base.base.parent wx.PostEvent(owner, StatusEvent(status="AnnulusSlicer: %s" % message)) # Draw a circle [self.inner_circle] = self.axes.plot([], [], linestyle='-', marker='', color=self.color) # the number of points that make the ring line self.npts = 40 self.connect_markers([self.inner_marker]) self.update()
[docs] def set_layer(self, n): """ Allow adding plot to the same panel :param n: the number of layer """ self.layernum = n self.update()
[docs] def clear(self): """ Clear the slicer and all connected events related to this slicer """ self.clear_markers() try: self.inner_marker.remove() self.inner_circle.remove() except: # Old version of matplotlib for item in range(len(self.axes.lines)): del self.axes.lines[0]
[docs] def get_radius(self): """ :return self._inner_mouse_x: the current radius of the ring """ return self._inner_mouse_x
[docs] def update(self): """ Draw the new roughness on the graph. """ # Plot inner circle x = [] y = [] for i in range(self.npts): phi = 2.0 * math.pi / (self.npts - 1) * i xval = 1.0 * self._inner_mouse_x * math.cos(phi) yval = 1.0 * self._inner_mouse_x * math.sin(phi) x.append(xval) y.append(yval) self.inner_marker.set(xdata=[self.sign*math.fabs(self._inner_mouse_x)], ydata=[0]) self.inner_circle.set_data(x, y)
[docs] def save(self, ev): """ Remember the roughness for this layer and the next so that we can restore on Esc. """ self._inner_save_x = self._inner_mouse_x self._inner_save_y = self._inner_mouse_y self.base.freeze_axes()
[docs] def moveend(self, ev): """ Called after a dragging motion """ self.base.moveend(ev)
[docs] def restore(self): """ Restore the roughness for this layer. """ self._inner_mouse_x = self._inner_save_x self._inner_mouse_y = self._inner_save_y
[docs] def move(self, x, y, ev): """ Process move to a new position, making sure that the move is allowed. """ self._inner_mouse_x = x self._inner_mouse_y = y self.base.base.update()
[docs] def set_cursor(self, x, y): """ draw the ring given x, y value """ self.move(x, y, None) self.update()
[docs] def get_params(self): """ Store a copy of values of parameters of the slicer into a dictionary. :return params: the dictionary created """ params = {} params["radius"] = math.fabs(self._inner_mouse_x) return params
[docs] def set_params(self, params): """ Receive a dictionary and reset the slicer with values contained in the values of the dictionary. :param params: a dictionary containing name of slicer parameters and values the user assigned to the slicer. """ x = params["radius"] self.set_cursor(x, self._inner_mouse_y)
[docs]class CircularMask(_BaseInteractor): """ Draw a ring Given a radius """ def __init__(self, base, axes, color='grey', zorder=3, side=None): """ :param: the color of the line that defined the ring :param r: the radius of the ring :param sign: the direction of motion the the marker """ _BaseInteractor.__init__(self, base, axes, color=color) self.markers = [] self.axes = axes self.base = base self.is_inside = side self.qmax = min(math.fabs(self.base.data.xmax), math.fabs(self.base.data.xmin)) #must be positive self.connect = self.base.connect #Cursor position of Rings (Left(-1) or Right(1)) self.xmaxd = self.base.data.xmax self.xmind = self.base.data.xmin if (self.xmaxd + self.xmind) > 0: self.sign = 1 else: self.sign = -1 # Inner circle self.outer_circle = RingInteractor(self, self.base.subplot, 'blue', zorder=zorder+1, r=self.qmax/1.8, sign=self.sign) self.outer_circle.qmax = self.qmax * 1.2 self.update() self._post_data() # Bind to slice parameter events #self.base.Bind(EVT_SLICER_PARS, self._onEVT_SLICER_PARS) def _onEVT_SLICER_PARS(self, event): """ receive an event containing parameters values to reset the slicer :param event: event of type SlicerParameterEvent with params as attribute """ wx.PostEvent(self.base, StatusEvent(status="AnnulusSlicer._onEVT_SLICER_PARS")) event.Skip() if event.type == self.__class__.__name__: self.set_params(event.params) self.base.update()
[docs] def set_layer(self, n): """ Allow adding plot to the same panel :param n: the number of layer """ self.layernum = n self.update()
[docs] def clear(self): """ Clear the slicer and all connected events related to this slicer """ self.clear_markers() self.outer_circle.clear() self.base.connect.clearall() #self.base.Unbind(EVT_SLICER_PARS)
[docs] def update(self): """ Respond to changes in the model by recalculating the profiles and resetting the widgets. """ # Update locations self.outer_circle.update() #if self.is_inside != None: out = self._post_data() return out
[docs] def save(self, ev): """ Remember the roughness for this layer and the next so that we can restore on Esc. """ self.base.freeze_axes() self.outer_circle.save(ev)
def _post_data(self): """ Uses annulus parameters to plot averaged data into 1D data. :param nbins: the number of points to plot """ #Data to average data = self.base.data # If we have no data, just return if data == None: return mask = data.mask from sas.dataloader.manipulations import Ringcut rmin = 0 rmax = math.fabs(self.outer_circle.get_radius()) ## create the data1D Q average of data2D mask = Ringcut(r_min=rmin, r_max= rmax) if self.is_inside: out = (mask(data) == False) else: out = (mask(data)) #self.base.data.mask=out return out
[docs] def moveend(self, ev): """ Called when any dragging motion ends. Post an event (type =SlicerParameterEvent) to plotter 2D with a copy slicer parameters Call _post_data method """ self.base.thaw_axes() # create a 1D data plot self._post_data()
[docs] def restore(self): """ Restore the roughness for this layer. """ self.outer_circle.restore()
[docs] def move(self, x, y, ev): """ Process move to a new position, making sure that the move is allowed. """ pass
[docs] def set_cursor(self, x, y): pass
[docs] def get_params(self): """ Store a copy of values of parameters of the slicer into a dictionary. :return params: the dictionary created """ params = {} params["outer_radius"] = math.fabs(self.outer_circle._inner_mouse_x) return params
[docs] def set_params(self, params): """ Receive a dictionary and reset the slicer with values contained in the values of the dictionary. :param params: a dictionary containing name of slicer parameters and values the user assigned to the slicer. """ outer = math.fabs(params["outer_radius"] ) ## Update the picture self.outer_circle.set_cursor(outer, self.outer_circle._inner_mouse_y) ## Post the data given the nbins entered by the user self._post_data()
[docs] def freeze_axes(self): self.base.freeze_axes()
[docs] def thaw_axes(self): self.base.thaw_axes()
[docs] def draw(self): self.base.update()