source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 30ccad1

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 30ccad1 was 30ccad1, checked in by Jae Cho <jhjcho@…>, 13 years ago

fixed plot symbol custom size dialog: had errors on cancel and invalid values

  • Property mode set to 100644
File size: 26.6 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 os
16import pylab
17import math
18import numpy
19import time
20
21from danse.common.plottools.PlotPanel import PlotPanel
22from danse.common.plottools.SizeDialog import SizeDialog
23from danse.common.plottools.LabelDialog import LabelDialog
24#from danse.common.plottools.plottables import Graph
25from sans.guiframe import dataFitting
26from sans.guiframe.events import EVT_NEW_PLOT
27from sans.guiframe.events import StatusEvent
28from sans.guiframe.events import NewPlotEvent
29from sans.guiframe.events import NewColorEvent
30from sans.guiframe.events import SlicerEvent
31from sans.guiframe.events import PanelOnFocusEvent
32from sans.guiframe.events import EVT_NEW_LOADED_DATA
33from sans.guiframe.utils import PanelMenu
34from sans.guiframe.dataFitting import Data1D
35from sans.guiframe.panel_base import PanelBase
36from binder import BindArtist
37
38DEFAULT_QMAX = 0.05
39DEFAULT_QSTEP = 0.001
40DEFAULT_BEAM = 0.005
41BIN_WIDTH = 1
42
43
44class ModelPanel1D(PlotPanel, PanelBase):
45    """
46    Plot panel for use with the GUI manager
47    """
48   
49    ## Internal name for the AUI manager
50    window_name = "plotpanel"
51    ## Title to appear on top of the window
52    window_caption = "Graph"
53    ## Flag to tell the GUI manager that this panel is not
54    #  tied to any perspective
55    ALWAYS_ON = True
56    ## Group ID
57    group_id = None
58   
59    def __init__(self, parent, id=-1, color = None,
60                 dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
61        PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
62        PanelBase.__init__(self, parent)
63        ## Reference to the parent window
64        self.parent = parent
65        ## Plottables
66        self.plots = {}
67        #context menu
68        self._slicerpop = None
69       
70        self._available_data = []
71        self._menu_add_ids = []
72        self._symbol_labels = self.get_symbol_label()
73        self._color_labels = self.get_color_label()
74        self.currColorIndex = ""
75        self._is_changed_legend_label = False
76     
77        self.hide_menu = None
78        ## Unique ID (from gui_manager)
79        self.uid = None
80        self.x_size = None
81        ## Default locations
82        #self._default_save_location = os.getcwd()
83        self.size = None       
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       
90        # In resizing event
91        self.resizing = False
92        self.canvas.set_resizing(self.resizing)
93        self.Bind(wx.EVT_SIZE, self._OnReSize)
94        self._add_more_tool()
95       
96    def get_symbol_label(self):
97        """
98        Associates label to symbol
99        """
100        _labels = {}
101        i = 0
102        _labels['Circle'] = i
103        i += 1
104        _labels['Cross X '] = i
105        i += 1
106        _labels['Triangle Down'] = i
107        i += 1
108        _labels['Triangle Up'] = i
109        i += 1
110        _labels['Triangle Left'] = i
111        i += 1
112        _labels['Triangle Right'] = i
113        i += 1
114        _labels['Cross +'] = i
115        i += 1
116        _labels['Square'] = i
117        i += 1
118        _labels['Diamond'] = i
119        i += 1
120        _labels['Diamond'] = i
121        i += 1
122        _labels['Hexagon1'] = i
123        i += 1
124        _labels['Hexagon2'] = i
125        i += 1
126        _labels['Pentagon'] = i
127        i += 1
128        _labels['Line'] = i
129        return _labels
130   
131    def get_color_label(self):
132        """
133        Associates label to a specific color
134        """
135        _labels = {}
136        i = 0
137        _labels['Blue'] = i
138        i += 1
139        _labels['Green'] = i
140        i += 1
141        _labels['Red'] = i
142        i += 1
143        _labels['Cyan'] = i
144        i += 1
145        _labels['Magenta'] = i
146        i += 1
147        _labels['Yellow'] = i
148        i += 1
149        _labels['Black'] = i
150        return _labels
151
152   
153    def set_data(self, list=None):
154        """
155        """
156        pass
157   
158    def _reset(self):
159        """
160        Resets internal data and graph
161        """   
162        self.graph.reset()
163        self.plots      = {}
164        if self.is_zoomed:
165                self.is_zoomed = False
166       
167    def _OnReSize(self, event):   
168        """
169        On response of the resize of a panel, set axes_visiable False
170        """
171        # It was found that wx >= 2.9.3 sends an event even if no size changed.
172        # So manually recode the size (=x_size) and compare here.
173        if self.x_size != None:
174            if self.x_size == self.GetSize():
175                self.resizing = False
176                self.canvas.set_resizing(self.resizing)
177                return
178        self.x_size = self.GetSize()
179
180        # Ready for another event
181        # Do not remove this Skip. Otherwise it will get runtime error on wx>=2.9.
182        event.Skip() 
183        # set the resizing flag
184        self.resizing = True
185        self.canvas.set_resizing(self.resizing)
186        self.parent.set_schedule(True)
187        pos_x, pos_y = self.GetPositionTuple()
188        if pos_x != 0 and pos_y != 0:
189            self.size, _ = self.GetClientSizeTuple()
190       
191    def set_resizing(self, resizing=False):
192        """
193        Set the resizing (True/False)
194        """
195        self.resizing = resizing
196        #self.canvas.set_resizing(resizing)
197   
198    def schedule_full_draw(self, func='append'):   
199        """
200        Put self in schedule to full redraw list
201        """
202        # append/del this panel in the schedule list
203        self.parent.set_schedule_full_draw(self, func)
204       
205
206    def remove_data_by_id(self, id):
207        """'
208        remove data from plot
209        """
210        if id in self.plots.keys():
211            data =  self.plots[id]
212            self.graph.delete(data)
213            data_manager = self._manager.parent.get_data_manager()
214            data_list, theory_list = data_manager.get_by_id(id_list=[id])
215           
216            if id in data_list.keys():
217                data = data_list[id]
218            if id in theory_list.keys():
219                data = theory_list[id]
220           
221            del self.plots[id]
222            self.graph.render(self)
223            self.subplot.figure.canvas.draw_idle()   
224            if len(self.graph.plottables) == 0:
225                #onRemove: graph is empty must be the panel must be destroyed
226                self.parent.delete_panel(self.uid)
227
228       
229    def plot_data(self, data):
230        """
231        Data is ready to be displayed
232       
233        :param event: data event
234        """
235        if data.id in self.plots.keys():
236            #Recover panel prop.s
237            xlo, xhi = self.subplot.get_xlim()
238            ylo, yhi = self.subplot.get_ylim()
239            old_data = self.plots[data.id]
240            if self._is_changed_legend_label:
241                data.label = old_data.label
242            data.custom_color = old_data.custom_color
243            data.symbol = old_data.symbol
244            data.markersize = old_data.markersize
245            # Replace data
246            self.graph.replace(data)
247            self.plots[data.id] = data
248            ## Set the view scale for all plots
249            self._onEVT_FUNC_PROPERTY()
250            # Check if zoomed
251            toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
252            if self.is_zoomed or toolbar_zoomed:
253                # Recover the x,y limits
254                self.subplot.set_xlim((xlo, xhi))     
255                self.subplot.set_ylim((ylo, yhi)) 
256            # Update Graph menu and help string       
257            pos = self.parent._window_menu.FindItem(self.window_caption)
258            helpString = 'Show/Hide Graph: '
259            for plot in  self.plots.itervalues():
260                helpString += (' ' + plot.label +';')
261            self.parent._window_menu.SetHelpString(pos, helpString)
262        else:
263            self.plots[data.id] = data
264            self.graph.add(self.plots[data.id]) 
265            ## Set the view scale for all plots
266            self._onEVT_FUNC_PROPERTY()
267            self.toolbar.update()
268            if self.is_zoomed:
269                self.is_zoomed = False
270       
271         
272    def draw_plot(self):
273        """
274        Draw plot
275        """
276        self.draw() 
277
278
279       
280    def onLeftDown(self,event): 
281        """
282        left button down and ready to drag
283        Display the position of the mouse on the statusbar
284        """
285        PlotPanel.onLeftDown(self, event)
286        ax = event.inaxes
287        if ax != None:
288            try:
289                pos_x = float(event.xdata)# / size_x
290                pos_y = float(event.ydata)# / size_y
291                pos_x = "%8.3g"% pos_x
292                pos_y = "%8.3g"% pos_y
293                self.position = str(pos_x), str(pos_y)
294                wx.PostEvent(self.parent, StatusEvent(status=self.position))
295            except:
296                self.position = None 
297        # unfocus all
298        self.parent.set_plot_unfocus() 
299        #post nd event to notify guiframe that this panel is on focus
300        wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
301
302       
303    def _ontoggle_hide_error(self, event):
304        """
305        Toggle error display to hide or show
306        """
307        # Check zoom
308        xlo, xhi = self.subplot.get_xlim()
309        ylo, yhi = self.subplot.get_ylim()
310
311        selected_plot = self.plots[self.graph.selected_plottable]
312        if self.hide_menu.GetText() == "Hide Error Bar":
313            selected_plot.hide_error = True
314        else:
315            selected_plot.hide_error = False
316        ## increment graph color
317        self.graph.render(self)
318        self.subplot.figure.canvas.draw_idle() 
319        # Check if zoomed
320        toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
321        if self.is_zoomed or toolbar_zoomed:
322            # Recover the x,y limits
323            self.subplot.set_xlim((xlo, xhi))     
324            self.subplot.set_ylim((ylo, yhi)) 
325
326         
327    def _onRemove(self, event):
328        """
329        Remove a plottable from the graph and render the graph
330       
331        :param event: Menu event
332       
333        """
334        ## Check if there is a selected graph to remove
335        if self.graph.selected_plottable in self.plots.keys():
336            selected_plot = self.plots[self.graph.selected_plottable]
337            id = self.graph.selected_plottable
338            self.remove_data_by_id(id)
339           
340    def onContextMenu(self, event):
341        """
342        1D plot context menu
343       
344        :param event: wx context event
345       
346        """
347        self._slicerpop = PanelMenu()
348        self._slicerpop.set_plots(self.plots)
349        self._slicerpop.set_graph(self.graph)     
350        # Various plot options
351        id = wx.NewId()
352        self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
353        wx.EVT_MENU(self, id, self.onSaveImage)
354        id = wx.NewId()
355        self._slicerpop.Append(id, '&Print Image', 'Print image ')
356        wx.EVT_MENU(self, id, self.onPrint)
357        id = wx.NewId()
358        self._slicerpop.Append(id, '&Print Preview', 'Print preview')
359        wx.EVT_MENU(self, id, self.onPrinterPreview)
360       
361        id = wx.NewId()
362        self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
363        wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
364               
365        self._slicerpop.AppendSeparator()
366
367        #add menu of other plugins
368        item_list = self.parent.get_context_menu(self)
369
370        if (not item_list == None) and (not len(item_list) == 0):
371            for item in item_list:
372                try:
373                    id = wx.NewId()
374                    self._slicerpop.Append(id, item[0], item[1])
375                    wx.EVT_MENU(self, id, item[2])
376                except:
377                    msg = "ModelPanel1D.onContextMenu: "
378                    msg += "bad menu item  %s" % sys.exc_value
379                    wx.PostEvent(self.parent, StatusEvent(status=msg))
380                    pass
381            self._slicerpop.AppendSeparator()
382        #id = wx.NewId()
383        #self._slicerpop.Append(id, '&Print image', 'Print image')
384        if self.graph.selected_plottable in self.plots:
385            plot = self.plots[self.graph.selected_plottable]
386           
387            id = wx.NewId()
388            name = plot.name
389            self._slicerpop.Append(id, "&Save Points as a File")
390            wx.EVT_MENU(self, id, self._onSave)
391            self._slicerpop.AppendSeparator()
392            if self.parent.ClassName.count('wxDialog') == 0: 
393                id = wx.NewId()
394                self._slicerpop.Append(id, '&Linear Fit')
395                wx.EVT_MENU(self, id, self.onFitting)
396                self._slicerpop.AppendSeparator()
397   
398                id = wx.NewId()
399                self._slicerpop.Append(id, "Remove %s Curve" % name)
400                wx.EVT_MENU(self, id, self._onRemove)
401                if not plot.is_data:
402                    id = wx.NewId()
403                    self._slicerpop.Append(id, '&Freeze', 'Freeze')
404                    wx.EVT_MENU(self, id, self.onFreeze)
405                self._slicerpop.AppendSeparator()   
406                symbol_menu = wx.Menu()
407                for label in self._symbol_labels:
408                    id = wx.NewId()
409                    symbol_menu.Append(id, str(label), str(label))
410                    wx.EVT_MENU(self, id, self.onChangeSymbol)
411                id = wx.NewId()
412                self._slicerpop.AppendMenu(id,'&Modify Symbol',  symbol_menu)
413               
414                color_menu = wx.Menu()
415                for label in self._color_labels:
416                    id = wx.NewId()
417                    color_menu.Append(id, str(label), str(label))
418                    wx.EVT_MENU(self, id, self.onChangeColor)
419                id = wx.NewId()
420                self._slicerpop.AppendMenu(id, '&Modify Symbol Color', color_menu)
421               
422                size_menu = wx.Menu()
423                for i in range(10):
424                    id = wx.NewId()
425                    size_menu.Append(id, str(i), str(i))
426                    wx.EVT_MENU(self, id, self.onChangeSize)
427                id = wx.NewId()
428                size_menu.Append(id, '&Custom', 'Custom')
429                wx.EVT_MENU(self, id, self.onChangeSize)
430                id = wx.NewId()
431                self._slicerpop.AppendMenu(id, '&Modify Symbol Size', size_menu)
432               
433                self.hide_menu = self._slicerpop.Append(id, "Hide Error Bar")
434   
435                if plot.dy is not None and plot.dy != []:
436                    if plot.hide_error :
437                        self.hide_menu.SetText('Show Error Bar')
438                    else:
439                        self.hide_menu.SetText('Hide Error Bar')
440                else:
441                    self.hide_menu.Enable(False)
442                wx.EVT_MENU(self, id, self._ontoggle_hide_error)
443               
444                self._slicerpop.AppendSeparator()
445                id = wx.NewId()
446                self._slicerpop.Append(id, '&Edit Legend Label', 'Edit Legend Label')
447                wx.EVT_MENU(self, id, self.onEditLabels)
448                # Option to hide
449                #TODO: implement functionality to hide a plottable (legend click)
450
451        loc_menu = wx.Menu()
452        for label in self._loc_labels:
453            id = wx.NewId()
454            loc_menu.Append(id, str(label), str(label))
455            wx.EVT_MENU(self, id, self.onChangeLegendLoc)
456        id = wx.NewId()
457        self._slicerpop.AppendMenu(id, '&Modify Legend Location', loc_menu)
458       
459        id = wx.NewId()
460        self._slicerpop.Append(id, '&Toggle Legend On/Off', 'Toggle Legend On/Off')
461        wx.EVT_MENU(self, id, self.onLegend)
462        self._slicerpop.AppendSeparator()
463       
464        id = wx.NewId()
465        self._slicerpop.Append(id, '&Edit Y Axis Label')
466        wx.EVT_MENU(self, id, self._on_yaxis_label)     
467        id = wx.NewId()
468        self._slicerpop.Append(id, '&Edit X Axis Label')
469        wx.EVT_MENU(self, id, self._on_xaxis_label)
470
471        id = wx.NewId()
472        self._slicerpop.Append(id, '&Toggle Grid On/Off', 'Toggle Grid On/Off')
473        wx.EVT_MENU(self, id, self.onGridOnOff)
474        self._slicerpop.AppendSeparator()
475       
476        if self.position != None:
477            id = wx.NewId()
478            self._slicerpop.Append(id, '&Add Text')
479            wx.EVT_MENU(self, id, self._on_addtext)
480            id = wx.NewId()
481            self._slicerpop.Append(id, '&Remove Text')
482            wx.EVT_MENU(self, id, self._on_removetext)
483            self._slicerpop.AppendSeparator()
484        id = wx.NewId()
485        self._slicerpop.Append(id, '&Change Scale')
486        wx.EVT_MENU(self, id, self._onProperties)
487        self._slicerpop.AppendSeparator()
488        id = wx.NewId()
489        self._slicerpop.Append(id, '&Reset Graph Range')
490        wx.EVT_MENU(self, id, self.onResetGraph) 
491        try:
492            pos_evt = event.GetPosition()
493            pos = self.ScreenToClient(pos_evt)
494        except:
495            pos_x, pos_y = self.toolbar.GetPositionTuple()
496            pos = (pos_x, pos_y + 5)
497       
498        if self.parent.ClassName.count('wxDialog') == 0:   
499            self._slicerpop.AppendSeparator()
500            id = wx.NewId()
501            self._slicerpop.Append(id, '&Window Title')
502            wx.EVT_MENU(self, id, self.onChangeCaption)
503       
504        self.PopupMenu(self._slicerpop, pos)
505       
506    def onFreeze(self, event):
507        """
508        """
509        plot = self.plots[self.graph.selected_plottable]
510        self.parent.onfreeze([plot.id])
511       
512    def onEditLabels(self, event):
513        """
514        Edit legend label
515        """
516        selected_plot = self.plots[self.graph.selected_plottable]
517        label = selected_plot.label
518        dial = LabelDialog(None, -1, 'Change Legend Label', label)
519        if dial.ShowModal() == wx.ID_OK:
520            newLabel = dial.getText() 
521            selected_plot.label = newLabel
522            # Updata Graph menu help string
523            pos = self.parent._window_menu.FindItem(self.window_caption)
524            helpString = 'Show/Hide Graph: '
525            for plot in  self.plots.itervalues():
526                helpString += (' ' + plot.label +';')
527            self.parent._window_menu.SetHelpString(pos, helpString)
528            self._is_changed_legend_label = True
529            #break
530        dial.Destroy()
531       
532        ## render the graph
533        self._onEVT_FUNC_PROPERTY() 
534       
535    def onChangeColor(self, event):
536        """
537        Changes the color of the graph when selected
538        """
539        menu = event.GetEventObject()
540        id = event.GetId()
541        label =  menu.GetLabel(id)
542        selected_plot = self.plots[self.graph.selected_plottable]
543        selected_plot.custom_color = self._color_labels[label]
544        ## Set the view scale for all plots
545        self._check_zoom_plot()
546        #self._onEVT_FUNC_PROPERTY()
547        #wx.PostEvent(self.parent,
548        #              NewColorEvent(color=selected_plot.custom_color,
549        #                                     id=selected_plot.id))
550   
551    def onChangeSize(self, event):
552       
553        menu = event.GetEventObject()
554        id = event.GetId()
555        label =  menu.GetLabel(id)
556        selected_plot = self.plots[self.graph.selected_plottable]
557       
558        if label == "&Custom":
559            sizedial = SizeDialog(None, -1, 'Change Marker Size')
560            if sizedial.ShowModal() == wx.ID_OK:
561                try:
562                    label = sizedial.getText()
563                    selected_plot.markersize = int(label)
564                    self._check_zoom_plot()
565                except:
566                    msg = 'Symbol Size: Got an invalid Value.'
567                    wx.PostEvent( self.parent, 
568                          StatusEvent(status= msg, info='error'))
569            sizedial.Destroy()
570
571
572   
573    def onChangeSymbol(self, event):
574        """
575        """
576        menu = event.GetEventObject()
577        id = event.GetId()
578        label =  menu.GetLabel(id)
579        selected_plot = self.plots[self.graph.selected_plottable]
580        selected_plot.symbol = self._symbol_labels[label]
581        ## Set the view scale for all plots
582        self._check_zoom_plot()
583        #self._onEVT_FUNC_PROPERTY()
584        ## render the graph
585        #self.graph.render(self)
586        #self.subplot.figure.canvas.draw_idle()
587       
588       
589       
590    def _onsaveTXT(self, path):
591        """
592        Save file as txt
593           
594        :TODO: Refactor and remove this method. See TODO in _onSave.
595       
596        """
597        data = self.plots[self.graph.selected_plottable]
598       
599        if not path == None:
600            out = open(path, 'w')
601            has_errors = True
602            if data.dy == None or data.dy == []:
603                has_errors = False
604            # Sanity check
605            if has_errors:
606                try:
607                    if len(data.y) != len(data.dy):
608                        has_errors = False
609                except:
610                    has_errors = False
611            if has_errors:
612                if data.dx != None:
613                    out.write("<X>   <Y>   <dY>   <dX>\n")
614                else:
615                    out.write("<X>   <Y>   <dY>\n")
616            else:
617                out.write("<X>   <Y>\n")
618               
619            for i in range(len(data.x)):
620                if has_errors:
621                    if data.dx != None:
622                        out.write("%g  %g  %g  %g\n" % (data.x[i], 
623                                                    data.y[i],
624                                                    data.dy[i],
625                                                    data.dx[i]))
626                    else:
627                        out.write("%g  %g  %g\n" % (data.x[i], 
628                                                    data.y[i],
629                                                    data.dy[i]))
630                else:
631                    out.write("%g  %g\n" % (data.x[i], 
632                                            data.y[i]))
633            out.close()                 
634            try:
635                self._default_save_location = os.path.dirname(path)
636                self.parent._default_save_location = self._default_save_location
637            except:
638                pass   
639               
640    def _onSave(self, evt):
641        """
642        Save a data set to a text file
643       
644        :param evt: Menu event
645       
646        """
647       
648        path = None
649        wildcard = "Text files (*.txt)|*.txt|"\
650                    "CanSAS 1D files(*.xml)|*.xml" 
651        default_name = self.plots[self.graph.selected_plottable].label
652        if default_name.count('.') > 0:
653            default_name = default_name.split('.')[0]
654        default_name += "_out"
655        if self.parent != None:
656            self._default_save_location = self.parent._default_save_location
657        dlg = wx.FileDialog(self, "Choose a file",
658                            self._default_save_location,
659                            default_name, wildcard , wx.SAVE)
660       
661        if dlg.ShowModal() == wx.ID_OK:
662            path = dlg.GetPath()
663            # ext_num = 0 for .txt, ext_num = 1 for .xml
664            # This is MAC Fix
665            ext_num = dlg.GetFilterIndex()
666            if ext_num == 0:
667                format = '.txt'
668            else:
669                format = '.xml'
670            path = os.path.splitext(path)[0] + format
671            mypath = os.path.basename(path)
672           
673            #TODO: This is bad design. The DataLoader is designed
674            #to recognize extensions.
675            # It should be a simple matter of calling the .
676            #save(file, data, '.xml') method
677            # of the sans.dataloader.loader.Loader class.
678            from sans.dataloader.loader import  Loader
679            #Instantiate a loader
680            loader = Loader() 
681            data = self.plots[self.graph.selected_plottable]
682            format = ".txt"
683            if os.path.splitext(mypath)[1].lower() == format:
684                # Make sure the ext included in the file name
685                # especially on MAC
686                fName = os.path.splitext(path)[0] + format
687                self._onsaveTXT(fName)
688            format = ".xml"
689            if os.path.splitext(mypath)[1].lower() == format:
690                # Make sure the ext included in the file name
691                # especially on MAC
692                fName = os.path.splitext(path)[0] + format
693                loader.save(fName, data, format)
694            try:
695                self._default_save_location = os.path.dirname(path)
696                self.parent._default_save_location = self._default_save_location
697            except:
698                pass   
699        dlg.Destroy()
700
701    def _add_more_tool(self):
702        """
703        Add refresh, add/delete button in the tool bar
704        """
705        if self.parent.__class__.__name__ != 'ViewerFrame':
706            return
707        self.toolbar.AddSeparator()
708        id_delete = wx.NewId()
709        delete =  wx.ArtProvider.GetBitmap(wx.ART_DELETE, wx.ART_TOOLBAR, 
710                                           size = (18,18))
711        self.toolbar.AddSimpleTool(id_delete, delete,
712                           'Delete', 'permanently Delete')
713
714        self.toolbar.Realize()
715        wx.EVT_TOOL(self, id_delete,  self._on_delete)
716       
717        """
718        self.toolbar.AddSeparator()
719        id_text = wx.NewId()
720        text =  wx.ArtProvider.GetBitmap(wx.ART_CUT, wx.ART_TOOLBAR)
721        self.toolbar.AddSimpleTool(id_text, text,
722                           'Remove Text from Plot', 'Removes text from plot')
723
724        self.toolbar.Realize()
725        wx.EVT_TOOL(self, id_text,  self._on_removetext)
726        """
727    def _on_delete(self, event): 
728        """
729        Refreshes the plotpanel on refresh tollbar button
730        """
731       
732        if self.parent is not None:
733            wx.PostEvent(self.parent, 
734                         NewPlotEvent(group_id=self.group_id,
735                                      action="delete"))
736           
737           
Note: See TracBrowser for help on using the repository browser.