File size: 359 Bytes
d547008 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | 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)
|