qlsdk2 0.4.0a3__py3-none-any.whl → 0.4.1__py3-none-any.whl

qlsdk/rsc/device/c64_rs.py

@@ -0,0 +1,364 @@
+
+from multiprocessing import Queue
+from threading import Thread
+from time import sleep, time_ns
+from typing import Any, Dict, Literal
+
+from loguru import logger
+from qlsdk.persist import RscEDFHandler
+from qlsdk.rsc.interface import IDevice, IParser
+from qlsdk.rsc.command import *
+from qlsdk.rsc.parser.base import TcpMessageParser
+from qlsdk.rsc.device.base import QLBaseDevice
+
+class C64RS(QLBaseDevice):
+    
+    device_type = 0x39  # C64RS设备类型标识符
+    
+    def __init__(self, socket):
+        self.socket = socket
+        
+        self._id = None
+        
+        # 设备信息
+        self.device_id = None
+        self.device_name = None
+        self._device_no = None
+        self.software_version = None
+        self.hardware_version = None
+        self.connect_time = None
+        self.current_time = None
+        # mV
+        self.voltage = None
+        # %
+        self.battery_remain = None
+        # %
+        self.battery_total = None
+        # persist
+        self._recording = False
+        self._storage_path = None
+        self._file_prefix = None
+        
+        # 可设置参数
+        # 采集：采样量程、采样率、采样通道
+        # 刺激：刺激电流、刺激频率、刺激时间、刺激通道
+        # 采样量程（mV）：188、375、563、750、1125、2250、4500
+        self._sample_range:Literal[188, 375, 563, 750, 1125, 2250, 4500] = 188        
+        # 采样率（Hz）：250、500、1000、2000、4000、8000、16000、32000
+        self._sample_rate:Literal[250, 500, 1000, 2000, 4000, 8000, 16000, 32000] = 500
+        self._physical_max = self._sample_range * 1000  # 物理最大值（uV）
+        self._physical_min = -self._sample_range * 1000  # 物理最小值（uV）
+        self._digital_max = 8388607
+        self._digital_min = -8388608
+        self._physical_range = self._physical_max - self._physical_min
+        self._digital_range = 16777215
+        self._acq_channels = None
+        self._acq_param = {
+            "sample_range": 188,
+            "sample_rate": 500,
+            "channels": [],
+        }
+        
+        self._stim_param = {
+            "stim_type": 0,                 # 刺激类型：0-所有通道参数相同, 1: 通道参数不同
+            "channels": [],
+            "param": [{
+                    "channel_id": 0,        #通道号 从0开始                          -- 必填
+                    "waveform": 3,          #波形类型：0-直流，1-交流 2-方波 3-脉冲   -- 必填
+                    "current": 1,           #电流强度(mA)                            -- 必填
+                    "duration": 30,         #平稳阶段持续时间(s)                      -- 必填
+                    "ramp_up": 5,           #上升时间(s) 默认0
+                    "ramp_down": 5,         #下降时间(s) 默认0
+                    "frequency": 500,       #频率(Hz) -- 非直流必填
+                    "phase_position": 0,    #相位 -- 默认0
+                    "duration_delay": "0",  #延迟启动时间(s) -- 默认0
+                    "pulse_width": 0,       #脉冲宽度(us) -- 仅脉冲类型电流有效， 默认100us
+                },
+                {
+                    "channel_id": 1,        #通道号 从0开始                          -- 必填
+                    "waveform": 3,          #波形类型：0-直流，1-交流 2-方波 3-脉冲   -- 必填
+                    "current": 1,           #电流强度(mA)                            -- 必填
+                    "duration": 30,         #平稳阶段持续时间(s)                      -- 必填
+                    "ramp_up": 5,           #上升时间(s) 默认0
+                    "ramp_down": 5,         #下降时间(s) 默认0
+                    "frequency": 500,       #频率(Hz) -- 非直流必填
+                    "phase_position": 0,    #相位 -- 默认0
+                    "duration_delay": "0",  #延迟启动时间(s) -- 默认0
+                    "pulse_width": 0,       #脉冲宽度(us) -- 仅脉冲类型电流有效， 默认100us
+                }
+            ]
+        }
+        
+        self.stim_paradigm = None
+                      
+        signal_info = {
+            "param" : None,
+            "start_time" : None,
+            "finished_time" : None,
+            "packet_total" : None,
+            "last_packet_time" : None,
+            "state" : 0
+        }
+        stim_info = {
+            
+        }
+        Impedance_info = {
+            
+        }
+        # 信号采集状态
+        # 信号数据包总数（一个信号采集周期内）
+        # 信号采集参数
+        # 电刺激状态
+        # 电刺激开始时间（最近一次）
+        # 电刺激结束时间（最近一次）
+        # 电刺激参数
+        # 启动数据解析线程
+        # 数据存储状态
+        # 存储目录
+        
+        # 
+        self.__signal_consumer: Dict[str, Queue[Any]]={}
+        self.__impedance_consumer: Dict[str, Queue[Any]]={}
+        
+        # EDF文件处理器
+        self._edf_handler = None       
+        self.storage_enable = True 
+        
+        self._parser: IParser = TcpMessageParser(self)
+        self._parser.start()
+        
+        # 启动数据接收线程
+        self._accept = Thread(target=self.accept)
+        self._accept.daemon = True
+        self._accept.start()
+        
+    @property
+    def device_no(self):
+        return self._device_no
+    
+    @device_no.setter
+    def device_no(self, value: str):
+        self._device_no = value
+    
+    @classmethod
+    def from_parent(cls, parent:IDevice) -> IDevice:
+        rlt = cls(parent.socket)
+        rlt.device_id = parent.device_id
+        rlt._device_no = parent.device_no
+        return rlt
+        
+        
+    def init_edf_handler(self):
+        self._edf_handler = RscEDFHandler(self.sample_rate,  self.sample_range * 1000 , - self.sample_range * 1000, self.resolution)  
+        self._edf_handler.set_device_type(self.device_type)
+        self._edf_handler.set_device_no(self.device_name)
+        self._edf_handler.set_storage_path(self._storage_path)
+        self._edf_handler.set_file_prefix(self._file_prefix)
+        
+    @property
+    def edf_handler(self):
+        if not self.storage_enable:
+            return None
+        
+        if self._edf_handler is None:
+            self.init_edf_handler() 
+            
+        return self._edf_handler  
+    
+    @property
+    def acq_channels(self):
+        if self._acq_channels is None:
+            self._acq_channels = [i for i in range(1, 63)]
+        return self._acq_channels
+    @property
+    def sample_range(self):
+        return self._sample_range if self._sample_range else 188
+    @property
+    def sample_rate(self):
+        return self._sample_rate if self._sample_rate else 500
+    @property
+    def resolution(self):
+        return 24
+    
+    @property
+    def signal_consumers(self):
+        return self.__signal_consumer
+    
+    @property
+    def impedance_consumers(self):
+        return self.__impedance_consumer
+    
+    # 设置记录文件路径
+    def set_storage_path(self, path):
+        self._storage_path = path
+       
+    # 设置记录文件名称前缀 
+    def set_file_prefix(self, prefix):
+        self._file_prefix = prefix
+        
+    # 接收socket消息
+    def accept(self):
+        while True:
+            # 缓冲去4M
+            data = self.socket.recv(4096*1024)
+            if not data:
+                logger.warning(f"设备{self.device_name}连接结束")
+                break
+            
+            self._parser.append(data)
+            
+       
+    # socket发送数据    
+    def send(self, data):
+        self.socket.sendall(data)
+    
+    # 设置刺激参数
+    def set_stim_param(self, param):
+        self.stim_paradigm = param
+    
+    # 设置采集参数
+    def set_acq_param(self, channels, sample_rate = 500, sample_range = 188):
+        self._acq_param["channels"] = channels 
+        self._acq_param["sample_rate"] = sample_rate
+        self._acq_param["sample_range"] = sample_range
+        self._acq_channels = channels
+        self._sample_rate = sample_rate
+        self._sample_range = sample_range
+    
+    # 通用配置-TODO
+    def set_config(self, key:str, val: str):
+        pass      
+        
+    def start_impedance(self):
+        logger.info("启动阻抗测量")  
+        msg = StartImpedanceCommand.build(self).pack()
+        logger.debug(f"start_impedance message is {msg.hex()}")
+        self.socket.sendall(msg)
+    
+    def stop_impedance(self):
+        logger.info("停止阻抗测量")  
+        msg = StopImpedanceCommand.build(self).pack()
+        logger.debug(f"stop_impedance message is {msg.hex()}")
+        self.socket.sendall(msg)
+    
+    def start_stimulation(self):
+        if self.stim_paradigm is None:
+            logger.warning("刺激参数未设置，请先设置刺激参数")
+            return
+        logger.info("启动电刺激")  
+        msg = StartStimulationCommand.build(self).pack()
+        logger.debug(f"start_stimulation message is {msg.hex()}")
+        self.socket.sendall(msg)
+        t = Thread(target=self._stop_stimulation_trigger, args=(self.stim_paradigm.duration,))
+        t.start()
+        
+    def _stop_stimulation_trigger(self, duration):
+        delay = duration
+        while delay > 0:
+            sleep(1)
+            delay -= 1
+        logger.info(f"_stop_stimulation_trigger duration: {duration}")
+        if self._edf_handler:
+            self._edf_handler.trigger("stimulation should be stopped")
+        else:
+            logger.warning("stop stim trigger fail. no edf writer alive")
+        
+    def stop_stimulation(self):
+        logger.info("停止电刺激")  
+        msg = StopStimulationCommand.pack()
+        logger.debug(f"stop_stimulation message is {msg.hex()}")
+        self.socket.sendall(msg)
+        
+    # 启动采集
+    def start_acquisition(self, recording = True):
+        logger.info("启动信号采集")  
+        self._recording = recording
+        # 设置数据采集参数
+        param_bytes = SetAcquisitionParamCommand.build(self).pack()
+        # 启动数据采集
+        start_bytes = StartAcquisitionCommand.build(self).pack()
+        msg = param_bytes + start_bytes
+        logger.debug(f"start_acquisition message is {msg.hex()}")
+        self.socket.sendall(msg)
+        
+    # 停止采集
+    def stop_acquisition(self):
+        logger.info("停止信号采集")  
+        msg = StopAcquisitionCommand.build(self).pack()
+        logger.debug(f"stop_acquisition message is {msg.hex()}")
+        self.socket.sendall(msg)
+        if self._edf_handler:
+            # 发送结束标识
+            self.edf_handler.write(None)
+        
+    # 订阅实时数据
+    def subscribe(self, topic:str=None, q : Queue=None, type : Literal["signal","impedance"]="signal"):  
+            
+        # 数据队列
+        if q is None:
+            q = Queue(maxsize=1000)
+            
+        # 队列名称     
+        if topic is None:
+            topic = f"{type}_{time_ns()}"
+        
+        # 订阅生理电信号数据
+        if type == "signal":
+            # topic唯一，用来区分不同的订阅队列（下同）
+            if topic in list(self.__signal_consumer.keys()):
+                logger.warning(f"exists {type} subscribe of {topic}")
+            else:
+                self.__signal_consumer[topic] = q
+            
+        # 订阅阻抗数据
+        if type == "impedance":
+            if topic in list(self.__signal_consumer.keys()):
+                logger.warning(f"exists {type} subscribe of {topic}")
+            else:
+                self.__impedance_consumer[topic] = q
+            
+        return topic, q
+    
+    def trigger(self, desc):
+        if self._edf_handler:
+            self.edf_handler.trigger(desc)
+        else:
+            logger.warning("no edf handler, no place to recording trigger")
+        
+    def __str__(self):
+        return f''' 
+            Device: 
+                Name: {self.device_name}, 
+                Type: {hex(self.device_type) if self.device_type else None}, 
+                ID: {self.device_id if self.device_id else None}, 
+                Software: {self.software_version}, 
+                Hardware: {self.hardware_version}, 
+                Connect Time: {self.connect_time},
+                Current Time: {self.current_time},
+                Voltage: {str(self.voltage) + "mV" if self.voltage else None},
+                Battery Remain: {str(self.battery_remain)+ "%" if self.battery_remain else None},
+                Battery Total: {str(self.battery_total) + "%" if self.battery_total else None}
+            '''
+
+    def __repr__(self):
+        return f''' 
+            Device: 
+                Name: {self.device_name}, 
+                Type: {hex(self.device_type)}, 
+                ID: {self.device_id}, 
+                Software: {self.software_version}, 
+                Hardware: {self.hardware_version}, 
+                Connect Time: {self.connect_time},
+                Current Time: {self.current_time},
+                Voltage: {self.voltage}mV,
+                Battery Remain: {self.battery_remain}%,
+                Battery Total: {self.battery_total}%
+            '''
+
+    def __eq__(self, other):
+        return self.device_name == other.device_name and self.device_type == other.device_type and self.device_id == other.device_id
+
+    def __hash__(self):
+        return hash((self.device_name, self.device_type, self.device_id))    
+    
+    

qlsdk/rsc/device/device_factory.py

@@ -0,0 +1,29 @@
+from typing import Dict, Type
+from qlsdk.rsc.interface import IDevice
+
+from qlsdk.rsc.device.c64_rs import C64RS
+
+from loguru import logger
+
+class DeviceFactory(object):    
+    """Registry for device implementations"""
+    _devices: Dict[str, Type[IDevice]] = {}
+    
+    @classmethod
+    def register(cls, device_type: int, device: Type[IDevice]):
+        cls._devices[device_type] = device
+    
+    @classmethod
+    def create_device(cls, device: IDevice) -> Type[IDevice]:
+        logger.debug(f"Creating device for device type: {hex(device.device_type)}, support types: {cls._devices.keys()}")
+        if device.device_type not in cls._devices.keys():
+            logger.warning(f"不支持的设备类型: {hex(device.device_type)}")
+            raise ValueError(f"Unsupported device type: {hex(device.device_type)}")
+        
+        instance = cls._devices[device.device_type]
+        return instance.from_parent(device) if hasattr(instance, 'from_parent') else instance(device.socket)
+
+# Register the C64RS device with the DeviceFactory    
+DeviceFactory.register(C64RS.device_type, C64RS) 
+    
+    

qlsdk/rsc/entity.py

@@ -2,7 +2,7 @@ from multiprocessing import Queue, Process
 from typing import Any, Dict, Literal
 from threading import Thread
 from loguru import logger
-from time import time_ns
+from time import time_ns, sleep
 
 from qlsdk.core import *
 from qlsdk.core.device import BaseDevice
@@ -26,6 +26,10 @@ class QLDevice(BaseDevice):
         self.battery_remain = None
         # %
         self.battery_total = None
+        # persist
+        self._recording = False
+        self._storage_path = None
+        self._file_prefix = None
         
         # 可设置参数
         # 采集：采样量程、采样率、采样通道
@@ -34,6 +38,12 @@ class QLDevice(BaseDevice):
         self._sample_range:Literal[188, 375, 563, 750, 1125, 2250, 4500] = 188        
         # 采样率（Hz）：250、500、1000、2000、4000、8000、16000、32000
         self._sample_rate:Literal[250, 500, 1000, 2000, 4000, 8000, 16000, 32000] = 500
+        self._physical_max = 188000
+        self._physical_min = -188000
+        self._digital_max = 8388607
+        self._digital_min = -8388608
+        self._physical_range = 376000
+        self._digital_range = 16777215
         self._acq_channels = None
         self._acq_param = {
             "sample_range": 188,
@@ -115,8 +125,17 @@ class QLDevice(BaseDevice):
         self._accept.start()
         
     def init_edf_handler(self):
-        self._edf_handler = RscEDFHandler(self.sample_rate,  self.sample_range, - self.sample_range, - 2 >> 24,  2 >> 24 - 1, self.resolution)  
-        
+        self._edf_handler = RscEDFHandler(self.sample_rate,  self._physical_max , self._physical_min, self.resolution)  
+        self._edf_handler.set_device_type(self.device_type)
+        self._edf_handler.set_device_no(self.device_name)
+        self._edf_handler.set_storage_path(self._storage_path)
+        self._edf_handler.set_file_prefix(self._file_prefix)
+        
+    # eeg数字值转物理值
+    def eeg2phy(self, digital:int):
+        # 向量化计算（自动支持广播）
+        return ((digital - self._digital_min) / self._digital_range) * self._physical_range + self._physical_min
+    
     @property
     def edf_handler(self):
         if not self.storage_enable:
@@ -127,17 +146,22 @@ class QLDevice(BaseDevice):
             
         return self._edf_handler  
     
+    # 采集通道列表， 从1开始
     @property
     def acq_channels(self):
         if self._acq_channels is None:
             self._acq_channels = [i for i in range(1, 63)]
         return self._acq_channels
+    
+    # 量程范围
     @property
     def sample_range(self):
+        
         return self._sample_range if self._sample_range else 188
     @property
     def sample_rate(self):
         return self._sample_rate if self._sample_rate else 500
+    
     @property
     def resolution(self):
         return 24
@@ -150,13 +174,17 @@ class QLDevice(BaseDevice):
     def impedance_consumers(self):
         return self.__impedance_consumer
     
+    # 设置记录文件路径
     def set_storage_path(self, path):
-        pass
+        self._storage_path = path
+       
+    # 设置记录文件名称前缀 
+    def set_file_prefix(self, prefix):
+        self._file_prefix = prefix
         
     def accept(self):
         while True:
             data = self.socket.recv(4096*1024)
-            # logger.debug(f"QLDevice接收到数据: {data.hex()}")
             if not data:
                 logger.warning(f"设备{self.device_name}连接结束")
                 break
@@ -175,35 +203,31 @@ class QLDevice(BaseDevice):
         self.stim_paradigm = param
     
     # 设置采集参数
-    def set_acq_param(self, channels, sample_rate = 500, sample_range = 188):
+    def set_acq_param(self, channels, sample_rate:Literal[250, 500, 1000, 2000, 4000, 8000, 16000, 32000] = 500, sample_range:Literal[188, 375, 563, 750, 1125, 2250, 4500] = 188):
         self._acq_param["channels"] = channels 
         self._acq_param["sample_rate"] = sample_rate
         self._acq_param["sample_range"] = sample_range
         self._acq_channels = channels
         self._sample_rate = sample_rate
         self._sample_range = sample_range
+        # 根据量程更新信号物理值范围
+        self._physical_max = self._sample_range * 1000
+        self._physical_min = -self._sample_range * 1000
+        self._physical_range = self._physical_max - self._physical_min
     
-    # 通用配置
+    # 通用配置-TODO
     def set_config(self, key:str, val: str):
         pass      
         
-    def set_impedance_param(self):
-        channels = bytes.fromhex("FFFFFFFFFFFFFFFF0000000000000000000000000000000000000000000000000000")
-        sample_rate = 1000
-        sample_len = 300
-        resolution = self.resolution
-        
     def start_impedance(self):
         logger.info("启动阻抗测量")  
         msg = StartImpedanceCommand.build(self).pack()
-        # msg = bytes.fromhex("5aa50239320013243f0000001104ffffffffffffffff000000000000000000000000000000000000000000000000e8030000fa00000010000164000000a11e5aa50239320013243a00000012040000000000000000000000000000000000000000000000000000000000000000000001000000000000000325")
         logger.debug(f"start_impedance message is {msg.hex()}")
         self.socket.sendall(msg)
     
     def stop_impedance(self):
         logger.info("停止阻抗测量")  
         msg = StopImpedanceCommand.build(self).pack()
-        # msg = bytes.fromhex("5aa5023932001324100000001304e9df")
         logger.debug(f"stop_impedance message is {msg.hex()}")
         self.socket.sendall(msg)
     
@@ -212,20 +236,33 @@ class QLDevice(BaseDevice):
             logger.warning("刺激参数未设置，请先设置刺激参数")
             return
         logger.info("启动电刺激")  
-        # conf = SetStimulationParamCommand.build(self).pack()
         msg = StartStimulationCommand.build(self).pack()
         logger.debug(f"start_stimulation message is {msg.hex()}")
         self.socket.sendall(msg)
+        t = Thread(target=self._stop_stimulation_trigger, args=(self.stim_paradigm.duration,))
+        t.start()
+        
+    def _stop_stimulation_trigger(self, duration):
+        delay = duration
+        while delay > 0:
+            sleep(1)
+            delay -= 1
+        logger.info(f"_stop_stimulation_trigger duration: {duration}")
+        if self._edf_handler:
+            self._edf_handler.trigger("stimulation should be stopped")
+        else:
+            logger.warning("trigger fail for 'stop stim'. no recording file to write")
         
     def stop_stimulation(self):
         logger.info("停止电刺激")  
-        msg = StopStimulationCommand.pack()
+        msg = StopStimulationCommand.build().pack()
         logger.debug(f"stop_stimulation message is {msg.hex()}")
         self.socket.sendall(msg)
         
     # 启动采集
-    def start_acquisition(self):
+    def start_acquisition(self, recording = True):
         logger.info("启动信号采集")  
+        self._recording = recording
         # 设置数据采集参数
         param_bytes = SetAcquisitionParamCommand.build(self).pack()
         # 启动数据采集
@@ -238,11 +275,11 @@ class QLDevice(BaseDevice):
     def stop_acquisition(self):
         logger.info("停止信号采集")  
         msg = StopAcquisitionCommand.build(self).pack()
-        logger.debug(f"stop_acquisition message is {msg}")
+        logger.debug(f"stop_acquisition message is {msg.hex()}")
         self.socket.sendall(msg)
         if self._edf_handler:
             # 发送结束标识
-            self.edf_handler.append(None)
+            self.edf_handler.write(None)
         
     # 订阅实时数据
     def subscribe(self, topic:str=None, q : Queue=None, type : Literal["signal","impedance"]="signal"):  
@@ -271,6 +308,12 @@ class QLDevice(BaseDevice):
                 self.__impedance_consumer[topic] = q
             
         return topic, q
+    
+    def trigger(self, desc):
+        if self._edf_handler:
+            self.edf_handler.trigger(desc)
+        else:
+            logger.warning("no edf handler, no place to recording trigger")
         
     def __str__(self):
         return f''' 
@@ -326,10 +369,6 @@ class RSC64R(QLDevice):
 class ARSKindling(QLDevice):
     def __init__(self, socket):
         super().__init__(socket)
-        
-        
-        
-
 
 class DeviceParser(object):
     def __init__(self, device : QLDevice):

qlsdk/rsc/interface/__init__.py

@@ -0,0 +1,3 @@
+from .device import *
+from .parser import *
+from .handler import *

qlsdk/rsc/interface/command.py

@@ -0,0 +1,10 @@
+from abc import ABC, abstractmethod
+
+class ICommand(ABC):
+    def __init__(self):
+        super().__init__()
+        
+        
+    @abstractmethod
+    def append(self, buffer):
+        pass