From 4e636b91b4697c9e9f92584211af3a8722985506 Mon Sep 17 00:00:00 2001
From: KCFeng425 <2100011801@stu.pku.edu.cn>
Date: Thu, 5 Jun 2025 14:08:30 +0800
Subject: [PATCH] Add Device MockHeater

---
 test/experiments/MockHeater.json              |  30 +++
 .../devices/Mock/MockHeater/MockHeater.py     | 199 ++++++++++++++++++
 unilabos/devices/Mock/MockHeater/__init__.py  |   0
 unilabos/registry/devices/MockHeater.yaml     |  65 ++++++
 4 files changed, 294 insertions(+)
 create mode 100644 test/experiments/MockHeater.json
 create mode 100644 unilabos/devices/Mock/MockHeater/MockHeater.py
 create mode 100644 unilabos/devices/Mock/MockHeater/__init__.py
 create mode 100644 unilabos/registry/devices/MockHeater.yaml

diff --git a/test/experiments/MockHeater.json b/test/experiments/MockHeater.json
new file mode 100644
index 0000000..680dbc5
--- /dev/null
+++ b/test/experiments/MockHeater.json
@@ -0,0 +1,30 @@
+{  
+    "nodes": [  
+        {  
+            "id": "MockHeater1",  
+            "name": "模拟加热器",  
+            "children": [],  
+            "parent": null,  
+            "type": "device",  
+            "class": "MockHeater",  
+            "position": {  
+                "x": 620.6111111111111,  
+                "y": 171,  
+                "z": 0  
+            },  
+            "config": {  
+                "port": "MOCK"  
+            },  
+            "data": {  
+                "current_temperature": 25.0,  
+                "target_temperature": 25.0,  
+                "status": "Idle",  
+                "power_on": false,  
+                "is_heating": false,  
+                "heating_power": 0.0,  
+                "max_temperature": 300.0  
+            }  
+        }  
+    ],  
+    "links": []  
+}
\ No newline at end of file
diff --git a/unilabos/devices/Mock/MockHeater/MockHeater.py b/unilabos/devices/Mock/MockHeater/MockHeater.py
new file mode 100644
index 0000000..be67353
--- /dev/null
+++ b/unilabos/devices/Mock/MockHeater/MockHeater.py
@@ -0,0 +1,199 @@
+import time  
+import threading  
+  
+  
+class MockHeater:  
+    def __init__(self, port: str = "MOCK"):  
+        self.port = port  
+        self._current_temperature: float = 25.0  # 室温开始  
+        self._target_temperature: float = 25.0  
+        self._status: str = "Idle"  
+        self._is_heating: bool = False  
+        self._power_on: bool = False  
+        self._heating_power: float = 0.0  # 加热功率百分比 0-100  
+        self._max_temperature: float = 300.0  # 最大加热温度  
+          
+        # 模拟加热过程的线程  
+        self._heating_thread = None  
+        self._running = False  
+      
+    @property  
+    def current_temperature(self) -> float:  
+        """当前温度 - 会被自动识别的设备属性"""  
+        return self._current_temperature  
+      
+    @property  
+    def target_temperature(self) -> float:  
+        """目标温度"""  
+        return self._target_temperature  
+      
+    @property  
+    def status(self) -> str:  
+        """设备状态 - 会被自动识别的设备属性"""  
+        return self._status  
+      
+    @property  
+    def power_on(self) -> bool:  
+        """电源状态"""  
+        return self._power_on  
+      
+    @property  
+    def is_heating(self) -> bool:  
+        """是否正在加热"""  
+        return self._is_heating  
+      
+    @property  
+    def heating_power(self) -> float:  
+        """加热功率百分比"""  
+        return self._heating_power  
+      
+    @property  
+    def max_temperature(self) -> float:  
+        """最大加热温度"""  
+        return self._max_temperature  
+      
+    def set_temperature(self, temperature: float):
+        """设置目标温度 - 需要在注册表添加的设备动作"""
+        try:
+            temperature = float(temperature)
+        except ValueError:
+            self._status = "Error: Invalid temperature value"
+            return False
+
+        if not self._power_on:
+            self._status = "Error: Power Off"
+            return False
+
+        if temperature > self._max_temperature:
+            self._status = f"Error: Temperature exceeds maximum ({self._max_temperature}°C)"
+            return False
+
+        self._target_temperature = temperature
+        self._status = "Setting Temperature"
+
+        # 启动加热控制
+        self._start_heating_control()
+        return True 
+      
+    def set_heating_power(self, power: float):
+        """设置加热功率"""
+        try:
+            power = float(power)
+        except ValueError:
+            self._status = "Error: Invalid power value"
+            return False
+
+        if not self._power_on:
+            self._status = "Error: Power Off"
+            return False
+
+        self._heating_power = max(0.0, min(100.0, power))  # 限制在0-100%
+        return True
+      
+    def power_on_off(self, power_state: str):
+        """开关机控制，接收字符串命令 "On" 或 "Off" """
+        power_state = power_state.capitalize()
+        if power_state not in ["On", "Off"]:
+            self._status = "Error: Invalid power state"
+            return "Error"
+        self._power_on = True if power_state == "On" else False
+        if self._power_on:
+            self._status = "Power On"
+            self._start_heating_control()
+        else:
+            self._status = "Power Off"
+            self._stop_heating_control()
+            self._is_heating = False
+            self._heating_power = 0.0
+        return "Success"
+      
+    def _start_heating_control(self):  
+        """启动加热控制线程"""  
+        if not self._running and self._power_on:  
+            self._running = True  
+            self._heating_thread = threading.Thread(target=self._heating_control_loop)  
+            self._heating_thread.daemon = True  
+            self._heating_thread.start()  
+      
+    def _stop_heating_control(self):  
+        """停止加热控制"""  
+        self._running = False  
+        if self._heating_thread:  
+            self._heating_thread.join(timeout=1.0)  
+      
+    def _heating_control_loop(self):  
+        """加热控制循环 - 模拟真实加热器的温度变化"""  
+        while self._running and self._power_on:  
+            temp_diff = self._target_temperature - self._current_temperature  
+              
+            if abs(temp_diff) < 0.5:  # 温度接近目标值  
+                self._status = "At Target Temperature"  
+                self._is_heating = False  
+                self._heating_power = 10.0  # 维持温度的最小功率  
+            elif temp_diff > 0:  # 需要加热  
+                self._status = "Heating"  
+                self._is_heating = True  
+                # 根据温差调整加热功率  
+                if temp_diff > 50:  
+                    self._heating_power = 100.0  
+                elif temp_diff > 20:  
+                    self._heating_power = 80.0  
+                elif temp_diff > 10:  
+                    self._heating_power = 60.0  
+                else:  
+                    self._heating_power = 40.0  
+                  
+                # 模拟加热过程，加热速度与功率成正比  
+                heating_rate = self._heating_power / 100.0 * 2.0  # 最大每秒升温2度  
+                self._current_temperature += heating_rate  
+            else:  # 目标温度低于当前温度，自然冷却  
+                self._status = "Cooling Down"  
+                self._is_heating = False  
+                self._heating_power = 0.0  
+                # 模拟自然冷却，每秒降低0.2度  
+                self._current_temperature -= 0.2  
+              
+            # 限制温度范围  
+            self._current_temperature = max(20.0, min(self._max_temperature, self._current_temperature))  
+              
+            time.sleep(1.0)  # 每秒更新一次  
+      
+    def emergency_stop(self):  
+        """紧急停止"""  
+        self._status = "Emergency Stop"  
+        self._stop_heating_control()  
+        self._is_heating = False  
+        self._heating_power = 0.0  
+      
+    def get_status_info(self) -> dict:  
+        """获取完整状态信息"""  
+        return {  
+            "current_temperature": self._current_temperature,  
+            "target_temperature": self._target_temperature,  
+            "status": self._status,  
+            "power_on": self._power_on,  
+            "is_heating": self._is_heating,  
+            "heating_power": self._heating_power,  
+            "max_temperature": self._max_temperature  
+        }  
+  
+  
+# 用于测试的主函数  
+if __name__ == "__main__":  
+    heater = MockHeater()  
+      
+    # 测试基本功能  
+    print("启动加热器测试...")  
+    heater.power_on_off(True)  
+    print(f"初始状态: {heater.get_status_info()}")  
+      
+    # 设置目标温度为80度  
+    heater.set_temperature(80.0)  
+      
+    # 模拟运行15秒  
+    for i in range(15):  
+        time.sleep(1)  
+        print(f"第{i+1}秒: 当前温度={heater.current_temperature:.1f}°C, 功率={heater.heating_power:.1f}%, 状态={heater.status}")  
+      
+    heater.emergency_stop()  
+    print("测试完成")
\ No newline at end of file
diff --git a/unilabos/devices/Mock/MockHeater/__init__.py b/unilabos/devices/Mock/MockHeater/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/unilabos/registry/devices/MockHeater.yaml b/unilabos/registry/devices/MockHeater.yaml
new file mode 100644
index 0000000..b366401
--- /dev/null
+++ b/unilabos/registry/devices/MockHeater.yaml
@@ -0,0 +1,65 @@
+MockHeater:  
+  description: Mock Heater Device  
+  class:  
+    module: unilabos.devices.Mock.MockHeater.MockHeater:MockHeater  
+    type: python  
+    status_types:  
+      current_temperature: Float64  
+      target_temperature: Float64  
+      status: String  
+      power_on: Bool  
+      is_heating: Bool  
+      heating_power: Float64  
+      max_temperature: Float64  
+    action_value_mappings:  
+      set_temperature:  
+        type: SendCmd  
+        goal:  
+          command: temperature  
+        feedback: {}  
+        result:  
+          success: success  
+      set_heating_power:  
+        type: SendCmd  
+        goal:  
+          command: power  
+        feedback: {}  
+        result:  
+          success: success  
+      power_on_off:  
+        type: SendCmd  
+        goal:  
+          command: power_state  
+        feedback: {}  
+        result:  
+          success: success  
+  schema:  
+    type: object  
+    properties:  
+      current_temperature:  
+        type: number  
+        description: Current temperature of the heater  
+      target_temperature:  
+        type: number  
+        description: Target temperature setting  
+      status:  
+        type: string  
+        description: Current status of the device  
+      power_on:  
+        type: boolean  
+        description: Power state of the device  
+      is_heating:  
+        type: boolean  
+        description: Whether the device is actively heating  
+      heating_power:  
+        type: number  
+        description: Current heating power percentage  
+      max_temperature:  
+        type: number  
+        description: Maximum heating temperature limit  
+    required:  
+      - current_temperature  
+      - target_temperature  
+      - status  
+      - power_on  
+    additionalProperties: false
\ No newline at end of file