Create iot_simulator.py

iot_simulator.py

@@ -0,0 +1,46 @@
+import random
+import time
+import numpy as np
+from typing import Dict, List
+
+class IoTSimulator:
+    """Simulates a robotic arm with sensors."""
+    def __init__(self, seed: int = 42):
+        random.seed(seed)
+        self.step = 0
+        self.base_temp = 35.0
+        self.base_vibration = 0.1
+        self.base_current = 2.0
+        self.base_position_error = 0.01
+        self.fault_mode = None  # 'overheat', 'vibration', 'stall', 'drift'
+
+    def set_fault(self, fault_type: str):
+        self.fault_mode = fault_type
+
+    def clear_fault(self):
+        self.fault_mode = None
+
+    def read(self) -> Dict[str, float]:
+        self.step += 1
+        # Normal variation
+        temp = self.base_temp + random.gauss(0, 0.5)
+        vib = self.base_vibration + random.gauss(0, 0.02)
+        current = self.base_current + random.gauss(0, 0.1)
+        pos_err = self.base_position_error + random.gauss(0, 0.005)
+
+        # Inject fault
+        if self.fault_mode == 'overheat':
+            temp += 5 + 0.1 * self.step
+        elif self.fault_mode == 'vibration':
+            vib += 0.5 + 0.02 * self.step
+        elif self.fault_mode == 'stall':
+            current += 1.0 + 0.1 * self.step
+        elif self.fault_mode == 'drift':
+            pos_err += 0.02 + 0.002 * self.step
+
+        return {
+            'temperature': max(0, temp),
+            'vibration': max(0, vib),
+            'motor_current': max(0, current),
+            'position_error': max(0, pos_err)
+        }