libs/pyglet

@@ -1 +1 @@
-Subproject commit 55b02b16d01bd0cafa5b0db3cd2e574299db9d34
+Subproject commit ea875640db0e529769fcfe597faa3c3267c2b1e3

mods/dr_game/Difficult_Rocket_rs/__init__.py

@@ -9,9 +9,6 @@ from .lib import *  # noqa: F403
 from typing import TYPE_CHECKING, Dict, Tuple, Optional, List
 
 if TYPE_CHECKING:
-    IdType = int
-    RawConnectionData = Tuple[int, int, IdType, IdType]
-
     def get_version_str() -> str:
         """
         获取版本号
@@ -166,42 +163,6 @@ if TYPE_CHECKING:
         def __init__(self, save_default: Optional[bool] = False) -> None:
             ...
 
-    class SR1Connection_rs:  # NOQA
-        """用于存储一堆连接的信息
-        同时提供了一些好用的 API
-        至少总比你在那里吭哧吭哧在 Python 里搜快
-        pub datas: Vec<Connection>"""
-
-        def search_connection_by_parent(
-            self, parent_id: IdType
-        ) -> List[RawConnectionData]:
-            """通过父节点搜索连接"""
-            ...
-
-        def search_connection_by_child(self, child_id: IdType) -> List[RawConnectionData]:
-            """通过子节点搜索连接"""
-            ...
-
-        def search_connection_by_id(self, any_id: IdType) -> List[RawConnectionData]:
-            """通过父子中任意一个 id 搜索连接"""
-            ...
-
-        def search_by_both_id(
-            self, parent_id: IdType, child_id: IdType
-        ) -> List[RawConnectionData]:
-            """通过父子双方 id 获取连接
-
-            保险起见, 我还是返回一个 Vec
-
-            万一真有 双/多 连接呢"""
-            ...
-
-        def get_raw_data(self) -> List[RawConnectionData]:
-            """获取原始数据
-
-            万一你确实需要吭哧吭哧去处理原始数据呢"""
-            ...
-
     class SR1Ship_rs:  # NOQA
         """用于高效且省内存的读取 SR1Ship"""
 
@@ -245,6 +206,11 @@ if TYPE_CHECKING:
             """-x -y +x +y  左下右上"""
             ...
 
+        @property
+        def connection(self) -> List[Tuple[int, int, int, int]]:
+            """获取所有连接信息"""
+            ...
+
         def get_part_box(
             self, part_id: int
         ) -> Optional[Tuple[Tuple[int, int], Tuple[int, int]]]:
@@ -263,16 +229,6 @@ if TYPE_CHECKING:
         ) -> None:
             ...
 
-        def connections(self) -> SR1Connection_rs:
-            """获取所有连接"""
-            ...
-
-        def disconnected_parts(
-            self
-        ) -> List[Tuple[List[Tuple[SR1PartType_rs, SR1PartData_rs]], SR1Connection_rs]]:
-            """获取所有未连接的零件"""
-            ...
-
     class Console_rs:  # NOQA
         def __init__(self) -> None:
             ...

mods/dr_game/Difficult_Rocket_rs/src/Cargo.lock

@@ -146,7 +146,7 @@ dependencies = [
 
 [[package]]
 name = "difficult_rocket_rs"
-version = "0.3.4"
+version = "0.3.3"
 dependencies = [
  "anyhow",
  "dict_derive",

mods/dr_game/Difficult_Rocket_rs/src/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "difficult_rocket_rs"
-version = "0.3.4"
+version = "0.3.3"
 edition = "2021"
 license-file = '../../LICENSE'
 authors = ["shenjackyuanjie <3695888@qq.com>"]

mods/dr_game/Difficult_Rocket_rs/src/src/lib.rs

@@ -31,7 +31,6 @@ fn module_init(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
     m.add_class::<python::data::PySR1PartType>()?;
     m.add_class::<python::data::PySR1PartData>()?;
     m.add_class::<python::data::PySaveStatus>()?;
-    m.add_class::<python::data::PySR1Connections>()?;
     m.add_class::<python::console::PyConsole>()?;
     // m.add_class::<python::editor::EditorArea>()?;
     Ok(())

mods/dr_game/Difficult_Rocket_rs/src/src/python/data.rs

@@ -5,7 +5,7 @@ use pyo3::prelude::*;
 
 use crate::dr_physics::math::{Point2D, Rotate};
 use crate::sr1_parse::part_list::RawPartList;
-use crate::sr1_parse::ship::{Connection, Connections, RawConnectionData, RawShip};
+use crate::sr1_parse::ship::RawShip;
 use crate::sr1_parse::SaveStatus;
 use crate::sr1_parse::{get_max_box, SR1PartData, SR1PartListTrait};
 use crate::sr1_parse::{SR1PartList, SR1PartType, SR1Ship};
@@ -207,47 +207,41 @@ impl PySR1PartData {
 }
 
 #[pyclass]
-#[derive(Debug, Clone)]
+#[derive(Clone, Debug)]
-#[pyo3(name = "SR1Connection_rs")]
+#[pyo3(name = "SR1PartTypeArrayIterator_rs")]
-pub struct PySR1Connections {
+pub struct PySR1PartDataIterator {
-    pub datas: Vec<Connection>,
+    pub datas: Vec<(PySR1PartType, PySR1PartData)>,
+    pub index: usize,
 }
 
 #[pymethods]
-impl PySR1Connections {
+impl PySR1PartDataIterator {
-    /// 通过父节点获取连接
+    fn __next__(mut slf: PyRefMut<'_, Self>) -> Option<(PySR1PartType, PySR1PartData)> {
-    fn search_connection_by_parent(&self, parent_id: IdType) -> Vec<RawConnectionData> {
+        if slf.index < slf.datas.len() {
-        self.datas.iter().filter(|x| x.parent_part == parent_id).map(|x| x.as_raw_data()).collect()
+            slf.index += 1;
+            Some(slf.datas[slf.index - 1].clone())
+        } else {
+            None
         }
-    /// 通过子节点获取连接
-    fn search_by_child(&self, child_id: IdType) -> Vec<RawConnectionData> {
-        self.datas.iter().filter(|x| x.child_part == child_id).map(|x| x.as_raw_data()).collect()
     }
-    /// 通过父子中任意一个 id 搜索连接
-    fn search_by_id(&self, any_id: IdType) -> Vec<RawConnectionData> {
-        self.datas
-            .iter()
-            .filter(|x| x.parent_part == any_id || x.child_part == any_id)
-            .map(|x| x.as_raw_data())
-            .collect()
 }
-    /// 通过父子双方 id 获取连接
+
-    ///
+/// 用来存一堆 PartData 的数组
-    /// 保险期间, 我还是返回一个 Vec
+#[pyclass]
-    ///
+#[derive(Clone, Debug)]
-    /// 万一真有 双/多 连接呢
+#[pyo3(name = "SR1PartArray_rs")]
-    fn search_by_both_id(&self, parent_id: IdType, child_id: IdType) -> Vec<RawConnectionData> {
+pub struct PySR1PartArray {
-        self.datas
+    pub datas: Vec<(PySR1PartType, PySR1PartData)>,
-            .iter()
+    pub part_list: SR1PartList,
-            .filter(|x| x.parent_part == parent_id && x.child_part == child_id)
+}
-            .map(|x| x.as_raw_data())
+
-            .collect()
+#[pymethods]
+impl PySR1PartArray {
+    fn __iter__(&self) -> PySR1PartDataIterator {
+        PySR1PartDataIterator {
+            datas: self.datas.clone(),
+            index: 0,
         }
-    /// 获取所有连接的原始数据
-    ///
-    /// 万一你确实需要吭哧吭哧去处理原始数据呢
-    fn get_raw_data(&self) -> Vec<RawConnectionData> {
-        self.datas.iter().map(|x| x.as_raw_data()).collect()
     }
 }
 
@@ -278,28 +272,41 @@ impl PySR1Ship {
         }
     }
 
-    fn disconnected_parts(&self) -> Vec<(Vec<(PySR1PartType, PySR1PartData)>, PySR1Connections)> {
+    fn parts(&self) -> PySR1PartArray {
+        let mut parts: Vec<(PySR1PartType, PySR1PartData)> = Vec::new();
+        for part_data in self.ship.parts.iter() {
+            if let Some(part_type) = self.part_list.get_part_type(&part_data.part_type_id) {
+                let part_type = PySR1PartType::new(part_type.clone());
+                let py_part_data = PySR1PartData::new(part_data.clone());
+                parts.push((part_type, py_part_data));
+            }
+        }
+        PySR1PartArray {
+            datas: parts,
+            part_list: self.part_list.clone(),
+        }
+    }
+
+    fn disconnected_parts(&self) -> Vec<PySR1PartArray> {
         match self.ship.disconnected.as_ref() {
             Some(parts) => {
                 if parts.is_empty() {
                     return Vec::new();
                 }
                 let mut result = Vec::with_capacity(parts.len());
-                for (part_group, connections) in parts.iter() {
+                for (part_group, _) in parts.iter() {
-                    let mut group_parts = Vec::with_capacity(part_group.len());
+                    let mut part_list = Vec::with_capacity(part_group.len());
                     for part_data in part_group.iter() {
                         if let Some(part_type) = self.part_list.get_part_type(&part_data.part_type_id) {
                             let part_type = PySR1PartType::new(part_type.clone());
                             let py_part_data = PySR1PartData::new(part_data.clone());
-                            group_parts.push((part_type, py_part_data));
+                            part_list.push((part_type, py_part_data));
                         }
                     }
-                    result.push((
+                    result.push(PySR1PartArray {
-                        group_parts,
+                        datas: part_list,
-                        PySR1Connections {
+                        part_list: self.part_list.clone(),
-                            datas: connections.clone().unwrap_or_default(),
+                    });
-                        },
-                    ));
                 }
                 result
             }
@@ -332,12 +339,12 @@ impl PySR1Ship {
 
     #[getter]
     fn get_lift_off(&self) -> bool {
-        self.ship.lift_off
+        self.ship.lift_off.to_owned()
     }
 
     #[getter]
     fn get_touch_ground(&self) -> bool {
-        self.ship.touch_ground
+        self.ship.touch_ground.to_owned()
     }
 
     #[getter]
@@ -350,10 +357,18 @@ impl PySR1Ship {
         mass
     }
 
-    fn connections(&self) -> PySR1Connections {
+    #[getter]
-        PySR1Connections {
+    fn get_connection(&self) -> Vec<(i32, i32, IdType, IdType)> {
-            datas: self.ship.connections.clone(),
+        let mut connections = Vec::new();
+        for connect in self.ship.connections.iter() {
+            connections.push((
+                connect.parent_attach_point,
+                connect.child_attach_point,
+                connect.parent_part,
+                connect.child_part,
+            ));
         }
+        connections
     }
 
     fn as_list(&self) -> Vec<(PySR1PartType, PySR1PartData)> {

mods/dr_game/Difficult_Rocket_rs/src/src/sr1_parse/data_structure/ship.rs

@@ -155,8 +155,6 @@ pub struct DisconnectedPart {
     pub connects: Connections,
 }
 
-pub type RawConnectionData = (i32, i32, IdType, IdType);
-
 #[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct Connection {
     #[serde(rename = "@parentAttachPoint")]
@@ -169,18 +167,6 @@ pub struct Connection {
     pub child_part: IdType,
 }
 
-impl Connection {
-    /// 真有地方需要
-    pub fn as_raw_data(&self) -> RawConnectionData {
-        (
-            self.parent_attach_point,
-            self.child_attach_point,
-            self.parent_part,
-            self.child_part,
-        )
-    }
-}
-
 impl SR1PartDataTrait for Part {
     fn to_sr_part_data(&self) -> SR1PartData {
         let attr = SR1PartDataAttr::from_raw(self, None, true);

mods/dr_game/__init__.py

@@ -16,7 +16,7 @@ from Difficult_Rocket.api.types import Options, Version
 
 from lib_not_dr import loggers
 
-DR_rust_version = Version("0.3.4")  # DR_mod 的 Rust 编写部分的兼容版本
+DR_rust_version = Version("0.3.3")  # DR_mod 的 Rust 编写部分的兼容版本
 
 logger = loggers.config.get_logger_from_old("client.dr_game", "client")
 

mods/dr_game/menu.py

@@ -10,18 +10,10 @@ from pyglet.graphics import Batch, Group
 from Difficult_Rocket.client import ClientWindow
 from Difficult_Rocket.api.screen import BaseScreen
 from Difficult_Rocket.main import Game
-from Difficult_Rocket.gui.widget.button import (
+from Difficult_Rocket.gui.widget.button import PressTextButton, MinecraftWikiButtonTheme, ButtonThemeOptions, BaseButtonTheme
-    PressTextButton,
-    MinecraftWikiButtonTheme,
-    ButtonThemeOptions,
-    BaseButtonTheme,
-)
-
-from lib_not_dr import loggers
 
 # from . import DR_mod_runtime
 
-logger = loggers.config.get_logger_from_old("client.dr_game_menu", "client")
 
 class Menu(BaseScreen):
     """
@@ -39,59 +31,59 @@ class Menu(BaseScreen):
         # 欸呀, 正好是 911 纪念日哦
         # 好, 高二 第一学期 期末都考完了, 我过来做测试了 (20240119)
 
-        # self.wiki_button1 = PressTextButton(
+        self.wiki_button1 = PressTextButton(
-        #     x=200,
+            x=200,
-        #     y=300,
+            y=300,
-        #     width=150,
+            width=150,
-        #     height=30,
+            height=30,
-        #     text="wiki button1",
+            text="wiki button1",
-        #     batch=self.main_batch,
+            batch=self.main_batch,
-        #     group=self.main_group,
+            group=self.main_group,
-        #     draw_theme=MinecraftWikiButtonTheme,
+            draw_theme=MinecraftWikiButtonTheme,
-        #     dict_theme={"pop_out": True},
+            dict_theme={"pop_out": True},
-        # )
+        )
-        # self.wiki_button2 = PressTextButton(
+        self.wiki_button2 = PressTextButton(
-        #     x=400,
+            x=400,
-        #     y=300,
+            y=300,
-        #     width=150,
+            width=150,
-        #     height=30,
+            height=30,
-        #     text="wiki button2",
+            text="wiki button2",
-        #     batch=self.main_batch,
+            batch=self.main_batch,
-        #     group=self.main_group,
+            group=self.main_group,
-        #     draw_theme=MinecraftWikiButtonTheme,
+            draw_theme=MinecraftWikiButtonTheme,
-        #     dict_theme={"pop_out": False},
+            dict_theme={"pop_out": False},
-        # )
+        )
-        # self.wiki_button3 = PressTextButton(
+        self.wiki_button3 = PressTextButton(
-        #     x=200,
+            x=200,
-        #     y=250,
+            y=250,
-        #     width=150,
+            width=150,
-        #     height=30,
+            height=30,
-        #     text="wiki button3",
+            text="wiki button3",
-        #     batch=self.main_batch,
+            batch=self.main_batch,
-        #     group=self.main_group,
+            group=self.main_group,
-        #     draw_theme=MinecraftWikiButtonTheme,
+            draw_theme=MinecraftWikiButtonTheme,
-        #     dict_theme={"pop_out": True, "drag_list": True},
+            dict_theme={"pop_out": True, "drag_list": True},
-        # )
+        )
-        # self.wiki_button4 = PressTextButton(
+        self.wiki_button4 = PressTextButton(
-        #     x=400,
+            x=400,
-        #     y=250,
+            y=250,
-        #     width=150,
+            width=150,
-        #     height=30,
+            height=30,
-        #     text="wiki button4",
+            text="wiki button4",
-        #     batch=self.main_batch,
+            batch=self.main_batch,
-        #     group=self.main_group,
+            group=self.main_group,
-        #     draw_theme=MinecraftWikiButtonTheme,
+            draw_theme=MinecraftWikiButtonTheme,
-        #     dict_theme={"pop_out": False, "drag_list": True},
+            dict_theme={"pop_out": False, "drag_list": True},
-        # )
+        )
-        # self.button3 = PressTextButton(
+        self.button3 = PressTextButton(
-        #     x=200,
+            x=200,
-        #     y=200,
+            y=200,
-        #     width=150,
+            width=150,
-        #     height=30,
+            height=30,
-        #     text="imgui button",
+            text="imgui button",
-        #     batch=self.main_batch,
+            batch=self.main_batch,
-        #     group=self.main_group,
+            group=self.main_group,
-        # )
+        )
         self.enter_ship_editor_button = PressEnterShipEditorButton(
             window=main_window,
             x=100,
@@ -101,19 +93,17 @@ class Menu(BaseScreen):
             text="进入编辑器",
             batch=self.main_batch,
             group=self.main_group,
-            draw_theme=MinecraftWikiButtonTheme
         )
-        # main_window.push_handlers(self.wiki_button1)
+        main_window.push_handlers(self.wiki_button1)
-        # main_window.push_handlers(self.wiki_button2)
+        main_window.push_handlers(self.wiki_button2)
-        # main_window.push_handlers(self.wiki_button3)
+        main_window.push_handlers(self.wiki_button3)
-        # main_window.push_handlers(self.wiki_button4)
+        main_window.push_handlers(self.wiki_button4)
-        # main_window.push_handlers(self.button3)
+        main_window.push_handlers(self.button3)
         main_window.push_handlers(self.enter_ship_editor_button)
 
     def on_draw(self, dt: float, window: ClientWindow):
         self.main_batch.draw()
 
-
 class PressEnterShipEditorButton(PressTextButton):
     def __init__(
         self,
@@ -127,14 +117,11 @@ class PressEnterShipEditorButton(PressTextButton):
         group: Optional[Group] = None,
         theme: Optional[ButtonThemeOptions] = None,
         draw_theme: Optional[type(BaseButtonTheme)] = None,
-        dict_theme: Optional[dict] = None,
+        dict_theme: Optional[dict] = None,):
-    ):
+        super().__init__(x,y,width,height,text,batch,group,theme,draw_theme,dict_theme)
-        super().__init__(
-            x, y, width, height, text, batch, group, theme, draw_theme, dict_theme
-        )
         self.window=window
         
-
+    #from lib_not_dr import loggers
     def on_mouse_release(self, x, y, buttons, modifiers):
         if self.pressed and (x, y) in self:
             if self.draw_theme:
@@ -144,7 +131,8 @@ class PressEnterShipEditorButton(PressTextButton):
             self.pressed = False
                     
             from .sr1_ship import SR1ShipRender
-
             self.window.remove_sub_screen("DR_game_menu")
             self.window.add_sub_screen("SR1_ship", SR1ShipRender)
-            logger.info("added SR1_ship screen", tag="dr_game")
+            #logger.info("added SR1_ship screen", tag="dr_game")
+
+

mods/dr_game/sr1_ship.py

@@ -38,7 +38,6 @@ if DR_mod_runtime.use_DR_rust:
         SR1Ship_rs,
         SR1PartData_rs,
         SR1PartType_rs,
-        # SR1Connection_rs
     )
 
 
@@ -292,7 +291,7 @@ class SR1ShipRender(BaseScreen):
                     count = 0
                     yield
             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.connection:
                 # 连接线
                 parent_part_data = cache[connect[2]][0][1]
                 child_part_data = cache[connect[3]][0][1]
@@ -372,7 +371,7 @@ class SR1ShipRender(BaseScreen):
             sr_tr()
             .sr1.ship.ship.load_time()
             .format((time.perf_counter_ns() - start_time) / 1000000000),
-            tag="ship",
+            tag="ship"
         )
         logger.info(
             sr_tr()
@@ -539,8 +538,7 @@ class SR1ShipRender(BaseScreen):
                 for index, sprite in enumerate(sprites):
                     sprite_img = sprite.image
                     print(
-                        f"sprite_img: {sprite_img} {part_data[part][index][1].x * 60} "
+                        f"sprite_img: {sprite_img} {part_data[part][index][1].x * 60} {part_data[part][index][1].y * 60}"
-                        f"{part_data[part][index][1].y * 60}"
                     )
                     img_data = sprite_img.get_image_data()
                     fmt = img_data.format