DR_Team/Difficult-Rocket
DR_Team/Difficult-Rocket
6
2
Fork 0
Code Issues Pull Requests Projects Releases 25 Wiki Activity

把渲染parts的部分包装成了draw_parts函数

Browse Source
This commit is contained in:
ssssssssboom 2024-05-23 19:46:05 +08:00
parent bb8d280892
commit c268d338c6
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
mods/dr_game/sr1_ship.py
View File

@ -189,12 +189,15 @@ class SR1ShipRender(BaseScreen):
self.logger.error(traceback.format_exc(), tag="load_xml") self.logger.error(traceback.format_exc(), tag="load_xml")
return False return False
def draw_parts(self, def draw_parts(self,
cache: Dict[int, List[Tuple[SR1PartType_rs, SR1PartData_rs]]], cache: List[Tuple[SR1PartType_rs, SR1PartData_rs]],
part_group: Group(2), count: int,
line_box_group: Group(6),
each_count: int, each_count: int,
draw_part_box: bool): draw_part_box: bool):
for p_id, parts in cache.items(): #渲染传入的parts
part_group = Group(2, parent=self.part_group)
line_box_group = Group(6, parent=self.part_group)
logger.info()
for p_id, parts in cache:
p_id: int p_id: int
parts: List[Tuple[SR1PartType_rs, SR1PartData_rs]] parts: List[Tuple[SR1PartType_rs, SR1PartData_rs]]
batch = [] batch = []
@ -281,7 +284,8 @@ class SR1ShipRender(BaseScreen):
count += 1 count += 1
if count >= each_count: if count >= each_count:
count = 0 count = 0
yield return count
return count
def gen_sprite(self, each_count: int = 100) -> Generator: def gen_sprite(self, each_count: int = 100) -> Generator:
""" """
@ -294,8 +298,6 @@ class SR1ShipRender(BaseScreen):
self.status.draw_done = False self.status.draw_done = False
# rust 渲染 # rust 渲染
if DR_mod_runtime.use_DR_rust: if DR_mod_runtime.use_DR_rust:
part_group = Group(2, parent=self.part_group)
line_box_group = Group(6, parent=self.part_group)
''' '''
#渲染所有未连接零件 #渲染所有未连接零件
@ -307,15 +309,20 @@ class SR1ShipRender(BaseScreen):
#渲染所有已连接零件 #渲染所有已连接零件
draw_part_box = False draw_part_box = True
cache = self.rust_ship.as_dict() cache = self.rust_ship.as_dict()
self.draw_parts(cache, logger.info(cache)
part_group, count=self.draw_parts(cache.items(),
line_box_group, count,
each_count, each_count,
draw_part_box) draw_part_box)
if count >= each_count:
count = 0
yield
connect_line_group = Group(7, parent=self.part_group) connect_line_group = Group(7, parent=self.part_group)
for connect in self.rust_ship.connections().get_raw_data: for connect in self.rust_ship.connections().get_raw_data():
# 连接线 # 连接线
parent_part_data = cache[connect[2]][0][1] parent_part_data = cache[connect[2]][0][1]
child_part_data = cache[connect[3]][0][1] child_part_data = cache[connect[3]][0][1]