qlsdk2 0.6.0a4__py3-none-any.whl → 0.6.0a6__py3-none-any.whl

qlsdk/rsc/command/__init__.py

@@ -233,7 +233,7 @@ class StartStimulationCommand(DeviceCommand):
     cmd_code = 0x48C
     cmd_desc = "启动刺激"
     def pack_body(self):
-        return self.device.stim_paradigm.to_bytes()
+        return self.device.get_stim_param()
         # return bytes.fromhex('01000000000000008813000000000000010000000000000000000140420f00640064000000803f0000010000000000000000000000000000000000000000008813000000000000')
     def parse_body(self, body: bytes):
         # time - 8B

qlsdk/rsc/device/base.py

@@ -9,6 +9,7 @@ from qlsdk.core.entity import RscPacket, ImpedancePacket
 from qlsdk.core.utils import to_bytes
 from qlsdk.rsc.interface import IDevice, IParser
 from qlsdk.rsc.command import StartImpedanceCommand, StopImpedanceCommand, StartStimulationCommand, StopStimulationCommand, SetAcquisitionParamCommand, StartAcquisitionCommand, StopAcquisitionCommand
+from qlsdk.rsc.paradigm import StimulationParadigm
 from qlsdk.rsc.parser.base import TcpMessageParser
 
 class QLBaseDevice(IDevice):
@@ -91,7 +92,7 @@ class QLBaseDevice(IDevice):
             ]
         }
         
-        self.stim_paradigm = None
+        self.stim_paradigm: StimulationParadigm = None
                       
         signal_info = {
             "param" : None,
@@ -389,6 +390,9 @@ class QLBaseDevice(IDevice):
         t = Thread(target=self._stop_stimulation_trigger, args=(self.stim_paradigm.duration,), daemon=True)
         t.start()
         
+    def get_stim_param(self) -> bytes:
+        return self.stim_paradigm.to_bytes()
+        
     def _stop_stimulation_trigger(self, duration):
         delay = duration
         while delay > 0:

qlsdk/rsc/device/c256_rs.py

@@ -1,9 +1,3 @@
-
-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.core.entity import C256RSPacket
 from qlsdk.persist import RscEDFHandler
@@ -18,130 +12,6 @@ class C256RS(QLBaseDevice):
     
     def __init__(self, socket):
         super().__init__(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 = RSCMessageParser(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
-            
         
     def init_edf_handler(self):
         self._edf_handler = RscEDFHandler(self.sample_rate,  self.sample_range * 1000 , - self.sample_range * 1000, self.resolution)  
@@ -149,191 +19,14 @@ class C256RS(QLBaseDevice):
         self._edf_handler.set_device_no(self.device_no)
         self._edf_handler.set_storage_path(self._storage_path)
         self._edf_handler.set_file_prefix(self._file_prefix if self._file_prefix else 'C256RS')
+        logger.debug(f"EDF Handler initialized")
         
-    @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, 256)]
-        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 send(self, data):
-        self.socket.sendall(data)
-    
-    # 设置刺激参数
-    def set_stim_param(self, param):
-        self.stim_paradigm = param
     
     def _signal_wrapper(self, body: bytes):
         return C256RSPacket().transfer(body)
         
-    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=10240000)  # 10M缓存
-            
-        # 队列名称     
-        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))    
-    
+    def get_stim_param(self) -> bytes:
+        return self.stim_paradigm.to_bytes_c256()    
     
 DeviceFactory.register(C256RS.device_type, C256RS) 
 DeviceFactory.register(C256RS.device_type_001, C256RS) 

qlsdk/rsc/device/c64_rs.py

@@ -1,14 +1,7 @@
-
-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.interface import IDevice
 from qlsdk.rsc.command import *
-from qlsdk.rsc.parser.base import TcpMessageParser
 from qlsdk.rsc.device.base import QLBaseDevice
 
 class C64RS(QLBaseDevice):
@@ -17,123 +10,6 @@ class C64RS(QLBaseDevice):
     
     def __init__(self, socket):
         super().__init__(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._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:
@@ -149,183 +25,5 @@ class C64RS(QLBaseDevice):
         self._edf_handler.set_device_no(self.device_no)
         self._edf_handler.set_storage_path(self._storage_path)
         self._edf_handler.set_file_prefix(self._file_prefix if self._file_prefix else 'C64RS')
-        
-    @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 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))    
-    
+        logger.debug(f"EDF Handler initialized")
     

qlsdk/rsc/paradigm.py

@@ -61,6 +61,22 @@ class StimulationChannel(ABC):
         
         return result
     
+    def to_bytes_c256(self):        
+        result = self.channel_id.to_bytes(1, 'little')
+        result += self.wave_form.to_bytes(1, 'little')
+        
+        result += self._to_bytes_c256()
+        
+        result += int(self.delay_time).to_bytes(4, 'little')
+        result += int(self.ramp_up * 1000).to_bytes(4, 'little')
+        result += int((self.duration + self.ramp_up)  * 1000).to_bytes(4, 'little')
+        result += int(self.ramp_down * 1000).to_bytes(4, 'little')
+        
+        return result
+    
+    def _to_bytes_c256(self):
+        return bytes.fromhex("00000000000000000000000000000000000000000000000000000000")
+    
     def _ext_bytes(self):
         return bytes.fromhex("00000000000000000000000000000000")
     
@@ -99,6 +115,9 @@ class DCStimulation(StimulationChannel):
             "delay_time": self.delay_time
         }
         
+    def _to_bytes_c256(self):
+        return int(self.current_max * 1000 ).to_bytes(2, 'little') + bytes.fromhex("0000000000000000000000000000000000000000000000000000")
+        
     def from_json(self, param):
         pass
     def __str__(self):
@@ -140,9 +159,19 @@ class SquareWaveStimulation(StimulationChannel):
         result += int((self.duration + self.ramp_up)  * 1000).to_bytes(4, 'little')
         result += int(self.ramp_down * 1000).to_bytes(4, 'little')
         
-        return result
-        
-# 刺激模式-交流
+        return result      
+    
+    def _to_bytes_c256(self):
+        result = int(self.current_max * 1000000 ).to_bytes(4, 'little')
+        result += int(self.frequency).to_bytes(2, 'little')
+        result += bytes.fromhex("0000")
+        result += struct.pack('<f', self.duty_cycle)
+        result = int(self.current_min * 1000000 ).to_bytes(4, 'little')
+        result += self._ext_bytes()
+        return  result
+    
+      
+# 刺激模式-正弦
 class ACStimulation(StimulationChannel):
     '''
         channel_id: int,  通道编号，从0开始
@@ -177,6 +206,15 @@ class ACStimulation(StimulationChannel):
         
     def from_json(self, param):
         pass
+    
+    def _to_bytes_c256(self):
+        result = int(self.current_max * 1000 ).to_bytes(2, 'little')
+        result += bytes.fromhex("0000")
+        result += struct.pack('<f', self.frequency)
+        result += struct.pack('<f', self.phase_position)
+        result += self._ext_bytes()
+        return  result
+    
     def __str__(self):
         return f"ACStimulation(channel_id={self.channel_id}, waveform={self.waveform}, current={self.current_max}, duration={self.duration}, ramp_up={self.ramp_up}, ramp_down={self.ramp_down}, frequency={self.frequency}, phase_position={self.phase_position}, duration_delay={self.duration_delay})"
         
@@ -230,6 +268,15 @@ class PulseStimulation(StimulationChannel):
         
         return result
     
+    def _to_bytes_c256(self):
+        result = int(self.current_max * 1000000 ).to_bytes(4, 'little')
+        result += int(self.frequency).to_bytes(2, 'little')
+        result += bytes.fromhex("0000")
+        result += struct.pack('<f', self.duty_cycle)
+        result = int(self.current_min * 1000000 ).to_bytes(4, 'little')
+        result += self._ext_bytes()
+        return  result
+    
     def to_json(self):
         return {
             "channel_id": self.channel_id,
@@ -254,7 +301,7 @@ class StimulationParadigm(object):
     def __init__(self):
         self.channels = None
         self.duration = None
-        self.interval_time = 0
+        self.interval_time = 25000
         self.characteristic = 1
         self.mode = 0
         self.repeats = 0
@@ -287,6 +334,17 @@ class StimulationParadigm(object):
             result += channel.to_bytes()
         return result
     
+    def to_bytes_c256(self):
+        result = to_bytes(list(self.channels.keys()), 256)
+        result += int(self.duration * 1000).to_bytes(4, 'little')
+        result += int(self.interval_time).to_bytes(4, 'little')
+        result += int(self.characteristic).to_bytes(4, 'little')
+        result += int(self.mode).to_bytes(1, 'little')
+        result += int(self.repeats).to_bytes(4, 'little')
+        for channel in self.channels.values():
+            result += channel.to_bytes()
+        return result
+    
     def to_json(self):
         # Convert the object to JSON for transmission
         return {
@@ -306,8 +364,70 @@ class StimulationParadigm(object):
     def clear(self):
         self.channels = None
         self.duration = None
-        self.interval_time = 0
+        self.interval_time = 25000
         self.characteristic = 1
         self.mode = 0
         self.repeats = 0
 
+
+# 刺激范式
+class C256StimulationParadigm(object):
+    def __init__(self):
+        self.channels = None
+        self.duration = None
+        self.interval_time = 0
+        self.characteristic = 1
+        self.mode = 0
+        self.repeats = 0
+        
+    def add_channel(self, channel: StimulationChannel, update=False):
+        if self.channels is None:
+            self.channels = {}
+        channel_id = channel.channel_id + 1
+        if channel_id in self.channels.keys():
+            logger.warning(f"Channel {channel_id} already exists")
+            if update:
+                self.channels[channel_id] = channel
+        else:
+            self.channels[channel_id] = channel  
+            
+        # 计算刺激时间
+        duration = channel.duration + channel.ramp_up + channel.ramp_down     
+        if self.duration is None or duration > self.duration:
+            self.duration = duration 
+        
+        
+    def to_bytes(self):
+        result = to_bytes(list(self.channels.keys()), 64)
+        result += int(self.duration * 1000).to_bytes(4, 'little')
+        result += int(self.interval_time).to_bytes(4, 'little')
+        result += int(self.characteristic).to_bytes(4, 'little')
+        result += int(self.mode).to_bytes(1, 'little')
+        result += int(self.repeats).to_bytes(4, 'little')
+        for channel in self.channels.values():
+            result += channel.to_bytes()
+        return result
+    
+    def to_json(self):
+        # Convert the object to JSON for transmission
+        return {
+            "channels": list(self.channels.keys()),
+            "duration": self.duration,
+            "interval_time": self.interval_time,
+            "characteristic": self.characteristic,
+            "mode": self.mode,
+            "repeats": self.repeats, 
+            "stim": [channel.to_json() for channel in self.channels.values()]
+        }
+        
+    # @staticmethod
+    # def from_json(param: Dict[str, Any]):
+    #     pass
+    
+    def clear(self):
+        self.channels = None
+        self.duration = None
+        self.interval_time = 0
+        self.characteristic = 1
+        self.mode = 0
+        self.repeats = 0

{qlsdk2-0.6.0a4.dist-info → qlsdk2-0.6.0a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qlsdk2
-Version: 0.6.0a4
+Version: 0.6.0a6
 Summary: SDK for quanlan device
 Home-page: https://github.com/hehuajun/qlsdk
 Author: hehuajun

{qlsdk2-0.6.0a4.dist-info → qlsdk2-0.6.0a6.dist-info}/RECORD

@@ -37,16 +37,16 @@ qlsdk/rsc/device_manager.py,sha256=1ucd-lzHkNeQPKPzXV6OBkAMqPp_vOcsLyS-9TJ7wRc,4
 qlsdk/rsc/discover.py,sha256=ONXN6YWY-OMU0sBoLqqKUyb-8drtAp1g_MvnpzaFvHQ,3124
 qlsdk/rsc/eegion.py,sha256=lxrktO-3Z_MYdFIwc4NxvgLM5AL5kU3UItjH6tsKmHY,11670
 qlsdk/rsc/entity.py,sha256=-fRWFkVWp9d8Y1uh6GiacXC5scdeEKNiNFf3aziGdCE,17751
-qlsdk/rsc/paradigm.py,sha256=DGfwY36sMdPIMRjbGo661GvUTEwsRRi3jrmG405mSTk,12840
+qlsdk/rsc/paradigm.py,sha256=dYkm0BCpx7tkL-1OE3Io48JgKT0j-lP-n3uC1AcFiWs,17561
 qlsdk/rsc/proxy.py,sha256=9CPdGNGWremwBUh4GvlXAykYB-x_BEPPLqsNvwuwIDE,2736
-qlsdk/rsc/command/__init__.py,sha256=TncLrFyanIvaE106EKd2kqHJloQTZee--XaPnADL-Sc,12233
+qlsdk/rsc/command/__init__.py,sha256=bnnPxn5mChrKCpJbnX718YuFEBDZ8KESYfMIf_8K7J0,12225
 qlsdk/rsc/command/message.py,sha256=nTdG-Vp4MBnltyrgedAWiKD6kzOaPrg58Z_hq6yjhys,12220
 qlsdk/rsc/device/__init__.py,sha256=BzY9lRfssGPUlJ1ys-v3CWNgGihg7mPa2T4X0tl0Vg4,214
 qlsdk/rsc/device/arskindling.py,sha256=owci6MEGjyWqohEXzPdKj_ESeVIZKgO53StVj6Tmi18,15002
-qlsdk/rsc/device/base.py,sha256=De0SjD84i0_apwPxdnnpGXaVdUGUGAy7YE1k5zNqdfs,20070
+qlsdk/rsc/device/base.py,sha256=cE4nGsOmw44gU4e1PiJmbv2KuQ0dFIoIIkXXjA4_44A,20239
 qlsdk/rsc/device/c16_rs.py,sha256=BHQRHOnsTMAKgqSXaAS2RjPIklZQAl2CVfe6i_iX-i4,5928
-qlsdk/rsc/device/c256_rs.py,sha256=hsv2MBDFkK2rfsNTkDdG3TBAfVMAaSnTgnsi_NiQB4o,13222
-qlsdk/rsc/device/c64_rs.py,sha256=AAwYcioJ741b22fifhSHTC3C-ESbRtXTe0b8A8OvNjg,12607
+qlsdk/rsc/device/c256_rs.py,sha256=EOzNqf4zZwmnzzLyb0KlOXmHhFjPJ3TUVGUDkGgI5Wc,1281
+qlsdk/rsc/device/c64_rs.py,sha256=x8wHdwATKDU34j9vXNEXsNSJg23RAWmAKL8pgIGamG8,1091
 qlsdk/rsc/device/c64s1.py,sha256=L7nKmsoMCGj6GMjHYfYkKgkBtrGfP516kQHQ5I1FAUE,13986
 qlsdk/rsc/device/device_factory.py,sha256=6cPhm3pPGrVXA1s1HePFLjZqmhNI1vOAucFI0VRD_Y0,1317
 qlsdk/rsc/interface/__init__.py,sha256=xeRzIlQSB7ZSf4r5kLfH5cDQLzCyWeJAReG8Xq5nOE0,70
@@ -70,7 +70,7 @@ qlsdk/sdk/libs/libAr4SDK.dll,sha256=kZp9_DRwPdAJ5OgTFQSqS8tEETxUs7YmmETuBP2g60U,
 qlsdk/sdk/libs/libwinpthread-1.dll,sha256=W77ySaDQDi0yxpnQu-ifcU6-uHKzmQpcvsyx2J9j5eg,52224
 qlsdk/x8/__init__.py,sha256=FDpDK7GAYL-g3vzfU9U_V03QzoYoxH9YLm93PjMlANg,4870
 qlsdk/x8m/__init__.py,sha256=cLeUqEEj65qXw4Qa4REyxoLh6T24anSqPaKe9_lR340,634
-qlsdk2-0.6.0a4.dist-info/METADATA,sha256=BBbbb11OYzAxMMoXHycDv36RDWPBTSQIjSJ0Aw51wL8,1882
-qlsdk2-0.6.0a4.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-qlsdk2-0.6.0a4.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
-qlsdk2-0.6.0a4.dist-info/RECORD,,
+qlsdk2-0.6.0a6.dist-info/METADATA,sha256=xrhOFC2qVtEuKyGCkA_rOjFRGbFGrI0XB-hU1vLrHKo,1882
+qlsdk2-0.6.0a6.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+qlsdk2-0.6.0a6.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
+qlsdk2-0.6.0a6.dist-info/RECORD,,