Changeset f1181977 in sasview for prview/perspectives/pr

Timestamp:

Aug 8, 2008 9:00:56 AM (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:

0eb801a

Parents:

08b9e586

Message:

Can now add multiple additional curves, normalization now affects all added curves, acknowledgment statement added in about box.

File:

• prview/perspectives/pr/pr.py (modified) (9 diffs)

prview/perspectives/pr/pr.py

@@ -12 +12 @@
 from sans.pr.invertor import Invertor
 
-PR_FIT_LABEL       = "P_{fit}(r)"
-PR_LOADED_LABEL    = "P_{loaded}(r)"
-IQ_DATA_LABEL      = "I_{obs}(q)"
-IQ_FIT_LABEL       = "I_{fit}(q)"
-IQ_SMEARED_LABEL   = "I_{smeared}(q)"
+PR_FIT_LABEL       = r"$P_{fit}(r)$"
+PR_LOADED_LABEL    = r"$P_{loaded}(r)$"
+IQ_DATA_LABEL      = r"$I_{obs}(q)$"
+IQ_FIT_LABEL       = r"$I_{fit}(q)$"
+IQ_SMEARED_LABEL   = r"$I_{smeared}(q)$"
 
 import wx.lib
@@ -80 +80 @@
         self._scale_output_unity = False
         
+        ## List of added P(r) plots
+        self._added_plots = {}
+        self._default_Iq  = {}
+        
         # Log startup
         logging.info("Pr(r) plug-in started")
@@ -318 +322 @@
         else:
             new_plot = Data1D(x, y, dy=dy)
-        new_plot.name = "P_{fit}(r)"
+        new_plot.name = PR_FIT_LABEL
         new_plot.xaxis("\\rm{r}", 'A')
         new_plot.yaxis("\\rm{P(r)} ","cm^{-3}")
@@ -516 +520 @@
         self.show_pr(self._last_out, self._last_pr, self._last_cov)
         
+        # Now replot the original added data
+        for plot in self._added_plots:
+            self._added_plots[plot].y = numpy.copy(self._default_Iq[plot])
+            wx.PostEvent(self.parent, NewPlotEvent(plot=self._added_plots[plot], 
+                                                   title=self._added_plots[plot].name,
+                                                   update=True))        
+        
+        # Need the update flag in the NewPlotEvent to protect against
+        # the plot no longer being there...
+        
     def _on_normalize(self, evt):
         """
-            Switch normalization ON/OFF
+            Normalize the area under the P(r) curve to 1.
+            This operation is done for all displayed plots.
+            
             @param evt: Menu event
         """
@@ -526 +542 @@
         self.show_pr(self._last_out, self._last_pr, self._last_cov)
         
+        # Now scale the added plots too
+        for plot in self._added_plots:
+            sum = numpy.sum(self._added_plots[plot].y)
+            npts = len(self._added_plots[plot].x)
+            sum *= self._added_plots[plot].x[npts-1]/npts
+            y = self._added_plots[plot].y/sum
+            
+            new_plot = Theory1D(self._added_plots[plot].x, y)
+            new_plot.name = self._added_plots[plot].name
+            new_plot.xaxis("\\rm{r}", 'A')
+            new_plot.yaxis("\\rm{P(r)} ","cm^{-3}")
+            
+            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, update=True,
+                                                   title=self._added_plots[plot].name))
+                
+        
+        
     def _on_scale_unity(self, evt):
         """
-            Switch normalization ON/OFF
+            Scale the maximum P(r) value on each displayed plot to 1.
+            
             @param evt: Menu event
         """
@@ -535 +569 @@
             
         self.show_pr(self._last_out, self._last_pr, self._last_cov)
+        
+        # Now scale the added plots too
+        for plot in self._added_plots:
+            _max = 0
+            for y in self._added_plots[plot].y:
+                if y>_max: 
+                    _max = y
+            y = self._added_plots[plot].y/_max
+            
+            new_plot = Theory1D(self._added_plots[plot].x, y)
+            new_plot.name = self._added_plots[plot].name
+            new_plot.xaxis("\\rm{r}", 'A')
+            new_plot.yaxis("\\rm{P(r)} ","cm^{-3}")
+            
+            wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, update=True,
+                                                   title=self._added_plots[plot].name))        
+        
         
     def _on_add_data(self, evt):
@@ -547 +598 @@
         x, y, err = self.parent.load_ascii_1D(path)
         
+        filename = os.path.basename(path)
+        
         #new_plot = Data1D(x, y, dy=err)
         new_plot = Theory1D(x, y)
-        new_plot.name = "P_{loaded}(r)"
+        new_plot.name = filename
         new_plot.xaxis("\\rm{r}", 'A')
         new_plot.yaxis("\\rm{P(r)} ","cm^{-3}")
             
-        wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title="P(r) fit"))
+        # Store a ref to the plottable for later use
+        self._added_plots[filename] = new_plot
+        self._default_Iq[filename]  = numpy.copy(y)
+        
+        wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title=filename))
         
         
@@ -653 +710 @@
             # 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.name = IQ_DATA_LABEL
             new_plot.xaxis("\\rm{Q}", 'A^{-1}')
             new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
@@ -689 +746 @@
         else:
             new_plot = Data1D(self.pr.x, self.pr.y, dy=self.pr.err)
-        new_plot.name = "I_{obs}(q)"
+        new_plot.name = IQ_DATA_LABEL
         new_plot.xaxis("\\rm{Q}", 'A^{-1}')
         new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")