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

qlsdk/ar4m/ar4sdk.py

@@ -3,12 +3,15 @@ from multiprocessing import Queue
 import platform
 from ctypes import (
     c_int, c_int32, c_int64, c_uint32, c_uint64, 
-    c_char, c_char_p, c_void_p, 
+    c_char, c_char_p, c_void_p, c_double,
     Structure, POINTER, CFUNCTYPE
 )
+from typing import Literal
 from loguru import logger
 from time import sleep, time
 import os 
+import numpy as np
+from .persist import EdfHandler
 
 real_path = os.path.realpath(__file__)
 dll_path = f'{os.path.dirname(real_path)}/libs/libAr4SDK.dll'
@@ -89,10 +92,34 @@ _dll.ar4_sdk_start_acq.restype = c_int
 _dll.ar4_sdk_stop_acq.argtypes = [c_void_p]
 _dll.ar4_sdk_stop_acq.restype = c_int
 _dll.ar4_sdk_get_record_sample_rate.argtypes = [c_void_p]
-_dll.ar4_sdk_get_record_sample_rate.restype = c_char_p
+_dll.ar4_sdk_get_record_sample_rate.restype = c_double
 _dll.ar4_sdk_get_acq_start_time.argtypes = [c_void_p]
 _dll.ar4_sdk_get_acq_start_time.restype = c_int64
 
+# eeg信号物理量转换
+_dll.ar4_sdk_get_eeg_phy_max.argtypes = [c_void_p]
+_dll.ar4_sdk_get_eeg_phy_max.restype = c_double
+_dll.ar4_sdk_get_eeg_phy_min.argtypes = [c_void_p]
+_dll.ar4_sdk_get_eeg_phy_min.restype = c_double
+_dll.ar4_sdk_get_eeg_digtal_max.argtypes = [c_void_p]
+_dll.ar4_sdk_get_eeg_digtal_max.restype = c_int32
+_dll.ar4_sdk_get_eeg_digtal_min.argtypes = [c_void_p]
+_dll.ar4_sdk_get_eeg_digtal_min.restype = c_int32
+_dll.ar4_sdk_get_eeg_phy_unit.argtypes = [c_void_p]
+_dll.ar4_sdk_get_eeg_phy_unit.restype = c_char_p
+
+# acc信号物理量转换
+_dll.ar4_sdk_get_acc_phy_max.argtypes = [c_void_p]
+_dll.ar4_sdk_get_acc_phy_max.restype = c_double
+_dll.ar4_sdk_get_acc_phy_min.argtypes = [c_void_p]
+_dll.ar4_sdk_get_acc_phy_min.restype = c_double
+_dll.ar4_sdk_get_acc_digtal_max.argtypes = [c_void_p]
+_dll.ar4_sdk_get_acc_digtal_max.restype = c_int32
+_dll.ar4_sdk_get_acc_digtal_min.argtypes = [c_void_p]
+_dll.ar4_sdk_get_acc_digtal_min.restype = c_int32
+_dll.ar4_sdk_get_acc_phy_unit.argtypes = [c_void_p]
+_dll.ar4_sdk_get_acc_phy_unit.restype = c_char_p
+
 # 回调注册
 _dll.ar4_sdk_register_data_notify.argtypes = [c_void_p, FuncAr4DataNotify]
 _dll.ar4_sdk_register_data_notify.restype = None
@@ -136,18 +163,17 @@ class AR4SDK:
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         pass
-  
 
 # 读取系统当前时间(ms)
 def _get_time():
     cur_time = int(round(time()) * 1000)
-    logger.debug(f"_get_time is {cur_time}")
     return cur_time  
-    
-# ar4设备对象
-class AR4(object):
+
+class LMDevice(object):
     def __init__(self, box_mac:str, slot:int, hub_name:str):
+        # 设备句柄
         self._handle = None
+        # 设备基本信息
         self._box_type = None
         self._box_mac = box_mac
         self._box_id = None
@@ -166,13 +192,41 @@ class AR4(object):
         self._conn_time = None
         self._last_time = None
         
+        self._recording = False
+        self._record_start_time = None
+        
+        ## eeg 数据
+        self._eeg_phy_max = None
+        self._eeg_phy_min = None
+        self._eeg_dig_max = None
+        self._eeg_dig_min = None
+        self._eeg_phy_unit = 'uV'
+        self._eeg_phy_range = None
+        self._eeg_dig_range = None
+        ## acc 数据
+        self._acc_phy_max = None    
+        self._acc_phy_min = None
+        self._acc_dig_max = None
+        self._acc_dig_min = None
+        self._acc_phy_unit = 'mG'
+        self._acc_phy_range = None
+        self._acc_dig_range = None
+        
         self._acq_info = {}
+        # 订阅者列表，数值为数字信号值
+        self._dig_subscriber: dict[str, Queue] = {}
+        # 订阅者列表，数值为物理信号值
+        self._phy_subscriber: dict[str, Queue] = {}
+        
         # 回调函数
         self._data_callback = FuncAr4DataNotify(self._wrap_data_accept())        
         self._trigger_callback = FuncAr4TriggerNotify(self._wrap_trigger_accept())
-        
-        # 消息订阅
-        self._consumers: dict[str, Queue] = {}
+        self._connected_notify_callback = FuncAr4RecorderConnected(self._wrap_connected_notify())
+        self._disconnected_notify_callback = FuncAr4RecorderDisconnected(self._wrap_disconnected_notify())
+        self._box_connected_notify_callback = FuncAr4RecorderConnected(self._wrap_box_connected_notify())
+        self._box_disconnected_notify_callback = FuncAr4RecorderDisconnected(self._wrap_box_disconnected_notify())
+        self._start_record_notify_callback = FuncAr4RecorderConnected(self._wrap_start_record_notify())
+        self._stop_record_notify_callback = FuncAr4RecorderDisconnected(self._wrap_stop_record_notify())
         
     @property
     def box_mac(self):
@@ -187,6 +241,34 @@ class AR4(object):
         # logger.info(f"init ar4 {self.box_mac}")
         if not self._handle:
             self._connected = self.connect()
+            ## eeg 参数
+            self._eeg_phy_max = _dll.ar4_sdk_get_eeg_phy_max(self._handle)
+            self._eeg_phy_min = _dll.ar4_sdk_get_eeg_phy_min(self._handle)
+            self._eeg_dig_max = _dll.ar4_sdk_get_eeg_digtal_max(self._handle)
+            self._eeg_dig_min = _dll.ar4_sdk_get_eeg_digtal_min(self._handle)
+            self._eeg_phy_range = self._eeg_phy_max - self._eeg_phy_min
+            self._eeg_dig_range = self._eeg_dig_max - self._eeg_dig_min
+            eeg_unit = _dll.ar4_sdk_get_eeg_phy_unit(self._handle)
+            if eeg_unit:
+                try:
+                    self._eeg_phy_unit = eeg_unit.decode("utf-8")
+                except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取eeg物理单位异常: {str(e)}")
+            
+            ## acc 参数
+            self._acc_phy_max = _dll.ar4_sdk_get_acc_phy_max(self._handle)    
+            self._acc_phy_min = _dll.ar4_sdk_get_acc_phy_min(self._handle)
+            self._acc_dig_max = _dll.ar4_sdk_get_acc_digtal_max(self._handle)
+            self._acc_dig_min = _dll.ar4_sdk_get_acc_digtal_min(self._handle)
+            self._acc_phy_range = self._acc_phy_max - self._acc_phy_min
+            self._acc_dig_range = self._acc_dig_max - self._acc_dig_min
+            acc_unit = _dll.ar4_sdk_get_acc_phy_unit(self._handle)
+            if acc_unit:
+                try:
+                    self._acc_phy_unit = eeg_unit.decode("utf-8")
+                except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取acc物理单位异常: {str(e)}")
+            
             if self._connected:
                 self._conn_time = _get_time()
                 self.get_box_name()
@@ -194,7 +276,18 @@ class AR4(object):
                 self.get_record_conn_state()
                 self._last_time = _get_time()
             logger.debug(self)
+            self._register_callback()
         return True
+    
+    def _register_callback(self):
+        try:
+            _dll.ar4_sdk_register_data_notify(self._handle, self._data_callback)
+            _dll.ar4_sdk_register_trigger_notify(self._handle, self._trigger_callback)
+            _dll.ar4_sdk_register_conn_notify(self._handle, self._connected_notify_callback, self._disconnected_notify_callback)
+            _dll.ar4_sdk_register_record_state_notify(self._handle, self._start_record_notify_callback, self._stop_record_notify_callback)
+            _dll.ar4_sdk_register_box_conn_notify(self._handle, self._box_connected_notify_callback, self._box_disconnected_notify_callback)
+        except Exception as e:
+            logger.error(f"回调函数注册异常: {str(e)}")
 
     def update_info(self):
         self._last_time = _get_time()
@@ -244,17 +337,6 @@ class AR4(object):
         self._acq_info["start_time"] = _get_time()
         
         if self._handle:
-            # 设置信号数据回调
-            try:
-                _dll.ar4_sdk_register_data_notify(self._handle, self._data_callback)
-            except Exception as e:
-                logger.error(f"ar4 {self._box_mac} 停止采集异常: {str(e)}")
-                
-            # 设置trigger数据回调
-            # try:
-            #     _dll.ar4_sdk_register_trigger_notify(self._handle, self._trigger_callback)
-            # except Exception as e:
-            #     logger.error(f"ar4 {self._box_mac} 停止采集异常: {str(e)}")
             # 启动采集
             try:
                 logger.debug(f"ar4 {self._box_mac} 启动采集: {self._handle}")
@@ -305,15 +387,22 @@ class AR4(object):
             logger.error(f"ar4 {self._box_mac} 获取采样开始时间异常: {str(e)}")
        
     # 订阅推送消息 
-    def subscribe(self, topic: str = None, q: Queue = Queue()):
+    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_{_get_time()}"
-        if topic in list(self._consumers.keys()):  
-            logger.warning(f"ar4 {self._box_mac} 订阅主题已存在: {topic}")
-            return topic, self._consumers[topic]
-        self._consumers[topic] = q
         
-        return topic, self._consumers[topic]
+        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 _wrap_data_accept(self):
         
@@ -322,17 +411,34 @@ class AR4(object):
             self._data_accept(data_ptr)
             
         return data_accept
-            
     def _data_accept(self, data_ptr):
         # logger.info(f"_eeg_accept 被调用")
         # logger.debug(f'handle:{self}, data_ptr:{data_ptr}')
         # data = cast(data_ptr, POINTER(Ar4NotifyData)).contents
-        if len(self._consumers) > 0:
+        if len(self._dig_subscriber) > 0 or len(self._phy_subscriber) > 0:
             packet = AR4Packet().transfer(data_ptr.contents)
-            for consumer in self._consumers.values():
+            
+            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)
+        
             # logger.debug(f"EEG数据: {packet}")
     
+    def eeg2phy(self, digtal):
+        # 向量化计算（自动支持广播）
+        return ((digtal - self._eeg_dig_min) / self._eeg_dig_range) * self._eeg_phy_range + self._eeg_phy_min
+    
+    def acc2phy(self, digtal):
+        return ((digtal - self._acc_dig_min) / self._acc_dig_range) * self._acc_phy_range + self._acc_phy_min
     def _wrap_trigger_accept(self):
         
         @FuncAr4DataNotify
@@ -344,7 +450,490 @@ class AR4(object):
     def _trigger_accept(self, time_ms, trigger_type, trigger_value):
         logger.info(f"_trigger_accept 被调用")
         logger.info(f"触发时间: {time_ms}, 触发类型: {trigger_type}, 触发值: {trigger_value}")
+        
+        
+    def _wrap_connected_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def connected_notify(handle):
+            self._connected_notify(handle)
+            
+        return connected_notify      
+    
+    def _connected_notify(self, handle):
+        logger.info(f"_connected_notify 被调用 handle: {handle}")
+        
+        
+    def _wrap_disconnected_notify(self):
+        @FuncAr4RecorderDisconnected
+        def disconnected_notify(handle):
+            self._disconnected_notify(handle)
+            
+        return disconnected_notify      
+    
+    def _disconnected_notify(self, handle):
+        logger.info(f"_disconnected_notify 被调用 handle: {handle}")
+        
+        
+    def _wrap_box_connected_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def box_connected_notify(handle):
+            self._box_connected_notify(handle)
+            
+        return box_connected_notify      
+    
+    def _box_connected_notify(self, handle):
+        logger.info(f"_box_connected_notify 被调用 handle: {handle}")
+        
+    def _wrap_box_disconnected_notify(self):
+        @FuncAr4RecorderDisconnected
+        def box_disconnected_notify(handle):
+            self._box_disconnected_notify(handle)
+            
+        return box_disconnected_notify      
+    
+    def _box_disconnected_notify(self, handle):
+        logger.info(f"_box_disconnected_notify 被调用 handle: {handle}")
+        
+    def _wrap_start_record_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def start_record_notify(handle):
+            self._start_record_notify(handle)
+            
+        return start_record_notify      
+    
+    def _start_record_notify(self, handle):
+        logger.info(f"_start_record_notify 被调用 handle: {handle}")        
+        self._recording = True
+        self._record_start_time = time()
+        logger.info(self)
+        
+        
+    def _wrap_stop_record_notify(self):
+        @FuncAr4RecorderDisconnected
+        def stop_record_notify(handle):
+            self._stop_record_notify(handle)
+            
+        return stop_record_notify      
+    
+    def _stop_record_notify(self, handle):
+        logger.info(f"_stop_record_notify 被调用 handle: {handle}")      
+        self._recording = False
+        
+    def __str__(self):
+        return f"""
+                box mac: {self._box_mac}, 
+                box name: {self._box_name}, 
+                box soc: {self._box_soc}
+                head conn state: {self._head_conn_state}
+                head mac: {self._head_mac},
+                head soc: {self._head_soc} 
+                connected: {self._connected}
+                connect time: {self._conn_time}
+                last time: {self._last_time}
+                [
+                eeg phy max: {self._eeg_phy_max}
+                eeg phy min: {self._eeg_phy_min}
+                eeg dig max: {self._eeg_dig_max}
+                eeg dig min: {self._eeg_dig_min}
+                eeg phy unit: {self._eeg_phy_unit}
+                ]
+                [
+                acc phy max: {self._acc_phy_max}
+                acc phy min: {self._acc_phy_min}
+                acc dig max: {self._acc_dig_max}
+                acc dig min: {self._acc_dig_min}
+                acc phy unit: {self._acc_phy_unit}
+                ]
+                dig -> ((dig - {self._eeg_dig_min}) / {self._eeg_dig_range}) * {self._eeg_phy_range} + {self._eeg_phy_min}
+                dig -> ((dig - {self._acc_dig_min}) / {self._acc_dig_range}) * {self._acc_phy_range} + {self._acc_phy_min}
+            """
+            
+    
+    
+# ar4设备对象
+class AR4(object):
+    def __init__(self, box_mac:str, slot:int=None, hub_name:str=None, is_persist:bool=True, storage_path:str=None):
+        # 是否持久化-保存为文件
+        self._is_persist = is_persist
+        self._storage_path = storage_path
+        self._edf_handler = None
+        # 设备句柄
+        self._handle = None
+        
+        self.box_connected = False
+        self.sample_frequency = 500
+        # 设备基本信息
+        self._box_type = None
+        self._box_mac = box_mac
+        self._box_id = None
+        self._box_soc = None
+        self._box_name = None
+        self._box_version = None
+        self._head_type = None
+        self._head_mac = None
+        self._head_version = None
+        self._head_conn_state = None
+        self._head_soc = None
+        self._net_state = None
+        self._hub_name = hub_name
+        self._slot = slot
+        self._connected = False
+        self._conn_time = None
+        self._last_time = None
+        
+        self._recording = False
+        self._record_start_time = None
+        
+        ## eeg 数据
+        self._eeg_phy_max = None
+        self._eeg_phy_min = None
+        self._eeg_dig_max = None
+        self._eeg_dig_min = None
+        self._eeg_phy_unit = 'uV'
+        self._eeg_phy_range = None
+        self._eeg_dig_range = None
+        ## acc 数据
+        self._acc_phy_max = None    
+        self._acc_phy_min = None
+        self._acc_dig_max = None
+        self._acc_dig_min = None
+        self._acc_phy_unit = 'mG'
+        self._acc_phy_range = None
+        self._acc_dig_range = None
+        
+        self._acq_info = {}
+        # 订阅者列表，数值为数字信号值
+        self._dig_subscriber: dict[str, Queue] = {}
+        # 订阅者列表，数值为物理信号值
+        self._phy_subscriber: dict[str, Queue] = {}
+        
+        # 回调函数
+        self._data_callback = FuncAr4DataNotify(self._wrap_data_accept())        
+        self._trigger_callback = FuncAr4TriggerNotify(self._wrap_trigger_accept())
+        self._connected_notify_callback = FuncAr4RecorderConnected(self._wrap_connected_notify())
+        self._disconnected_notify_callback = FuncAr4RecorderDisconnected(self._wrap_disconnected_notify())
+        self._box_connected_notify_callback = FuncAr4RecorderConnected(self._wrap_box_connected_notify())
+        self._box_disconnected_notify_callback = FuncAr4RecorderDisconnected(self._wrap_box_disconnected_notify())
+        self._start_record_notify_callback = FuncAr4RecorderConnected(self._wrap_start_record_notify())
+        self._stop_record_notify_callback = FuncAr4RecorderDisconnected(self._wrap_stop_record_notify())
+        
+    @property
+    def box_mac(self):
+        return self._box_mac
+    @property
+    def slot(self):
+        return self._slot
+    @property
+    def hub_name(self):
+        return self._hub_name
+    
+    def set_storage_path(self, storage_path):
+        self._storage_path = storage_path
+
+    def init(self):
+        # logger.info(f"init ar4 {self.box_mac}")
+        if not self._handle:
+            self._connected = self.connect()
+            ## eeg 参数
+            self._eeg_phy_max = _dll.ar4_sdk_get_eeg_phy_max(self._handle)
+            self._eeg_phy_min = _dll.ar4_sdk_get_eeg_phy_min(self._handle)
+            self._eeg_dig_max = _dll.ar4_sdk_get_eeg_digtal_max(self._handle)
+            self._eeg_dig_min = _dll.ar4_sdk_get_eeg_digtal_min(self._handle)
+            self._eeg_phy_range = self._eeg_phy_max - self._eeg_phy_min
+            self._eeg_dig_range = self._eeg_dig_max - self._eeg_dig_min
+            eeg_unit = _dll.ar4_sdk_get_eeg_phy_unit(self._handle)
+            if eeg_unit:
+                try:
+                    self._eeg_phy_unit = eeg_unit.decode("utf-8")
+                except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取eeg物理单位异常: {str(e)}")
+            
+            ## acc 参数
+            self._acc_phy_max = _dll.ar4_sdk_get_acc_phy_max(self._handle)    
+            self._acc_phy_min = _dll.ar4_sdk_get_acc_phy_min(self._handle)
+            self._acc_dig_max = _dll.ar4_sdk_get_acc_digtal_max(self._handle)
+            self._acc_dig_min = _dll.ar4_sdk_get_acc_digtal_min(self._handle)
+            self._acc_phy_range = self._acc_phy_max - self._acc_phy_min
+            self._acc_dig_range = self._acc_dig_max - self._acc_dig_min
+            acc_unit = _dll.ar4_sdk_get_acc_phy_unit(self._handle)
+            if acc_unit:
+                try:
+                    self._acc_phy_unit = eeg_unit.decode("utf-8")
+                except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取acc物理单位异常: {str(e)}")
+            
+            if self._connected:
+                self._conn_time = _get_time()
+                self.get_box_name()
+                self.get_head_mac()
+                self.get_record_conn_state()
+                self._last_time = _get_time()
+            logger.debug(self)
+            self._register_callback()
+        return True
+    
+    def _register_callback(self):
+        try:
+            _dll.ar4_sdk_register_data_notify(self._handle, self._data_callback)
+            _dll.ar4_sdk_register_trigger_notify(self._handle, self._trigger_callback)
+            _dll.ar4_sdk_register_conn_notify(self._handle, self._connected_notify_callback, self._disconnected_notify_callback)
+            _dll.ar4_sdk_register_record_state_notify(self._handle, self._start_record_notify_callback, self._stop_record_notify_callback)
+            _dll.ar4_sdk_register_box_conn_notify(self._handle, self._box_connected_notify_callback, self._box_disconnected_notify_callback)
+        except Exception as e:
+            logger.error(f"回调函数注册异常: {str(e)}")
+
+    @property
+    def device_type(self):
+        return "ar4" if self._box_type == 4  else "x8" if self._box_type == 8 else "lm"
+    
+    def update_info(self):
+        logger.debug(f"update info {self._box_mac}")
+        self._last_time = _get_time()
+        
+    # 设备连接
+    def connect(self)-> bool:
+        
+            if self._box_mac:
+                """连接设备"""
+                # callback = FuncAr4GetTime(self._get_time)
+                try:
+                    self._handle = _dll.ar4_sdk_connect(int(self._box_mac, 16), c_void_p(0))
+                    # logger.info(f"conn handle is {self._handle}")
+                    self._handle = c_void_p(self._handle)     
+                    logger.debug(f"ar4 {self._box_mac} 连接: {self._handle}")                   
+                except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 连接异常: {str(e)}")
+            return self._handle is not None
+    
+    # 读取盒子名称
+    def get_box_name(self):
+        if self._handle:    
+            try:        
+                self._head_mac = _dll.ar4_sdk_get_recoder_mac(self._handle)
+            except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取盒子记录子mac异常: {str(e)}")
+    # 读取记录子mac
+    def get_head_mac(self):
+        if self._handle:    
+            try:        
+                self._box_name = _dll.ar4_sdk_get_record_conn_state(self._handle)
+            except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取盒子名称异常: {str(e)}")
+    # 读取记录子连接状态
+    def get_record_conn_state(self):
+        if self._handle:    
+            try:        
+                self._head_conn_state = _dll.ar4_sdk_get_record_conn_state(self._handle)
+            except Exception as e:
+                    logger.error(f"ar4 {self._box_mac} 获取盒子记录子连接状态异常: {str(e)}")
+
+    # 数据采集启动
+    def start_acquisition(self):
+        logger.info(f"ar4 {self._box_mac} 启动数据采集...")
+        """启动数据采集"""
+        
+        self._acq_info["start_time"] = _get_time()
+        
+        if self._handle:
+            # 启动采集
+            try:
+                logger.debug(f"ar4 {self._box_mac} 启动采集: {self._handle}")
+                ret = _dll.ar4_sdk_start_acq(self._handle)
+                logger.debug(f"ar4 {self._box_mac} 启动采集结果: {ret}")
+                if (self._is_persist):
+                    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)
+                    
+                return ret == 0
+            except Exception as e:
+                logger.error(f"ar4 {self._box_mac} 启动数据采集异常: {str(e)}")
+        else:
+            logger.info(f"ar4 {self._box_mac} 启动数据采集失败， 设备未连接")
+            return False
+
+    def stop_acquisition(self):
+        """停止采集"""
+        if self._handle:
+            try:                
+                self.get_acq_start_time() 
+            except Exception as e:
+                logger.error(f"ar4 {self._box_mac} 获取开始采集时间异常: {str(e)}")
+            try:
+                ret = _dll.ar4_sdk_stop_acq(self._handle)
+                if ret == 0 and self._edf_handler:
+                    self._edf_handler.append(None)
                 
+                return ret == 0
+            except Exception as e:
+                logger.error(f"ar4 {self._box_mac} 停止采集异常: {str(e)}")
+        else:
+            return False
+
+    def disconnect(self):
+        """断开连接"""
+        if self._handle:
+            _dll.ar4_sdk_disconnect(self._handle)
+
+    def get_sample_rate(self):
+        try:
+            ret = _dll.ar4_sdk_get_record_sample_rate(self._handle)
+            logger.debug(f"ar4 {self._box_mac} 获取采样率: {ret.contents.decode('utf-8')}")
+        except Exception as e:
+            logger.error(f"ar4 {self._box_mac} 获取采样率异常: {str(e)}")
+            
+    def get_acq_start_time(self):
+        try:
+            ret = _dll.ar4_sdk_get_acq_start_time(self._handle)
+            # 更新采样开始时间
+            if ret:
+                self._acq_info["start_time"] = ret
+            logger.debug(f"ar4 {self._box_mac} 获取采样开始时间: {ret}")
+        except Exception as e:
+            logger.error(f"ar4 {self._box_mac} 获取采样开始时间异常: {str(e)}")
+       
+    # 订阅推送消息 
+    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_{_get_time()}"
+        
+        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 _wrap_data_accept(self):
+        
+        @FuncAr4DataNotify
+        def data_accept(handle, data_ptr):
+            self._data_accept(data_ptr)
+            
+        return data_accept
+    def _data_accept(self, data_ptr):
+        # logger.info(f"_eeg_accept 被调用")
+        # logger.debug(f'handle:{self}, data_ptr:{data_ptr}')
+        # data = cast(data_ptr, POINTER(Ar4NotifyData)).contents
+        # if len(self._dig_subscriber) > 0 or len(self._phy_subscriber) > 0:
+        packet = AR4Packet().transfer(data_ptr.contents)
+        if self._edf_handler:
+            self._edf_handler.append(packet)
+        
+        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)
+        
+            # logger.debug(f"EEG数据: {packet}")
+    
+    def eeg2phy(self, digtal):
+        # 向量化计算（自动支持广播）
+        return ((digtal - self._eeg_dig_min) / self._eeg_dig_range) * self._eeg_phy_range + self._eeg_phy_min
+    
+    def acc2phy(self, digtal):
+        return ((digtal - self._acc_dig_min) / self._acc_dig_range) * self._acc_phy_range + self._acc_phy_min
+    def _wrap_trigger_accept(self):
+        
+        @FuncAr4DataNotify
+        def trigger_accept(handle, time_ms, trigger_type, trigger_value):
+            self._trigger_accept(time_ms, trigger_type, trigger_value)
+            
+        return trigger_accept
+        
+    def _trigger_accept(self, time_ms, trigger_type, trigger_value):
+        logger.info(f"_trigger_accept 被调用")
+        logger.info(f"触发时间: {time_ms}, 触发类型: {trigger_type}, 触发值: {trigger_value}")
+        
+        
+    def _wrap_connected_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def connected_notify(handle):
+            self._connected_notify(handle)
+            
+        return connected_notify      
+    
+    def _connected_notify(self, handle):
+        logger.info(f"_connected_notify 被调用 handle: {handle}")
+        
+        
+    def _wrap_disconnected_notify(self):
+        @FuncAr4RecorderDisconnected
+        def disconnected_notify(handle):
+            self._disconnected_notify(handle)
+            
+        return disconnected_notify      
+    
+    def _disconnected_notify(self, handle):
+        logger.info(f"_disconnected_notify 被调用 handle: {handle}")
+        
+        
+    def _wrap_box_connected_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def box_connected_notify(handle):
+            self._box_connected_notify(handle)
+            
+        return box_connected_notify      
+    
+    def _box_connected_notify(self, handle):
+        self.box_connected = True
+        logger.info(f"_box_connected_notify 被调用 handle: {handle}")
+        
+    def _wrap_box_disconnected_notify(self):
+        @FuncAr4RecorderDisconnected
+        def box_disconnected_notify(handle):
+            self._box_disconnected_notify(handle)
+            
+        return box_disconnected_notify      
+    
+    def _box_disconnected_notify(self, handle):        
+        self.box_connected = False
+        logger.info(f"_box_disconnected_notify 被调用 handle: {handle}")
+        
+    def _wrap_start_record_notify(self):
+        
+        @FuncAr4RecorderConnected
+        def start_record_notify(handle):
+            self._start_record_notify(handle)
+            
+        return start_record_notify      
+    
+    def _start_record_notify(self, handle):
+        logger.info(f"_start_record_notify 被调用 handle: {handle}")        
+        self._recording = True
+        self._record_start_time = time()
+        logger.info(self)
+        
+        
+    def _wrap_stop_record_notify(self):
+        @FuncAr4RecorderDisconnected
+        def stop_record_notify(handle):
+            self._stop_record_notify(handle)
+            
+        return stop_record_notify      
+    
+    def _stop_record_notify(self, handle):
+        logger.info(f"_stop_record_notify 被调用 handle: {handle}")      
+        self._recording = False
+        
     def __str__(self):
         return f"""
                 box mac: {self._box_mac}, 
@@ -356,6 +945,22 @@ class AR4(object):
                 connected: {self._connected}
                 connect time: {self._conn_time}
                 last time: {self._last_time}
+                [
+                eeg phy max: {self._eeg_phy_max}
+                eeg phy min: {self._eeg_phy_min}
+                eeg dig max: {self._eeg_dig_max}
+                eeg dig min: {self._eeg_dig_min}
+                eeg phy unit: {self._eeg_phy_unit}
+                ]
+                [
+                acc phy max: {self._acc_phy_max}
+                acc phy min: {self._acc_phy_min}
+                acc dig max: {self._acc_dig_max}
+                acc dig min: {self._acc_dig_min}
+                acc phy unit: {self._acc_phy_unit}
+                ]
+                dig -> ((dig - {self._eeg_dig_min}) / {self._eeg_dig_range}) * {self._eeg_phy_range} + {self._eeg_phy_min}
+                dig -> ((dig - {self._acc_dig_min}) / {self._acc_dig_range}) * {self._acc_phy_range} + {self._acc_phy_min}
             """
             
 class Packet(object):
@@ -383,11 +988,6 @@ class AR4Packet(Packet):
         self.acc_count = data.acc_count
         # 读eeg数据
         if self.eeg_ch_count and self.eeg_count:
-            # self.eeg = [[] for _ in range(self.eeg_ch_count)]
-            # for i in range(self.eeg_ch_count):
-                # self.eeg[i] = [data.eeg[j + (i * self.eeg_count)] for j in range(self.eeg_count)]
-            # tmp = data.eeg[:self.eeg_ch_count * self.eeg_count]
-            # logger.info(tmp)
             self.eeg = [data.eeg[i:self.eeg_count*self.eeg_ch_count:self.eeg_ch_count] for i in range(self.eeg_ch_count)]
         # 读acc数据
         if self.acc_ch_count and self.acc_count:

qlsdk/ar4m/persist.py

@@ -0,0 +1,178 @@
+from datetime import datetime
+from multiprocessing import Lock, Queue
+from pyedflib import FILETYPE_BDFPLUS, FILETYPE_EDFPLUS, EdfWriter
+from threading import Thread
+from loguru import logger
+import numpy as np
+import os
+
+class EdfHandler(object):
+    def __init__(self, sample_frequency, physical_max, physical_min, digital_max, digital_min, resolution=16, storage_path = None):
+        self.physical_max = physical_max
+        self.physical_min = physical_min
+        self.digital_max = digital_max
+        self.digital_min = digital_min
+        self.channels = None
+        self._cache = Queue()
+        self.resolution = resolution
+        self.sample_frequency = sample_frequency
+        # bytes per second
+        self.bytes_per_second = 0
+        self._edf_writer = None
+        self._cache2 = tuple()
+        self._recording = False
+        self._edf_writer = None
+        self.annotations = None
+        # 每个数据块大小
+        self._chunk = np.array([])
+        self._Lock = Lock()
+        self._duration = 0
+        self._points = 0
+        self._first_pkg_id = None
+        self._last_pkg_id = None
+        self._first_timestamp = None
+        self._end_time = None
+        self._patient_code = "patient_code"
+        self._patient_name = "patient_name"
+        self._device_type = None
+        self._total_packets = 0
+        self._lost_packets = 0
+        self._storage_path = storage_path
+        
+    @property
+    def file_name(self): 
+        if self._storage_path:
+            try:
+                os.makedirs(self._storage_path, exist_ok=True)  # 自动创建目录，存在则忽略
+                return f"{self._storage_path}/{self._device_type}_{self._first_timestamp}.edf"
+            except Exception as e:
+                logger.error(f"创建目录[{self._storage_path}]失败: {e}")
+            
+        return f"{self._device_type}_{self._first_timestamp}.edf"
+         
+    @property
+    def file_type(self):
+        return FILETYPE_BDFPLUS if self.resolution == 24 else FILETYPE_EDFPLUS
+    
+    def set_device_type(self, device_type):
+        self._device_type = device_type
+        
+    def set_storage_path(self, storage_path):
+        self._storage_path = storage_path
+        
+    def set_patient_code(self, patient_code):
+        self._patient_code = patient_code
+        
+    def set_patient_name(self, patient_name):
+        self._patient_name = patient_name
+    
+    def append(self, data):
+        if data:
+            # 通道数
+            if self._first_pkg_id is None:
+                self.channels = data.eeg_ch_count
+                self._first_pkg_id = data.pkg_id
+                self._first_timestamp = data.time_stamp
+                
+            if self._last_pkg_id and self._last_pkg_id != data.pkg_id - 1:  
+                self._lost_packets += data.pkg_id - self._last_pkg_id - 1
+                logger.warning(f"数据包丢失: {self._last_pkg_id} -> {data.pkg_id}, 丢包数: {data.pkg_id - self._last_pkg_id - 1}")
+                
+            self._last_pkg_id = data.pkg_id
+            self._total_packets += 1
+            
+        # 数据
+        self._cache.put(data)
+        if not self._recording:
+            self.start()
+    
+    def trigger(self, data):
+        pass        
+    
+    def start(self):
+        self._recording = True
+        record_thread = Thread(target=self._consumer)
+        record_thread.start()
+        
+    def _consumer(self):
+        logger.debug(f"开始消费数据 _consumer: {self._cache.qsize()}")
+        while True:
+            if self._recording or (not self._cache.empty()):
+                data = self._cache.get()
+                if data is None:
+                    break
+                # 处理数据
+                self._points += len(data.eeg[0])
+                self._write_file(data.eeg)
+            else:
+                break
+            
+        self.close()
+        
+    def _write_file(self, data):
+        try:
+            if self._edf_writer is None:
+                self.initialize_edf()
+            
+            if (self._chunk.size == 0):
+                self._chunk = np.asarray(data)
+            else:                
+                self._chunk = np.hstack((self._chunk, data))
+                
+            if self._chunk.size >= self.sample_frequency * self.channels:                
+                self._write_chunk(self._chunk[:self.sample_frequency])
+                self._chunk = self._chunk[self.sample_frequency:]            
+            
+        except Exception as e:
+            logger.error(f"写入数据异常: {str(e)}")
+        
+    def close(self):
+        self._recording = False
+        if self._edf_writer:            
+            self._end_time = datetime.now().timestamp()
+            self._edf_writer.writeAnnotation(0, 1, "start recording ")
+            self._edf_writer.writeAnnotation(self._duration, 1, "recording end")
+            self._edf_writer.close()
+        
+        logger.info(f"文件: {self.file_name}完成记录, 总点数: {self._points}, 总时长: {self._duration}秒 丢包数: {self._lost_packets}/{self._total_packets + self._lost_packets}")
+        
+    
+        
+    def initialize_edf(self):
+        # 创建EDF+写入器
+        self._edf_writer = EdfWriter(
+            self.file_name,
+            self.channels,
+            file_type=self.file_type
+        )
+        
+        # 设置头信息
+        self._edf_writer.setPatientCode(self._patient_code)
+        self._edf_writer.setPatientName(self._patient_name)
+        self._edf_writer.setEquipment(self._device_type)
+        self._edf_writer.setStartdatetime(datetime.now())
+        
+        # 配置通道参数
+        signal_headers = []
+        for ch in range(self.channels):
+            signal_headers.append({
+                "label": f'channels {ch + 1}',
+                "dimension": 'uV',
+                "sample_frequency": self.sample_frequency,
+                "physical_min": self.physical_min,
+                "physical_max": self.physical_max,
+                "digital_min": self.digital_min,
+                "digital_max": self.digital_max
+            })
+        
+        self._edf_writer.setSignalHeaders(signal_headers)
+    
+    def _write_chunk(self, chunk):
+        logger.debug(f"写入数据: {chunk}")
+        # 转换数据类型为float64（pyedflib要求）
+        data_float64 = chunk.astype(np.float64)
+        # 写入时转置为(样本数, 通道数)格式
+        self._edf_writer.writeSamples(data_float64)
+        self._duration += 1
+
+    

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: qlsdk2
-Version: 0.1.7
+Version: 0.2.0a1
 Summary: SDK for quanlan device
 Home-page: https://github.com/hehuajun/qlsdk
 Author: hehuajun
@@ -11,6 +11,8 @@ Classifier: Operating System :: OS Independent
 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.2.0a1.dist-info/RECORD

@@ -0,0 +1,10 @@
+qlsdk/__init__.py,sha256=pHMMwFiz7AGYlWvxo-81Obd7gIkojhLAEUGe0yuiAJI,372
+qlsdk/ar4m/__init__.py,sha256=HCFv58t9vYoH2nNN9EiswwdRngiDmh_7h-uwhW-id1I,1784
+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
+qlsdk2-0.2.0a1.dist-info/METADATA,sha256=r8hXXxFJJkgVi6W7XiYlTdMu3FjnP8vk1rYBUjLHX9k,7066
+qlsdk2-0.2.0a1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+qlsdk2-0.2.0a1.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
+qlsdk2-0.2.0a1.dist-info/RECORD,,

qlsdk2-0.1.7.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=M244in6ytN8f-PI7lIbt8BfEE_DcjAsE06iLxI7xNeY,15358
-qlsdk/ar4m/libs/libAr4SDK.dll,sha256=kZp9_DRwPdAJ5OgTFQSqS8tEETxUs7YmmETuBP2g60U,15402132
-qlsdk/ar4m/libs/libwinpthread-1.dll,sha256=W77ySaDQDi0yxpnQu-ifcU6-uHKzmQpcvsyx2J9j5eg,52224
-qlsdk2-0.1.7.dist-info/METADATA,sha256=ynZ7zTy2_H7Hney0pXE-GgVERk9dgvaLoCierdAim8U,7001
-qlsdk2-0.1.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-qlsdk2-0.1.7.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
-qlsdk2-0.1.7.dist-info/RECORD,,