source: sasview/guiframe/local_perspectives/plotting/Plotter1D.py @ 1f3655a

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 1f3655a was ffd23b5, checked in by Gervaise Alina <gervyh@…>, 16 years ago
  • Property mode set to 100644
File size: 12.7 KB
RevLine 
[1bf33c1]1"""
2This software was developed by the University of Tennessee as part of the
3Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
4project funded by the US National Science Foundation.
5
6See the license text in license.txt
7
8copyright 2008, University of Tennessee
9"""
10
11
12import wx
13import sys
14import danse.common.plottools
15from danse.common.plottools.PlotPanel import PlotPanel
16from danse.common.plottools.plottables import Graph,Data1D
17from sans.guicomm.events import EVT_NEW_PLOT
18from sans.guicomm.events import StatusEvent ,NewPlotEvent
19
20
21from SlicerParameters import SlicerEvent
22from binder import BindArtist
23(InternalEvent, EVT_INTERNAL)   = wx.lib.newevent.NewEvent()
24#from SlicerParameters import SlicerEvent
25#(InternalEvent, EVT_INTERNAL)   = wx.lib.newevent.NewEvent()
26DEFAULT_QMAX = 0.05
27
28DEFAULT_QSTEP = 0.001
29DEFAULT_BEAM = 0.005
30BIN_WIDTH =1
31import pylab
32
33class PanelMenu(wx.Menu):
34    plots = None
35    graph = None
36   
37    def set_plots(self, plots):
38        self.plots = plots
39   
40    def set_graph(self, graph):
41        self.graph = graph
42class ModelPanel1D(PlotPanel):
43    """
44        Plot panel for use with the GUI manager
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   
57    def __init__(self, parent, id = -1, color = None,\
58        dpi = None, style = wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
59        """
60            Initialize the panel
61        """
62        PlotPanel.__init__(self, parent, id = id, style = style, **kwargs)
63       
64        ## Reference to the parent window
65        self.parent = parent
66        ## Plottables
67        self.plots = {}
68       
69        ## Unique ID (from gui_manager)
70        self.uid = None
71       
72        ## Action IDs for internal call-backs
73        self.action_ids = {}
74       
75        ## Graph       
76        self.graph = Graph()
77        self.graph.xaxis("\\rm{Q}", 'A^{-1}')
78        self.graph.yaxis("\\rm{Intensity} ","cm^{-1}")
79        self.graph.render(self)
80   
81    def _reset(self):
82        """
83            Resets internal data and graph
84        """   
85        self.graph.reset()
86        self.plots      = {}
87        self.action_ids = {}
88   
89    def _onEVT_1DREPLOT(self, event):
90        """
91            Data is ready to be displayed
92            @param event: data event
93        """
[ffd23b5]94       
[1bf33c1]95        #TODO: Check for existence of plot attribute
96
97        # Check whether this is a replot. If we ask for a replot
98        # and the plottable no longer exists, ignore the event.
99        if hasattr(event, "update") and event.update==True \
100            and event.plot.name not in self.plots.keys():
101            return
102       
103        if hasattr(event, "reset"):
104            self._reset()
105       
106        is_new = True
107        if event.plot.name in self.plots.keys():
108            # Check whether the class of plottable changed
109            if not event.plot.__class__==self.plots[event.plot.name].__class__:
[ab8f936]110                #overwrite a plottable using the same name
[1bf33c1]111                self.graph.delete(self.plots[event.plot.name])
112            else:
[ab8f936]113                # plottable is already draw on the panel
[1bf33c1]114                is_new = False
[ffd23b5]115       
116           
[ab8f936]117           
[1bf33c1]118        if is_new:
[ab8f936]119            # a new plottable overwrites a plotted one  using the same id
120            for plottable in self.plots.itervalues():
121                if event.plot.id==plottable.id :
122                    self.graph.delete(plottable)
123           
[1bf33c1]124            self.plots[event.plot.name] = event.plot
125            self.graph.add(self.plots[event.plot.name])
126        else:
[ab8f936]127            #replot the graph
[1bf33c1]128            self.plots[event.plot.name].x = event.plot.x   
129            self.plots[event.plot.name].y = event.plot.y   
130            self.plots[event.plot.name].dy = event.plot.dy 
131            if hasattr(event.plot, 'dx') and hasattr(self.plots[event.plot.name], 'dx'):
132                self.plots[event.plot.name].dx = event.plot.dx   
133 
134       
135        # Check axis labels
136        #TODO: Should re-factor this
137        #if event.plot._xunit != self.graph.prop["xunit"]:
[ffd23b5]138       
[1bf33c1]139        self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
140           
141        #if event.plot._yunit != self.graph.prop["yunit"]:
142        self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
143     
144        # Set the view scale for all plots
[ef0c170]145       
[1bf33c1]146        self._onEVT_FUNC_PROPERTY()
[ef0c170]147   
[1bf33c1]148        self.graph.render(self)
149        self.subplot.figure.canvas.draw_idle()
150
151    def onLeftDown(self,event): 
152        """ left button down and ready to drag"""
153           
154        PlotPanel.onLeftDown(self, event)
155        ax = event.inaxes
156        if ax != None:
157            position = "x: %8.3g    y: %8.3g" % (event.xdata, event.ydata)
158            wx.PostEvent(self.parent, StatusEvent(status=position))
159
160    def _onRemove(self, event):
161        """
162        """
163        if not self.graph.selected_plottable == None:
164            print self.graph.selected_plottable
165           
166           
167            self.graph.delete(self.plots[self.graph.selected_plottable])
168            del self.plots[self.graph.selected_plottable]
169            self.graph.render(self)
170            self.subplot.figure.canvas.draw_idle()   
171           
172
173    def onContextMenu(self, event):
174        """
175            1D plot context menu
176            @param event: wx context event
177        """
178        #slicerpop = wx.Menu()
179        slicerpop = PanelMenu()
180        slicerpop.set_plots(self.plots)
181        slicerpop.set_graph(self.graph)
182               
183        # Option to save the data displayed
184       
185        #for plot in self.graph.plottables:
186        if self.graph.selected_plottable in self.plots:
187            plot = self.plots[self.graph.selected_plottable]
188            id = wx.NewId()
189            name = plot.name
190            slicerpop.Append(id, "&Save %s points" % name)
191            self.action_ids[str(id)] = plot
192            wx.EVT_MENU(self, id, self._onSave)
193               
194            # Option to delete plottable
195            id = wx.NewId()
196            slicerpop.Append(id, "Remove %s curve" % name)
197            self.action_ids[str(id)] = plot
198            wx.EVT_MENU(self, id, self._onRemove)
199           
200            # Option to hide
201            #TODO: implement functionality to hide a plottable (legend click)
202            slicerpop.AppendSeparator()
203               
204        # Various plot options
205        id = wx.NewId()
206        slicerpop.Append(id,'&Save image', 'Save image as PNG')
207        wx.EVT_MENU(self, id, self.onSaveImage)
208       
209       
210        item_list = self.parent.get_context_menu(self.graph)
211        if (not item_list==None) and (not len(item_list)==0):
212                slicerpop.AppendSeparator()
213                for item in item_list:
214                    try:
215                        id = wx.NewId()
216                        slicerpop.Append(id, item[0], item[1])
217                        wx.EVT_MENU(self, id, item[2])
218                    except:
219                        print sys.exc_value
220                        print RuntimeError, "View1DPanel.onContextMenu: bad menu item"
221       
222        slicerpop.AppendSeparator()
223       
224        if self.graph.selected_plottable in self.plots:
225            if self.plots[self.graph.selected_plottable].__class__.__name__=="Theory1D":
226                id = wx.NewId()
227                slicerpop.Append(id, '&Add errors to data')
[ffd23b5]228                #print "panel scale before  ",self.xLabel, self.yLabel
229                #print "cyllinder before adding error", self.plots[self.graph.selected_plottable].x
[1bf33c1]230                wx.EVT_MENU(self, id, self._on_add_errors)
231            else:
232                id = wx.NewId()
233                slicerpop.Append(id, '&Linear fit')
234                wx.EVT_MENU(self, id, self.onFitting)
235               
236       
237
238        id = wx.NewId()
239        slicerpop.Append(id, '&Change scale')
240        wx.EVT_MENU(self, id, self._onProperties)
241       
242        id = wx.NewId()
243        #slicerpop.AppendSeparator()
244        slicerpop.Append(id, '&Reset Graph')
245        wx.EVT_MENU(self, id, self.onResetGraph)       
246
247        pos = event.GetPosition()
248        pos = self.ScreenToClient(pos)
249        self.PopupMenu(slicerpop, pos)
250   
251   
252    def _on_add_errors(self, evt):
253        """
254            Compute reasonable errors for a data set without
255            errors and transorm the plottable to a Data1D
256        """
257        import math
258        import numpy
259        import time
260       
261        if not self.graph.selected_plottable == None:
262            length = len(self.plots[self.graph.selected_plottable].x)
263            dy = numpy.zeros(length)
264            for i in range(length):
265                dy[i] = math.sqrt(self.plots[self.graph.selected_plottable].y[i])
266               
267            new_plot = Data1D(self.plots[self.graph.selected_plottable].x,
268                              self.plots[self.graph.selected_plottable].y,
269                              dy=dy)
270            new_plot.interactive = True
271            new_plot.name = self.plots[self.graph.selected_plottable].name
272            if hasattr(self.plots[self.graph.selected_plottable], "group_id"):
273                new_plot.group_id = self.plots[self.graph.selected_plottable].group_id
[d1dd9d4]274                new_plot.id = self.plots[self.graph.selected_plottable].id
[1bf33c1]275            else:
276                new_plot.group_id = str(time.time())
[d1dd9d4]277                new_plot.id = str(time.time())
[1bf33c1]278           
279            label, unit = self.plots[self.graph.selected_plottable].get_xaxis()
280            new_plot.xaxis(label, unit)
281            label, unit = self.plots[self.graph.selected_plottable].get_yaxis()
282            new_plot.yaxis(label, unit)
[ffd23b5]283            #print "panel scale ",self.xLabel, self.yLabel
[1bf33c1]284            self.graph.delete(self.plots[self.graph.selected_plottable])
285           
286            self.graph.add(new_plot)
[ffd23b5]287            # transforming the view of the new data into the same of the previous data
288            self._onEVT_FUNC_PROPERTY()
289            #print "cyllinder", self.plots[self.graph.selected_plottable].x,self.plots[self.graph.selected_plottable].view.x, new_plot.x, new_plot.view.x
[1bf33c1]290            self.plots[self.graph.selected_plottable]=new_plot
291           
292            self.graph.render(self)
293            self.subplot.figure.canvas.draw_idle()   
294   
295    def _onSave(self, evt):
296        """
297            Save a data set to a text file
298            @param evt: Menu event
299        """
300        import os
301        id = str(evt.GetId())
302        if id in self.action_ids:         
303           
304            path = None
305            dlg = wx.FileDialog(self, "Choose a file", os.getcwd(), "", "*.txt", wx.SAVE)
306            if dlg.ShowModal() == wx.ID_OK:
307                path = dlg.GetPath()
308                mypath = os.path.basename(path)
309                print path
310            dlg.Destroy()
311           
312            if not path == None:
313                out = open(path, 'w')
314                has_errors = True
315                if self.action_ids[id].dy==None or self.action_ids[id].dy==[]:
316                    has_errors = False
317                   
318                # Sanity check
319                if has_errors:
320                    try:
321                        if len(self.action_ids[id].y) != len(self.action_ids[id].dy):
322                            print "Y and dY have different lengths"
323                            has_errors = False
324                    except:
325                        has_errors = False
326               
327                if has_errors:
328                    out.write("<X>   <Y>   <dY>\n")
329                else:
330                    out.write("<X>   <Y>\n")
331                   
332                for i in range(len(self.action_ids[id].x)):
333                    if has_errors:
334                        out.write("%g  %g  %g\n" % (self.action_ids[id].x[i], 
335                                                    self.action_ids[id].y[i],
336                                                    self.action_ids[id].dy[i]))
337                    else:
338                        out.write("%g  %g\n" % (self.action_ids[id].x[i], 
339                                                self.action_ids[id].y[i]))
340                       
341                out.close()
342   
343   
344    def _onToggleScale(self, event):
345        if self.get_yscale() == 'log':
346            self.set_yscale('linear')
347        else:
348            self.set_yscale('log')
349        self.subplot.figure.canvas.draw_idle()   
350       
Note: See TracBrowser for help on using the repository browser.