ó \™xTc @ sÚddlmZddlZddlmZmZddlmZdefd„ƒYZ defd„ƒYZ d efd „ƒYZ d efd „ƒYZ d efd„ƒYZ edkrÖe ddƒZejGHndS(iÿÿÿÿ(tdivisionN(tmatrixtlbfgs(tflextfit_peakcB s;eZdZddddd„Zd„Zddd„ZRS(s ============================================================================= fit_peak Class This class fits a list of points (x,y,z) and their heights to a certain shape using the LBFGS minimizer and the function sum{(p_i - p_i0)^2}. ----------------------------------------------------------------------------- Arguments: height_list - the list of heights (list of floats) xyz_list - the list of positions (list of tuples) shape - flag for shape to be fitted (parabola, quadratic, gaussian) max_iterations - the maximum number of minimization steps (int) Useful accessible attributes: self.x - the final parameters from the fit (list) self.vertex - the peak position (tuple) Notes: The first point should be the point closest to the peak. If the peak is found outside the 26 nearest neighbors, the center point is returned. The order of the points is very important because the initial guess for minimization is obtained by solving a linear system. See pick_map_neighbors for the required order. To use "gaussian," height_list should actually be the ln(height) since the least squares function is sum{[ln(p_i) - ln(p_i0)]^2} for that case. ----------------------------------------------------------------------------- tparabolaicC së||_||_|dkrStƒj|_t|_|jd|d|ƒg}g}x×t|jƒD]Æ}|j||ƒ|j||d||dƒ|j||d||dƒ|j||d||dƒ|j||dƒ|j||dƒ|j||dƒ|jdƒqhWtj |ƒ}tj |ƒ} nÔ|dksk|d krt ƒj|_|dkr’t |_n t |_|jd|d|ƒg}g}x:t|jƒD])}|j||ƒ|j||d||dƒ|j||d||dƒ|j||d||dƒ|j||dƒ|j||dƒ|j||dƒ|j||d||dƒ|j||d||dƒ|j||d||dƒ|jdƒqÍWtj |ƒ}tj |ƒ} n d GHt ƒtj|jƒ| ƒ|_t|ƒ|jkr‚tjd |d tjd |ƒƒ|_n|jd|jƒ} | j|_|dd} |dd} |dd} |dd}|dd}|dd}| | ks t‚| |kst‚||ks-t‚|jd| ksS|jd| krk|dd|jds ‚747