diff --git a/sim_scenes/solar_system/solar_system_reality.py b/sim_scenes/solar_system/solar_system_reality.py index 0dc84b6f539e94fab95c31ebf2e44da5d51a66aa..8dff780150f938613a5fa247030241ef607f20af 100644 --- a/sim_scenes/solar_system/solar_system_reality.py +++ b/sim_scenes/solar_system/solar_system_reality.py @@ -54,34 +54,37 @@ def get_bodies_names(bodies): names += body.__class__.__name__ + "," return names[0:-1] + def are_planets_in_line(positions, line_width): # 检查行星的数量是否足够判断是否在一条线上 if len(positions) < 3: return False - # 获取第一个行星的坐标 + # 获取第一个行星的坐标 x1, y1, z1 = positions[0] - # 计算行星之间的向量差 + # 计算行星之间的向量差 dx = positions[1][0] - x1 dy = positions[1][1] - y1 dz = positions[1][2] - z1 - # 计算线的宽度的平方 - line_width_squared = line_width**2 - # 遍历剩余的行星 + # 计算线的宽度的平方 + line_width_squared = line_width ** 2 + # 遍历剩余的行星 for i in range(2, len(positions)): # 获取当前行星的坐标 x, y, z = positions[i] - # 计算当前行星与第一个行星之间的向量差 + # 计算当前行星与第一个行星之间的向量差 current_dx = x - x1 current_dy = y - y1 current_dz = z - z1 - # 计算当前行星与线之间的距离的平方 - distance_squared = (current_dy * dz - current_dz * dy)**2 + (current_dz * dx - current_dx * dz)**2 + (current_dx * dy - current_dy * dx)**2 - # 如果距离的平方大于线的宽度的平方，则行星不在一条线上 + # 计算当前行星与线之间的距离的平方 + distance_squared = (current_dy * dz - current_dz * dy) ** 2 + (current_dz * dx - current_dx * dz) ** 2 + ( + current_dx * dy - current_dy * dx) ** 2 + # 如果距离的平方大于线的宽度的平方，则行星不在一条线上 if distance_squared > line_width_squared: return False - # 所有行星都在一条线上 + # 所有行星都在一条线上 return True + def are_planets_in_line(planets, line_width): if len(planets) < 2: return False @@ -94,10 +97,14 @@ def are_planets_in_line(planets, line_width): y_diff = [abs(y2 - y1) for (y1, y2) in zip(y_coords, y_coords[1:])] z_diff = [abs(z2 - z1) for (z1, z2) in zip(z_coords, z_coords[1:])] - widths = [max(d1, d2, line_width) for (d1, d2) in zip(x_diff, y_diff)] + [line_width] + [max(d1, d2, line_width) for (d1, d2) in zip(x_diff[::-1], y_diff[::-1])] + widths = [max(d1, d2, line_width) for (d1, d2) in zip(x_diff, y_diff)] + [line_width] + [max(d1, d2, line_width) for + (d1, d2) in + zip(x_diff[::-1], + y_diff[::-1])] return all(w == widths[0] for w in widths) + def are_planets_in_line(planets, line_width): if len(planets) < 2: return False @@ -110,10 +117,12 @@ def are_planets_in_line(planets, line_width): y_diff = [abs(y2 - y1) for (y1, y2) in zip(y_coords, y_coords[1:])] z_diff = [abs(z2 - z1) for (z1, z2) in zip(z_coords, z_coords[1:])] - widths = [max(d1, d2, line_width**2) for (d1, d2) in zip(x_diff, y_diff)] + [line_width**2] + [max(d1, d2, line_width**2) for (d1, d2) in zip(x_diff[::-1], y_diff[::-1])] + widths = [max(d1, d2, line_width ** 2) for (d1, d2) in zip(x_diff, y_diff)] + [line_width ** 2] + [ + max(d1, d2, line_width ** 2) for (d1, d2) in zip(x_diff[::-1], y_diff[::-1])] return all(w == widths[0] for w in widths) + current_time = Time.now() in_line_datetimes = [] @@ -127,12 +136,12 @@ if __name__ == '__main__': # ===================================================================== # 以下展示的效果为太阳系真实的距离 # 由于宇宙空间尺度非常大，如果按照实际的天体大小，则无法看到天体，因此需要对天体的尺寸进行放大 - sun = Sun(name="太阳", size_scale=0.4e2) # 太阳放大 40 倍，距离保持不变 + sun = Sun(name="太阳", size_scale=0.4e2) # 太阳放大 40 倍，距离保持不变（size_scale=0.04e2）TODO:进行调试大小 bodies = [ sun, Mercury(name="水星", size_scale=1.5e3), # 水星 Venus(name="金星", size_scale=1e3), # 金星 - Earth(name="地球", size_scale=1e3), # 地球 + Earth(name="地球", size_scale=1e3), # 地球（size_scale=10e3）TODO:进行调试大小 Moon(name="月球", size_scale=2e3), # 月球 Mars(name="火星", size_scale=1.2e3), # 火星 # Asteroids(size_scale=1e2, parent=sun, rotate_angle=-20), @@ -144,6 +153,7 @@ if __name__ == '__main__': earth = bodies[3] earth.rotate_axis_color = (255, 255, 50) + # earth.set_light_disable(True) # TODO:进行调试 names = get_bodies_names(bodies) names = names.replace("Asteroids,", "") @@ -160,6 +170,9 @@ if __name__ == '__main__': def on_ready(): # 运行前触发 camera.rotation_z = -20 + # 需要按照时间和日期控制地球的自转，不能随意转动 + delattr(earth.planet, "rotation_speed") + delattr(earth.planet, "rotspeed") def on_timer_changed(time_data: TimeData): @@ -168,6 +181,7 @@ if __name__ == '__main__': # earth_loc = None earth_pos = None sun_pos = None + positions = [] for body in bodies: if isinstance(body, Asteroids): @@ -204,6 +218,19 @@ if __name__ == '__main__': dt = time_data.get_datetime(str(current_time)) + # # 日期当天的偏转角度+误差 + # angle_of_day = day_of_year * (360 / 365) + 75 + # # 控制地球的自转 + # earth.planet.rotation_y = -(time_data.total_hours) * 15 + angle_of_day + + # 需要按照时间和日期控制地球的自转，不能随意转动 + # 日期是当年的第几天 + day_of_year = dt.timetuple().tm_yday + # 计算出：日期当天的偏转角度 - 贴图的误差 + angle_of_day = day_of_year * (360 / 365) - 120 + # 控制地球的自转速度和方向，保障白天，中国面对太阳（会存在一点点的误差，可以通过上面“贴图的误差”进行调整）。 + earth.planet.rotation_y = -(time_data.total_hours) * 15 - angle_of_day + # if len(in_line_datetimes) == 0: # in_line = are_planets_in_line(positions, 5*AU) # if in_line: