Skip to content

宇宙模拟器
宇宙模拟器

Python_超人 / 宇宙模拟器

通知 19
Star 2
Fork 0
宇宙模拟器
宇宙模拟器
提交 5c5180a4 编写于 作者: 三月三net's avatar 三月三net
浏览文件
操作

Python超人-宇宙模拟器(代码备份，代码的清理未删除)

上级 a191211d
隐藏空白更改
内联 并排
Showing with 60 addition and 101 deletion
common/celestial_data_service.py
浏览文件 @ 5c5180a4
......@@ -7,10 +7,12 @@
# python_version :3.8
# ==============================================================================
import math
from common.consts import G, AU, SECONDS_PER_DAY
from bodies import Body, Sun, Asteroids, Moon, Earth
import numpy as np
from bodies import Body, Sun, Asteroids, Moon, Earth
from common.consts import G, AU, SECONDS_PER_DAY
def calc_solar_acceleration(body_or_pos, big_body):
"""
......@@ -286,9 +288,9 @@ def gen_real_pos_vel_bodies(body_names, dt):
pos.y.to(u.km).value,
pos.z.to(u.km).value)
gen_body_str += f" init_pos_vels['{body_name.lower()}']['vel']=[%.4f,%.4f,%.4f]\n" % (
vel.x.to(u.km / u.second).value,
vel.y.to(u.km / u.second).value,
vel.z.to(u.km / u.second).value)
vel.x.to(u.km / u.second).value,
vel.y.to(u.km / u.second).value,
vel.z.to(u.km / u.second).value)
return gen_body_str
# pos, vel = earth_pos_vel[0], earth_pos_vel[1]
......@@ -302,6 +304,10 @@ def gen_real_pos_vel_bodies(body_names, dt):
def get_init_pos_vels():
"""
获取初始位置和速度（方便模拟）
@return:
"""
init_pos_vels = {}
init_pos_vels['sun'] = {}
init_pos_vels['sun']['pos'] = [-4638653.0000, 0.0000, 1314332.6250]
......
sim_scenes/tri_bodies/two_way_foil.py
浏览文件 @ 5c5180a4
......@@ -16,7 +16,6 @@ from common.consts import SECONDS_PER_WEEK, AU
from objs import QuadObj, CircleObj, Obj
from sim_scenes.func import camera_look_at, two_bodies_colliding
from sim_scenes.func import ursina_run, create_sphere_sky
from simulators.ursina.entities.entity_utils import create_rings
from simulators.ursina.ursina_event import UrsinaEvent
......@@ -26,9 +25,6 @@ class TwoWayFoilSim:
"""
def __init__(self):
# self.three_dim_bodies = []
# self.two_dim_bodies = []
self.current_stage = self.stage_01
def build_solar_system(self):
......@@ -48,6 +44,7 @@ class TwoWayFoilSim:
]
# endregion
# 获取模拟的初始位置和速度
init_pos_vels = get_init_pos_vels()
for body in self.bodies:
......@@ -58,13 +55,18 @@ class TwoWayFoilSim:
body.init_velocity = pos_vels['vel']
def build_two_way_foil(self):
"""
创建二向箔（一个原始的方形二向箔，一个不断扩展的圆形二向箔）
@return:
"""
# 原始的方形二向箔
self.two_way_foil = QuadObj(texture='two_way_foil.png',
# size_scale=4e7,
size_scale=1e7,
init_velocity=[0, -6, 150],
init_position=[0, 0.5 * AU, -10 * AU]) \
.set_light_disable(True).set_ignore_gravity(True)
# 不断扩展的圆形二向箔
self.two_way_foil_circle = CircleObj(texture="two_way_foil_circle.png",
size_scale=self.two_way_foil.size_scale * 2,
) \
......@@ -73,33 +75,9 @@ class TwoWayFoilSim:
self.bodies.append(self.two_way_foil)
self.bodies.append(self.two_way_foil_circle)
# def create_two_dim(self, body):
# body_2dim = CircleObj(texture=body.texture,
# size_scale=body.diameter * body.size_scale,
# distance_scale=body.distance_scale,
# init_position=body.position) \
# .set_light_disable(True).set_ignore_gravity(True)
#
# body.two_dim = body_2dim
# body_2dim.three_dim = body
# return body_2dim
# def build_two_dim_bodies(self):
# for body in self.bodies:
# # two_dim_body = self.create_two_dim(body)
# self.three_dim_bodies.append(body)
# # self.two_dim_bodies.append(two_dim_body)
#
# self.bodies.append(self.two_way_foil)
# self.bodies.append(self.two_way_foil_circle)
#
# for body in self.two_dim_bodies:
# self.bodies.append(body)
def build(self):
self.build_solar_system()
self.build_two_way_foil()
# self.build_two_dim_bodies()
def on_ready(self):
"""
......@@ -111,129 +89,104 @@ class TwoWayFoilSim:
camera.clip_plane_far = 1000000
create_sphere_sky(scale=200000)
application.time_scale = 5
# for body in self.two_dim_bodies:
# body.planet.enabled = False
# model = body.planet.model
# if body.three_dim.has_rings:
# rings = create_rings(body.planet)
# rings.scale *= 2e10
# 圆形二向箔初始化（一开始不显示）
self.two_way_foil_circle.planet.rotation_x = 90
self.two_way_foil_circle.planet.enabled = False
# self.two_way_foil.planet.alpha = 0.6
# for b in self.two_dim_bodies:
# b.planet.rotation_x = 90
def flatten_animation(self, body):
"""
天体二维化的动画
@param body: 天体
@return:
"""
def flatten_update(planet):
def warp():
# 原始的 update 方法中有计算天体的运行
planet.init_update()
# time.sleep(0.2)
# 对Y轴进行压平动画，如果压平大小不足 1/20，则继续压缩，直到压缩到 1/20 就不压缩了
# （如果觉得 1/20 的厚度压的不够，还可以继续压缩，基本上就够了）
if planet.scale_y > planet.init_scale_y / 20:
planet.scale_y_v /= 1.01
# planet.enabled = False
# flatten_planet.enabled = True
# planet.update = planet.init_update
# elif planet.scale_y < planet.init_scale_y / 1.01:
# planet.set_light_off(True)
# 灯光关闭，不然压到2纬就会是黑色
planet.set_light_off(True)
planet.scale_y = planet.scale_y_v
return warp
# 压平时，转速将为以前的 1/20
body.planet.rotation_speed /= 20
# 记录原始的厚度大小
body.planet.scale_y_v = body.planet.scale_y
body.planet.init_scale_y = body.planet.scale_y
# 原始的 update 方法中有计算天体的运行，需要保留
body.planet.init_update = body.planet.update
# 替换 update
body.planet.update = flatten_update(body.planet)
# for i in range(10):
# # body.planet.scale_y /= 2
# body.init_scale = [body.init_scale[0], body.init_scale[1] / 2, body.init_scale[2]]
# time.sleep(0.2)
# body.planet.enabled = False
# flatten_body.planet.enabled = True
# flatten_body.planet.x += 100000
def stage_01(self):
"""
二向箔飞向太阳
@return:
"""
# 如果二向箔和太阳碰撞
if two_bodies_colliding(self.two_way_foil, self.sun):
# t = threading.Thread(target=self.flatten_animation, args=[self.sun, self.sun.two_dim])
# t.start()
# for i in range(10):
# self.sun.planet.scale_y /= 0.5
# time.sleep(1)
# self.sun.planet.enabled = False
# self.sun.two_dim.planet.enabled = True
# self.sun.two_dim.planet.x += 100000
# two_way_foil.explode(sun)
# 隐藏原始二向箔，保持在原地，不在飞行
self.two_way_foil.planet.enabled = False
self.two_way_foil.init_velocity = [0, 0, 0]
# 圆形二向箔显示并设置透明度为0.9
self.two_way_foil_circle.planet.alpha = 0.9
self.two_way_foil_circle.planet.enabled = True
# 当前阶段为 stage_02：二向箔压平天体的阶段
self.current_stage = self.stage_02
def blink(self, body):
return
if not hasattr(body, "blink_d"):
body.blink_d = 1
body.blink_v = 1
if body.blink_d > 0:
body.blink_v -= 0.05
body.planet.alpha = body.blink_v
if body.blink_v <= 0.7:
body.blink_d = -1
elif body.blink_d < 0:
body.blink_v += 0.05
body.planet.alpha = body.blink_v
if body.blink_v >= 1:
body.blink_d = 1
def stage_02(self):
"""
二向箔压平天体（二维化）的阶段
@return:
"""
# self.sun.two_dim.planet.init_scale += 0.05
# 圆形二向箔不断扩展变大
self.two_way_foil_circle.planet.init_scale += 0.8
# self.blink(self.sun.two_dim)
# 二维化延时的时间
# 调整天体二维化的时间，一般需要延时，保证扩展和二维化同步的真实效果
two_way_delay_times = [0.5, # 太阳
0.5, 0.8, 1.0, # 水星 金星 地球
1.2, 2.0, 3.0, # 火星 木星 土星
4.0, 4.5, 5.5] # 天王星 海王星 冥王星
for idx, b in enumerate(self.bodies):
if isinstance(b, Obj):
# 二向箔不处理
continue
if hasattr(b, "two_way_time"):
# 二向箔和天体碰撞的时间不为空，则说明已经碰撞
if b.two_way_time is not None:
# 如果碰撞后的延时时间到，则进行压平天体处理（二维化）
if time.time() - b.two_way_time > two_way_delay_times[idx]:
self.flatten_animation(b)
# b.planet.enabled = False
# b.two_dim.planet.enabled = True
# 二向箔和天体碰撞的时间设置为空，就是说明二维化结束
b.two_way_time = None
elif two_bodies_colliding(self.two_way_foil_circle, b):
# 二向箔和天体碰撞，但暂时先不二维化，记下时间，延时二维化
b.two_way_time = time.time()
# b.planet.enabled = False
# b.two_dim.planet.enabled = True
# if b.two_dim.planet.enabled:
# # b.two_dim.planet.init_scale += 0.05
# self.blink(b.two_dim)
# 圆形二向箔不断旋转的效果
self.two_way_foil_circle.planet.rotation_z += 0.4
def on_timer_changed(self, time_data):
self.two_way_foil.planet.rotation_x += 0.1
self.two_way_foil.planet.rotation_y += 1
# self.two_way_foil.planet.rotation_z += 10
"""
@param time_data:
@return:
"""
# 原始方形二向箔飞行的翻转效果
if self.two_way_foil.planet.enabled:
self.two_way_foil.planet.rotation_x += 0.1
self.two_way_foil.planet.rotation_y += 1
# 摄像机始终看向二向箔
camera_look_at(self.two_way_foil)
self.current_stage()
# for b in self.two_dim_bodies:
# b.planet.rotation_y -= 0.5
# b.init_position = b.three_dim.position
if __name__ == '__main__':
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或