source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 9d52bcb

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

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