Changeset 84f2962 in sasmodels


Ignore:
Timestamp:
Sep 8, 2018 5:27:56 AM (2 months ago)
Author:
Torin Cooper-Bennun <torin.cooper-bennun@…>
Branches:
beta_approx, py3, ticket-1015-gpu-mem-error, ticket-1157, ticket-608-user-defined-weights, ticket_1156
Children:
c88f983
Parents:
c0131d4
Message:

calculate_Iq returns intermediate results from ProductKernel?

File:
1 edited

Legend:

Unmodified
Added
Removed
  • sasmodels/sasview_model.py

    r117c02a r84f2962  
    381381                continue 
    382382            self.params[p.name] = p.default 
    383             if p.limits is not None: 
    384                 self.details[p.id] = [p.units if not p.choices else p.choices, p.limits[0], p.limits[1]] 
    385             else: 
    386                 self.details[p.id] = [p.units if not p.choices else p.choices, None, None] 
     383            self.details[p.id] = [p.units, p.limits[0], p.limits[1]] 
    387384            if p.polydisperse: 
    388385                self.details[p.id+".width"] = [ 
     
    543540            # pylint: disable=unpacking-non-sequence 
    544541            q, phi = x 
    545             return self.calculate_Iq([q*math.cos(phi)], [q*math.sin(phi)])[0] 
    546         else: 
    547             return self.calculate_Iq([x])[0] 
     542            result, _ = self.calculate_Iq([q*math.cos(phi)], [q*math.sin(phi)]) 
     543            return result[0] 
     544        else: 
     545            result, _ = self.calculate_Iq([x]) 
     546            return result[0] 
    548547 
    549548 
     
    560559        """ 
    561560        if isinstance(x, (list, tuple)): 
    562             return self.calculate_Iq([x[0]], [x[1]])[0] 
    563         else: 
    564             return self.calculate_Iq([x])[0] 
     561            result, _ = self.calculate_Iq([x[0]], [x[1]]) 
     562            return result[0] 
     563        else: 
     564            result, _ = self.calculate_Iq([x]) 
     565            return result[0] 
    565566 
    566567    def evalDistribution(self, qdist): 
     
    596597            # Check whether we have a list of ndarrays [qx,qy] 
    597598            qx, qy = qdist 
    598             return self.calculate_Iq(qx, qy) 
     599            result, _ = self.calculate_Iq(qx, qy) 
     600            return result 
    599601 
    600602        elif isinstance(qdist, np.ndarray): 
    601603            # We have a simple 1D distribution of q-values 
    602             return self.calculate_Iq(qdist) 
     604            result, _ = self.calculate_Iq(qdist) 
     605            return result 
    603606 
    604607        else: 
     
    640643        if composition and composition[0] == 'product': # only P*S for now 
    641644            with calculation_lock: 
    642                 self._calculate_Iq(qx) 
    643                 # for compatibility with sasview 4.3 
    644                 results = self._intermediate_results() 
    645                 return results["P(Q)"], results["S(Q)"] 
     645                _, lazy_results = self._calculate_Iq(qx) 
     646                # for compatibility with sasview 4.x 
     647                results = lazy_results() 
     648                pq_data = results.get("P(Q)") 
     649                sq_data = results.get("S(Q)") 
     650                return pq_data, sq_data 
    646651        else: 
    647652            return None 
    648653 
    649     def calculate_Iq(self, qx, qy=None): 
    650         # type: (Sequence[float], Optional[Sequence[float]]) -> np.ndarray 
     654    def calculate_Iq(self, 
     655                     qx,     # type: Sequence[float] 
     656                     qy=None # type: Optional[Sequence[float]] 
     657                     ): 
     658        # type: (...) -> Tuple[np.ndarray, Callable[[], collections.OrderedDict[str, np.ndarray]]] 
    651659        """ 
    652660        Calculate Iq for one set of q with the current parameters. 
     
    656664        This should NOT be used for fitting since it copies the *q* vectors 
    657665        to the card for each evaluation. 
     666 
     667        The returned tuple contains the scattering intensity followed by a 
     668        callable which returns a dictionary of intermediate data from 
     669        ProductKernel. 
    658670        """ 
    659671        ## uncomment the following when trying to debug the uncoordinated calls 
     
    688700        result = calculator(call_details, values, cutoff=self.cutoff, 
    689701                            magnetic=is_magnetic) 
     702        lazy_results = getattr(calculator, 'results', 
     703                               lambda: collections.OrderedDict()) 
    690704        #print("result", result) 
    691         self._intermediate_results = getattr(calculator, 'results', None) 
     705 
    692706        calculator.release() 
    693707        #self._model.release() 
    694         return result 
     708 
     709        return result, lazy_results 
    695710 
    696711    def calculate_ER(self): 
     
    881896    q = np.linspace(-0.35, 0.35, 500) 
    882897    qx, qy = np.meshgrid(q, q) 
    883     result = model.calculate_Iq(qx.flatten(), qy.flatten()) 
     898    result, _ = model.calculate_Iq(qx.flatten(), qy.flatten()) 
    884899    result = result.reshape(qx.shape) 
    885900 
Note: See TracChangeset for help on using the changeset viewer.