np.linspace(-1, 1, nby), np.linspace(-1, 1, nbx), indexing="ij", ) window = multivariate_normal.pdf( np.column_stack([grid_z.flat, grid_y.flat, grid_x.flat]), mean=mu, cov=covariance, ) window = window.reshape((nbz, nby, nbx)) elif window_type == "tukey": comment = comment + "_tukey" window = pu.tukey_window(data.shape, alpha=alpha) elif window_type == "blackman": comment = comment + "_blackman" window = pu.blackman_window(data.shape) else: print("invalid window type") sys.exit() if debug: fig, ax0 = plt.subplots(1, 1) plt.imshow(window[:, :, nbx // 2], vmin=0, vmax=window.max()) plt.title("Window at middle frame") fig, ax0 = plt.subplots(1, 1) plt.plot(window[nbz // 2, nby // 2, :]) plt.plot(window[:, nby // 2, nbx // 2]) plt.plot(window[nbz // 2, :, nbx // 2]) plt.title("Window linecuts at array center")
sum_frames=True, title='Bragg peaks positions', slice_position=pivot, vmin=0, vmax=1, scale='linear', cmap=my_cmap, is_orthogonal=True, reciprocal_space=True) plt.pause(0.1) ######################### # define the peak shape # ######################### peak_shape = pu.blackman_window(shape=(kernel_length, kernel_length, kernel_length), normalization=100) ##################################### # define the list of angles to test # ##################################### angles_qx = np.linspace( start=angles_ranges[0], stop=angles_ranges[1], num=max(1, np.rint((angles_ranges[1] - angles_ranges[0]) / angular_step) + 1)) angles_qz = np.linspace( start=angles_ranges[2], stop=angles_ranges[3], num=max(1, np.rint((angles_ranges[3] - angles_ranges[2]) / angular_step) + 1))