From 9bbece1da4e497790dd2d5ba6bffaf8d7eb8d243 Mon Sep 17 00:00:00 2001
From: shenjack <3695888@qq.com>
Date: Wed, 22 May 2024 23:16:06 +0800
Subject: [PATCH] =?UTF-8?q?=E7=9C=8B=E7=9D=80=E8=BF=98=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/src/python/data.rs                    | 37 ++++++++++++++++++-
 1 file changed, 35 insertions(+), 2 deletions(-)

diff --git a/mods/dr_game/Difficult_Rocket_rs/src/src/python/data.rs b/mods/dr_game/Difficult_Rocket_rs/src/src/python/data.rs
index 754636d..051db48 100644
--- a/mods/dr_game/Difficult_Rocket_rs/src/src/python/data.rs
+++ b/mods/dr_game/Difficult_Rocket_rs/src/src/python/data.rs
@@ -238,7 +238,10 @@ pub struct PySR1PartArray {
 #[pymethods]
 impl PySR1PartArray {
     fn __iter__(&self) -> PySR1PartDataIterator {
-        PySR1PartDataIterator { datas: self.datas.clone(), index: 0 }
+        PySR1PartDataIterator {
+            datas: self.datas.clone(),
+            index: 0,
+        }
     }
 }
 
@@ -278,7 +281,37 @@ impl PySR1Ship {
                 parts.push((part_type, py_part_data));
             }
         }
-        PySR1PartArray { datas: parts, part_list: self.part_list.clone() }
+        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, _) in parts.iter() {
+                    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());
+                            part_list.push((part_type, py_part_data));
+                        }
+                    }
+                    result.push(PySR1PartArray {
+                        datas: part_list,
+                        part_list: self.part_list.clone(),
+                    });
+                }
+                result
+            }
+            None => Vec::new(),
+        }
     }
 
     #[getter]