Changeset 9624cda in sasview for calculatorview/src/sans

Timestamp:

Mar 13, 2013 11:10:51 AM (12 years ago)

Author:

Jae Cho <jhjcho@…>

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:

49a8843

Parents:

5f248f6

Message:

fixed debye averging in gensans

File:

• calculatorview/src/sans/perspectives/calculator/gen_scatter_panel.py (modified) (19 diffs)

calculatorview/src/sans/perspectives/calculator/gen_scatter_panel.py

@@ -21 +21 @@
 from data_util.calcthread import CalcThread
 from sans.guiframe.local_perspectives.plotting.SimplePlot import PlotFrame
-from sans.guiframe.dataFitting import Data2D
+from sans.guiframe.dataFitting import Data2D 
+from sans.guiframe.dataFitting import Data1D
 from sans.dataloader.data_info import Detector
 from sans.dataloader.data_info import Source
@@ -46 +47 @@
 _QMAX_DEFAULT = 0.3
 _NPTS_DEFAULT = 50 
+_Q1D_MIN = 0.001
 
 def add_icon(parent, frame):
@@ -84 +86 @@
                  completefn = None,
                  updatefn   = None,
-                 elapsed = 0,
-                 yieldtime  = 0.005,
+                 #elapsed = 0,
+                 yieldtime  = 0.01,
                  worktime   = 0.01):
         """
@@ -96 +98 @@
         self.id = id 
         self.input = input 
+        self.update_fn = updatefn
         
     def compute(self):
@@ -101 +104 @@
         excuting computation
         """
-        elapsed = time.time() - self.starttime
-       
-        self.complete(input=self.input,
-                      elapsed = elapsed)
+        #elapsed = time.time() - self.starttime
+        self.starttime = time.time()
+        self.complete(input=self.input, update=self.update_fn)
             
 class SasGenPanel(ScrolledPanel, PanelBase):
@@ -140 +142 @@
         self.data = None
         self.scale2d = None
+        self.is_avg = False
         self.plot_frame = None
         self.qmax_x = _QMAX_DEFAULT
@@ -299 +302 @@
                                       style=wx.CB_READONLY) 
         orient_combo.Append('Fixed orientation')
-        orient_combo.Append('No orientation ')
+        orient_combo.Append('Debye full avg.')
+        #orient_combo.Append('Debye sph. sym.')
         
         orient_combo.Bind(wx.EVT_COMBOBOX, self._on_orient_select)
@@ -311 +315 @@
         event.Skip()
         cb = event.GetEventObject()
-        is_avg = cb.GetCurrentSelection() == 1
-        self.model.set_is_avg(is_avg)         
+        if cb.GetCurrentSelection() == 2:
+            self.is_avg = None
+        else:
+            is_avg = cb.GetCurrentSelection() == 1
+            self.is_avg = is_avg
+        self.model.set_is_avg(self.is_avg)
+        self.set_est_time()    
            
     def _layout_qrange(self):
@@ -360 +369 @@
             Do the layout for the button widgets
         """ 
-        self.est_time = '*Estimated Computation time : %s sec'
-        self.time_text = wx.StaticText(self, -1, self.est_time% str(2) )
+        self.est_time = '*Estimated Computation time : %s'
+        self.time_text = wx.StaticText(self, -1, self.est_time% str('2 sec') )
         self.orient_combo = self._fill_orient_combo()
         self.orient_combo.Show(False)
@@ -368 +377 @@
         self.bt_compute.SetToolTipString("Compute 2D Scattering Pattern.")
         self.button_sizer.AddMany([(self.time_text , 0, wx.LEFT, 20),
-                                   (self.orient_combo , 0, wx.LEFT, 20),
+                                   (self.orient_combo , 0, wx.LEFT, 40),
                                    (self.bt_compute, 0, wx.LEFT, 20)])
         
@@ -378 +387 @@
         n_qbins *= n_qbins
         n_pixs = float(self.parent.get_npix())
+        if self.is_avg:
+            n_pixs *= (n_pixs / 200)
         x_in = n_qbins * n_pixs / 100000
         # magic equation: not very accurate
@@ -387 +398 @@
         Set text for est. computation time
         """
+        unit = 'sec'
         if self.time_text != None:
+            self.time_text.SetForegroundColour('black')
             etime = self.estimate_ctime()
-            self.time_text.SetLabel(self.est_time% str(etime))
+            if etime > 60:
+                etime /= 60
+                unit = 'min'
+                self.time_text.SetForegroundColour('red')
+            time_str = str(etime) + ' ' + unit
+            self.time_text.SetLabel(self.est_time% time_str)
         
     def _do_layout(self):
@@ -541 +559 @@
             self.file_name = filename.split('.')[0]
             self.orient_combo.SetSelection(0)
+            self.is_avg = False
             if self.ext in self.omfreader.ext:
                 gen = sans_gen.OMF2SLD()
@@ -577 +596 @@
         Set sld data helper
         """
-        is_avg = self.orient_combo.GetCurrentSelection() == 1
-        self.model.set_is_avg(is_avg)
+        #is_avg = self.orient_combo.GetCurrentSelection() == 1
+        self.model.set_is_avg(self.is_avg)
         self.model.set_sld_data(self.sld_data)
         
@@ -741 +760 @@
             graph_title += " - Magnetic Vector as Arrow -" 
             panel = self.plot_frame.plotpanel
-            def _draw_arrow(input=None, elapsed=0.1):
+            def _draw_arrow(input=None, update=None):
                 """
                 draw magnetic vectors w/arrow
@@ -828 +847 @@
             self.model.set_sld_data(self.sld_data)
             self.set_input_params()
-            self._create_default_2d_data()
-            i_out = numpy.zeros(len(self.data.data))
+            if self.is_avg or self.is_avg == None:
+                self._create_default_1d_data()
+                i_out = numpy.zeros(len(self.data.y))
+                inputs = [self.data.x, [], i_out]
+            else:
+                self._create_default_2d_data()
+                i_out = numpy.zeros(len(self.data.data))
+                inputs=[self.data.qx_data, self.data.qy_data, i_out]
+                
             msg = "Computation is in progress..."
             status_type = 'progress'
             self._status_info(msg, status_type)
-            
-            cal_out = CalcGen(input=[self.data.qx_data, 
-                                     self.data.qy_data, i_out], 
+            cal_out = CalcGen(input=inputs, 
                               completefn=self.complete, 
                               updatefn=self._update)
-            cal_out.queue()
+            cal_out.queue()              
             
         except:
@@ -900 +924 @@
     def _update(self, time=None):
         """
-        Update the output of plotting model
-        """
-        if self.parent.parent != None:
-            msg = "Computation/drawing is in progress..."
-            wx.PostEvent(self.parent.parent, 
-                         StatusEvent(status=msg, type="update"))
+        Update the progress bar
+        """
+        if self.parent.parent == None:
+            return
+        type = "progress"
+        msg = "Please wait. Computing... (Note: Window may look frozen.)"
+        wx.PostEvent(self.parent.parent, StatusEvent(status=msg,
+                                                  type=type))
                                 
-    def complete(self, input, elapsed=None):   
+    def complete(self, input, update=None):   
         """
         Gen compute complete function
         :Param input: input list [qx_data, qy_data, i_out]
         """
-        out = self.model.runXY(input)
-        self._draw2D(out)
+        out = numpy.empty(0)
+        #s = time.time()
+        for ind in range(len(input[0])):
+            if self.is_avg:
+                if ind % 1 == 0 and update != None:
+                    update()
+                    time.sleep(0.1)
+                inputi = [input[0][ind:ind+1], [], input[2][ind:ind+1]]
+                outi = self.model.run(inputi)
+                out = numpy.append(out, outi)
+            else:
+                if ind % 50 == 0  and update != None:
+                    update()
+                    time.sleep(0.001)
+                inputi = [input[0][ind:ind+1], input[1][ind:ind+1], 
+                          input[2][ind:ind+1]]
+                outi = self.model.runXY(inputi)
+                out = numpy.append(out, outi)
+        #print time.time() - s
+        if self.is_avg or self.is_avg == None:
+            self._draw1D(out)
+        else:
+            #out = self.model.runXY(input)
+            self._draw2D(out)
+            
         msg = "Gen computation completed.\n"
         status_type = 'stop'
         self._status_info(msg, status_type)
-        
+               
     def _create_default_2d_data(self):
         """
@@ -977 +1026 @@
         self.data.ymin = ymin
         self.data.ymax = ymax
-        
+
+    def _create_default_1d_data(self):
+        """
+        Create 2D data by default
+        Only when the page is on theory mode.
+        :warning: This data is never plotted.
+                    residuals.x = data_copy.x[index]
+            residuals.dy = numpy.ones(len(residuals.y))
+            residuals.dx = None
+            residuals.dxl = None
+            residuals.dxw = None
+        """
+        self.qmax_x = float(self.qmax_ctl.GetValue())
+        self.npts_x = int(float(self.npt_ctl.GetValue()))
+        qmax = self.qmax_x #/ numpy.sqrt(2)
+        ## Default values
+        xmax = qmax
+        xmin = qmax * _Q1D_MIN
+        qstep = self.npts_x
+        x = numpy.linspace(start=xmin, stop=xmax, num=qstep, endpoint=True)
+        # store x and y bin centers in q space
+        #self.data.source = Source()
+        y = numpy.ones(len(x))
+        dy = numpy.zeros(len(x))
+        dx = numpy.zeros(len(x))
+        self.data = Data1D(x=x, y=y)
+        self.data.dx = dx
+        self.data.dy = dy
+
+    def _draw1D(self, y_out):
+        """
+        Complete get the result of modelthread and create model 2D
+        that can be plot.
+        """
+        page_id = self.id
+        data = self.data
+        
+        model = self.model
+        state = None
+        
+        new_plot = Data1D(x=data.x, y=y_out)
+        new_plot.dx = data.dx
+        new_plot.dy = data.dy
+        new_plot.xaxis('\\rm{Q_{x}}', '\AA^{-1}')
+        new_plot.yaxis('\\rm{Intensity}', 'cm^{-1}')
+        new_plot.is_data = False
+        new_plot.id = str(self.uid) + " GenData1D"
+        new_plot.group_id = str(self.uid) + " Model1D"
+        new_plot.name = model.name + '1d'
+        new_plot.title = "Generic model1D "
+        new_plot.id = str(page_id) + ': ' + self.file_name \
+                        + ' #%s'% str(self.graph_num) + "_1D"
+        new_plot.group_id = str(page_id) + " Model1D"  +\
+                             ' #%s'% str(self.graph_num) + "_1D"
+        new_plot.is_data = False
+
+        title = new_plot.title
+        _yaxis, _yunit = new_plot.get_yaxis()
+        _xaxis, _xunit = new_plot.get_xaxis()
+        new_plot.xaxis(str(_xaxis), str(_xunit))
+        new_plot.yaxis(str(_yaxis), str(_yunit))
+        
+        if new_plot.is_data:
+            data_name = str(new_plot.name)
+        else:
+            data_name = str(model.__class__.__name__) + '1d'
+
+        if len(title) > 1:
+            new_plot.title = "Gen Theory for %s "% model.name + data_name
+        new_plot.name = new_plot.id
+        new_plot.label = new_plot.id
+        #theory_data = deepcopy(new_plot)
+        if self.parent.parent != None:
+            self.parent.parent.update_theory(data_id=new_plot.id,
+                                           theory=new_plot,
+                                           state=state)
+        title = new_plot.title
+        num_graph = str(self.graph_num)
+        wx.CallAfter(self.parent.draw_graph, new_plot, 
+                     title="Graph %s: "% num_graph + new_plot.id )
+        self.graph_num += 1
+                
     def _draw2D(self, image):
         """
@@ -995 +1125 @@
         new_plot.title = "Generic model 2D "
         new_plot.id = str(page_id) + ': ' + self.file_name \
-                        + ' #%s'% str(self.graph_num)
-        new_plot.group_id = str(page_id) + " Model2D"
+                        + ' #%s'% str(self.graph_num) + "_2D"
+        new_plot.group_id = str(page_id) + " Model2D" \
+                        + ' #%s'% str(self.graph_num) + "_2D"
         new_plot.detector = data.detector
         new_plot.source = data.source