Changeset 9624cda in sasview for sanscalculator/src/sans/calculator/sans_gen.py

Timestamp:

Mar 13, 2013 11:10:51 AM (11 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:

• sanscalculator/src/sans/calculator/sans_gen.py (modified) (6 diffs)

sanscalculator/src/sans/calculator/sans_gen.py

@@ -30 +30 @@
     sld_m = factor * mag
     return sld_m
+
+def transform_center(pos_x, pos_y, pos_z):
+    """
+    re-center 
+    
+    :return: posx, posy, posz   [arrays]
+    """ 
+    posx = pos_x - (min(pos_x) + max(pos_x)) / 2.0
+    posy = pos_y - (min(pos_y) + max(pos_y)) / 2.0
+    posz = pos_z - (min(pos_z) + max(pos_z)) / 2.0  
+    return posx, posy, posz   
 
 class GenSAS(BaseComponent):
@@ -101 +112 @@
         :return: function value
         """
-        len_x = len(self.data_x)
-        if self.is_avg:
+        pos_x = self.data_x
+        pos_y = self.data_y
+        pos_z = self.data_z
+        len_x = len(pos_x)
+        if self.is_avg == None:
             len_x *= -1
+            pos_x, pos_y, pos_z = transform_center(pos_x, pos_y, pos_z)
         len_q = len(x)
         sldn = copy.deepcopy(self.data_sldn)
         sldn -= self.params['solvent_SLD']
-        model = mod.new_GenI(len_x, self.data_x, self.data_y, self.data_z, 
+        model = mod.new_GenI(len_x, pos_x, pos_y, pos_z, 
                              sldn, self.data_mx, self.data_my, 
                              self.data_mz, self.data_vol,
@@ -113 +128 @@
                              self.params['Up_frac_out'], 
                              self.params['Up_theta'])
-
-        mod.genicom(model, len_q, x, y, i)
+        if y == []:
+            mod.genicom(model, len_q, x, i)
+        else:
+            mod.genicomXY(model, len_q, x, y, i)
         vol_correction = self.data_total_volume / self.params['total_volume']
         return  self.params['scale'] * vol_correction * i + \
@@ -151 +168 @@
         """
         if x.__class__.__name__ == 'list':
+            if len(x[1]) > 0:
+                msg = "Not a 1D."
+                raise ValueError, msg
             i_out = numpy.zeros_like(x[0])
-            y_in = numpy.zeros_like(x[0])
+            #import time
+            #s = time.time()
             # 1D I is found at y =0 in the 2D pattern
-            return self._gen(x[0], y_in, i_out )
+            out = self._gen(x[0], [], i_out )
+            #print "i_out_time", time.time() - s
+            return out
         else:
             msg = "Q must be given as list of qx's and qy's"
@@ -168 +191 @@
         if x.__class__.__name__ == 'list':
             i_out = numpy.zeros_like(x[0])
-            import time
-            s = time.time()
+            #import time
+            #s = time.time()
             out = self._gen(x[0], x[1], i_out)
-            print "i_out_time", time.time() - s
+            #print "i_out_time", time.time() - s
             return out
         else:
@@ -189 +212 @@
         """
         if qdist.__class__.__name__ == 'list':
-            out = self.runXY(qdist)
+            if len(qdist[1]) < 1:
+                out = self.run(qdist)
+            else:
+                out = self.runXY(qdist)
             return out
         else: