atomcloud.utils.fit_utils module

Created on Fri Mar 18 17:29:59 2022

@author: hofer

atomcloud.utils.fit_utils.approximate_jonquieress_function(z, gamma, n_max=10)

atomcloud.utils.fit_utils.calc_diff_elements(coords)

Calculate the difference between each element in the x and y arrays. This is used to calculate the average pixel size along each axis.

atomcloud.utils.fit_utils.coordinate_transformation2D(XY_tuple, x0=0, y0=0, theta=0)

Transforms coordinates by first performing a translation and then a rotation

atomcloud.utils.fit_utils.generate_elliptical_mask2D(XY_tuple, x0, y0, sig_x, sig_y, theta, scale=1)

Currently does mask edge at wx (2 * sigmax), but change it to sigmax

atomcloud.utils.fit_utils.get_array_bounds(data)

atomcloud.utils.fit_utils.get_lab_widths(px, py, theta)

Given two orthogonal parameters which are rotated by theta with respect to the x and y axes, this function returns the widths in the lab frame.

atomcloud.utils.fit_utils.get_masked_data(XY_tuple, data, mask)

atomcloud.utils.fit_utils.get_wrapped_angle(angle, angle_range=3.141592653589793, min_angle=-1.5707963267948966)

wrap angles to be within a certain angle range

atomcloud.utils.fit_utils.is_nested_list(obj, level)

atomcloud.utils.fit_utils.polylog_val(z, gamma, n_max=100)

atomcloud.utils.fit_utils.rotate_coordinates2D(coordinates, theta)

Rotates coordinates by theta radians

Parameters:

• coordinates (tuple[numpy.ndarray, numpy.ndarray]) – tuple of x and y coordinates

• theta (float) – angle in radians

Returns:

rotated x and y coordinates

Return type:

tuple

atomcloud.utils.fit_utils.translate_coordinates2D(coords, x0, y0)

Translates coordinates by x0 and y0