rtc_util.py

 
 
import numpy as np
 
 
def create_interp_mat(dimx: int, dimy: int):
    """TODO doc
 
    Args:
 
        dimx: (int) :
 
        dimy: (int) :
    """
    n = max(dimx, dimy)
    tmp2, tmp1 = np.indices((n, n))
    tmp1 = tmp1[:dimx, :dimy] - (dimx // 2)
    tmp2 = tmp2[:dimx, :dimy] - (dimy // 2)
 
    tmp = np.zeros((tmp1.size, 6), np.int32)
    tmp[:, 0] = (tmp1**2).flatten()
    tmp[:, 1] = (tmp2**2).flatten()
    tmp[:, 2] = (tmp1 * tmp2).flatten()
    tmp[:, 3] = tmp1.flatten()
    tmp[:, 4] = tmp2.flatten()
    tmp[:, 5] = 1
 
    return np.dot(np.linalg.inv(np.dot(tmp.T, tmp)), tmp.T).T
 
 
def centroid_gain(E, F):
    """ Returns the mean centroid gain
 
    Args:
 
        E : (np.array(dtype=np.float32)) : measurements from WFS
 
        F : (np.array(dtype=np.float32)) : geometric measurements
 
    :return:
 
        cgain : (float) : mean centroid gain between the sets of WFS measurements and geometric ones
    """
    if E.ndim == 1:
        cgains = np.polyfit(E, F, 1)[0]
    elif E.ndim == 2:
        cgains = np.zeros(E.shape[1])
        for k in range(E.shape[1]):
            cgains[k] = np.polyfit(E[:, k], F[:, k], 1)[0]
        cgains = np.mean(cgains)
    else:
        raise ValueError("E and F must 1D or 2D arrays")
 
    return cgains