# -*- coding:utf-8 -*- # title :ursina天体Planet # description :ursina天体Planet # author :Python超人 # date :2023-02-11 # link :https://gitcode.net/pythoncr/ # python_version :3.8 # ============================================================================== # pip install -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com ursina from ursina import application, Entity, camera, color, Vec3, Text, load_texture, destroy, PointLight from simulators.ursina.entities.entity_utils import create_name_text, create_trails, clear_trails, create_rings, \ trail_init, create_fixed_star_lights from simulators.ursina.ursina_config import UrsinaConfig from simulators.ursina.ursina_event import UrsinaEvent from common.color_utils import adjust_brightness, conv_to_vec4_color, get_inverse_color from common.func import find_file from simulators.views.body_view import BodyView from simulators.ursina.ursina_mesh import create_sphere, create_torus, create_arrow_line, create_line, create_label import math class Planet(Entity): @property def body(self): return self.body_view.body def on_reset(self): # 删除拖尾 clear_trails(self) self.body.reset() @property def init_scale(self): return self.__init_scale @init_scale.setter def init_scale(self, val): self.body_scale *= val / self.__init_scale self.__init_scale = val def __init__(self, body_view: BodyView): self.body_view = body_view if hasattr(self.body, "rotation_speed"): self.rotation_speed = self.body.rotation_speed self.rotMode = 'x' # random.choice(["x", "y", "z"]) self.name = body_view.name pos = body_view.position * self.body.distance_scale * UrsinaConfig.SCALE_FACTOR if hasattr(self.body, "diameter"): scale = self.body.diameter * self.body.size_scale * UrsinaConfig.SCALE_FACTOR else: scale = self.body.size_scale * UrsinaConfig.SCALE_FACTOR self.__init_scale = scale self.body_scale = 1.0 if hasattr(body_view, "texture"): texture = load_texture(body_view.texture) # color.white self.plant_color = color.white else: texture = None b_color = self.body_view.color if len(b_color) == 3: b_color = (b_color[0], b_color[1], b_color[2], 1.0) self.plant_color = color.rgba(*b_color) collider = "sphere" if hasattr(self.body, "torus_stars"): # 创建一个星环小天体群（主要模拟小行星群，非一个天体） model = create_torus(0.83, 1.05, 64, 1) rotation = (90, 0, 0) elif hasattr(self.body, "torus_zone"): # 创建一个星环小天体群（主要模拟环形带，非一个天体） inner_radius, outer_radius, subdivisions = self.body.torus_zone if subdivisions is None: subdivisions = 64 model = create_torus(inner_radius, outer_radius, subdivisions, 1) rotation = (90, 0, 0) else: # 创建一个天体 subdivisions = 32 if self.body.resolution is not None: subdivisions = self.body.resolution if hasattr(self.body, "model"): if self.body.model is None: model = create_sphere(0.5, subdivisions) elif isinstance(self.body.model, str): model = self.body.model else: model = self.body.model if hasattr(self.body, "rotation"): if self.body.rotation is None: rotation = (0, 0, 0) else: rotation = tuple(self.body.rotation) else: model = create_sphere(0.5, subdivisions) rotation = (0, 0, 0) # 按键盘的 “O” 重置键会触发 on_reset UrsinaEvent.on_reset_subscription(self.on_reset) # 当对天体大小进行缩放的时候会触发 change_body_scale UrsinaEvent.on_body_size_changed_subscription(self.change_body_scale) # if model == 'circle': # from ursina.models.procedural.circle import Circle # model = Circle() super().__init__( # model="sphere", model=model, scale=scale, texture=texture, color=self.plant_color, collider=collider, position=pos, rotation=rotation, ignore_paused=True, double_sided=True ) if hasattr(self.body, "rotate_angle"): if self.body.rotate_angle != 0: # 为了给天体增加一个倾斜角，增加了一个Entity self.create_rotate_entity() # self.rotate_angle = self.body.rotate_angle # self.main_entity = Entity() # self.main_entity.rotation_x = self.rotate_angle # self.main_entity.body_view = self.body_view # self.main_entity.body = self.body # self.parent = self.main_entity # self.position = [0, 0, 0] else: self.rotate_angle = 0 self.main_entity = self else: self.rotate_angle = 0 self.main_entity = self # Rotation axis color if hasattr(self.body, "rotate_axis_color"): if self.body.rotate_axis_color is not None: axis_color = self.body.rotate_axis_color axis_color = (axis_color[0] / 255, axis_color[1] / 255, axis_color[2] / 255, 1.0) axis_color = color.rgba(*axis_color) self.create_rotate_line(axis_color) if hasattr(self.body, "torus_stars") or hasattr(self.body, "torus_zone"): # 星环小天体群（主要模拟小行星群，非一个天体） self.set_light_off() self.double_sided = True else: # 一个天体 # 拖尾球体的初始化 trail_init(self.main_entity, self.scale_x) if hasattr(self.body, "is_fixed_star"): if self.body.is_fixed_star: # 如果是恒星，开启恒星的发光的效果、并作为灯光源 create_fixed_star_lights(self) elif self.body.light_disable: # 如果是非恒星，并且禁用灯光 self.set_light_off() elif self.body.light_disable: # 如果是非恒星，并且禁用灯光 self.set_light_off() if self.body.show_name: create_name_text(self) if hasattr(self.body, "has_rings"): if self.body.has_rings: # 创建行星环（目前只有土星环） create_rings(self) def create_rotate_entity(self): """ @return: """ self.rotate_angle = self.body.rotate_angle self.main_entity = Entity() self.main_entity.rotation_x = self.rotate_angle self.main_entity.body_view = self.body_view self.main_entity.body = self.body self.parent = self.main_entity self.position = [0, 0, 0] def create_rotate_line(self, line_color): from_pos = Vec3(0, 1, 0) to_pos = Vec3(0, -1, 0) # UrsinaConfig.SCALE_FACTOR * 10000000 = 5 # UrsinaConfig.auto_scale_factor = 1 # UrsinaConfig.body_size_factor = 1 # if self.main_entity is self: # # 没有偏转角度 # line_scale = math.pow(self.main_entity.scale_x, 1 / 10) / 1.5 # else: # # 有偏转角度 # # line_scale = math.pow(self.main_entity.scale_x, 1 / 10) # line_scale = self.scale_x # camera.scale_x line_scale = self.scale_x / self.main_entity.scale_x # print(self.body.name, self.main_entity.scale_x, self.scale_x, line_scale) # 0.006373216398060322 0.006373216398060322 0.40210975353339423 # 1.0 0.006373216398060322 0.006373216398060322 if hasattr(self.body, "rotate_axis_scale"): line_scale = self.body.rotate_axis_scale * line_scale self.rotate_axis_line = create_line(from_pos, to_pos, parent=self.main_entity, len_scale=line_scale, color=line_color, thickness=2) def change_body_scale(self): if hasattr(self.body, "torus_stars") or hasattr(self.body, "torus_zone"): # 星环小天体群（主要模拟小行星群，非一个天体）不受 body_size_factor 影响 self.scale = self.init_scale else: self.scale = self.init_scale * UrsinaConfig.body_size_factor if hasattr(self, "rotate_axis_line"): if not hasattr(self, "rotate_axis_init_scale"): self.rotate_axis_init_scale = self.rotate_axis_line.scale self.rotate_axis_line.scale = self.rotate_axis_init_scale * UrsinaConfig.body_size_factor # def self_rotate(self, dt): # rotspeed = self.rotation_speed * (dt / 3600) / 2.4 * \ # UrsinaConfig.ROTATION_SPEED_FACTOR * UrsinaConfig.body_spin_factor # self.rotation_y -= rotspeed def update(self): self.change_body_scale() if application.paused: return pos = self.body_view.position * UrsinaConfig.SCALE_FACTOR if self.body.parent is None: # self.x = -pos[1] # self.y = pos[2] # self.z = pos[0] self.main_entity.x = pos[0] self.main_entity.y = pos[1] self.main_entity.z = pos[2] else: self.follow_parent() dt = 0 if hasattr(self.body, "dt"): dt = self.body.dt if hasattr(self, "rotation_speed"): if self.rotation_speed is None or dt == 0: self.rotspeed = 0 # 旋转速度和大小成反比（未使用真实数据） # self.rotspeed = 30000 / self.body_view.radius # random.uniform(1.0, 2.0) else: # 是通过月球保持一面面对地球，调整得到 self.rotspeed = self.rotation_speed * (dt / 3600) / 2.4 * \ UrsinaConfig.ROTATION_SPEED_FACTOR * UrsinaConfig.body_spin_factor # rotation_speed 度/小时 dt 秒 = (dt / 3600)小时 # if self.rotation_y < 0: # self.rotation_y += 360 try: if hasattr(self, "rotspeed"): # 天体旋转 self.rotation_y -= self.rotspeed except Exception as e: print(self.body) self.destroy_all() return # 如果有行星环 if hasattr(self, "ring"): # 如果有行星环，则不让行星环跟随行星转动 self.ring.rotation = -Vec3(self.rotation_x - self.ring_rotation_x, self.rotation_y, self.rotation_z) if not hasattr(self.body, "show_trail") or self.body.show_trail: if UrsinaConfig.show_trail: # 有时候第一个位置不正确，所以判断一下有历史记录后在创建 if len(self.body.his_position()) > 1: create_trails(self.main_entity) else: clear_trails(self.main_entity) if hasattr(self, "name_text"): d = (camera.world_position - self.name_text.world_position).length() if d < pow(self.scale_x, 1.02) * 1.2: self.name_text.visible = False else: self.name_text.visible = True # print(d, self.name_text.text, self.scale_x ,self.scale_x*1.23) # # 计算相机和实体之间的距离 # distance = (camera.world_position - self.world_position).length() # # 根据距离设置文本缩放比例 # self.name_text.scale = distance / 10 def follow_parent(self): if not hasattr(self, "f_parent"): if not hasattr(self.body_view, "bodies_system"): return sys = self.body_view.bodies_system for b in sys.bodies: if self.body.parent == b: self.f_parent = b break pos = self.f_parent.position * UrsinaConfig.SCALE_FACTOR # self.x = -pos[1] # self.y = pos[2] # self.z = pos[0] self.main_entity.x = pos[0] self.main_entity.y = pos[1] self.main_entity.z = pos[2] def destroy_all(self): # 从天体系统中移除自己（TODO:暂时还不能移除） # self.body_view.bodies_system.bodies.remove(self.body) # 删除拖尾 clear_trails(self.main_entity) # 如果有行星环，则删除行星环 if hasattr(self, "ring"): destroy(self.main_entity.ring) self.body.appeared = False self.body_view.appeared = False # 最后删除自己 destroy(self) def input(self, key): if self.hovered: if key == 'left mouse down': # print(key, self) # self.show_name() pass def show_name(self): if hasattr(self, "label_name"): destroy(self.label_name) delattr(self, "label_name") else: self.label_name = create_label(parent=self, label=self.body.name, pos=Vec3(-0.5, -0.5, -0.5), color=color.red) self.label_name.set_light_off()