import os
import numpy as np
import matplotlib.pyplot as plt
from scipy.ndimage import zoom
import gc
from tqdm import tqdm
from zernike import RZerndef wrap_phase(phase):"""將相位包裹到[-π, π]區間"""return np.angle(np.exp(1j * phase))def clean_nan_values(arr):"""清理數組中的NaN值"""arr_copy = arr.copy()if np.any(np.isnan(arr_copy)):arr_copy = np.nan_to_num(arr_copy, nan=0.0)return arr_copydef fit_zernike_coefficients(phase, mask, n_coef=37):"""使用RZern對相位進行Zernike多項式擬合參數:phase: 相位圖mask: 掩碼n_coef: Zernike系