qlsdk2 0.3.0a2__py3-none-any.whl → 0.4.0__py3-none-any.whl

qlsdk/ar4m/persist.py

@@ -1,178 +0,0 @@
-from datetime import datetime
-from multiprocessing import Lock, Queue
-from pyedflib import FILETYPE_BDFPLUS, FILETYPE_EDFPLUS, EdfWriter
-from threading import Thread
-from loguru import logger
-import numpy as np
-import os
-
-class EdfHandler(object):
-    def __init__(self, sample_frequency, physical_max, physical_min, digital_max, digital_min, resolution=16, storage_path = None):
-        self.physical_max = physical_max
-        self.physical_min = physical_min
-        self.digital_max = digital_max
-        self.digital_min = digital_min
-        self.channels = None
-        self._cache = Queue()
-        self.resolution = resolution
-        self.sample_frequency = sample_frequency
-        # bytes per second
-        self.bytes_per_second = 0
-        self._edf_writer = None
-        self._cache2 = tuple()
-        self._recording = False
-        self._edf_writer = None
-        self.annotations = None
-        # 每个数据块大小
-        self._chunk = np.array([])
-        self._Lock = Lock()
-        self._duration = 0
-        self._points = 0
-        self._first_pkg_id = None
-        self._last_pkg_id = None
-        self._first_timestamp = None
-        self._end_time = None
-        self._patient_code = "patient_code"
-        self._patient_name = "patient_name"
-        self._device_type = None
-        self._total_packets = 0
-        self._lost_packets = 0
-        self._storage_path = storage_path
-        
-    @property
-    def file_name(self): 
-        if self._storage_path:
-            try:
-                os.makedirs(self._storage_path, exist_ok=True)  # 自动创建目录，存在则忽略
-                return f"{self._storage_path}/{self._device_type}_{self._first_timestamp}.edf"
-            except Exception as e:
-                logger.error(f"创建目录[{self._storage_path}]失败: {e}")
-            
-        return f"{self._device_type}_{self._first_timestamp}.edf"
-         
-    @property
-    def file_type(self):
-        return FILETYPE_BDFPLUS if self.resolution == 24 else FILETYPE_EDFPLUS
-    
-    def set_device_type(self, device_type):
-        self._device_type = device_type
-        
-    def set_storage_path(self, storage_path):
-        self._storage_path = storage_path
-        
-    def set_patient_code(self, patient_code):
-        self._patient_code = patient_code
-        
-    def set_patient_name(self, patient_name):
-        self._patient_name = patient_name
-    
-    def append(self, data):
-        if data:
-            # 通道数
-            if self._first_pkg_id is None:
-                self.channels = data.eeg_ch_count
-                self._first_pkg_id = data.pkg_id
-                self._first_timestamp = data.time_stamp
-                
-            if self._last_pkg_id and self._last_pkg_id != data.pkg_id - 1:  
-                self._lost_packets += data.pkg_id - self._last_pkg_id - 1
-                logger.warning(f"数据包丢失: {self._last_pkg_id} -> {data.pkg_id}, 丢包数: {data.pkg_id - self._last_pkg_id - 1}")
-                
-            self._last_pkg_id = data.pkg_id
-            self._total_packets += 1
-            
-        # 数据
-        self._cache.put(data)
-        if not self._recording:
-            self.start()
-    
-    def trigger(self, data):
-        pass        
-    
-    def start(self):
-        self._recording = True
-        record_thread = Thread(target=self._consumer)
-        record_thread.start()
-        
-    def _consumer(self):
-        logger.debug(f"开始消费数据 _consumer: {self._cache.qsize()}")
-        while True:
-            if self._recording or (not self._cache.empty()):
-                data = self._cache.get()
-                if data is None:
-                    break
-                # 处理数据
-                self._points += len(data.eeg[0])
-                self._write_file(data.eeg)
-            else:
-                break
-            
-        self.close()
-        
-    def _write_file(self, data):
-        try:
-            if self._edf_writer is None:
-                self.initialize_edf()
-            
-            if (self._chunk.size == 0):
-                self._chunk = np.asarray(data)
-            else:                
-                self._chunk = np.hstack((self._chunk, data))
-                
-            if self._chunk.size >= self.sample_frequency * self.channels:                
-                self._write_chunk(self._chunk[:self.sample_frequency])
-                self._chunk = self._chunk[self.sample_frequency:]            
-            
-        except Exception as e:
-            logger.error(f"写入数据异常: {str(e)}")
-        
-    def close(self):
-        self._recording = False
-        if self._edf_writer:            
-            self._end_time = datetime.now().timestamp()
-            self._edf_writer.writeAnnotation(0, 1, "start recording ")
-            self._edf_writer.writeAnnotation(self._duration, 1, "recording end")
-            self._edf_writer.close()
-        
-        logger.info(f"文件: {self.file_name}完成记录, 总点数: {self._points}, 总时长: {self._duration}秒 丢包数: {self._lost_packets}/{self._total_packets + self._lost_packets}")
-        
-    
-        
-    def initialize_edf(self):
-        # 创建EDF+写入器
-        self._edf_writer = EdfWriter(
-            self.file_name,
-            self.channels,
-            file_type=self.file_type
-        )
-        
-        # 设置头信息
-        self._edf_writer.setPatientCode(self._patient_code)
-        self._edf_writer.setPatientName(self._patient_name)
-        self._edf_writer.setEquipment(self._device_type)
-        self._edf_writer.setStartdatetime(datetime.now())
-        
-        # 配置通道参数
-        signal_headers = []
-        for ch in range(self.channels):
-            signal_headers.append({
-                "label": f'channels {ch + 1}',
-                "dimension": 'uV',
-                "sample_frequency": self.sample_frequency,
-                "physical_min": self.physical_min,
-                "physical_max": self.physical_max,
-                "digital_min": self.digital_min,
-                "digital_max": self.digital_max
-            })
-        
-        self._edf_writer.setSignalHeaders(signal_headers)
-    
-    def _write_chunk(self, chunk):
-        logger.debug(f"写入数据: {chunk}")
-        # 转换数据类型为float64（pyedflib要求）
-        data_float64 = chunk.astype(np.float64)
-        # 写入时转置为(样本数, 通道数)格式
-        self._edf_writer.writeSamples(data_float64)
-        self._duration += 1
-
-    

qlsdk2-0.3.0a2.dist-info/RECORD

@@ -1,18 +0,0 @@
-qlsdk/__init__.py,sha256=w0UCnZJekAq4orDahQXRL8s5JJGrWMz0De5LVIf4X_c,426
-qlsdk/ar4/__init__.py,sha256=iR0bFanDwqSxSEM1nCcge1l-GSIWjwnMeRQa47wGrds,4752
-qlsdk/ar4m/__init__.py,sha256=Z13X53WKmKp6MW_fA1FUPdQgJxhxxuyzemJWPS_CEHA,639
-qlsdk/ar4m/ar4sdk.py,sha256=OtmlvbO7XAgk5FNPFYIv_rXGj14maaWtgi2lQ7zQkvY,41418
-qlsdk/ar4m/persist.py,sha256=YrISKVfY-r1m2GZOw-JLuqTbEajA5wYa31MejGlz14I,6696
-qlsdk/ar4m/libs/libAr4SDK.dll,sha256=kZp9_DRwPdAJ5OgTFQSqS8tEETxUs7YmmETuBP2g60U,15402132
-qlsdk/ar4m/libs/libwinpthread-1.dll,sha256=W77ySaDQDi0yxpnQu-ifcU6-uHKzmQpcvsyx2J9j5eg,52224
-qlsdk/persist/__init__.py,sha256=UYaNL_RSczIZL_YrBWAifoAnNxpgZHEeSp-goHqyqv4,27
-qlsdk/persist/edf.py,sha256=usgFjZJVcfAUwSy6cJtRdgjdqMgOS4viLCDGcwAGMZE,7106
-qlsdk/sdk/__init__.py,sha256=v9LKP-5qXCqnAsCkiRE9LDb5Tagvl_Qd_fqrw7y9yd4,68
-qlsdk/sdk/ar4sdk.py,sha256=ZILuUVgUZN7-LNQm4xdItxmZWGwvFt4oWUKGez1BCLI,27101
-qlsdk/sdk/hub.py,sha256=uEOGZBZtMDCWlV8G2TZe6FAo6eTPcwHAW8zdqr1eq_0,1571
-qlsdk/x8/__init__.py,sha256=XlQn3Af_HA7bGrxhv3RZKWMob_SnmissXiso7OH1UgI,4750
-qlsdk/x8m/__init__.py,sha256=cLeUqEEj65qXw4Qa4REyxoLh6T24anSqPaKe9_lR340,634
-qlsdk2-0.3.0a2.dist-info/METADATA,sha256=o6ms9sTAOlaWyWyGuLc0IWYvFYu1oXXVDhqNMixvpqI,7066
-qlsdk2-0.3.0a2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-qlsdk2-0.3.0a2.dist-info/top_level.txt,sha256=2CHzn0SY-NIBVyBl07Suh-Eo8oBAQfyNPtqQ_aDatBg,6
-qlsdk2-0.3.0a2.dist-info/RECORD,,