kdl.py 文件源码

def cell_fd_extention(fname_org='sheet.gz/cell_db.cvs.gz', camera_bit_resolution=14):
    cell_df = pd.read_csv(fname_org)
    Limg, Lx, Ly = cell_fd_info(cell_df)

    cell_df_ext = cell_df.copy()

    # Fresnel diffraction
    cell_img_fd_a = cell_fd_conv(cell_df)
    cell_df_ext['freznel image'] = cell_img_fd_a.reshape(-1)

    # max_v, min_v = np.max(cell_df["image"]), np.min(cell_df["image"])
    cell_img_fd_a_2d = cell_img_fd_a.reshape(Limg, -1)
    cell_img_fd_a_2d_scale = preprocessing.minmax_scale(
        np.abs(cell_img_fd_a_2d)) * (2**camera_bit_resolution)
    cell_img_fd_a_2d_scale_200x144x144 = cell_img_fd_a_2d_scale.reshape(
        Limg, Lx, Ly).astype(int)
    cell_df_ext[
        'mag freznel image'] = cell_img_fd_a_2d_scale_200x144x144.reshape(-1)

    return cell_df_ext