qldsdk-serial 0.1.0__tar.gz

qldsdk_serial-0.1.0/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.1
+Name: qldsdk-serial
+Version: 0.1.0
+Summary: SDK for quanlan device with serial communication
+Home-page: https://github.com/hehuajun/qlsdk
+Author: hehuajun
+Author-email: hehuajun@eegion.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: Microsoft :: Windows :: Windows 10
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: numpy>=1.23.5
+Requires-Dist: bitarray>=1.5.3
+Requires-Dist: pyserial>=3.5.0
+Requires-Dist: pyedflib>=0.1.42
+Requires-Dist: scipy>=1.10.1
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: twine>=3.0; extra == "dev"
+
+## 1. 项目名称
+   qldsdk-serial
+
+## 2. 项目说明
+
+qldsdk-serial 是一个功能完整的脑电设备控制和数据采集工具包，采用模块化设计，支持多种设备类型和功能。通过工厂模式和接口定义，实现了设备的统一管理和扩展。系统架构清晰，核心功能完善，为脑电设备的使用提供了便捷的编程接口。
+
+该SDK不仅支持基本的数据采集和存储功能，还提供了丰富的刺激控制选项，满足不同实验场景的需求。未来通过完善接口定义、增强错误处理、添加更多设备支持等方式进一步提升系统的稳定性和可扩展性。

qldsdk_serial-0.1.0/README.md

@@ -0,0 +1,8 @@
+## 1. 项目名称
+   qldsdk-serial
+
+## 2. 项目说明
+
+qldsdk-serial 是一个功能完整的脑电设备控制和数据采集工具包，采用模块化设计，支持多种设备类型和功能。通过工厂模式和接口定义，实现了设备的统一管理和扩展。系统架构清晰，核心功能完善，为脑电设备的使用提供了便捷的编程接口。
+
+该SDK不仅支持基本的数据采集和存储功能，还提供了丰富的刺激控制选项，满足不同实验场景的需求。未来通过完善接口定义、增强错误处理、添加更多设备支持等方式进一步提升系统的稳定性和可扩展性。

qldsdk_serial-0.1.0/setup.cfg

@@ -0,0 +1,32 @@
+[metadata]
+name = qldsdk-serial
+version = 0.1.0
+author = hehuajun
+author_email = hehuajun@eegion.com
+description = Quan Lan Device SDK for Serial Communication
+long_description = file: README.md
+long_description_content_type = text/markdown
+url = https://github.com/yourusername/qlsdk
+
+[options]
+package_dir = = src
+packages = find:
+python_requires = >=3.10
+install_requires = 
+	loguru>=0.6.0
+
+[options.packages.find]
+where = src
+exclude = 
+	tests*
+	examples*
+
+[options.extras_require]
+dev = 
+	pytest>=6.0
+	twine>=3.0
+
+[egg_info]
+tag_build = 
+tag_date = 0
+

qldsdk_serial-0.1.0/setup.py

@@ -0,0 +1,26 @@
+# setup.py
+import setuptools
+
+with open("README.md", "r", encoding='utf-8' ) as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="qldsdk-serial",
+    version="0.1.0",
+    author="hehuajun",
+    author_email="hehuajun@eegion.com",
+    description="SDK for quanlan device with serial communication",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/hehuajun/qlsdk",
+    package_dir={"": "src"},
+    packages=setuptools.find_packages(where="src"),
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: Microsoft :: Windows :: Windows 10",
+    ],
+    python_requires='>=3.10',
+    install_requires=open("requirements.txt").read().splitlines(),
+    include_package_data=True,
+)

qldsdk_serial-0.1.0/src/qldsdk/__init__.py

@@ -0,0 +1,4 @@
+
+from .core import *
+from .serial import *
+from .third import *

qldsdk_serial-0.1.0/src/qldsdk/core/__init__.py

@@ -0,0 +1,8 @@
+from qldsdk.core.crc import *
+from qldsdk.core.message import *
+from qldsdk.core.local import *
+from qldsdk.core.utils import *
+from qldsdk.core.entity import *
+from qldsdk.core.serial import *
+from qldsdk.core.dict_utils import filter_dict_keys
+from qldsdk.core.rsc import *

qldsdk_serial-0.1.0/src/qldsdk/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])

qldsdk_serial-0.1.0/src/qldsdk/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
+        

qldsdk_serial-0.1.0/src/qldsdk/core/device.py

@@ -0,0 +1,34 @@
+# 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 device_type(self) -> int:
+        return self._device_type
+    
+    @device_type.setter
+    def device_type(self, value: int):
+        self._device_type = value
+        
+    @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
+    

qldsdk_serial-0.1.0/src/qldsdk/core/dict_utils.py

@@ -0,0 +1,12 @@
+def filter_dict_keys(source_dict: dict, keys_to_keep: set) -> dict:
+    """
+    从源字典中提取指定key集合
+    
+    Args:
+        source_dict: 源字典，key不固定
+        keys_to_keep: 要保留的key集合
+    
+    Returns:
+        只包含指定key的新字典
+    """
+    return {k: v for k, v in source_dict.items() if k in keys_to_keep}

qldsdk_serial-0.1.0/src/qldsdk/core/entity/__init__.py

@@ -0,0 +1,222 @@
+from qldsdk.core.utils import to_channels
+from loguru import logger
+
+class Packet(object):
+    def __init__(self):
+        self.time_stamp = None
+        self.pkg_id = None
+        self.result = None
+        self.channels = None
+        
+            
+class RscPacket(Packet):
+    def __init__(self):
+        super().__init__()
+        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 # 数字信号
+        self.eeg_p = None # 物理值
+    
+    @staticmethod
+    def transfer(body: bytes) -> 'RscPacket':
+        packet = RscPacket()
+        packet.time_stamp = int.from_bytes(body[0:8], 'little')
+        packet.result = body[8]
+        packet.pkg_id = int.from_bytes(body[9: 13], 'little')
+        # logger.trace(f"pkg_id: {packet.pkg_id}")
+        packet.channels = to_channels(body[13: 45])
+        packet.origin_sample_rate = int.from_bytes(body[45: 49], 'little')
+        packet.sample_rate = int.from_bytes(body[49: 53], 'little')
+        packet.sample_num = int.from_bytes(body[53: 57], 'little')
+        packet.resolution = int(int(body[57]) / 8)
+        packet.filter = body[58]
+        packet.data_len = int.from_bytes(body[59: 63], 'little')
+        # 步径 相同通道的点间隔
+        step = int(len(packet.channels) * packet.resolution + 4)
+        b_eeg = body[63:]
+        ch_num = len(packet.channels) 
+        # 字节序列(4Cn{channel_size}){sample_num})
+        packet.trigger = [int.from_bytes(b_eeg[j * step : j * step + 4], 'little', signed=False) for j in range(packet.sample_num)]
+        
+        packet.eeg = [
+            [
+                int.from_bytes(b_eeg[i * packet.resolution + 4 + j * step:i * packet.resolution + 4 + j * step + 3], 'big', signed=True)
+                for j in range(packet.sample_num)
+            ]
+            for i in range(ch_num)
+        ]
+                
+        # logger.trace(packet)
+        
+        return packet
+
+    def copy(self) -> 'RscPacket':
+        packet = RscPacket()
+        packet.time_stamp = self.time_stamp
+        packet.pkg_id = self.pkg_id
+        packet.result = self.result
+        packet.channels = self.channels
+        packet.origin_sample_rate = self.origin_sample_rate
+        packet.sample_rate = self.sample_rate
+        packet.sample_num = self.sample_num
+        packet.resolution = self.resolution
+        packet.filter = self.filter
+        packet.data_len = self.data_len
+        packet.trigger = self.trigger
+        packet.eeg = self.eeg
+        return packet  
+        
+    def __str__(self):
+        return f"""[
+            "time_stamp": {self.time_stamp}
+            "pkg_id": {self.pkg_id}
+            "channels": {self.channels}
+            "trigger": {self.trigger}
+            "eeg": {self.eeg}
+        ]"""
+        
+class ImpedancePacket(Packet):
+    def __init__(self):
+        super().__init__()
+        self.impedance = None
+        
+    @staticmethod
+    def transfer(body:bytes) -> 'ImpedancePacket':
+        packet = ImpedancePacket()
+        packet.time_stamp = int.from_bytes(body[0:8], 'little')
+        # packet.result = body[8]
+        packet.pkg_id = int.from_bytes(body[9: 13], 'little')
+        packet.channels = to_channels(body[13: 45])
+        # packet.sample_rate = int.from_bytes(body[45: 49], 'little')
+        # packet.sample_len = int.from_bytes(body[49: 53], 'little')
+        # packet.resolution = int(int(body[53]) / 8)
+        # packet.filter = int(int(body[54]) / 8)
+        # packet.wave_type = int(int(body[55]) / 8)
+        # packet.wave_freq = int.from_bytes(body[56: 60], 'little')
+        # packet.data_len = int.from_bytes(body[60: 64], 'little')
+        b_impedance = body[64:]
+        packet.impedance = [int.from_bytes(b_impedance[j * 4 : j * 4 + 4], 'little', signed=False) for j in range(len(packet.channels))]
+
+        # logger.trace(f"impedance: {packet}")
+        
+        return packet
+
+    def __str__(self):
+        return f"""[
+            "time_stamp": {self.time_stamp}
+            "pkg_id": {self.pkg_id}
+            "channels": {self.channels}
+            "impedance": {self.impedance}
+        ]"""
+        
+    
+class C256RSPacket(Packet):
+    def __init__(self):
+        super().__init__()
+        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
+    
+    @staticmethod
+    def transfer(body: bytes) -> 'RscPacket':
+        packet = RscPacket()
+        packet.time_stamp = int.from_bytes(body[0:8], 'little')
+        packet.result = body[8]
+        packet.pkg_id = int.from_bytes(body[9: 13], 'little')
+        packet.channels = to_channels(body[13: 45])
+        logger.trace(f"pkg_id: {packet.pkg_id}, channels: {packet.channels}")
+            
+        packet.origin_sample_rate = int.from_bytes(body[45: 49], 'little')
+        packet.sample_rate = int.from_bytes(body[49: 53], 'little')
+        packet.sample_num = int.from_bytes(body[53: 57], 'little')
+        packet.resolution = int(int(body[57]) / 8)
+        packet.filter = body[58]
+        packet.data_len = int.from_bytes(body[59: 63], 'little')
+                
+        # 数据块
+        b_eeg = body[63:]
+        # 根据值域分割数组-代表设备的4个模块
+        ranges = [(1, 64), (65, 128), (129, 192), (193, 256)]
+        sub_channels = [[x for x in packet.channels if low <= x <= high] for low, high in ranges]
+        # 只处理选中的模块
+        sub_channels = [_ for _ in sub_channels if len(_) > 0]
+        # 步径 相同通道的点间隔
+        step = int(len(packet.channels) * packet.resolution + 4 * len(sub_channels))
+        offset = 0
+        
+        # 分按子模块处理
+        for channels in sub_channels:
+            logger.trace(f"子数组: {channels} 长度: {len(channels)}")
+            channel_size = len(channels) 
+            
+            # 模块没有选中通道的，跳过
+            if channel_size == 0:
+                continue
+            
+            # 只保留第一个有效模块的trigger，其他的模块是冗余信息，无实际含义
+            if packet.trigger is None:
+                packet.trigger = [int.from_bytes(b_eeg[j * step : j * step + 4], 'little', signed=False) for j in range(packet.sample_num)]
+                logger.trace(f"trigger: {packet.trigger}")
+                trigger_positions = [index for index, value in enumerate(packet.trigger) if value != 0]
+                if len(trigger_positions) > 0:
+                    logger.debug(f"Trigger触发点位置: {trigger_positions}, 触发点时间戳: {[packet.time_stamp + int(pos * 1000 / packet.sample_rate) for pos in trigger_positions]}")
+                
+            eeg = [
+                [
+                    int.from_bytes(b_eeg[offset + step * j + 4 +  k * packet.resolution : offset + step * j + 7 +  k * packet.resolution], 'big', signed=True)
+                    for j in range(packet.sample_num)
+                ]
+                for k in range(channel_size)
+            ]
+            packet.eeg = packet.eeg + eeg if packet.eeg else eeg
+            
+            offset += 4 + channel_size * packet.resolution 
+            
+        # logger.trace(packet)
+        return packet
+    
+  
+class C256ImpedancePacket(Packet):
+    def __init__(self):
+        super().__init__()
+        self.impedance = None
+        
+    @staticmethod
+    def transfer(body:bytes) -> 'ImpedancePacket':
+        packet = ImpedancePacket()
+        packet.time_stamp = int.from_bytes(body[0:8], 'little')
+        # packet.result = body[8]
+        packet.pkg_id = int.from_bytes(body[9: 13], 'little')
+        packet.channels = to_channels(body[13: 45])
+        # packet.sample_rate = int.from_bytes(body[45: 49], 'little')
+        # packet.sample_len = int.from_bytes(body[49: 53], 'little')
+        # packet.resolution = int(int(body[53]) / 8)
+        # packet.filter = int(int(body[54]) / 8)
+        # packet.wave_type = int(int(body[55]) / 8)
+        # packet.wave_freq = int.from_bytes(body[56: 60], 'little')
+        # packet.data_len = int.from_bytes(body[60: 64], 'little')
+        b_impedance = body[64:]
+        packet.impedance = [int.from_bytes(b_impedance[j : j + 4], 'little', signed=False) for j in range(len(packet.channels))]
+
+        # logger.trace(f"impedance: {packet}")
+
+    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}
+        """
+            

qldsdk_serial-0.1.0/src/qldsdk/core/filter/__init__.py

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

qldsdk_serial-0.1.0/src/qldsdk/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

qldsdk_serial-0.1.0/src/qldsdk/core/interface/__init__.py

@@ -0,0 +1,10 @@
+from .parser import IParser
+
+from .device import IDevice
+
+
+'''
+    设备消息解析器
+'''
+class DeviceMessageParser:
+    pass

qldsdk_serial-0.1.0/src/qldsdk/core/interface/analyzer.py

@@ -0,0 +1,2 @@
+class IAnalyzer:
+    pass

qldsdk_serial-0.1.0/src/qldsdk/core/interface/collector.py

@@ -0,0 +1,10 @@
+class ICollector:
+    def __init__(self):
+        pass
+
+    def collect_data(self):
+        pass
+    
+class Collector(ICollector):
+    def __init__(self):
+        super().__init__()

qldsdk_serial-0.1.0/src/qldsdk/core/interface/device.py

@@ -0,0 +1,2 @@
+class IDevice:
+    pass

qldsdk_serial-0.1.0/src/qldsdk/core/interface/parser.py

@@ -0,0 +1,13 @@
+class IParser:
+    pass
+
+class Parser(IParser):
+    def __init__(self):
+        super().__init__()
+        
+class MessageParser(Parser):
+    def __init__(self):
+        super().__init__()
+        
+    def parse(self, data):
+        pass

qldsdk_serial-0.1.0/src/qldsdk/core/interface/stimulator.py

@@ -0,0 +1,2 @@
+class IStimulator:
+    pass

qldsdk_serial-0.1.0/src/qldsdk/core/interface/store.py

@@ -0,0 +1,2 @@
+class IStore:
+    pass

qldsdk_serial-0.1.0/src/qldsdk/core/local.py

@@ -0,0 +1,37 @@
+import os
+import socket
+from time import time
+
+# 读取本机全部ip列表
+# return list
+def get_ips():
+    ips = socket.gethostbyname_ex(socket.gethostname())[-1]
+    # 排除环回地址
+    valid_ips = [ip for ip in ips if not ip.startswith('127.')]
+    return valid_ips
+
+# 读取本机的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()

qldsdk_serial-0.1.0/src/qldsdk/core/message/__init__.py

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