qlsdk2 0.2.0__py3-none-any.whl → 0.3.0a1__py3-none-any.whl

qlsdk/sdk/hub.py

@@ -0,0 +1,51 @@
+from qlsdk.sdk.ar4sdk import AR4SDK, LMDevice
+
+
+from time import sleep, time
+from threading import Lock, Timer
+from loguru import logger
+
+class Hub(object):
+    def __init__(self):
+        self._lock = Lock()
+        self._search_timer = None 
+        self._search_running = False
+        self._devices: dict[str, LMDevice] = {}
+    
+    @property
+    def devices(self):
+        return self._devices
+    
+    def search(self):
+        if not self._search_running:
+            self._search_running = True
+            self._search()
+        
+    def _search(self):
+        if self._search_running:
+
+            self._search_timer = Timer(2, self._search_device)
+            self._search_timer.daemon = True
+            self._search_timer.start()
+        
+        
+    def _search_device(self):
+        try:                        
+            devices = AR4SDK.enum_devices()
+            # logger.debug(f"_search_ar4 devices size: {len(devices)}")
+            for dev in devices:
+                self.add_device(hex(dev.mac))
+        except Exception as e:
+            logger.error(f"_search_device 异常: {str(e)}")
+        finally:
+            self._search()
+            
+    def add_device(self, mac: str):
+        if mac in list(self._devices.keys()):
+            self._devices[mac].update_info()
+            logger.info(f"update device mac: {mac}")
+        else:
+            dev = LMDevice(mac)
+            if dev.init():
+                self._devices[mac] = dev
+                logger.info(f"add device mac: {dev}")

qlsdk/x8/__init__.py

@@ -0,0 +1,128 @@
+from multiprocessing import Queue
+from typing import Literal
+from loguru import logger
+from qlsdk.sdk import LMDevice, LMPacket
+import numpy as np
+from qlsdk.persist import EdfHandler
+import time
+    
+# 设备对象
+class X8(LMDevice):
+    def __init__(self, box_mac:str, is_persist:bool=True, storage_path:str=None):
+        # 是否持久化-保存为文件
+        self._is_persist = is_persist
+        self._storage_path = storage_path
+        self._edf_handler = None
+        
+        self._recording = False
+        self._record_start_time = None
+        
+        self._acq_info = {}
+        # 订阅者列表，数值为数字信号值
+        self._dig_subscriber: dict[str, Queue] = {}
+        # 订阅者列表，数值为物理信号值
+        self._phy_subscriber: dict[str, Queue] = {}
+        
+        super().__init__(box_mac)
+    
+    def set_storage_path(self, storage_path):
+        self._storage_path = storage_path
+
+    @property
+    def device_type(self):
+        return "x8"
+    
+       
+    def eeg_accept(self, packet):        
+        if len(self._dig_subscriber) > 0 or len(self._phy_subscriber) > 0:            
+            for consumer in self._dig_subscriber.values():
+                consumer.put(packet)
+             
+            if len(self._phy_subscriber) > 0:
+                logger.info(f"dig data eeg: {packet.eeg}")
+                logger.info(f"dig data acc: {packet.acc}")
+                packet.eeg = self.eeg2phy(np.array(packet.eeg))
+                packet.acc = self.acc2phy(np.array(packet.acc))
+                logger.info(f"phy data eeg: {packet.eeg}")
+                logger.info(f"phy data acc: {packet.acc}")
+                for consumer2 in self._phy_subscriber.values():
+                    consumer2.put(packet)
+            
+            if self._is_persist:
+                if self._edf_handler is None:
+                    self.start_record()
+                    self._recording = True
+                    self._record_start_time = packet.time_stamp
+                    logger.info(f"开始记录数据: {self.box_mac}")
+            
+                # 处理数据包
+                self._edf_handler.append(packet)
+    
+    def start_record(self):
+        if self._is_persist:
+            if self._edf_handler is None:
+                self._edf_handler = EdfHandler(self.sample_frequency, self.eeg_phy_max, self.eeg_phy_min, self.eeg_dig_max, self.eeg_dig_min, storage_path=self._storage_path)
+                self._edf_handler.set_device_type(self.device_type)
+                self._edf_handler.set_storage_path(self._storage_path)
+                
+    def stop_record(self):
+        if self._edf_handler:
+            # 等待设备端数据传输完成
+            time.sleep(0.5)
+            # 添加结束标识
+            self._edf_handler.append(None)
+            self._edf_handler = None
+            self._recording = False
+            logger.info(f"停止记录数据: {self.box_mac}")
+
+    # 订阅推送消息 
+    def subscribe(self, topic: str = None, q: Queue = Queue(), value_type: Literal['phy', 'dig'] = 'phy') -> tuple[str, Queue]:
+        if topic is None:
+            topic = f"msg_{self.box_mac}"
+        
+        if value_type == 'dig':
+            if topic in list(self._dig_subscriber.keys()):  
+                logger.warning(f"ar4 {self.box_mac} 订阅主题已存在: {topic}")
+                return topic, self._dig_subscriber[topic]
+            self._dig_subscriber[topic] = q
+        else:
+            if topic in list(self._phy_subscriber.keys()):  
+                logger.warning(f"ar4 {self.box_mac} 订阅主题已存在: {topic}")
+                return topic, self._phy_subscriber[topic]
+            self._phy_subscriber[topic] = q
+        
+        return topic, q    
+    
+    def eeg2phy(self, digtal):
+        # 向量化计算（自动支持广播）
+        return super().eeg2phy(digtal)
+    
+    def acc2phy(self, digtal):
+        return super().eeg2phy(digtal)
+            
+class X8Packet(LMPacket):
+    def __init__(self, data: bytes):
+        super().__init__(data)
+        self._data = data
+        self._head = None
+        self._body = None
+        
+    @property
+    def head(self):
+        return self._head
+    
+    @property
+    def body(self):
+        return self._body
+    
+    def parse(self):
+        # 解析数据包头部和数据体
+        if len(self._data) < 4:
+            logger.error(f"数据包长度不足: {len(self._data)}")
+            return False
+        
+        # 解析头部和数据体
+        self._head = self._data[:4]
+        self._body = self._data[4:]
+        
+        return True

qlsdk/x8m/__init__.py

@@ -0,0 +1,21 @@
+from qlsdk.sdk import Hub
+from qlsdk.x8 import X8
+from loguru import logger
+
+
+class X8M(Hub):
+    def __init__(self):
+        super().__init__()
+        self._devices: dict[str, X8] = {}
+        self._search_running = False
+        self._search_timer = None
+    
+    def add_device(self, mac: str):
+        if mac in list(self._devices.keys()):
+            self._devices[mac].update_info()
+            logger.debug(f"update x8 device mac: {mac}")
+        else:
+            dev = X8(mac)
+            if dev.connected:
+                self._devices[mac] = dev
+                logger.info(f"add x8 device mac: {dev}")

{qlsdk2-0.2.0.dist-info → qlsdk2-0.3.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: qlsdk2
-Version: 0.2.0
+Version: 0.3.0a1
 Summary: SDK for quanlan device
 Home-page: https://github.com/hehuajun/qlsdk
 Author: hehuajun
@@ -12,6 +12,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: loguru>=0.6.0
 Requires-Dist: numpy>=1.23.5
+Requires-Dist: pyedflib>=0.1.40
 Provides-Extra: dev
 Requires-Dist: pytest>=6.0; extra == "dev"
 Requires-Dist: twine>=3.0; extra == "dev"

qlsdk2-0.3.0a1.dist-info/RECORD

@@ -0,0 +1,18 @@
+qlsdk/__init__.py,sha256=w0UCnZJekAq4orDahQXRL8s5JJGrWMz0De5LVIf4X_c,426
+qlsdk/ar4/__init__.py,sha256=iR0bFanDwqSxSEM1nCcge1l-GSIWjwnMeRQa47wGrds,4752
+qlsdk/ar4m/__init__.py,sha256=Z13X53WKmKp6MW_fA1FUPdQgJxhxxuyzemJWPS_CEHA,639
+qlsdk/ar4m/ar4sdk.py,sha256=OtmlvbO7XAgk5FNPFYIv_rXGj14maaWtgi2lQ7zQkvY,41418
+qlsdk/ar4m/persist.py,sha256=YrISKVfY-r1m2GZOw-JLuqTbEajA5wYa31MejGlz14I,6696
+qlsdk/ar4m/libs/libAr4SDK.dll,sha256=kZp9_DRwPdAJ5OgTFQSqS8tEETxUs7YmmETuBP2g60U,15402132
+qlsdk/ar4m/libs/libwinpthread-1.dll,sha256=W77ySaDQDi0yxpnQu-ifcU6-uHKzmQpcvsyx2J9j5eg,52224
+qlsdk/persist/__init__.py,sha256=UYaNL_RSczIZL_YrBWAifoAnNxpgZHEeSp-goHqyqv4,27
+qlsdk/persist/edf.py,sha256=usgFjZJVcfAUwSy6cJtRdgjdqMgOS4viLCDGcwAGMZE,7106
+qlsdk/sdk/__init__.py,sha256=v9LKP-5qXCqnAsCkiRE9LDb5Tagvl_Qd_fqrw7y9yd4,68
+qlsdk/sdk/ar4sdk.py,sha256=ZILuUVgUZN7-LNQm4xdItxmZWGwvFt4oWUKGez1BCLI,27101
+qlsdk/sdk/hub.py,sha256=uEOGZBZtMDCWlV8G2TZe6FAo6eTPcwHAW8zdqr1eq_0,1571
+qlsdk/x8/__init__.py,sha256=XlQn3Af_HA7bGrxhv3RZKWMob_SnmissXiso7OH1UgI,4750
+qlsdk/x8m/__init__.py,sha256=cLeUqEEj65qXw4Qa4REyxoLh6T24anSqPaKe9_lR340,634
+qlsdk2-0.3.0a1.dist-info/METADATA,sha256=4VqeQbubQOwzp3fwRsmNXgVRiWjvDXE6bCe7lcOGlNk,7066
+qlsdk2-0.3.0a1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+qlsdk2-0.3.0a1.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
+qlsdk2-0.3.0a1.dist-info/RECORD,,

qlsdk2-0.2.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-qlsdk/__init__.py,sha256=pHMMwFiz7AGYlWvxo-81Obd7gIkojhLAEUGe0yuiAJI,372
-qlsdk/ar4m/__init__.py,sha256=HCFv58t9vYoH2nNN9EiswwdRngiDmh_7h-uwhW-id1I,1784
-qlsdk/ar4m/ar4sdk.py,sha256=DlPCVF5eFA500wXGk-gRqcQXGnoCdUhHe1yoLsOGu7c,23847
-qlsdk/ar4m/libs/libAr4SDK.dll,sha256=kZp9_DRwPdAJ5OgTFQSqS8tEETxUs7YmmETuBP2g60U,15402132
-qlsdk/ar4m/libs/libwinpthread-1.dll,sha256=W77ySaDQDi0yxpnQu-ifcU6-uHKzmQpcvsyx2J9j5eg,52224
-qlsdk2-0.2.0.dist-info/METADATA,sha256=V6UWoXTs44Ulwtb_HvDLyv8A6o6ACgqDTr_dy2XFU6A,7031
-qlsdk2-0.2.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-qlsdk2-0.2.0.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
-qlsdk2-0.2.0.dist-info/RECORD,,