source: sasview/guiframe/local_perspectives/plotting/Plotter1D.py @ 95b2bad

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 95b2bad was 0275276, checked in by Jae Cho <jhjcho@…>, 14 years ago

fixed plot_panel highlight problem

  • Property mode set to 100644
File size: 15.7 KB
RevLine 
[1bf33c1]1
[d955bf19]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################################################################################
[1bf33c1]11
12
13import wx
[4ac8556]14import sys
15import os
16import pylab
17import math
18import numpy
19import time
[a07e72f]20
[1bf33c1]21from danse.common.plottools.PlotPanel import PlotPanel
[e5664f2]22from danse.common.plottools.plottables import Graph
[3b69ca6]23from sans.guiframe import dataFitting
[df7046f]24from sans.guiframe.events import EVT_NEW_PLOT
25from sans.guiframe.events import StatusEvent
26from sans.guiframe.events import NewPlotEvent
27from sans.guiframe.events import SlicerEvent
28from sans.guiframe.events import RemoveDataEvent
[a45037aa]29from sans.guiframe.events import PanelOnFocusEvent
[52b8b74]30from sans.guiframe.events import EVT_NEW_LOADED_DATA
[0d9dae8]31from sans.guiframe.utils import PanelMenu
[4ac8556]32from sans.guiframe.dataFitting import Data1D
[691643c]33from sans.guiframe.panel_base import PanelBase
[1bf33c1]34from binder import BindArtist
35
[0d9dae8]36DEFAULT_QMAX = 0.05
[1bf33c1]37DEFAULT_QSTEP = 0.001
38DEFAULT_BEAM = 0.005
[32c0841]39BIN_WIDTH = 1
[1bf33c1]40
[0d9dae8]41
[691643c]42class ModelPanel1D(PlotPanel, PanelBase):
[1bf33c1]43    """
[d955bf19]44    Plot panel for use with the GUI manager
[1bf33c1]45    """
46   
47    ## Internal name for the AUI manager
48    window_name = "plotpanel"
49    ## Title to appear on top of the window
50    window_caption = "Plot Panel"
51    ## Flag to tell the GUI manager that this panel is not
52    #  tied to any perspective
53    ALWAYS_ON = True
54    ## Group ID
55    group_id = None
56   
[32c0841]57    def __init__(self, parent, id=-1, color = None,
58                 dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
59        PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
[a45037aa]60        PanelBase.__init__(self, parent)
[1bf33c1]61        ## Reference to the parent window
62        self.parent = parent
63        ## Plottables
64        self.plots = {}
[52b8b74]65        #context menu
66        self._slicerpop = None
[a07e72f]67       
[52b8b74]68        self._available_data = []
69        self._menu_add_ids = []
[a07e72f]70        self._symbol_labels = self.get_symbol_label()
71     
72        self.hide_menu = None
[1bf33c1]73        ## Unique ID (from gui_manager)
74        self.uid = None
[a07e72f]75       
[6063b16]76        ## Default locations
77        self._default_save_location = os.getcwd()       
[1bf33c1]78        ## Graph       
79        self.graph = Graph()
80        self.graph.xaxis("\\rm{Q}", 'A^{-1}')
[32c0841]81        self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
[1bf33c1]82        self.graph.render(self)
[6d727ae]83       
84        # In resizing event
85        self.resizing = False
86        self.canvas.set_resizing(self.resizing)
87        self.Bind(wx.EVT_SIZE, self._OnReSize)
[a07e72f]88       
89    def get_symbol_label(self):
[52b8b74]90        """
[a07e72f]91        Associates label to symbol
[52b8b74]92        """
[a07e72f]93        _labels = {}
94        i = 0
[6d727ae]95        _labels['Circle'] = i
[a07e72f]96        i += 1
[6d727ae]97        _labels['Cross X '] = i
[a07e72f]98        i += 1
99        _labels['Triangle Down'] = i
100        i += 1
101        _labels['Triangle Up'] = i
102        i += 1
103        _labels['Triangle Left'] = i
104        i += 1
105        _labels['Triangle Right'] = i
106        i += 1
[6d727ae]107        _labels['Cross +'] = i
[a07e72f]108        i += 1
109        _labels['Square'] = i
110        i += 1
[6d727ae]111        _labels['Diamond'] = i
[a07e72f]112        i += 1
113        _labels['Diamond'] = i
114        i += 1
115        _labels['Hexagon1'] = i
116        i += 1
117        _labels['Hexagon2'] = i
118        i += 1
119        _labels['Pentagon'] = i
120        i += 1
[6d727ae]121        _labels['Line'] = i
[a07e72f]122        return _labels
123
[52b8b74]124   
[32c0841]125    def set_data(self, list=None):
[3c44c66]126        """
127        """
128        pass
129   
[1bf33c1]130    def _reset(self):
131        """
[d955bf19]132        Resets internal data and graph
[1bf33c1]133        """   
134        self.graph.reset()
135        self.plots      = {}
[6d727ae]136       
137    def _OnReSize(self, event):   
138        """
139        On response of the resize of a panel, set axes_visiable False
140        """
141        # ready for another event
142        event.Skip() 
143        # set the resizing flag
144        self.resizing = True
145        self.canvas.set_resizing(self.resizing)
146        self.parent.set_schedule(True)
147       
148       
149    def set_resizing(self, resizing=False):
150        """
151        Set the resizing (True/False)
152        """
153        self.resizing = resizing
154        #self.canvas.set_resizing(resizing)
[1bf33c1]155   
[6d727ae]156    def schedule_full_draw(self, func='append'):   
157        """
158        Put self in schedule to full redraw list
159        """
160        # append/del this panel in the schedule list
161        self.parent.set_schedule_full_draw(self, func)
162       
163
[a07e72f]164    def remove_data_by_id(self, id):
165        """'
166        remove data from plot
167        """
168        if id in self.plots.keys():
169            data =  self.plots[id]
170            self.graph.delete(data)
171            data_manager = self._manager.parent.get_data_manager()
[df22224]172            data_list, theory_list = data_manager.get_by_id(id_list=[id])
173           
174            if id in data_list.keys():
175                data = data_list[id]
176            else:
177                data = theory_list[id]
178           
[a07e72f]179            del self.plots[id]
180            self.graph.render(self)
181            self.subplot.figure.canvas.draw_idle()   
[df22224]182            event = RemoveDataEvent(data=data)
183            wx.PostEvent(self.parent, event)
[a07e72f]184            if len(self.graph.plottables) == 0:
[df22224]185                #onRemove: graph is empty must be the panel must be destroyed
186                self.parent.delete_panel(self.uid)
[a07e72f]187       
188    def plot_data(self, data):
[1bf33c1]189        """
[d955bf19]190        Data is ready to be displayed
191       
192        :param event: data event
[1bf33c1]193        """
[a07e72f]194        if data.id in self.plots.keys():
195            #replace
196            self.graph.replace(data)
197            self.plots[data.id] = data
198        else:
199            self.plots[data.id] = data
200            self.graph.add(self.plots[data.id]) 
[ffd23b5]201       
[a07e72f]202        x_label, x_unit = data.get_xaxis()
203        y_label, y_unit = data.get_yaxis()
204        self.graph.xaxis(x_unit, x_label)
205        self.graph.yaxis(y_unit, y_label)
[6c0568b]206        ## Set the view scale for all plots
[1bf33c1]207        self._onEVT_FUNC_PROPERTY()
[6d727ae]208        ## render the graph<=No need this done in canvas
209        #self.graph.render(self)
210        #self.subplot.figure.canvas.draw_idle()
211   
212    def draw_plot(self):
213        """
214        Draw plot
215        """
216        self.draw() 
217
[a07e72f]218
219       
[1bf33c1]220    def onLeftDown(self,event): 
[6c0568b]221        """
[d955bf19]222        left button down and ready to drag
223        Display the position of the mouse on the statusbar
[6c0568b]224        """
[1bf33c1]225        PlotPanel.onLeftDown(self, event)
226        ax = event.inaxes
227        if ax != None:
228            position = "x: %8.3g    y: %8.3g" % (event.xdata, event.ydata)
229            wx.PostEvent(self.parent, StatusEvent(status=position))
[0275276]230        # unfocus all
231        self.parent.set_plot_unfocus() 
[4ed210f4]232        #post nd event to notify guiframe that this panel is on focus
[a45037aa]233        wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
[df68da1]234
[4ed210f4]235       
[a07e72f]236    def _ontoggle_hide_error(self, event):
237        """
238        Toggle error display to hide or show
239        """
240       
241        selected_plot = self.plots[self.graph.selected_plottable]
242        if self.hide_menu.GetText() == "Hide Error":
243            selected_plot.hide_error = True
244        else:
245            selected_plot.hide_error = False
246        ## increment graph color
247        self.graph.render(self)
248        self.subplot.figure.canvas.draw_idle() 
249         
[1bf33c1]250    def _onRemove(self, event):
251        """
[d955bf19]252        Remove a plottable from the graph and render the graph
253       
254        :param event: Menu event
255       
[1bf33c1]256        """
[6c0568b]257        ## Check if there is a selected graph to remove
[a07e72f]258        if self.graph.selected_plottable in self.plots.keys():
259            selected_plot = self.plots[self.graph.selected_plottable]
260            id = self.graph.selected_plottable
[df22224]261            self.remove_data_by_id(id)
262           
[1bf33c1]263    def onContextMenu(self, event):
264        """
[d955bf19]265        1D plot context menu
266       
267        :param event: wx context event
268       
[1bf33c1]269        """
[52b8b74]270        self._slicerpop = PanelMenu()
271        self._slicerpop.set_plots(self.plots)
272        self._slicerpop.set_graph(self.graph)     
[9a585d0]273        # Various plot options
274        id = wx.NewId()
[6d727ae]275        self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
[9a585d0]276        wx.EVT_MENU(self, id, self.onSaveImage)
277        id = wx.NewId()
[6d727ae]278        self._slicerpop.Append(id, '&Print Image', 'Print image ')
[18eba35]279        wx.EVT_MENU(self, id, self.onPrint)
280        id = wx.NewId()
[6d727ae]281        self._slicerpop.Append(id, '&Print Preview', 'Print preview')
[18eba35]282        wx.EVT_MENU(self, id, self.onPrinterPreview)
[52b8b74]283       
[6d727ae]284        id = wx.NewId()
285        self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
286        wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
287       
288        self._slicerpop.AppendSeparator()
289
[52b8b74]290        #add menu of other plugins
[a07e72f]291        item_list = self.parent.get_context_menu(self)
[6d727ae]292
[32c0841]293        if (not item_list == None) and (not len(item_list) == 0):
[9a585d0]294            for item in item_list:
295                try:
296                    id = wx.NewId()
[52b8b74]297                    self._slicerpop.Append(id, item[0], item[1])
[9a585d0]298                    wx.EVT_MENU(self, id, item[2])
299                except:
[32c0841]300                    msg = "ModelPanel1D.onContextMenu: "
[a07e72f]301                    msg += "bad menu item  %s" % sys.exc_value
[32c0841]302                    wx.PostEvent(self.parent, StatusEvent(status=msg))
[9a585d0]303                    pass
[52b8b74]304            self._slicerpop.AppendSeparator()
[6d727ae]305        #id = wx.NewId()
306        #self._slicerpop.Append(id, '&Print image', 'Print image')
[1bf33c1]307        if self.graph.selected_plottable in self.plots:
308            plot = self.plots[self.graph.selected_plottable]
[6d727ae]309
310            id = wx.NewId()
311            self._slicerpop.Append(id, '&Linear Fit')
312            wx.EVT_MENU(self, id, self.onFitting)
313            self._slicerpop.AppendSeparator()
314            id = wx.NewId()
315            name = plot.name
316            self._slicerpop.Append(id, "&Save Points as a File")
317            wx.EVT_MENU(self, id, self._onSave)
318            id = wx.NewId()
319            self._slicerpop.Append(id, "Remove %s Curve" % name)
320            wx.EVT_MENU(self, id, self._onRemove)
[e6a93df]321            if not plot.is_data:
322                id = wx.NewId()
323                self._slicerpop.Append(id, '&Freeze', 'Freeze')
324                wx.EVT_MENU(self, id, self.onFreeze)
[a3c96f7a]325            symbol_menu = wx.Menu()
326            for label in self._symbol_labels:
327                id = wx.NewId()
328                symbol_menu.Append(id, str(label), str(label))
329                wx.EVT_MENU(self, id, self.onChangeSymbol)
330            id = wx.NewId()
331            self._slicerpop.AppendMenu(id,'&Modify Symbol',  symbol_menu)
332            self._slicerpop.AppendSeparator()
[6d727ae]333
[a07e72f]334            id = wx.NewId()
335            self.hide_menu = self._slicerpop.Append(id, "Hide Error")
336            if plot.dy is not None or plot.dy != []:
337                if plot.hide_error :
338                    self.hide_menu.SetText('Show Error')
339                else:
340                    self.hide_menu.SetText('Hide Error')
341            else:
342                self.hide_menu.Disable()
343            wx.EVT_MENU(self, id, self._ontoggle_hide_error)
344           
[52b8b74]345            self._slicerpop.AppendSeparator()
[1bf33c1]346            # Option to hide
347            #TODO: implement functionality to hide a plottable (legend click)
[df22224]348       
[a07e72f]349       
[1bf33c1]350        id = wx.NewId()
[52b8b74]351        self._slicerpop.Append(id, '&Change scale')
[1bf33c1]352        wx.EVT_MENU(self, id, self._onProperties)
353        id = wx.NewId()
[52b8b74]354        self._slicerpop.Append(id, '&Reset Graph')
[d468daa]355        wx.EVT_MENU(self, id, self.onResetGraph) 
[1bf33c1]356        pos = event.GetPosition()
357        pos = self.ScreenToClient(pos)
[52b8b74]358        self.PopupMenu(self._slicerpop, pos)
[a07e72f]359     
[e6a93df]360    def onFreeze(self, event):
361        """
362        """
363        plot = self.plots[self.graph.selected_plottable]
364        self.parent.onfreeze([plot.id])
365       
[a07e72f]366    def onChangeSymbol(self, event):
[6c0568b]367        """
368        """
[a07e72f]369        menu = event.GetEventObject()
370        id = event.GetId()
371        label =  menu.GetLabel(id)
372        selected_plot = self.plots[self.graph.selected_plottable]
373        selected_plot.symbol = self._symbol_labels[label]
374        ## Set the view scale for all plots
375        self._onEVT_FUNC_PROPERTY()
376        ## render the graph
[6d727ae]377        #self.graph.render(self)
378        #self.subplot.figure.canvas.draw_idle()
[c81140c]379       
[42d27f2]380    def _onsaveTXT(self, path):
381        """
[d955bf19]382        Save file as txt
[1abcb04]383           
[d955bf19]384        :TODO: Refactor and remove this method. See TODO in _onSave.
385       
[42d27f2]386        """
387        data = self.plots[self.graph.selected_plottable]
388       
389        if not path == None:
390            out = open(path, 'w')
391            has_errors = True
[32c0841]392            if data.dy == None or data.dy == []:
[42d27f2]393                has_errors = False
394            # Sanity check
395            if has_errors:
396                try:
397                    if len(data.y) != len(data.dy):
398                        has_errors = False
399                except:
400                    has_errors = False
401            if has_errors:
[6d727ae]402                if data.dx != None:
403                    out.write("<X>   <Y>   <dY>   <dX>\n")
404                else:
405                    out.write("<X>   <Y>   <dY>\n")
[42d27f2]406            else:
407                out.write("<X>   <Y>\n")
408               
409            for i in range(len(data.x)):
410                if has_errors:
[6d727ae]411                    if data.dx != None:
412                        out.write("%g  %g  %g  %g\n" % (data.x[i], 
413                                                    data.y[i],
414                                                    data.dy[i],
415                                                    data.dx[i]))
416                    else:
417                        out.write("%g  %g  %g\n" % (data.x[i], 
418                                                    data.y[i],
419                                                    data.dy[i]))
[42d27f2]420                else:
421                    out.write("%g  %g\n" % (data.x[i], 
422                                            data.y[i]))
423            out.close()                 
[6063b16]424            try:
425                self._default_save_location = os.path.dirname(path)
426            except:
427                pass   
[8bd764d]428               
[1bf33c1]429    def _onSave(self, evt):
430        """
[d955bf19]431        Save a data set to a text file
432       
433        :param evt: Menu event
434       
[1bf33c1]435        """
[a07e72f]436       
437        path = None
438        wildcard = "Text files (*.txt)|*.txt|"\
439        "CanSAS 1D files(*.xml)|*.xml" 
440        dlg = wx.FileDialog(self, "Choose a file",
441                            self._default_save_location,
442                             "", wildcard , wx.SAVE)
443       
444        if dlg.ShowModal() == wx.ID_OK:
445            path = dlg.GetPath()
446            mypath = os.path.basename(path)
[1bf33c1]447           
[a07e72f]448            #TODO: This is bad design. The DataLoader is designed
449            #to recognize extensions.
450            # It should be a simple matter of calling the .
451            #save(file, data, '.xml') method
452            # of the DataLoader.loader.Loader class.
453            from DataLoader.loader import  Loader
454            #Instantiate a loader
455            loader = Loader() 
456            data = self.plots[self.graph.selected_plottable]
457            format = ".txt"
458            if os.path.splitext(mypath)[1].lower() == format:
459                 self._onsaveTXT( path)
460            format = ".xml"
461            if os.path.splitext(mypath)[1].lower() == format:
462                loader.save(path, data, format)
463            try:
464                self._default_save_location = os.path.dirname(path)
465            except:
466                pass   
467        dlg.Destroy()
Note: See TracBrowser for help on using the repository browser.