source: sasview/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ f468791

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.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since f468791 was f468791, checked in by Mathieu Doucet <doucetm@…>, 11 years ago

Move plottools under sans

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