qlsdk2 0.6.0a8__py3-none-any.whl → 0.6.0a9__py3-none-any.whl

qlsdk/persist/ars_edf.py

@@ -214,7 +214,7 @@ class ARSKindlingEDFHandler(object):
         
         if channel_pos is None or len(channel_pos) == 0 :
             logger.trace(f"没有指定分区{name}的通道，跳过")
-            pass
+            return
         
         # 分区数据包写入
         if name not in self._edf_handler.keys():
@@ -234,7 +234,8 @@ class ARSKindlingEDFHandler(object):
         data = RscPacket()
         data.time_stamp = packet.time_stamp
         data.pkg_id = packet.pkg_id
-        data.channels = channels
+        # data.channels = channels
+        data.channels = self.channel_display(channels)
         data.origin_sample_rate = packet.origin_sample_rate
         data.sample_rate = packet.sample_rate
         data.sample_num = packet.sample_num
@@ -251,3 +252,6 @@ class ARSKindlingEDFHandler(object):
         # trigger现在（20250702）多个分区共享， 分发到所有分区文件中标记
         for k in self._edf_handler.keys():
             self._edf_handler[k].trigger(desc, cur_time) 
+            
+    def channel_display(self, channel):
+        return [self._channel_mapping.get(channel_id, []) for channel_id in channel]

qlsdk/rsc/device/arskindling.py

@@ -77,19 +77,4 @@ class ARSKindling(QLBaseDevice):
         
         return body
     
-    def __str__(self):
-        return f''' 
-            Device: 
-                Name: {self.device_no}, 
-                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}
-            '''
-    
     

qlsdk/rsc/device/base.py

@@ -144,14 +144,14 @@ class QLBaseDevice(IDevice):
     def _produce_signal(self, body: bytes):        
         
         # 处理信号数据
-        data = self._signal_wrapper(body)   
+        data = self._signal_wrapper(body)  
         # logger.debug("pkg_id: {}, eeg len: {}".format(data.pkg_id, len(data.eeg)))
         # 
         trigger_positions = [index for index, value in enumerate(data.trigger) if value != 0]
         if len(trigger_positions) > 0:
             # logger.debug(f"Trigger触发点位置: {trigger_positions}, 触发点时间戳: {[data.time_stamp + int(pos * 1000 / data.sample_rate) for pos in trigger_positions]}")
             for pos in trigger_positions:
-                self.trigger(data.trigger[pos])
+                self.trigger(trigger_info(data.trigger[pos]))
         # 存储
         if self.storage_enable:            
             # 确保记录线程启动
@@ -402,8 +402,8 @@ class QLBaseDevice(IDevice):
         msg = StartStimulationCommand.build(self).pack()
         logger.trace(f"start_stimulation message is {msg.hex()}")
         self.socket.sendall(msg)
-        t = Thread(target=self._stop_stimulation_trigger, args=(self.stim_paradigm.duration,), daemon=True)
-        t.start()
+        # 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()
@@ -421,7 +421,7 @@ class QLBaseDevice(IDevice):
         
     def stop_stimulation(self):
         logger.info(f"[设备-{self.device_no}]停止电刺激")  
-        msg = StopStimulationCommand.pack()
+        msg = StopStimulationCommand.build(self).pack()
         logger.trace(f"stop_stimulation message is {msg.hex()}")
         self.socket.sendall(msg)
         
@@ -537,4 +537,15 @@ class QLBaseDevice(IDevice):
         return self.device_type == other.device_type and self.device_no == other.device_no
 
     def __hash__(self):
-        return hash((self.device_type, self.device_no))    
+        return hash((self.device_type, self.device_no))    
+    
+__TRIGGER_MAPPING = {
+    0x3E8: "Start of stimulation",
+    0x3E9: "End of stimulation",
+    0x3EA: "Ascending end of stimulation",
+    0x3EB: "Descending start of stimulation ",
+    0x3EC: "刺激参数有误",
+    0x3ED: "End of stimulation (by force)",
+}    
+def trigger_info(code: int) -> str:
+    return __TRIGGER_MAPPING.get(code, "未知触发事件")    

qlsdk/rsc/parser/base.py

@@ -131,10 +131,15 @@ class TcpMessageParser(IParser):
     
     def unpack(self, packet):        
         # 提取指令码
-        cmd_code = int.from_bytes(packet[self.cmd_pos : self.cmd_pos + 2], 'little')
-        cmd_class = CommandFactory.create_command(cmd_code)
-        instance = cmd_class(self.device)
-        instance.parse_body(packet[self.header_len:-2])
+        cmd_code = None
+        try:
+            cmd_code = int.from_bytes(packet[self.cmd_pos : self.cmd_pos + 2], 'little')
+            cmd_class = CommandFactory.create_command(cmd_code)
+            instance = cmd_class(self.device)
+            instance.parse_body(packet[self.header_len:-2])
+        except Exception as e:
+            logger.error(f"指令[{cmd_code}]数据包[len={len(packet)}]解析异常: {e}")
+            
         return instance
             
     def start(self):

{qlsdk2-0.6.0a8.dist-info → qlsdk2-0.6.0a9.dist-info}/METADATA

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

{qlsdk2-0.6.0a8.dist-info → qlsdk2-0.6.0a9.dist-info}/RECORD

@@ -28,7 +28,7 @@ qlsdk/interface/parser.py,sha256=00Ds01Xp9eUPwx7okL4B1Bl0_fRbgaxfm_2GOeWoZiM,248
 qlsdk/interface/stimulator.py,sha256=MJbgiI-4qTMjbdhcBAZe8zb10hV9x8yhFCPL3Iq_QyI,28
 qlsdk/interface/store.py,sha256=694WQnnrtXThP44JkQlBXzghvqA0PwHKK3uA640aiUo,23
 qlsdk/persist/__init__.py,sha256=b8qk1aOU6snEMCQNYDl1ijV3-2gwBmMt76fiAzNk1E8,107
-qlsdk/persist/ars_edf.py,sha256=H5dhdoQyJps5WM23vZoY1EKj0PYx0OY9AZeui72TjaY,8930
+qlsdk/persist/ars_edf.py,sha256=EI0V3kcjBduqk6UDzPR99ua_5YED-LHsC6lpKV411kc,9131
 qlsdk/persist/edf.py,sha256=ETngb86CfkIUJYWmw86QR445MvTFC7Edk_CH9nyNgtY,7857
 qlsdk/persist/rsc_edf.py,sha256=djR_qXScb_XOOXaoMsvnK_qO2NCPzqEeKxKJCm4xnpk,13989
 qlsdk/persist/stream.py,sha256=TCVF1sqDrHiYBsJC27At66AaCs-_blXeXA_WXdJiIVA,5828
@@ -42,8 +42,8 @@ qlsdk/rsc/proxy.py,sha256=9CPdGNGWremwBUh4GvlXAykYB-x_BEPPLqsNvwuwIDE,2736
 qlsdk/rsc/command/__init__.py,sha256=iPrPor7NwBIrkQDxz061gVbnlLYK9KE7VXy-NKJy1XQ,12277
 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=tVih2D3Hhhal3WP0yqjKpzkwFPi6KLenshFStoP-vsw,4123
-qlsdk/rsc/device/base.py,sha256=fCGzjAbagTTGRBfHhbiYOqN3TLOViEd8il8Qp6RiVJs,21106
+qlsdk/rsc/device/arskindling.py,sha256=j1t-_49MNpi2cNYCWpJaBgj6s_NGVQCuMrdAB7rpPfg,3358
+qlsdk/rsc/device/base.py,sha256=G3o9Pgvy4zoTFvh_mX1ihw0AzglgCcFAmfwAXbmIKlk,21515
 qlsdk/rsc/device/c16_rs.py,sha256=qXt8m5vwcKQsN8JBllWnAsda5_Y6qkEhfHQQX101TMQ,5826
 qlsdk/rsc/device/c256_rs.py,sha256=7vAEzf_ggNcwrXKmcZMylnKzLFD5ZqtAIfkkI3lQ1iI,1682
 qlsdk/rsc/device/c64_rs.py,sha256=x8wHdwATKDU34j9vXNEXsNSJg23RAWmAKL8pgIGamG8,1091
@@ -61,7 +61,7 @@ qlsdk/rsc/network/__init__.py,sha256=PfYiqXS2pZV__uegQ1TjaeYhY1pefZ_shwE_X5HNVbs
 qlsdk/rsc/network/discover.py,sha256=GRXP0WxxIorDZWXq1X5CPAV60raSRvNiVwMQE8647XA,3044
 qlsdk/rsc/parser/__init__.py,sha256=cVKk06bRYOnwE3XMoksGlatKJSaZE1GVrYypQm-9aro,69
 qlsdk/rsc/parser/base-new.py,sha256=cAOy1V_1fAJyGq7bm7uLxpW41DbkllWOprnfWKpjtsQ,5116
-qlsdk/rsc/parser/base.py,sha256=tH4r_1PVkYfF2alpxKdXiFHmhTDPW_NKbq6DUqU6LVg,5702
+qlsdk/rsc/parser/base.py,sha256=EelI7jYLQFJ-frN_X8X0bc7rWSe-pJWCCylSvJdr0TM,5897
 qlsdk/rsc/parser/rsc.py,sha256=RuBqsg5KZNio7mTyM14svQEeU0_0CLhnRmfbP0NBM2o,4724
 qlsdk/sdk/__init__.py,sha256=v9LKP-5qXCqnAsCkiRE9LDb5Tagvl_Qd_fqrw7y9yd4,68
 qlsdk/sdk/ar4sdk.py,sha256=tugH3UUeNebdka78AzLyrtAXbYQQE3iFJ227zUit6tY,27261
@@ -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.0a8.dist-info/METADATA,sha256=RweCHF8Y0Gv3rDVK58bgKl7dgQAmZreBAUcelU_rLbM,1882
-qlsdk2-0.6.0a8.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-qlsdk2-0.6.0a8.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
-qlsdk2-0.6.0a8.dist-info/RECORD,,
+qlsdk2-0.6.0a9.dist-info/METADATA,sha256=As8KCL909tZ_a7nQ2H7lqircEwZx_Jxah768hCKkIWQ,1882
+qlsdk2-0.6.0a9.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+qlsdk2-0.6.0a9.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
+qlsdk2-0.6.0a9.dist-info/RECORD,,