source: sasview/src/sas/sasgui/guiframe/local_perspectives/plotting/Plotter1D.py @ 245ae18

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 245ae18 was 245ae18, checked in by lewis, 5 years ago

Don't log coords on corfunc QRange plot, only on drag

  • Property mode set to 100644
File size: 31.5 KB
Line 
1
2################################################################################
3# This software was developed by the University of Tennessee as part of the
4# Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
5# project funded by the US National Science Foundation.
6#
7# See the license text in license.txt
8#
9# copyright 2008, University of Tennessee
10################################################################################
11
12
13import wx
14import sys
15import math
16import numpy
17import logging
18from sas.sasgui.plottools.PlotPanel import PlotPanel
19from sas.sasgui.guiframe.events import StatusEvent
20from sas.sasgui.guiframe.events import PanelOnFocusEvent
21from sas.sasgui.guiframe.utils import PanelMenu, IdList
22from sas.sasgui.guiframe.panel_base import PanelBase
23from sas.sasgui.guiframe.gui_style import GUIFRAME_ICON
24from appearanceDialog import appearanceDialog
25from graphAppearance import graphAppearance
26
27DEFAULT_QMAX = 0.05
28DEFAULT_QSTEP = 0.001
29DEFAULT_BEAM = 0.005
30BIN_WIDTH = 1
31IS_MAC = (sys.platform == 'darwin')
32
33
34def find_key(dic, val):
35    """return the key of dictionary dic given the value"""
36    return [k for k, v in dic.iteritems() if v == val][0]
37
38class ModelPanel1D(PlotPanel, PanelBase):
39    """
40    Plot panel for use with the GUI manager
41    """
42
43    ## Internal name for the AUI manager
44    window_name = "plotpanel"
45    ## Title to appear on top of the window
46    window_caption = "Graph"
47    ## Flag to tell the GUI manager that this panel is not
48    #  tied to any perspective
49    ALWAYS_ON = True
50    ## Group ID
51    group_id = None
52    _menu_ids = IdList()
53
54    def __init__(self, parent, id=-1, color=None,
55                 dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
56        PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
57        PanelBase.__init__(self, parent)
58        ## Reference to the parent window
59        self.parent = parent
60        if hasattr(parent, "parent"):
61            self.parent = self.parent.parent
62        ## Plottables
63        self.plots = {}
64        self.frame = None
65        # context menu
66        self._slicerpop = None
67
68        self._available_data = []
69        self._symbol_labels = self.get_symbol_label()
70        self._color_labels = self.get_color_label()
71        self.currColorIndex = ""
72        self._is_changed_legend_label = False
73        self.is_xtick = False
74        self.is_ytick = False
75
76        self.hide_menu = None
77        ## Unique ID (from gui_manager)
78        self.uid = None
79        self.x_size = None
80        ## Default locations
81        # self._default_save_location = os.getcwd()
82        self.size = None
83        self.vl_ind = 0
84        ## Graph
85        # self.graph = Graph()
86        self.graph.xaxis("\\rm{Q}", 'A^{-1}')
87        self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
88        self.graph.render(self)
89        self.cursor_id = None
90
91        # In resizing event
92        self.resizing = False
93        self.canvas.set_resizing(self.resizing)
94        self.Bind(wx.EVT_SIZE, self._OnReSize)
95        self.parent.SetFocus()
96
97        # If true, there are 3 qrange bars
98        self.is_corfunc = False
99
100
101    def get_symbol_label(self):
102        """
103        Associates label to symbol
104        """
105        _labels = {}
106        i = 0
107        _labels['Circle'] = i
108        i += 1
109        _labels['Cross X '] = i
110        i += 1
111        _labels['Triangle Down'] = i
112        i += 1
113        _labels['Triangle Up'] = i
114        i += 1
115        _labels['Triangle Left'] = i
116        i += 1
117        _labels['Triangle Right'] = i
118        i += 1
119        _labels['Cross +'] = i
120        i += 1
121        _labels['Square'] = i
122        i += 1
123        _labels['diamond'] = i
124        i += 1
125        _labels['Diamond'] = i
126        i += 1
127        _labels['Hexagon1'] = i
128        i += 1
129        _labels['Hexagon2'] = i
130        i += 1
131        _labels['Pentagon'] = i
132        i += 1
133        _labels['Line'] = i
134        i += 1
135        _labels['Dash'] = i
136        i += 1
137        _labels['Vline'] = i
138        i += 1
139        _labels['Step'] = i
140        return _labels
141
142    def get_color_label(self):
143        """
144        Associates label to a specific color
145        """
146        _labels = {}
147        i = 0
148        _labels['Blue'] = i
149        i += 1
150        _labels['Green'] = i
151        i += 1
152        _labels['Red'] = i
153        i += 1
154        _labels['Cyan'] = i
155        i += 1
156        _labels['Magenta'] = i
157        i += 1
158        _labels['Yellow'] = i
159        i += 1
160        _labels['Black'] = i
161        return _labels
162
163
164    def set_data(self, list=None):
165        """
166        """
167        pass
168
169    def _reset(self):
170        """
171        Resets internal data and graph
172        """
173        self.graph.reset()
174        self.plots = {}
175        self.is_zoomed = False
176
177    def _OnReSize(self, event):
178        """
179        On response of the resize of a panel, set axes_visiable False
180        """
181        # It was found that wx >= 2.9.3 sends an event even if no size changed.
182        # So manually recode the size (=x_size) and compare here.
183        # Massy code to work around:<
184        if self.parent._mgr != None:
185            max_panel = self.parent._mgr.GetPane(self)
186            if max_panel.IsMaximized():
187                self.parent._mgr.RestorePane(max_panel)
188                max_panel.Maximize()
189        if self.x_size != None:
190            if self.x_size == self.GetSize():
191                self.resizing = False
192                self.canvas.set_resizing(self.resizing)
193                return
194        self.x_size = self.GetSize()
195
196        # Ready for another event
197        # Do not remove this Skip. Otherwise it will get runtime error on wx>=2.9.
198        event.Skip()
199        # set the resizing flag
200        self.resizing = True
201        self.canvas.set_resizing(self.resizing)
202        self.parent.set_schedule(True)
203        pos_x, pos_y = self.GetPositionTuple()
204        if pos_x != 0 and pos_y != 0:
205            self.size, _ = self.GetClientSizeTuple()
206        self.SetSizer(self.sizer)
207        wx.CallAfter(self.parent.disable_app_menu, self)
208
209    def on_plot_qrange(self, event=None):
210        """
211        On Qmin Qmax vertical line event
212        """
213        if event == None:
214            return
215        event.Skip()
216        active_ctrl = event.active
217        if active_ctrl == None:
218            return
219        if hasattr(event, 'is_corfunc'):
220            self.is_corfunc = event.is_corfunc
221        if event.id in self.plots.keys():
222            ctrl = event.ctrl
223            self.cursor_id = event.id
224            # Set line position and color
225            colors = ['red', 'purple']
226            x_data = self.plots[self.cursor_id].x
227            values = [max(x_data.min(), float(ctrl[0].GetValue())),
228                      min(x_data.max(), float(ctrl[1].GetValue()))]
229            if len(ctrl) == 3:
230                colors.append('purple')
231                values.append(min(x_data.max(), float(ctrl[2].GetValue())))
232            if self.ly == None:
233                self.ly = []
234                for c, v in zip(colors, values):
235                    h = self.subplot.axvline(x=v, color=c, lw=2.5, alpha=0.7)
236                    h.set_rasterized(True)
237                    self.ly.append(h)
238            try:
239                # Display x,y in the status bar if possible
240                xval = float(active_ctrl.GetValue())
241                position = self.get_data_xy_vals(xval)
242                if position != None and not self.is_corfunc:
243                    wx.PostEvent(self.parent, StatusEvent(status=position))
244            except:
245                logging.error(sys.exc_value)
246            if not event.leftdown:
247                # text event
248                try:
249                    is_moved = False
250                    for h, v in zip(self.ly, values):
251                        # check if vline moved
252                        if h.get_xdata() != v:
253                            h.set_xdata(v)
254                            is_moved = True
255                    if is_moved:
256                        self.canvas.draw()
257                except:
258                    logging.error(sys.exc_value)
259                event.Skip()
260                return
261            self.q_ctrl = ctrl
262            for h, c, v in zip(self.ly, colors, values):
263                h.set_color(c)
264                h.set_xdata(v)
265            self.canvas.draw()
266        else:
267            self.q_ctrl = None
268
269    def get_data_xy_vals(self, xval):
270        """
271        Get x, y data values near x = x_val
272        """
273        try:
274            x_data = self.plots[self.cursor_id].x
275            y_data = self.plots[self.cursor_id].y
276            indx = self._find_nearest(x_data, xval)
277            pos_x = x_data[indx]
278            pos_y = y_data[indx]
279            position = str(pos_x), str(pos_y)
280            return position
281        except:
282            return None
283
284    def _find_nearest(self, array, value):
285        """
286        Find and return the nearest value in array to the value.
287        Used in cusor_line()
288        :Param array: numpy array
289        :Param value: float
290        """
291        idx = (numpy.abs(array - value)).argmin()
292        return int(idx)  # array.flat[idx]
293
294    def _check_line_positions(self, pos_x=None, nop=None):
295        """
296        Check vertical line positions
297        :Param pos_x: position of the current line [float]
298        :Param nop: number of plots [int]
299        """
300        ly = self.ly
301        ly0x = ly[0].get_xdata()
302        ly1x = ly[1].get_xdata()
303        ly2x = None
304        if self.is_corfunc: ly2x = ly[2].get_xdata()
305        self.q_ctrl[0].SetBackgroundColour('white')
306        self.q_ctrl[1].SetBackgroundColour('white')
307        if ly0x >= ly1x:
308            if self.vl_ind == 0:
309                ly[1].set_xdata(pos_x)
310                ly[1].set_zorder(nop)
311                self.q_ctrl[1].SetValue(str(pos_x))
312                self.q_ctrl[0].SetBackgroundColour('pink')
313            elif self.vl_ind == 1:
314                ly[0].set_xdata(pos_x)
315                ly[0].set_zorder(nop)
316                self.q_ctrl[0].SetValue(str(pos_x))
317                self.q_ctrl[1].SetBackgroundColour('pink')
318        elif ly2x is not None and ly1x >= ly2x:
319            if self.vl_ind == 1:
320                ly[2].set_xdata(posx)
321                ly[2].set_zorder(nop)
322                self.q_ctrl[2].SetValue(str(pos_x))
323            elif self.vl_ind == 2:
324                ly[1].set_xdata(posx)
325                ly[1].set_zorder(nop)
326                self.q_ctrl[1].SetValue(str(pos_x))
327
328
329    def _get_cusor_lines(self, event):
330        """
331        Revmove or switch cursor line if drawn
332        :Param event: LeftClick mouse event
333        """
334        ax = event.inaxes
335        if hasattr(event, "action"):
336            dclick = event.action == 'dclick'
337            if ax == None or dclick:
338                # remove the vline
339                self._check_zoom_plot()
340                self.canvas.draw()
341                self.q_ctrl = None
342                return
343        if self.ly != None and event.xdata != None:
344            # Selecting a new line if cursor lines are displayed already
345            dqmin = math.fabs(event.xdata - self.ly[0].get_xdata())
346            dqmax = math.fabs(event.xdata - self.ly[1].get_xdata())
347            if not self.is_corfunc:
348                is_qmax = dqmin > dqmax
349                if is_qmax:
350                    self.vl_ind = 1
351                else:
352                    self.vl_ind = 0
353            else:
354                dqmax2 = math.fabs(event.xdata - self.ly[2].get_xdata())
355                closest = min(dqmin, dqmax, dqmax2)
356                self.vl_ind = { dqmin: 0, dqmax: 1, dqmax2: 2 }.get(closest)
357
358    def cusor_line(self, event):
359        """
360        Move the cursor line to write Q range
361        """
362        if self.q_ctrl == None:
363            return
364        # release a q range vline
365        if self.ly != None and not self.leftdown:
366            for ly in self.ly:
367                ly.set_alpha(0.7)
368                self.canvas.draw()
369            return
370        ax = event.inaxes
371        if ax == None or not hasattr(event, 'action'):
372            return
373        end_drag = event.action != 'drag' and event.xdata != None
374        nop = len(self.plots)
375        pos_x, _ = float(event.xdata), float(event.ydata)
376        try:
377            ly = self.ly
378            ly0x = ly[0].get_xdata()
379            ly1x = ly[1].get_xdata()
380            if ly0x == ly1x:
381                if ly[0].get_zorder() > ly[1].get_zorder():
382                    self.vl_ind = 0
383                else:
384                    self.vl_ind = 1
385            vl_ind = self.vl_ind
386            x_data = self.plots[self.cursor_id].x
387            xmin = x_data.min()
388            xmax = x_data.max()
389            indx = self._find_nearest(x_data, pos_x)
390            # Need to hold LeftButton to drag
391            if end_drag:
392                if event.button:
393                    self._check_line_positions(pos_x, nop)
394                return
395            if indx >= len(x_data):
396                indx = len(x_data) - 1
397            pos_x = x_data[indx]
398            if xmin == ly1x:
399                vl_ind = 1
400            elif xmax == ly0x:
401                vl_ind = 0
402            else:
403                ly[vl_ind].set_xdata(pos_x)
404                ly[vl_ind].set_zorder(nop + 1)
405                self._check_line_positions(pos_x, nop)
406            ly[vl_ind].set_xdata(pos_x)
407            ly[vl_ind].set_alpha(1.0)
408            ly[vl_ind].set_zorder(nop + 1)
409            self.canvas.draw()
410            self.q_ctrl[vl_ind].SetValue(str(pos_x))
411        except:
412            logging.error(sys.exc_value)
413
414    def set_resizing(self, resizing=False):
415        """
416        Set the resizing (True/False)
417        """
418        self.resizing = resizing
419        # self.canvas.set_resizing(resizing)
420
421    def schedule_full_draw(self, func='append'):
422        """
423        Put self in schedule to full redraw list
424        """
425        # append/del this panel in the schedule list
426        self.parent.set_schedule_full_draw(self, func)
427
428    def remove_data_by_id(self, id):
429        """
430            Remove data from plot
431        """
432        if id in self.plots.keys():
433            data = self.plots[id]
434            self.graph.delete(data)
435            data_manager = self._manager.parent.get_data_manager()
436            data_list, theory_list = data_manager.get_by_id(id_list=[id])
437
438            if id in data_list.keys():
439                data = data_list[id]
440            if id in theory_list.keys():
441                data = theory_list[id]
442
443            del self.plots[id]
444            self.graph.render(self)
445            self.subplot.figure.canvas.draw_idle()
446            if len(self.graph.plottables) == 0:
447                # onRemove: graph is empty must be the panel must be destroyed
448                self.parent.delete_panel(self.uid)
449
450    def plot_data(self, data):
451        """
452        Data is ready to be displayed
453
454        :param event: data event
455        """
456        if data.__class__.__name__ == 'Data2D':
457            return
458        plot_keys = self.plots.keys()
459        if data.id in plot_keys:
460            # Recover panel prop.s
461            xlo, xhi = self.subplot.get_xlim()
462            ylo, yhi = self.subplot.get_ylim()
463            old_data = self.plots[data.id]
464            if self._is_changed_legend_label:
465                data.label = old_data.label
466            if old_data.__class__.__name__ == 'Data1D':
467                data.custom_color = old_data.custom_color
468                data.symbol = old_data.symbol
469                data.markersize = old_data.markersize
470                data.zorder = len(plot_keys)
471            # Replace data
472            self.graph.replace(data)
473            self.plots[data.id] = data
474            ## Set the view scale for all plots
475            try:
476                self._onEVT_FUNC_PROPERTY()
477            except Exception, exc:
478                wx.PostEvent(self.parent,
479                             StatusEvent(status="Plotting Error: %s" % str(exc), info="error"))
480            if self.is_zoomed:
481                # Recover the x,y limits
482                self.subplot.set_xlim((xlo, xhi))
483                self.subplot.set_ylim((ylo, yhi))
484        else:
485            self.plots[data.id] = data
486            self.graph.add(self.plots[data.id])
487            data.zorder = len(plot_keys)
488            ## Set the view scale for all plots
489            try:
490                self._onEVT_FUNC_PROPERTY()
491                if IS_MAC:
492                    # MAC: forcing to plot 2D avg
493                    self.canvas._onDrawIdle()
494            except Exception, exc:
495                wx.PostEvent(self.parent, StatusEvent(status=\
496                    "Plotting Error: %s" % str(exc), info="error"))
497            self.toolbar.update()
498            self.is_zoomed = False
499
500    def draw_plot(self):
501        """
502        Draw plot
503        """
504        self.draw()
505
506    def onLeftDown(self, event):
507        """
508        left button down and ready to drag
509        Display the position of the mouse on the statusbar
510        """
511        # self.parent.set_plot_unfocus()
512        self._get_cusor_lines(event)
513        ax = event.inaxes
514        PlotPanel.onLeftDown(self, event)
515        if ax != None:
516            try:
517                pos_x = float(event.xdata)  # / size_x
518                pos_y = float(event.ydata)  # / size_y
519                pos_x = "%8.3g" % pos_x
520                pos_y = "%8.3g" % pos_y
521                self.position = str(pos_x), str(pos_y)
522                wx.PostEvent(self.parent, StatusEvent(status=self.position))
523            except:
524                self.position = None
525        # unfocus all
526        self.parent.set_plot_unfocus()
527        # post nd event to notify guiframe that this panel is on focus
528        wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
529
530
531    def _ontoggle_hide_error(self, event):
532        """
533        Toggle error display to hide or show
534        """
535        menu = event.GetEventObject()
536        event_id = event.GetId()
537        self.set_selected_from_menu(menu, event_id)
538        # Check zoom
539        xlo, xhi = self.subplot.get_xlim()
540        ylo, yhi = self.subplot.get_ylim()
541
542        selected_plot = self.plots[self.graph.selected_plottable]
543        if self.hide_menu.GetText() == "Hide Error Bar":
544            selected_plot.hide_error = True
545        else:
546            selected_plot.hide_error = False
547        ## increment graph color
548        self.graph.render(self)
549        self.subplot.figure.canvas.draw_idle()
550        if self.is_zoomed:
551            # Recover the x,y limits
552            self.subplot.set_xlim((xlo, xhi))
553            self.subplot.set_ylim((ylo, yhi))
554        self.graph.selected_plottable = None
555
556
557    def _onRemove(self, event):
558        """
559        Remove a plottable from the graph and render the graph
560
561        :param event: Menu event
562
563        """
564        menu = event.GetEventObject()
565        event_id = event.GetId()
566        self.set_selected_from_menu(menu, event_id)
567        ## Check if there is a selected graph to remove
568        if self.graph.selected_plottable in self.plots.keys():
569            graph_id = self.graph.selected_plottable
570            self.remove_data_by_id(graph_id)
571
572    def onContextMenu(self, event):
573        """
574        1D plot context menu
575
576        :param event: wx context event
577
578        """
579        self._slicerpop = PanelMenu()
580        self._slicerpop.set_plots(self.plots)
581        self._slicerpop.set_graph(self.graph)
582        ids = iter(self._menu_ids)
583
584        # Various plot options
585        wx_id = ids.next()
586        self._slicerpop.Append(wx_id, '&Save Image', 'Save image as PNG')
587        wx.EVT_MENU(self, wx_id, self.onSaveImage)
588        wx_id = ids.next()
589        self._slicerpop.Append(wx_id, '&Print Image', 'Print image ')
590        wx.EVT_MENU(self, wx_id, self.onPrint)
591
592        wx_id = ids.next()
593        self._slicerpop.Append(wx_id, '&Copy to Clipboard',
594                               'Copy to the clipboard')
595        wx.EVT_MENU(self, wx_id, self.OnCopyFigureMenu)
596
597        self._slicerpop.AppendSeparator()
598
599        for plot in self.plots.values():
600            # title = plot.title
601            name = plot.name
602            plot_menu = wx.Menu()
603            if self.graph.selected_plottable:
604                if not self.graph.selected_plottable in self.plots.keys():
605                    continue
606                if plot != self.plots[self.graph.selected_plottable]:
607                    continue
608
609            wx_id = ids.next()
610            plot_menu.Append(wx_id, "&DataInfo", name)
611            wx.EVT_MENU(self, wx_id, self. _onDataShow)
612            wx_id = ids.next()
613            plot_menu.Append(wx_id, "&Save Points as a File", name)
614            wx.EVT_MENU(self, wx_id, self._onSave)
615            plot_menu.AppendSeparator()
616
617            # add menu of other plugins
618            item_list = self.parent.get_current_context_menu(self)
619            if (not item_list == None) and (not len(item_list) == 0):
620                for item, wx_id in zip(item_list, [ids.next() for i in range(len(item_list))]):
621
622                    try:
623                        plot_menu.Append(wx_id, item[0], name)
624                        wx.EVT_MENU(self, wx_id, item[2])
625                    except:
626                        msg = "ModelPanel1D.onContextMenu: "
627                        msg += "bad menu item  %s" % sys.exc_value
628                        wx.PostEvent(self.parent, StatusEvent(status=msg))
629                plot_menu.AppendSeparator()
630
631            if self.parent.ClassName.count('wxDialog') == 0:
632                if plot.id != 'fit':
633                    wx_id = ids.next()
634                    plot_menu.Append(wx_id, '&Linear Fit', name)
635                    wx.EVT_MENU(self, wx_id, self.onFitting)
636                    plot_menu.AppendSeparator()
637
638                wx_id = ids.next()
639                plot_menu.Append(wx_id, "Remove", name)
640                wx.EVT_MENU(self, wx_id, self._onRemove)
641                if not plot.is_data:
642                    wx_id = ids.next()
643                    plot_menu.Append(wx_id, '&Freeze', name)
644                    wx.EVT_MENU(self, wx_id, self.onFreeze)
645                plot_menu.AppendSeparator()
646
647                if plot.is_data:
648                    wx_id = ids.next()
649                    self.hide_menu = plot_menu.Append(wx_id, "Hide Error Bar", name)
650
651                    if plot.dy is not None and plot.dy != []:
652                        if plot.hide_error:
653                            self.hide_menu.SetText('Show Error Bar')
654                        else:
655                            self.hide_menu.SetText('Hide Error Bar')
656                    else:
657                        self.hide_menu.Enable(False)
658                    wx.EVT_MENU(self, wx_id, self._ontoggle_hide_error)
659
660                    plot_menu.AppendSeparator()
661
662                wx_id = ids.next()
663                plot_menu.Append(wx_id, '&Modify Plot Property', name)
664                wx.EVT_MENU(self, wx_id, self.createAppDialog)
665            wx_id = ids.next()
666            # plot_menu.SetTitle(name)
667            self._slicerpop.AppendMenu(wx_id, '&%s' % name, plot_menu)
668            # Option to hide
669            # TODO: implement functionality to hide a plottable (legend click)
670
671        self._slicerpop.AppendSeparator()
672        loc_menu = wx.Menu()
673        for label in self._loc_labels:
674            wx_id = ids.next()
675            loc_menu.Append(wx_id, str(label), str(label))
676            wx.EVT_MENU(self, wx_id, self.onChangeLegendLoc)
677
678        wx_id = ids.next()
679        self._slicerpop.Append(wx_id, '&Modify Graph Appearance',
680                               'Modify graph appearance')
681        wx.EVT_MENU(self, wx_id, self.modifyGraphAppearance)
682        self._slicerpop.AppendSeparator()
683
684
685        if self.position != None:
686            wx_id = ids.next()
687            self._slicerpop.Append(wx_id, '&Add Text')
688            wx.EVT_MENU(self, wx_id, self._on_addtext)
689            wx_id = ids.next()
690            self._slicerpop.Append(wx_id, '&Remove Text')
691            wx.EVT_MENU(self, wx_id, self._on_removetext)
692            self._slicerpop.AppendSeparator()
693        wx_id = ids.next()
694        self._slicerpop.Append(wx_id, '&Change Scale')
695        wx.EVT_MENU(self, wx_id, self._onProperties)
696        self._slicerpop.AppendSeparator()
697        wx_id = ids.next()
698        self._slicerpop.Append(wx_id, '&Set Graph Range')
699        wx.EVT_MENU(self, wx_id, self.onSetRange)
700        wx_id = ids.next()
701        self._slicerpop.Append(wx_id, '&Reset Graph Range')
702        wx.EVT_MENU(self, wx_id, self.onResetGraph)
703
704        if self.parent.ClassName.count('wxDialog') == 0:
705            self._slicerpop.AppendSeparator()
706            wx_id = ids.next()
707            self._slicerpop.Append(wx_id, '&Window Title')
708            wx.EVT_MENU(self, wx_id, self.onChangeCaption)
709        try:
710            pos_evt = event.GetPosition()
711            pos = self.ScreenToClient(pos_evt)
712        except:
713            pos_x, pos_y = self.toolbar.GetPositionTuple()
714            pos = (pos_x, pos_y + 5)
715        self.PopupMenu(self._slicerpop, pos)
716
717    def onSetRange(self, event):
718        # Display dialog
719        # self.subplot.set_xlim((low, high))
720        # self.subplot.set_ylim((low, high))
721        from sas.sasgui.plottools.RangeDialog import RangeDialog
722        d = RangeDialog(self, -1)
723        xlim = self.subplot.get_xlim()
724        ylim = self.subplot.get_ylim()
725        d.SetXRange(xlim)
726        d.SetYRange(ylim)
727        if d.ShowModal() == wx.ID_OK:
728            x_range = d.GetXRange()
729            y_range = d.GetYRange()
730            if x_range is not None and y_range is not None:
731                self.subplot.set_xlim(x_range)
732                self.subplot.set_ylim(y_range)
733                self.subplot.figure.canvas.draw_idle()
734        d.Destroy()
735
736    def onFreeze(self, event):
737        """
738        on Freeze data
739        """
740        menu = event.GetEventObject()
741        wx_id = event.GetId()
742        self.set_selected_from_menu(menu, wx_id)
743        plot = self.plots[self.graph.selected_plottable]
744        self.parent.onfreeze([plot.id])
745
746    def _onSave(self, evt):
747        """
748        Save a data set to a text file
749
750        :param evt: Menu event
751
752        """
753        menu = evt.GetEventObject()
754        event_id = evt.GetId()
755        self.set_selected_from_menu(menu, event_id)
756        data = self.plots[self.graph.selected_plottable]
757        default_name = data.label
758        if default_name.count('.') > 0:
759            default_name = default_name.split('.')[0]
760        default_name += "_out"
761        if self.parent != None:
762            self.parent.save_data1d(data, default_name)
763
764    def _onDataShow(self, evt):
765        """
766        Show the data set in text
767
768        :param evt: Menu event
769
770        """
771        menu = evt.GetEventObject()
772        event_id = evt.GetId()
773        self.set_selected_from_menu(menu, event_id)
774        data = self.plots[self.graph.selected_plottable]
775        default_name = data.label
776        if default_name.count('.') > 0:
777            default_name = default_name.split('.')[0]
778        # default_name += "_out"
779        if self.parent != None:
780            self.parent.show_data1d(data, default_name)
781
782    def _on_hide(self, event):
783        """
784        Hides the plot when button is pressed
785        """
786        if self.parent is not None:
787            self.parent.hide_panel(self.uid)
788
789    def on_close(self, event):
790        """
791        On Close Event
792        """
793        ID = self.uid
794        self.parent.delete_panel(ID)
795
796    def createAppDialog(self, event):
797        """
798        Create the custom dialog for fit appearance modification
799        """
800        menu = event.GetEventObject()
801        event_id = event.GetId()
802        self.set_selected_from_menu(menu, event_id)
803        self.appearance_selected_plot = \
804                        self.plots[self.graph.selected_plottable]
805        # find current properties
806        curr_color = self.appearance_selected_plot.custom_color
807        curr_symbol = self.appearance_selected_plot.symbol
808        curr_size = self.appearance_selected_plot.markersize
809        curr_label = self.appearance_selected_plot.label
810
811        if curr_color == None:
812            curr_color = self._color_labels['Blue']
813            curr_symbol = 13
814
815        self.appD = appearanceDialog(self, 'Modify Plot Property')
816        icon = self.parent.GetIcon()
817        self.appD.SetIcon(icon)
818        self.appD.set_defaults(float(curr_size), int(curr_color),
819                               str(appearanceDialog.find_key(self.get_symbol_label(),
820                                                             int(curr_symbol))), curr_label)
821        self.appD.Bind(wx.EVT_CLOSE, self.on_AppDialog_close)
822        self.graph.selected_plottable = None
823
824    def on_AppDialog_close(self, event):
825        """
826        on_Modify Plot Property_close
827        """
828        if self.appD.okay_clicked == True:
829            info = self.appD.get_current_values()
830            self.appearance_selected_plot.custom_color = \
831                        self._color_labels[info[1].encode('ascii', 'ignore')]
832
833            self.appearance_selected_plot.markersize = float(info[0])
834            self.appearance_selected_plot.symbol = \
835                        self.get_symbol_label()[info[2]]
836            self.appearance_selected_plot.label = str(info[3])
837        self.appD.Destroy()
838        self._check_zoom_plot()
839
840    def modifyGraphAppearance(self, event):
841        """
842        On Modify Graph Appearance
843        """
844        self.graphApp = graphAppearance(self, 'Modify Graph Appearance')
845        icon = self.parent.GetIcon()
846        self.graphApp.SetIcon(icon)
847        self.graphApp.setDefaults(self.grid_on, self.legend_on,
848                                  self.xaxis_label, self.yaxis_label,
849                                  self.xaxis_unit, self.yaxis_unit,
850                                  self.xaxis_font, self.yaxis_font,
851                                  find_key(self.get_loc_label(), self.legendLoc),
852                                  self.xcolor, self.ycolor,
853                                  self.is_xtick, self.is_ytick)
854        self.graphApp.Bind(wx.EVT_CLOSE, self.on_graphApp_close)
855
856    def on_graphApp_close(self, event):
857        """
858        Gets values from graph appearance dialog and sends them off
859        to modify the plot
860        """
861        graph_app = self.graphApp
862        toggle_grid = graph_app.get_togglegrid()
863        legend_loc = graph_app.get_legend_loc()
864        toggle_legend = graph_app.get_togglelegend()
865
866        self.onGridOnOff(toggle_grid)
867        self.ChangeLegendLoc(legend_loc)
868        self.onLegend(toggle_legend)
869
870        self.xaxis_label = graph_app.get_xlab()
871        self.yaxis_label = graph_app.get_ylab()
872        self.xaxis_unit = graph_app.get_xunit()
873        self.yaxis_unit = graph_app.get_yunit()
874        self.xaxis_font = graph_app.get_xfont()
875        self.yaxis_font = graph_app.get_yfont()
876        self.is_xtick = graph_app.get_xtick_check()
877        self.is_ytick = graph_app.get_ytick_check()
878        if self.is_xtick:
879            self.xaxis_tick = self.xaxis_font
880        if self.is_ytick:
881            self.yaxis_tick = self.yaxis_font
882
883        self.xaxis(self.xaxis_label, self.xaxis_unit,
884                   graph_app.get_xfont(), graph_app.get_xcolor(),
885                   self.xaxis_tick)
886        self.yaxis(self.yaxis_label, self.yaxis_unit,
887                   graph_app.get_yfont(), graph_app.get_ycolor(),
888                   self.yaxis_tick)
889
890        graph_app.Destroy()
Note: See TracBrowser for help on using the repository browser.