Changeset 5d2e4c1 in sasview for sansmodels/src/sans/models

Timestamp:

Aug 26, 2011 3:15:04 PM (13 years ago)

Author:

Gervaise Alina <gervyh@…>

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:

6e4c9fe

Parents:

32cd80c

Message:

make sure smear is pickelable

File:

• sansmodels/src/sans/models/qsmearing.py (modified) (11 diffs)

sansmodels/src/sans/models/qsmearing.py

@@ -11 +11 @@
 import logging
 import sys
-import sans.models.sans_extension.smearer as smearer 
+import copy
+import sans_extension.smearer as smearer 
 from sans.models.smearing_2d import Smearer2D
 
@@ -94 +95 @@
         self._smearer = None
         self.model = None
-        
-    def __deepcopy__(self, memo={}):
-        """
-        Return a valid copy of self.
-        Avoid copying the _smearer C object and force a matrix recompute
-        when the copy is used.  
-        """
-        result = _BaseSmearer()
-        result.nbins = self.nbins
-        return result
-
+    
     def _compute_matrix(self):
         """
@@ -125 +116 @@
         if q_max == None:
             q_max = self.max
-
         _qmin_unsmeared, _qmax_unsmeared = self.get_unsmeared_range(q_min,
                                                                      q_max)
@@ -172 +162 @@
         iq_in_temp = iq_in
         if self.model != None:
-            temp_first, temp_last = self._get_extrapolated_bin( \
-                                                        first_bin, last_bin)
+            temp_first, temp_last = self._get_extrapolated_bin(first_bin, 
+                                                               last_bin)
             if self.nbins_low > 0:
-                iq_in_low = self.model.evalDistribution( \
+                iq_in_low = self.model.evalDistribution(
                                     numpy.fabs(self.qvalues[0:self.nbins_low]))
-            iq_in_high = self.model.evalDistribution( \
+            iq_in_high = self.model.evalDistribution(
                                             self.qvalues[(len(self.qvalues) - \
                                             self.nbins_high - 1):])
@@ -191 +181 @@
             temp_first = first_bin
             temp_last = last_bin
-            #iq_in_temp = iq_in
-
         # Sanity check
         if len(iq_in_temp) != self.nbins:
@@ -300 +288 @@
         self.qvalues  = None
         
+   
     def _initialize_smearer(self):
         """
@@ -343 +332 @@
         self.nbins_low = 0
         self.nbins_high = 0
+        self.data = data1D
         self.model = model
         if data1D.dxw is not None and len(data1D.dxw) == len(data1D.x):
@@ -387 +377 @@
         self.qvalues = data1d_x
         
+    def __deepcopy__(self, memo={}):
+        """
+        Return a valid copy of self.
+        Avoid copying the _smearer C object and force a matrix recompute
+        when the copy is used.  
+        """
+        result = SlitSmearer(self.data, self.model)
+        result.width = self.width
+        result.height = self.height
+        result.nbins_low = self.nbins_low
+        result.nbins_high = self.nbins_high
+        result.data = copy.deepcopy(self.data)
+        result.model = copy.deepcopy(self.model)
+        result.nbins = self.nbins
+        result.nbins_low = self.nbins_low
+        result.nbins_high = self.nbins_high
+        result._weights = self._weights
+        ## Internal flag to keep track of C++ smearer initialization
+        result._init_complete = self._init_complete
+        import sans_extension.smearer as smearer 
+        result._smearer =  smearer.new_slit_smearer_with_q(self.width, 
+                                                    self.height, self.qvalues)
+        return result
+
+        
+    def __setstate__(self, state):
+        """
+        """
+        self.__dict__, self.model,  self.data = state
+        import sans_extension.smearer as smearer 
+        self._smearer = smearer.new_slit_smearer_with_q(self.width, 
+                                                    self.height, self.qvalues)
+        
+    def __reduce_ex__(self, proto):
+        """
+        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of 
+        c model.
+        """
+        model = copy.deepcopy(self.model)
+        data = copy.deepcopy(self.data)
+        dict = self.__dict__
+        if "_smearer" in dict.keys():
+            del dict["_smearer"]
+        state = (dict, model, data)
+        return (QSmearer, (data, model), state, None, None)
+        
 
 class _QSmearer(_BaseSmearer):
@@ -457 +493 @@
         super(QSmearer, self).__init__()
         data1d_x = []
+        self.data =  copy.deepcopy(data1D)
         self.nbins_low = 0
         self.nbins_high = 0
-        self.model = model
+        self.model = copy.deepcopy(model)
         ## Resolution
         #self.width = numpy.zeros(len(data1D.x))
@@ -470 +507 @@
             self.nbins_low, self.nbins_high, self.width, data1d_x = \
                                 get_qextrapolate(self.width, data1D.x)
-
         ## Number of Q bins
         self.nbins = len(data1d_x)
@@ -479 +515 @@
         ## Q-values
         self.qvalues = data1d_x
+        
+    def __deepcopy__(self, memo={}):
+        """
+        Return a valid copy of self.
+        Avoid copying the _smearer C object and force a matrix recompute
+        when the copy is used.  
+        """
+        result = QSmearer(self.data, self.model)
+        result.nbins = self.nbins
+        result.min = self.min
+        result.max = self.max
+        result.nbins_low = self.nbins_low
+        result.nbins_high = self.nbins_high
+        result.width = copy.deepcopy(self.width)
+        result._weights = copy.deepcopy(self._weights)
+        result.qvalues = copy.deepcopy(self.qvalues)
+        ## Internal flag to keep track of C++ smearer initialization
+        result._init_complete = self._init_complete
+        import sans_extension.smearer as smearer 
+        result._smearer =  smearer.new_q_smearer_with_q(numpy.asarray(result.width),
+                                                      result.qvalues)
+        return result
+
+        
+    def __setstate__(self, state):
+        """
+        """
+        self.__dict__, self.model,  self.data = state
+        import sans_extension.smearer as smearer 
+        self._smearer =  smearer.new_q_smearer_with_q(numpy.asarray(self.width),
+                                                      self.qvalues)
+ 
+    def __reduce_ex__(self, proto):
+        """
+        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of 
+        c model.
+        """
+        model = copy.deepcopy(self.model)
+        data = copy.deepcopy(self.data)
+        dict = self.__dict__
+        if "_smearer" in dict.keys():
+            del dict["_smearer"]
+        state = (dict, model, data)
+        return (QSmearer, (data, model), state, None, None)
+