qlsdk2 0.6.0a4__tar.gz → 0.6.0a5__tar.gz

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/PKG-INFO

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

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/setup.py

@@ -6,7 +6,7 @@ with open("README.md", "r", encoding='utf-8' ) as fh:
 
 setuptools.setup(
     name="qlsdk2",
-    version="0.6.0a4",
+    version="0.6.0a5",
     author="hehuajun",
     author_email="hehuajun@eegion.com",
     description="SDK for quanlan device",

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/src/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

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/src/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:

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/src/qlsdk/rsc/device/c256_rs.py

@@ -225,6 +225,9 @@ class C256RS(QLBaseDevice):
         t = Thread(target=self._stop_stimulation_trigger, args=(self.stim_paradigm.duration,))
         t.start()
         
+    def get_stim_param(self) -> bytes:
+        return self.stim_paradigm.to_bytes_c256()
+        
     def _stop_stimulation_trigger(self, duration):
         delay = duration
         while delay > 0:

{qlsdk2-0.6.0a4 → qlsdk2-0.6.0a5}/src/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,
@@ -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 → qlsdk2-0.6.0a5}/src/qlsdk2.egg-info/PKG-INFO

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