Ignore:
File:
1 edited

Legend:

Unmodified
Added
Removed
  • src/sas/sascalc/pr/invertor.py

    r463e7ffc r9a5097c  
    77""" 
    88 
    9 import numpy 
     9import numpy as np 
    1010import sys 
    1111import math 
     
    1818from scipy import optimize 
    1919from sas.sascalc.pr.core.pr_inversion import Cinvertor 
    20  
    21 logger = logging.getLogger(__name__) 
    2220 
    2321def help(): 
     
    191189        #import numpy 
    192190        if name == 'x': 
    193             out = numpy.ones(self.get_nx()) 
     191            out = np.ones(self.get_nx()) 
    194192            self.get_x(out) 
    195193            return out 
    196194        elif name == 'y': 
    197             out = numpy.ones(self.get_ny()) 
     195            out = np.ones(self.get_ny()) 
    198196            self.get_y(out) 
    199197            return out 
    200198        elif name == 'err': 
    201             out = numpy.ones(self.get_nerr()) 
     199            out = np.ones(self.get_nerr()) 
    202200            self.get_err(out) 
    203201            return out 
     
    327325            raise RuntimeError, msg 
    328326 
    329         p = numpy.ones(nfunc) 
     327        p = np.ones(nfunc) 
    330328        t_0 = time.time() 
    331329        out, cov_x, _, _, _ = optimize.leastsq(self.residuals, p, full_output=1) 
     
    343341 
    344342        if cov_x is None: 
    345             cov_x = numpy.ones([nfunc, nfunc]) 
     343            cov_x = np.ones([nfunc, nfunc]) 
    346344            cov_x *= math.fabs(chisqr) 
    347345        return out, cov_x 
     
    360358            raise RuntimeError, msg 
    361359 
    362         p = numpy.ones(nfunc) 
     360        p = np.ones(nfunc) 
    363361        t_0 = time.time() 
    364362        out, cov_x, _, _, _ = optimize.leastsq(self.pr_residuals, p, full_output=1) 
     
    437435        """ 
    438436        # Note: To make sure an array is contiguous: 
    439         # blah = numpy.ascontiguousarray(blah_original) 
     437        # blah = np.ascontiguousarray(blah_original) 
    440438        # ... before passing it to C 
    441439 
     
    458456            nfunc += 1 
    459457 
    460         a = numpy.zeros([npts + nq, nfunc]) 
    461         b = numpy.zeros(npts + nq) 
    462         err = numpy.zeros([nfunc, nfunc]) 
     458        a = np.zeros([npts + nq, nfunc]) 
     459        b = np.zeros(npts + nq) 
     460        err = np.zeros([nfunc, nfunc]) 
    463461 
    464462        # Construct the a matrix and b vector that represent the problem 
     
    478476        self.chi2 = chi2 
    479477 
    480         inv_cov = numpy.zeros([nfunc, nfunc]) 
     478        inv_cov = np.zeros([nfunc, nfunc]) 
    481479        # Get the covariance matrix, defined as inv_cov = a_transposed * a 
    482480        self._get_invcov_matrix(nfunc, nr, a, inv_cov) 
     
    492490 
    493491        try: 
    494             cov = numpy.linalg.pinv(inv_cov) 
     492            cov = np.linalg.pinv(inv_cov) 
    495493            err = math.fabs(chi2 / float(npts - nfunc)) * cov 
    496494        except: 
    497495            # We were not able to estimate the errors 
    498496            # Return an empty error matrix 
    499             logger.error(sys.exc_value) 
     497            logging.error(sys.exc_value) 
    500498 
    501499        # Keep a copy of the last output 
     
    507505            self.background = c[0] 
    508506 
    509             err_0 = numpy.zeros([nfunc, nfunc]) 
    510             c_0 = numpy.zeros(nfunc) 
     507            err_0 = np.zeros([nfunc, nfunc]) 
     508            c_0 = np.zeros(nfunc) 
    511509 
    512510            for i in range(nfunc_0): 
     
    543541            # number of terms 
    544542            best_alpha, _, _ = self.estimate_alpha(self.nfunc) 
    545             logger.warning("Invertor.estimate_numterms: %s" % sys.exc_value) 
     543            logging.warning("Invertor.estimate_numterms: %s" % sys.exc_value) 
    546544            return self.nfunc, best_alpha, "Could not estimate number of terms" 
    547545 
     
    664662                                                   str(self.cov[i][i]))) 
    665663        file.write("<r>  <Pr>  <dPr>\n") 
    666         r = numpy.arange(0.0, self.d_max, self.d_max / npts) 
     664        r = np.arange(0.0, self.d_max, self.d_max / npts) 
    667665 
    668666        for r_i in r: 
     
    696694                        toks = line.split('=') 
    697695                        self.nfunc = int(toks[1]) 
    698                         self.out = numpy.zeros(self.nfunc) 
    699                         self.cov = numpy.zeros([self.nfunc, self.nfunc]) 
     696                        self.out = np.zeros(self.nfunc) 
     697                        self.cov = np.zeros([self.nfunc, self.nfunc]) 
    700698                    elif line.startswith('#alpha='): 
    701699                        toks = line.split('=') 
Note: See TracChangeset for help on using the changeset viewer.