| import numpy as np | |
| ROT_LIGHT_NUM_LIGHT = 30 | |
| ROT_LIGHT_NUM_LOOP = 3 | |
| def gen_rot_light__light_pos(num_light, num_loop): | |
| theta = np.linspace(0, np.pi, num_light) | |
| phi = np.linspace(0, 2 * np.pi * num_loop, num_light) | |
| x = np.sin(theta) * np.cos(phi) | |
| z = np.sin(theta) * np.sin(phi) | |
| y = np.cos(theta) | |
| return np.stack([x, y, z], axis=-1) | |