qlsdk2 0.3.0a2__tar.gz → 0.4.0__tar.gz

{qlsdk2-0.3.0a2 → qlsdk2-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: qlsdk2
-Version: 0.3.0a2
+Version: 0.4.0
 Summary: SDK for quanlan device
 Home-page: https://github.com/hehuajun/qlsdk
 Author: hehuajun
@@ -12,7 +12,7 @@ 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
+Requires-Dist: bitarray>=1.5.3
 Provides-Extra: dev
 Requires-Dist: pytest>=6.0; extra == "dev"
 Requires-Dist: twine>=3.0; extra == "dev"

{qlsdk2-0.3.0a2 → qlsdk2-0.4.0}/setup.py

@@ -6,7 +6,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="qlsdk2",
-    version="0.3.0a2",
+    version="0.4.0",
     author="hehuajun",
     author_email="hehuajun@eegion.com",
     description="SDK for quanlan device",
@@ -24,13 +24,7 @@ setuptools.setup(
     install_requires=open("requirements.txt").read().splitlines(),
     include_package_data=True,
     package_data={
-        # "src/qlsdk/ar4m/": ["libs/*.dll"],
-        "qlsdk/sdk": ["libs/*.dll"],
-        "":["*.txt", "*.md"]
-    },
-    # entry_points={
-    #     'console_scripts': [
-    #         'qlsdk-cli=qlsdk.cli:main',  # 如果有命令行工具
-    #     ],
-    # },
+        # "qlsdk2": ["/**/*.dll"],
+        "qlsdk": ["./**/*.dll"]
+    }
 )

{qlsdk2-0.3.0a2 → qlsdk2-0.4.0}/src/qlsdk/__init__.py

@@ -5,6 +5,7 @@ from .ar4m import AR4M
 from .ar4 import AR4
 from .x8 import X8
 from .x8m import X8M
+from .rsc import *
 
 __all__ = ['AR4M', 'AR4', 'AR4Packet', 'X8']
 

qlsdk2-0.4.0/src/qlsdk/core/__init__.py

@@ -0,0 +1,5 @@
+from qlsdk.core.crc import *
+from qlsdk.core.message import *
+from qlsdk.core.local import *
+from qlsdk.core.utils import *
+from qlsdk.core.entity import *

qlsdk2-0.4.0/src/qlsdk/core/crc/__init__.py

@@ -0,0 +1,5 @@
+from .crctools import crc16
+
+# packet crc validate
+def check_crc(data):
+    return int.from_bytes(data[-2:], 'little')  == crc16(data[:-2])

qlsdk2-0.4.0/src/qlsdk/core/crc/crctools.py

@@ -0,0 +1,95 @@
+# len=256
+CRC_HI = [
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41,
+    0x00, 0xC1, 0x81, 0x40, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0,
+    0x80, 0x41, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1, 0x81, 0x40,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1,
+    0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0, 0x80, 0x41,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1,
+    0x81, 0x40, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x00, 0xC1, 0x81, 0x40,
+    0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1,
+    0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1, 0x81, 0x40,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x00, 0xC1, 0x81, 0x40,
+    0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0,
+    0x80, 0x41, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1, 0x81, 0x40,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41,
+    0x00, 0xC1, 0x81, 0x40, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41,
+    0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40, 0x00, 0xC1, 0x81, 0x40,
+    0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0, 0x80, 0x41, 0x00, 0xC1,
+    0x81, 0x40, 0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41,
+    0x00, 0xC1, 0x81, 0x40, 0x01, 0xC0, 0x80, 0x41, 0x01, 0xC0,
+    0x80, 0x41, 0x00, 0xC1, 0x81, 0x40
+]
+
+# len=256
+CRC_LO = [
+    0x00, 0xC0, 0xC1, 0x01, 0xC3, 0x03, 0x02, 0xC2, 0xC6, 0x06,
+    0x07, 0xC7, 0x05, 0xC5, 0xC4, 0x04, 0xCC, 0x0C, 0x0D, 0xCD,
+    0x0F, 0xCF, 0xCE, 0x0E, 0x0A, 0xCA, 0xCB, 0x0B, 0xC9, 0x09,
+    0x08, 0xC8, 0xD8, 0x18, 0x19, 0xD9, 0x1B, 0xDB, 0xDA, 0x1A,
+    0x1E, 0xDE, 0xDF, 0x1F, 0xDD, 0x1D, 0x1C, 0xDC, 0x14, 0xD4,
+    0xD5, 0x15, 0xD7, 0x17, 0x16, 0xD6, 0xD2, 0x12, 0x13, 0xD3,
+    0x11, 0xD1, 0xD0, 0x10, 0xF0, 0x30, 0x31, 0xF1, 0x33, 0xF3,
+    0xF2, 0x32, 0x36, 0xF6, 0xF7, 0x37, 0xF5, 0x35, 0x34, 0xF4,
+    0x3C, 0xFC, 0xFD, 0x3D, 0xFF, 0x3F, 0x3E, 0xFE, 0xFA, 0x3A,
+    0x3B, 0xFB, 0x39, 0xF9, 0xF8, 0x38, 0x28, 0xE8, 0xE9, 0x29,
+    0xEB, 0x2B, 0x2A, 0xEA, 0xEE, 0x2E, 0x2F, 0xEF, 0x2D, 0xED,
+    0xEC, 0x2C, 0xE4, 0x24, 0x25, 0xE5, 0x27, 0xE7, 0xE6, 0x26,
+    0x22, 0xE2, 0xE3, 0x23, 0xE1, 0x21, 0x20, 0xE0, 0xA0, 0x60,
+    0x61, 0xA1, 0x63, 0xA3, 0xA2, 0x62, 0x66, 0xA6, 0xA7, 0x67,
+    0xA5, 0x65, 0x64, 0xA4, 0x6C, 0xAC, 0xAD, 0x6D, 0xAF, 0x6F,
+    0x6E, 0xAE, 0xAA, 0x6A, 0x6B, 0xAB, 0x69, 0xA9, 0xA8, 0x68,
+    0x78, 0xB8, 0xB9, 0x79, 0xBB, 0x7B, 0x7A, 0xBA, 0xBE, 0x7E,
+    0x7F, 0xBF, 0x7D, 0xBD, 0xBC, 0x7C, 0xB4, 0x74, 0x75, 0xB5,
+    0x77, 0xB7, 0xB6, 0x76, 0x72, 0xB2, 0xB3, 0x73, 0xB1, 0x71,
+    0x70, 0xB0, 0x50, 0x90, 0x91, 0x51, 0x93, 0x53, 0x52, 0x92,
+    0x96, 0x56, 0x57, 0x97, 0x55, 0x95, 0x94, 0x54, 0x9C, 0x5C,
+    0x5D, 0x9D, 0x5F, 0x9F, 0x9E, 0x5E, 0x5A, 0x9A, 0x9B, 0x5B,
+    0x99, 0x59, 0x58, 0x98, 0x88, 0x48, 0x49, 0x89, 0x4B, 0x8B,
+    0x8A, 0x4A, 0x4E, 0x8E, 0x8F, 0x4F, 0x8D, 0x4D, 0x4C, 0x8C,
+    0x44, 0x84, 0x85, 0x45, 0x87, 0x47, 0x46, 0x86, 0x82, 0x42,
+    0x43, 0x83, 0x41, 0x81, 0x80, 0x40
+]
+
+def crc16(data):
+    crcHi = 0xFF
+    crcLo = 0xFF
+    for i in range(len(data)):
+        index = crcHi ^ data[i]
+        crcHi = crcLo ^ CRC_HI[index]
+        crcLo = CRC_LO[index]
+
+    return crcHi << 8 | crcLo
+
+class CRCEnum(object):
+    CRC8 = 8
+    CRC16 = 16
+    CRC32 = 32
+
+class CRC(object):
+    def __init__(self, width=CRCEnum.CRC16) :
+        self.width = width
+        self.crcHi = 0xFF
+        self.crcLo = 0xFF
+    
+    def _init(self):
+        if self.width == CRCEnum.CRC16:
+            self.crcHi = 0xFF
+            self.crcLo = 0xFF
+
+    def calc(self, data):
+        for i in range(len(data)):
+            index = self.crcHi ^ data[i]
+            self.crcHi = self.crcLo ^ CRC_HI[index]
+            self.crcLo = CRC_LO[index]
+
+    def checksum(self):
+        return self.crcHi << 8 | self.crcLo
+        

qlsdk2-0.4.0/src/qlsdk/core/device.py

@@ -0,0 +1,25 @@
+# from abc import ABC, abstractmethod
+
+class BaseDevice(object):
+    def __init__(self, socket = None):
+        self.socket = socket
+        self.device_name = None
+        self.device_type = None
+        self.device_id = None
+        
+    @property
+    def acq_channels(self) :
+        return None
+    @property
+    def sample_range(self) -> int:
+        return None
+    @property
+    def sample_rate(self) -> int:
+        return None
+    @property
+    def sample_num(self) -> int:
+        return 10
+    @property
+    def resolution(self):
+        return 24
+    

qlsdk2-0.4.0/src/qlsdk/core/entity/__init__.py

@@ -0,0 +1,92 @@
+from qlsdk.core.utils import to_channels
+from loguru import logger
+
+class DataPacket(object):
+    def __init__(self, device_type, device_id, channels, data):
+        self.data = data
+        self.channels = None
+        
+            
+class RscPacket(object):
+    def __init__(self):
+        self.time_stamp = None
+        self.pkg_id = None
+        self.result = None
+        self.channels = None
+        self.origin_sample_rate = None
+        self.sample_rate = None
+        self.sample_num = None
+        self.resolution = None
+        self.filter = None
+        self.data_len = None
+        self.trigger = None
+        self.eeg = None
+    
+    def transfer(self, body: bytes) -> 'RscPacket':
+        self.time_stamp = int.from_bytes(body[0:8], 'little')
+        self.result = body[8]
+        self.pkg_id = int.from_bytes(body[9: 13], 'little')
+        self.channels = to_channels(body[13: 45])
+        self.origin_sample_rate = int.from_bytes(body[45: 49], 'little')
+        self.sample_rate = int.from_bytes(body[49: 53], 'little')
+        self.sample_num = int.from_bytes(body[53: 57], 'little')
+        self.resolution = int(int(body[57]) / 8)
+        self.filter = body[58]
+        self.data_len = int.from_bytes(body[59: 63], 'little')
+        # 步径 相同通道的点间隔
+        step = int(len(self.channels) * self.resolution + 4)
+        self.trigger = [int.from_bytes(body[i:i+4], 'little') for i in range(63, len(body) - 3, step)]
+        b_eeg = body[63:]
+        ch_num = len(self.channels) 
+        self.eeg = [
+            [
+                int.from_bytes(b_eeg[i * self.resolution + 4 + j * step:i * self.resolution + 4 + j * step + 3], 'big', signed=True)
+                for j in range(self.sample_num)
+            ]
+            for i in range(ch_num)
+        ]
+                
+        # logger.trace(self)
+        return self
+        
+    def __str__(self):
+        return f"""
+            time_stamp: {self.time_stamp}
+            pkg_id: {self.pkg_id}
+            origin_sample_rate: {self.origin_sample_rate}
+            sample_rate: {self.sample_rate}
+            sample_num: {self.sample_num}
+            resolution: {self.resolution}
+            filter: {self.filter}
+            channels: {self.channels}
+            data len: {self.data_len}
+            trigger: {self.trigger}
+            eeg: {self.eeg}
+        """
+        
+class ImpedancePacket(object):
+    def __init__(self):
+        self.time_stamp = None
+        self.pkg_id = None
+        self.result = None
+        self.channels = None
+        self.data_len = None
+        self.impedance = None
+        
+    def transfer(self, body:bytes) -> 'ImpedancePacket':
+        self.time_stamp = int.from_bytes(body[0:8], 'little')
+        self.result = body[8]
+        self.pkg_id = int.from_bytes(body[9: 13], 'little')
+        self.channels = to_channels(body[13: 45])
+        
+        logger.debug(f"impedance: {self}")
+        
+    def __str__(self):
+        return f"""
+            time_stamp: {self.time_stamp}
+            pkg_id: {self.pkg_id}
+            result: {self.result}
+            channels: {self.channels}
+            data len: {self.data_len}
+            impedance: {self.impedance}
+        """

qlsdk2-0.4.0/src/qlsdk/core/filter/__init__.py

@@ -0,0 +1 @@
+from .norch import notch_filter_50hz

qlsdk2-0.4.0/src/qlsdk/core/filter/norch.py

@@ -0,0 +1,59 @@
+import numpy as np
+
+def notch_filter_50hz(data: np.ndarray, 
+                     fs: float,
+                     notch_width: float = 2.0,
+                     max_harmonics: int = 5) -> np.ndarray:
+    """
+    多通道50Hz谐波陷波滤波器
+    
+    参数：
+    data : 输入信号，形状为 [通道数, 采样点数] 的二维数组
+    fs : 采样频率 (Hz)
+    notch_width : 陷波带宽 (Hz)，默认2Hz
+    max_harmonics : 最大谐波次数，默认处理前10次谐波
+    
+    返回：
+    滤波后的信号，形状与输入相同
+    """
+    # 输入校验
+    if data.ndim != 2:
+        raise ValueError("输入必须为二维数组 [channels, samples]")
+    if fs <= 0:
+        raise ValueError("采样频率必须为正数")
+
+    n_channels, n_samples = data.shape
+    nyquist = fs / 2
+    processed = np.empty_like(data)
+    
+    # 生成频率轴
+    freqs = np.fft.fftfreq(n_samples, 1/fs)
+    
+    for ch in range(n_channels):
+        # FFT变换
+        fft_data = np.fft.fft(data[ch])
+        
+        # 生成陷波掩模
+        mask = np.ones(n_samples, dtype=bool)
+        
+        # 计算需要消除的谐波
+        for k in range(1, max_harmonics+1):
+            target_freq = 50 * k
+            
+            # 超过奈奎斯特频率则停止
+            if target_freq > nyquist:
+                break
+                
+            # 生成陷波范围
+            notch_range = (np.abs(freqs - target_freq) <= notch_width/2) | \
+                          (np.abs(freqs + target_freq) <= notch_width/2)
+            
+            mask &= ~notch_range
+        
+        # 应用频域滤波
+        filtered_fft = fft_data * mask
+        
+        # 逆变换并取实数部分
+        processed[ch] = np.real(np.fft.ifft(filtered_fft))
+    
+    return processed

qlsdk2-0.4.0/src/qlsdk/core/local.py

@@ -0,0 +1,34 @@
+import os
+import socket
+from time import time
+
+# 读取本机全部ip列表
+# return list
+def get_ips():
+    return socket.gethostbyname_ex(socket.gethostname())[-1]
+
+# 读取本机的ip地址
+# return str
+def get_ip():
+    # 优先读取活跃ip地址
+    routes = os.popen('route print').readlines()
+    for idx, item in enumerate(routes):
+        if ' 0.0.0.0 ' in item and len(item.split()) > 2:
+            return item.split()[-2]
+    
+    # 取第一个地址
+    ips = get_ips()
+    if len(ips) > 0 :
+        return ips[0]
+
+    raise ValueError("Ip address not exists.")
+
+def get_cache(fname=None):
+    if fname is None:
+        fname = int(time())
+    
+    cpath = os.path.abspath(os.path.abspath(__file__))
+    print(cpath)
+
+if __name__ == '__main__':
+    get_cache()

qlsdk2-0.4.0/src/qlsdk/core/message/__init__.py

@@ -0,0 +1,2 @@
+from .command import *
+from .udp import UDPMessage