代码拉取完成，页面将自动刷新
# -*- coding: utf-8 -*-
"""
Created on Sun Dec 26 22:09:04 2021
@author: Xcz
"""
# %%
import math
import numpy as np
from fun_img_Resize import if_image_Add_black_border
from fun_global_var import init_GLV_DICT, tree_print, init_GLV_rmw, end_AST, Get, Set, init_EVV, \
fget, fkey, fGHU_plot_save, dset, dget, Fun3, fU_EVV_plot
from fun_thread import my_thread
from fun_pump import pump_pic_or_U
from fun_linear import init_AST
np.seterr(divide='ignore', invalid='ignore')
# %%
def AST_EVV(U_name="",
img_full_name="Grating.png",
is_phase_only=0,
# %%
z_pump=0,
is_LG=0, is_Gauss=0, is_OAM=0,
l=0, p=0,
theta_x=0, theta_y=0,
# %%
is_random_phase=0,
is_H_l=0, is_H_theta=0, is_H_random_phase=0,
# %%
U_size=1, w0=0.3,
z0=1,
# %%
lam1=0.8, is_air_pump=0, is_air=0, T=25,
# %%
is_save=0, is_save_txt=0, dpi=100,
# %%
cmap_2d='viridis',
# %%
ticks_num=6, is_contourf=0,
is_title_on=1, is_axes_on=1, is_mm=1,
# %%
fontsize=9,
font={'family': 'serif',
'style': 'normal', # 'normal', 'italic', 'oblique'
'weight': 'normal',
'color': 'black', # 'black','gray','darkred'
},
# %%
is_colorbar_on=1, is_colorbar_log=0,
is_energy=0,
# %%
is_print=1,
# %% 该程序 独有 -------------------------------
is_EVV_SSI=1, is_stored=1, sheets_stored_num=10,
# %%
sample=1, cmap_3d='rainbow',
elev=10, azim=-65, alpha=2,
# %%
is_plot_EVV=1, is_plot_3d_XYz=0, is_plot_selective=0,
X=0, Y=0, is_plot_YZ_XZ=1, is_plot_3d_XYZ=0,
# %%
plot_group="UGa", is_animated=1,
loop=0, duration=0.033, fps=5,
# %%
**kwargs, ):
# %%
if_image_Add_black_border(U_name, img_full_name,
__name__ == "__main__", is_print, **kwargs, )
# %%
info = "AST_线性角谱"
is_print and print(tree_print(kwargs.get("is_end", 0), add_level=2) + info)
kwargs.pop("is_end", None);
kwargs.pop("add_level", None) # 该 def 子分支 后续默认 is_end = 0，如果 kwargs 还会被 继续使用 的话。
# kwargs['ray'] = init_GLV_rmw(U_name, "~", "", "AST", **kwargs)
init_GLV_rmw(U_name, "l", "AST", "", **kwargs)
img_name, img_name_extension, img_squared, \
size_PerPixel, size_fig, Ix, Iy, \
U_0, g_shift = pump_pic_or_U(U_name,
img_full_name,
is_phase_only,
# %%
z_pump,
is_LG, is_Gauss, is_OAM,
l, p,
theta_x, theta_y,
# %%
is_random_phase,
is_H_l, is_H_theta, is_H_random_phase,
# %%
U_size, w0,
# %%
lam1, is_air_pump, T,
# %%
is_save, is_save_txt, dpi,
cmap_2d,
# %%
ticks_num, is_contourf,
is_title_on, is_axes_on, is_mm,
# %%
fontsize, font,
# %%
is_colorbar_on, is_energy,
# %%
is_print,
# %%
ray_pump='1', **kwargs, )
# %%
if "h" in Get("ray"): # 如果 ray 中含有 倍频 标识符
lam1 = lam1 / 2
if "lam3" in kwargs:
lam1 = kwargs["lam3"]
n1_inc, n1, k1_inc, k1, k1_z, k1_xy = init_AST(Ix, Iy, size_PerPixel,
lam1, is_air, T,
theta_x, theta_y,
**kwargs)
# %%
iz = z0 / size_PerPixel
zj = kwargs.get("zj_AST_EVV", np.linspace(0, z0, sheets_stored_num + 1)) # 防止 后续函数 接收的 kwargs 里 出现 关键字 zj 后重名
izj = zj / size_PerPixel
# print(izj)
if is_EVV_SSI == 1:
izj_delay_dz = [0] + list(izj)
dizj = list(np.array(izj_delay_dz)[1:] - np.array(izj_delay_dz)[:-1]) # 为循环 里使用
izj_delay_dz.pop(-1) # 可 pop 可不 pop 掉 最后一个元素，反正没啥用
# dizj = [izj[0] - 0] + dizj
# Set("is_EVV_SSI", 1)
# print(izj_delay_dz)
# print(dizj)
Set("zj", zj)
Set("izj", izj)
sheets_stored_num = len(zj) - 1
init_EVV(g_shift, U_0,
0, is_stored,
sheets_stored_num, sheets_stored_num,
X, Y, iz, size_PerPixel, )
# %%
def H_zdz(diz):
return np.power(math.e, k1_z * diz * 1j)
def Fun1(for_th2, fors_num2, *args, **kwargs, ):
if is_EVV_SSI == 1:
iz = izj_delay_dz[for_th2]
H1_z = H_zdz(iz)
G1_z = g_shift * H1_z
# U_z = ifft2(G1_z)
diz = dizj[for_th2]
else:
iz = izj[for_th2]
G1_z = g_shift
# U_z = U_0
diz = iz
G_z = G1_z * H_zdz(diz)
return G_z
def Fun2(for_th2, fors_num, G_z, *args, **kwargs, ):
dset("G", G_z)
return dget("G")
my_thread(10, sheets_stored_num + 1,
Fun1, Fun2, Fun3,
is_ordered=1, is_print=is_print, )
# %%
end_AST(z0, size_PerPixel,
g_shift, k1_z, )
fGHU_plot_save(0, # 默认 全自动 is_auto = 1
img_name_extension, is_print,
# %%
zj, sample, size_PerPixel,
is_save, is_save_txt, dpi, size_fig,
# %%
"b", cmap_2d,
ticks_num, is_contourf,
is_title_on, is_axes_on, is_mm,
fontsize, font,
# %%
is_colorbar_on, is_energy, # 默认无法 外界设置 vmax 和 vmin，因为 同时画 振幅 和 相位 得 传入 2*2 个 v
# %% 何况 一般默认 is_self_colorbar = 1...
z0,
# %%
is_end=1, **kwargs, )
# %%
fU_EVV_plot(img_name_extension,
is_save_txt,
# %%
sample, size_PerPixel,
is_save, dpi, size_fig,
elev, azim, alpha,
# %%
cmap_2d, cmap_3d,
ticks_num, is_contourf,
is_title_on, is_axes_on, is_mm,
fontsize, font,
# %%
is_colorbar_on, is_colorbar_log,
is_energy,
# %%
plot_group, is_animated,
loop, duration, fps,
# %%
is_plot_EVV, is_plot_3d_XYz, is_plot_selective,
is_plot_YZ_XZ, is_plot_3d_XYZ,
# %%
z0,
# %%
**kwargs, )
return fget("U"), fget("G"), Get("ray"), Get("method_and_way"), fkey("U")
if __name__ == '__main__':
kwargs = \
{"U_name": "",
"img_full_name": "lena1.png",
"U_pixels_x": 0, "U_pixels_y": 0,
"is_phase_only": 0,
# %%
"z_pump": 0,
"is_LG": 0, "is_Gauss": 0, "is_OAM": 0,
"l": 0, "p": 0,
"theta_x": 0, "theta_y": 0,
# %%
"is_random_phase": 0,
"is_H_l": 0, "is_H_theta": 0, "is_H_random_phase": 0,
# %%
"U_size": 1, "w0": 0,
"z0": 15,
# %%
"lam1": 1, "is_air_pump": 1, "is_air": 0, "T": 25,
# %%
"is_save": 0, "is_no_data_save": 0,
"is_save_txt": 0, "dpi": 100,
# %%
"cmap_2d": 'viridis',
# %%
"ticks_num": 6, "is_contourf": 0,
"is_title_on": 1, "is_axes_on": 1, "is_mm": 1,
# %%
"fontsize": 10,
"font": {'family': 'serif',
'style': 'normal', # 'normal', 'italic', 'oblique'
'weight': 'normal',
'color': 'black', # 'black','gray','darkred'
},
# %%
"is_colorbar_on": 1, "is_colorbar_log": 0,
"is_energy": 1,
# %%
"is_print": 1,
# %% 该程序 独有 -------------------------------
"is_EVV_SSI": 0, "is_stored": 1, "sheets_stored_num": 10,
# %%
"sample": 1, "cmap_3d": 'rainbow',
"elev": 10, "azim": -65, "alpha": 2,
# %%
"is_plot_EVV": 0, "is_plot_3d_XYz": 0, "is_plot_selective": 1,
"X": 0, "Y": 0, "is_plot_YZ_XZ": 1, "is_plot_3d_XYZ": 1,
# %%
"plot_group": "Ua", "is_animated": 1,
"loop": 0, "duration": 0.033, "fps": 5,
# %% 该程序 作为 主入口时 -------------------------------
"kwargs_seq": 0, "root_dir": r'1',
"border_percentage": 0.1, "is_end": -1,
# %%
"theta_z": 90, "phi_z": 0, "phi_c": 24.3,
# KTP 50 度 ：deff 最高： 90, ~, 24.3，（24.3 - 2002, 25.3 - 2000）
# 1994 ：68.8, ~, 90，（68.8 - 2002, 68.9 - 2000）
# KTP 25 度 ：deff 最高： 90, ~, 23.7，（23.7 - 2002, 24.8 - 2000）
# 1994 ：68.8, ~, 90，（68.8 - 2002, 68.7 - 2000）
# LN 25 度 ：90, ~, ~
"polar": "e", "ray": "1",
}
kwargs = init_GLV_DICT(**kwargs)
AST_EVV(**kwargs)
# AST(U_name="",
# img_full_name="Grating.png",
# is_phase_only=0,
# # %%
# z_pump=0,
# is_LG=0, is_Gauss=0, is_OAM=0,
# l=0, p=0,
# theta_x=0, theta_y=0,
# # %%
# is_random_phase=0,
# is_H_l=0, is_H_theta=0, is_H_random_phase=0,
# # %%
# U_size=1, w0=0.1,
# z0=1,
# # %%
# lam1=0.8, is_air_pump=0, is_air=0, T=25,
# # %%
# is_save=1, is_save_txt=0, dpi=100,
# # %%
# cmap_2d='viridis',
# # %%
# ticks_num=6, is_contourf=0,
# is_title_on=1, is_axes_on=1, is_mm=1,
# # %%
# fontsize=9,
# font={'family': 'serif',
# 'style': 'normal', # 'normal', 'italic', 'oblique'
# 'weight': 'normal',
# 'color': 'black', # 'black','gray','darkred'
# },
# # %%
# is_colorbar_on=1, is_energy=0,
# # %%
# is_print=1,
# # %%
# root_dir=r'',
# border_percentage=0.1, ray="1", is_end=-1, )
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手