From 7dfbe2222f9d99fd69755d2a5e82e12bc3344bfe Mon Sep 17 00:00:00 2001 From: GaugeAndGravity Date: Fri, 10 Nov 2023 13:43:30 -0500 Subject: [PATCH] fix bug --- rainbow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rainbow.py b/rainbow.py index 67d7c06..db3c510 100644 --- a/rainbow.py +++ b/rainbow.py @@ -105,7 +105,7 @@ def water_refraction_index(t, wavelength): def sun_spectral(wavelength): return 1e16/(np.power(wavelength,5)*(np.exp(6.62607015e6*2.99792458/(wavelength*1.380649*(5250+273.15)))-1)) -def modified_trace(wavelength, temp, center, r, dx, N, n_theta, d_theta, min_intensity, max_ray, max_angle): +def modified_trace(wavelength, temp, center, r, N, n_theta, d_theta, min_intensity, max_ray, max_angle): n = water_refraction_index(temp, wavelength) disk = Disk(center, r, n) stack = [] @@ -150,7 +150,7 @@ def rainbow(n_theta, max_theta, temp): # 使用 ThreadPoolExecutor 并行执行 with ProcessPoolExecutor() as executor: - futures = [executor.submit(modified_trace, wavelength, temp, center, r, dx, N, n_theta, d_theta, min_intensity, max_ray, max_theta) for wavelength in colorspace.data_wavelength] + futures = [executor.submit(modified_trace, wavelength, temp, center, r, N, n_theta, d_theta, min_intensity, max_ray, max_theta) for wavelength in colorspace.data_wavelength] for future in as_completed(futures): result = future.result()