Changeset aa4b8379 in sasview for prview/perspectives/pr

Timestamp:

Jun 18, 2008 6:10:06 PM (16 years ago)

Author:

Mathieu Doucet <doucetm@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

3fd1ebc

Parents:

4972de2

Message:

Updated for interactive graphs. Improved for standalone use.

Location:

prview/perspectives/pr

Files:

• inversion_panel.py (modified) (8 diffs)
• pr.py (modified) (11 diffs)

prview/perspectives/pr/inversion_panel.py

@@ -115 +115 @@
     pos_err  = 1.0
     
-    def __init__(self, parent, id = -1, plots = None, **kwargs):
+    def __init__(self, parent, id = -1, plots = None, standalone=False, **kwargs):
         wx.Panel.__init__(self, parent, id = id, **kwargs)
         
@@ -147 +147 @@
         ## Data manager
         self.manager   = None
+        
+        ## Standalone flage
+        self.standalone = standalone
         
         self._do_layout()
@@ -181 +184 @@
             #self.label_sugg.Show()
         elif name=='plotname':
-            self.plot_data.SetValue(str(value))
-            self.plot_radio.SetValue(True)
+            if self.standalone==False:
+                self.plot_data.SetValue(str(value))
+                self.plot_radio.SetValue(True)
+                self._on_pars_changed(None)
+        elif name=='datafile':
+            self.data_file.SetValue(str(value))
+            self.file_radio.SetValue(True)
             self._on_pars_changed(None)
         else:
@@ -214 +222 @@
             self.alpha_estimate_ctl.GetValue()
         elif name=='plotname':
-            self.plot_data.GetValue()
+            if self.standalone==False:
+                self.plot_data.GetValue()
+        elif name=='datafile':
+            self.data_file.GetValue()
         else:
             wx.Panel.__getattr__(self, name)
@@ -230 +241 @@
 
         # ----- I(q) data -----
-        databox = wx.StaticBox(self, -1, "I(q) data")
+        databox = wx.StaticBox(self, -1, "I(q) data source")
         
         boxsizer1 = wx.StaticBoxSizer(databox, wx.VERTICAL)
@@ -248 +259 @@
         pars_sizer.Add(choose_button, (iy,3), (1,1), wx.RIGHT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
         
-        iy += 1
-        self.plot_radio = wx.RadioButton(self, -1, "Plot data:")
-        self.plot_data = wx.TextCtrl(self, -1, size=(100,20))
-        self.plot_data.SetEditable(False)
-        pars_sizer.Add(self.plot_radio, (iy,0), (1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
-        pars_sizer.Add(self.plot_data, (iy,1), (1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 15)
+        if self.standalone==False:
+            iy += 1
+            self.plot_radio = wx.RadioButton(self, -1, "Plot data:")
+            self.plot_data = wx.TextCtrl(self, -1, size=(100,20))
+            self.plot_data.SetEditable(False)
+            pars_sizer.Add(self.plot_radio, (iy,0), (1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
+            pars_sizer.Add(self.plot_data, (iy,1), (1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 15)
         
         boxsizer1.Add(pars_sizer, 0, wx.EXPAND)
@@ -476 +488 @@
         # If the pars are valid, estimate alpha
         if flag:
-            if self.plot_radio.GetValue():
+            if self.standalone==False and self.plot_radio.GetValue():
                 dataset = self.plot_data.GetValue()
                 self.manager.estimate_plot_inversion(alpha=alpha, nfunc=nfunc, 
@@ -571 +583 @@
         
         if flag:
-            if self.plot_radio.GetValue():
+            if self.standalone==False and self.plot_radio.GetValue():
                 dataset = self.plot_data.GetValue()
-                self.manager.setup_plot_inversion(alpha=alpha, nfunc=nfunc, 
-                                                  d_max=dmax,
-                                                  q_min=qmin, q_max=qmax)
+                if len(dataset.strip())==0:
+                    message = "No data to invert. Select a data set before proceeding with P(r) inversion."
+                    wx.PostEvent(self.manager.parent, StatusEvent(status=message))
+                else:
+                    self.manager.setup_plot_inversion(alpha=alpha, nfunc=nfunc, 
+                                                      d_max=dmax,
+                                                      q_min=qmin, q_max=qmax)
             else:
                 path = self.data_file.GetValue()
-                self.manager.setup_file_inversion(alpha=alpha, nfunc=nfunc, 
-                                                  d_max=dmax, path=path,
-                                                  q_min=qmin, q_max=qmax
-                                                  )
+                if len(path.strip())==0:
+                    message = "No data to invert. Select a data set before proceeding with P(r) inversion."
+                    wx.PostEvent(self.manager.parent, StatusEvent(status=message))
+                else:
+                    self.manager.setup_file_inversion(alpha=alpha, nfunc=nfunc, 
+                                                      d_max=dmax, path=path,
+                                                      q_min=qmin, q_max=qmax
+                                                      )
                 
         else:

prview/perspectives/pr/pr.py

@@ -6 +6 @@
 import sys
 import wx
+import logging
 from sans.guitools.plottables import Data1D, Theory1D
 from sans.guicomm.events import NewPlotEvent, StatusEvent    
 import math, numpy
 from sans.pr.invertor import Invertor
+
+PR_FIT_LABEL    = "P_{fit}(r)"
+PR_LOADED_LABEL = "P_{loaded}(r)"
+IQ_DATA_LABEL   = "I_{obs}(q)"
+
+import wx.lib
+(NewPrFileEvent, EVT_PR_FILE) = wx.lib.newevent.NewEvent()
+
 
 class Plugin:
@@ -41 +50 @@
         # Start with a good default
         self.elapsed = 0.022
+        self.iq_data_shown = False
         
         ## Current invertor
@@ -54 +64 @@
         ## Number of P(r) points to display on the output plot
         self._pr_npts = 51
+        ## Flag to let the plug-in know that it is running standalone
+        self.standalone = True
+        
+        # Log startup
+        logging.info("Pr(r) plug-in started")
+        
+        
 
     def populate_menu(self, id, owner):
@@ -332 +349 @@
         """
         # Look whether this Graph contains P(r) data
+        #if graph.selected_plottable==IQ_DATA_LABEL:
         for item in graph.plottables:
-            if item.name=="P_{fit}(r)":
-                
-                return [["Compute P(r)", "Compute P(r) from distribution", self._on_context_inversion],
-                       ["Add P(r) data", "Load a data file and display it on this plot", self._on_add_data],
+            if item.name==PR_FIT_LABEL:
+                return [["Add P(r) data", "Load a data file and display it on this plot", self._on_add_data],
                        ["Change number of P(r) points", "Change the number of points on the P(r) output", self._on_pr_npts]]
-                
-        return [["Compute P(r)", "Compute P(r) from distribution", self._on_context_inversion]]
+
+            elif item.name==graph.selected_plottable:
+                return [["Compute P(r)", "Compute P(r) from distribution", self._on_context_inversion]]      
+                
+        return []
 
     def _on_add_data(self, evt):
@@ -429 +448 @@
         
         # Make a plot of I(q) data
-        new_plot = Data1D(self.pr.x, self.pr.y, dy=self.pr.err)
-        new_plot.name = "I_{obs}(q)"
-        new_plot.xaxis("\\rm{Q}", 'A^{-1}')
-        new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
-        #new_plot.group_id = "test group"
-        wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title="Iq"))
+        if False:
+            new_plot = Data1D(self.pr.x, self.pr.y, dy=self.pr.err)
+            new_plot.name = "I_{obs}(q)"
+            new_plot.xaxis("\\rm{Q}", 'A^{-1}')
+            new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
+            #new_plot.group_id = "test group"
+            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title="Iq"))
                 
         # Show I(q) fit
@@ -454 +474 @@
         new_plot.xaxis("\\rm{Q}", 'A^{-1}')
         new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
+        new_plot.interactive = True
         #new_plot.group_id = "test group"
         wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title="Iq"))
         
+        # Get Q range
+        self.control_panel.q_min = self.pr.x.min()
+        self.control_panel.q_max = self.pr.x.max()
+            
+
         
     def setup_plot_inversion(self, alpha, nfunc, d_max, q_min=None, q_max=None):
@@ -510 +536 @@
             
         self.pr = pr
+        self.iq_data_shown = True
 
           
@@ -624 +651 @@
         
         # If we have more than one displayed plot, make the user choose
-        if len(panel.plots)>1:
-            dialog = InversionDlg(None, -1, "P(r) Inversion", panel.plots, pars=False)
-            dialog.set_content(self.last_data, self.nfunc, self.alpha, self.max_length)
-            if dialog.ShowModal() == wx.ID_OK:
-                dataset = dialog.get_content()
-                dialog.Destroy()
-            else:
-                dialog.Destroy()
-                return
+        if len(panel.plots)>1 and panel.graph.selected_plottable in panel.plots:
+            dataset = panel.graph.selected_plottable
+            if False:
+                dialog = InversionDlg(None, -1, "P(r) Inversion", panel.plots, pars=False)
+                dialog.set_content(self.last_data, self.nfunc, self.alpha, self.max_length)
+                if dialog.ShowModal() == wx.ID_OK:
+                    dataset = dialog.get_content()
+                    dialog.Destroy()
+                else:
+                    dialog.Destroy()
+                    return
         elif len(panel.plots)==1:
             dataset = panel.plots.keys()[0]
@@ -646 +675 @@
         self.control_panel.alpha = self.alpha
         self.parent.set_perspective(self.perspective)
+        self.control_panel._on_invert(None)
             
     def get_panels(self, parent):
@@ -654 +684 @@
         
         self.parent = parent
-        self.control_panel = InversionControl(self.parent, -1, style=wx.RAISED_BORDER)
+        self.control_panel = InversionControl(self.parent, -1, 
+                                              style=wx.RAISED_BORDER,
+                                              standalone=self.standalone)
         self.control_panel.set_manager(self)
         self.control_panel.nfunc = self.nfunc
@@ -662 +694 @@
         self.perspective = []
         self.perspective.append(self.control_panel.window_name)
+        
+        self.parent.Bind(EVT_PR_FILE, self._on_new_file)
+        
         return [self.control_panel]
+    
+    def _on_new_file(self, evt):
+        """
+            Called when the application manager posted an
+            EVT_PR_FILE event. Just prompt the control
+            panel to load a new data file.
+        """
+        self.control_panel._change_file(None)
     
     def get_perspective(self):