source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 650fd318

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

now graph scale doesn't get reset the markersize on updating

  • Property mode set to 100644
File size: 26.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 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                label = sizedial.getText()
562            sizedial.Destroy()
563
564        selected_plot.markersize = int(label)
565        self._check_zoom_plot()
566        #self._onEVT_FUNC_PROPERTY()
567        ## Set the view scale for all plots
568       
569        ## render the graph
570        #self.graph.render(self)
571        #self.subplot.figure.canvas.draw_idle()
572
573   
574    def onChangeSymbol(self, event):
575        """
576        """
577        menu = event.GetEventObject()
578        id = event.GetId()
579        label =  menu.GetLabel(id)
580        selected_plot = self.plots[self.graph.selected_plottable]
581        selected_plot.symbol = self._symbol_labels[label]
582        ## Set the view scale for all plots
583        self._check_zoom_plot()
584        #self._onEVT_FUNC_PROPERTY()
585        ## render the graph
586        #self.graph.render(self)
587        #self.subplot.figure.canvas.draw_idle()
588       
589       
590       
591    def _onsaveTXT(self, path):
592        """
593        Save file as txt
594           
595        :TODO: Refactor and remove this method. See TODO in _onSave.
596       
597        """
598        data = self.plots[self.graph.selected_plottable]
599       
600        if not path == None:
601            out = open(path, 'w')
602            has_errors = True
603            if data.dy == None or data.dy == []:
604                has_errors = False
605            # Sanity check
606            if has_errors:
607                try:
608                    if len(data.y) != len(data.dy):
609                        has_errors = False
610                except:
611                    has_errors = False
612            if has_errors:
613                if data.dx != None:
614                    out.write("<X>   <Y>   <dY>   <dX>\n")
615                else:
616                    out.write("<X>   <Y>   <dY>\n")
617            else:
618                out.write("<X>   <Y>\n")
619               
620            for i in range(len(data.x)):
621                if has_errors:
622                    if data.dx != None:
623                        out.write("%g  %g  %g  %g\n" % (data.x[i], 
624                                                    data.y[i],
625                                                    data.dy[i],
626                                                    data.dx[i]))
627                    else:
628                        out.write("%g  %g  %g\n" % (data.x[i], 
629                                                    data.y[i],
630                                                    data.dy[i]))
631                else:
632                    out.write("%g  %g\n" % (data.x[i], 
633                                            data.y[i]))
634            out.close()                 
635            try:
636                self._default_save_location = os.path.dirname(path)
637                self.parent._default_save_location = self._default_save_location
638            except:
639                pass   
640               
641    def _onSave(self, evt):
642        """
643        Save a data set to a text file
644       
645        :param evt: Menu event
646       
647        """
648       
649        path = None
650        wildcard = "Text files (*.txt)|*.txt|"\
651                    "CanSAS 1D files(*.xml)|*.xml" 
652        default_name = self.plots[self.graph.selected_plottable].label
653        if default_name.count('.') > 0:
654            default_name = default_name.split('.')[0]
655        default_name += "_out"
656        if self.parent != None:
657            self._default_save_location = self.parent._default_save_location
658        dlg = wx.FileDialog(self, "Choose a file",
659                            self._default_save_location,
660                            default_name, wildcard , wx.SAVE)
661       
662        if dlg.ShowModal() == wx.ID_OK:
663            path = dlg.GetPath()
664            # ext_num = 0 for .txt, ext_num = 1 for .xml
665            # This is MAC Fix
666            ext_num = dlg.GetFilterIndex()
667            if ext_num == 0:
668                format = '.txt'
669            else:
670                format = '.xml'
671            path = os.path.splitext(path)[0] + format
672            mypath = os.path.basename(path)
673           
674            #TODO: This is bad design. The DataLoader is designed
675            #to recognize extensions.
676            # It should be a simple matter of calling the .
677            #save(file, data, '.xml') method
678            # of the sans.dataloader.loader.Loader class.
679            from sans.dataloader.loader import  Loader
680            #Instantiate a loader
681            loader = Loader() 
682            data = self.plots[self.graph.selected_plottable]
683            format = ".txt"
684            if os.path.splitext(mypath)[1].lower() == format:
685                # Make sure the ext included in the file name
686                # especially on MAC
687                fName = os.path.splitext(path)[0] + format
688                self._onsaveTXT(fName)
689            format = ".xml"
690            if os.path.splitext(mypath)[1].lower() == format:
691                # Make sure the ext included in the file name
692                # especially on MAC
693                fName = os.path.splitext(path)[0] + format
694                loader.save(fName, data, format)
695            try:
696                self._default_save_location = os.path.dirname(path)
697                self.parent._default_save_location = self._default_save_location
698            except:
699                pass   
700        dlg.Destroy()
701
702    def _add_more_tool(self):
703        """
704        Add refresh, add/delete button in the tool bar
705        """
706        if self.parent.__class__.__name__ != 'ViewerFrame':
707            return
708        self.toolbar.AddSeparator()
709        id_delete = wx.NewId()
710        delete =  wx.ArtProvider.GetBitmap(wx.ART_DELETE, wx.ART_TOOLBAR)
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.