sensor-sdk 0.0.10__py3-none-any.whl → 0.0.13__py3-none-any.whl

sensor/gforce.py

@@ -103,6 +103,8 @@ class Command(IntEnum):
     CMD_GET_BLE_MTU_INFO = (0xAE,)
     CMD_GET_BRT_CONFIG = (0xB3,)
 
+    CMD_SET_FRIMWARE_FILTER_SWITCH = (0xAA,)
+    CMD_GET_FRIMWARE_FILTER_SWITCH = (0xA9,)
     # Partial command packet, format: [CMD_PARTIAL_DATA, packet number in reverse order, packet content]
     MD_PARTIAL_DATA = 0xFF
 
@@ -737,6 +739,15 @@ class GForce:
             )
         )
 
+    async def set_firmware_filter_switch(self, switchStatus: int):
+        body = [0xFF & switchStatus]
+        body = bytes(body)
+        await self._send_request(Request(cmd=Command.CMD_SET_FRIMWARE_FILTER_SWITCH, body=body, has_res=True))
+
+    async def get_firmware_filter_switch(self):
+        buf = await self._send_request(Request(cmd=Command.CMD_GET_FRIMWARE_FILTER_SWITCH, has_res=True))
+        return buf[0]
+
     async def set_emg_raw_data_config(self, cfg=EmgRawDataConfig()):
         body = cfg.to_bytes()
         await self._send_request(

sensor/sensor_data_context.py

@@ -1,10 +1,12 @@
 import asyncio
 from collections import deque
+import os
 import platform
 from queue import Queue
 import struct
 from typing import Deque, List
-
+from concurrent.futures import ThreadPoolExecutor
+import csv
 from sensor import utils
 from sensor.gforce import DataSubscription, GForce
 from sensor.sensor_data import DataType, Sample, SensorData
@@ -56,9 +58,16 @@ class SensorProfileDataCtx:
             self.sensorDatas.append(SensorData())
         self.impedanceData: List[float] = list()
         self.saturationData: List[float] = list()
+        self.dataPool = ThreadPoolExecutor(1, "data")
+        self.init_map = {"NTF_EMG": "ON", "NTF_EEG": "ON", "NTF_ECG": "ON", "NTF_IMU": "ON", "NTF_BRTH": "ON"}
+        self.filter_map = {"FILTER_50Hz": "ON", "FILTER_60Hz": "ON", "FILTER_HPF": "ON", "FILTER_LPF": "ON"}
+        self.debugCSVWriter = None
+        self.debugCSVPath = None
 
     def close(self):
         self._is_running = False
+        if self.debugCSVWriter != None:
+            self.debugCSVWriter = None
 
     def clear(self):
         for sensorData in self.sensorDatas:
@@ -225,22 +234,22 @@ class SensorProfileDataCtx:
             if self.hasImpedance():
                 self.notifyDataFlag |= DataSubscription.DNF_IMPEDANCE
 
-            if self.hasEEG():
+            if self.hasEEG() & (self.init_map["NTF_EEG"] == "ON"):
                 # print("initEEG")
                 info.EegChannelCount = await self.initEEG(packageCount)
                 info.EegSampleRate = self.sensorDatas[SensorDataType.DATA_TYPE_EEG].sampleRate
 
-            if self.hasECG():
+            if self.hasECG() & (self.init_map["NTF_ECG"] == "ON"):
                 # print("initECG")
                 info.EcgChannelCount = await self.initECG(packageCount)
                 info.EcgSampleRate = self.sensorDatas[SensorDataType.DATA_TYPE_ECG].sampleRate
 
-            if self.hasBrth():
+            if self.hasBrth() & (self.init_map["NTF_BRTH"] == "ON"):
                 # print("initBrth")
                 info.BrthChannelCount = await self.initBrth(packageCount)
                 info.BrthSampleRate = self.sensorDatas[SensorDataType.DATA_TYPE_BRTH].sampleRate
 
-            if self.hasIMU():
+            if self.hasIMU() & (self.init_map["NTF_IMU"] == "ON"):
                 # print("initIMU")
                 imuChannelCount = await self.initIMU(packageCount)
                 info.AccChannelCount = imuChannelCount
@@ -285,9 +294,47 @@ class SensorProfileDataCtx:
             await self.gForce.set_subscription(0)
             return True
 
+    async def setFilter(self, filter: str, value: str) -> str:
+        self.filter_map[filter] = value
+        switch = 0
+        for filter in self.filter_map.keys():
+            value = self.filter_map[filter]
+            if filter == "FILTER_50Hz":
+                if value == "ON":
+                    switch |= 1
+            elif filter == "FILTER_60Hz":
+                if value == "ON":
+                    switch |= 2
+            elif filter == "FILTER_HPF":
+                if value == "ON":
+                    switch |= 4
+            elif filter == "FILTER_LPF":
+                if value == "ON":
+                    switch |= 8
+        try:
+            await self.gForce.set_firmware_filter_switch(switch)
+            return "OK"
+        except Exception as e:
+            return "ERROR: " + str(e)
+
+    async def setDebugCSV(self, debugFilePath) -> str:
+        if self.debugCSVWriter != None:
+            self.debugCSVWriter = None
+        if debugFilePath != None:
+            self.debugCSVPath = debugFilePath
+            try:
+                if self.debugCSVPath != "":
+                    CSVWriter = csv.writer(open(self.debugCSVPath, "w", newline=""), delimiter=",")
+                    CSVWriter = None
+            except Exception as e:
+                return "ERROR: " + str(e)
+        return "OK"
+
+    ####################################################################################
+
     async def process_data(self, buf: Queue[SensorData], sensor, callback):
         while self._is_running and self._rawDataBuffer.empty():
-            await asyncio.sleep(0.01)
+            await asyncio.sleep(0.1)
 
             while self._is_running and not self._rawDataBuffer.empty():
                 try:
@@ -306,7 +353,7 @@ class SensorProfileDataCtx:
                         break
                     if sensorData != None and callback != None:
                         try:
-                            asyncio.get_event_loop().run_in_executor(None, callback, sensor, sensorData)
+                            asyncio.get_event_loop().run_in_executor(self.dataPool, callback, sensor, sensorData)
                         except Exception as e:
                             print(e)
 
@@ -525,6 +572,31 @@ class SensorProfileDataCtx:
             sensorDataResult.minPackageSampleCount = sensorData.minPackageSampleCount
             sensorDataList.append(sensorDataResult)
 
+            if self.debugCSVPath != None and self.debugCSVPath != "" and self.debugCSVWriter == None:
+                try:
+                    self.debugCSVWriter = csv.writer(open(self.debugCSVPath, "w", newline="", encoding="utf-8"))
+                    header_append_keys = ["dataType", "sampleRate"]
+                    channel_samples_header = list(vars(sensorDataResult.channelSamples[0][0]).keys())
+                    for key_item in header_append_keys:
+                        channel_samples_header.append(key_item)
+                    self.debugCSVWriter.writerow(channel_samples_header)
+                except Exception as e:
+                    print(e)
+
+            if self.debugCSVWriter != None:
+                try:
+                    for i, channel_sample_list in enumerate(sensorDataResult.channelSamples):
+                        for channel_sample in channel_sample_list:
+                            row_data = []
+
+                            for key in vars(channel_sample).keys():
+                                row_data.append(getattr(channel_sample, key))
+                            row_data.append(sensorDataResult.dataType)
+                            row_data.append(sensorDataResult.sampleRate)
+                            self.debugCSVWriter.writerow(row_data)
+                except Exception as e:
+                    print(e)
+
             startIndex += sensorData.minPackageSampleCount
 
         leftChannelSamples = []
@@ -545,7 +617,7 @@ class SensorProfileDataCtx:
 
         while self._is_running:
             while self._is_running and self._rawDataBuffer.empty():
-                await asyncio.sleep(0.01)
+                await asyncio.sleep(0.1)
                 continue
 
             try:
@@ -588,7 +660,7 @@ class SensorProfileDataCtx:
                                 break
                             if sensorData != None and callback != None:
                                 try:
-                                    asyncio.get_event_loop().run_in_executor(None, callback, sensor, sensorData)
+                                    asyncio.get_event_loop().run_in_executor(self.dataPool, callback, sensor, sensorData)
                                 except Exception as e:
                                     print(e)
 

sensor/sensor_profile.py

@@ -18,7 +18,7 @@ import asyncio
 
 from sensor.sensor_data_context import SensorProfileDataCtx
 from sensor.sensor_device import BLEDevice, DeviceInfo, DeviceStateEx
-from sensor.utils import async_call, sync_call, async_exec, timer
+from sensor.utils import async_call, sync_call, async_exec
 
 SERVICE_GUID = "0000ffd0-0000-1000-8000-00805f9b34fb"
 OYM_CMD_NOTIFY_CHAR_UUID = "f000ffe1-0451-4000-b000-000000000000"
@@ -289,7 +289,7 @@ class SensorProfile:
 
     async def _waitForDisconnect(self) -> bool:
         while self.deviceState != DeviceStateEx.Disconnected:
-            asyncio.sleep(1)
+            await asyncio.sleep(1)
         return True
 
     async def _disconnect(self) -> bool:
@@ -473,6 +473,23 @@ class SensorProfile:
             return self._data_ctx._device_info
         return None
 
+    async def _setParam(self, key: str, value: str) -> str:
+        if self.deviceState != DeviceStateEx.Ready:
+            return "Please connect first"
+
+        if key in ["NTF_EMG", "NTF_EEG", "NTF_ECG", "NTF_IMU", "NTF_BRTH"]:
+            if value in ["ON", "OFF"]:
+                self._data_ctx.init_map[key] = value
+                return "OK"
+
+        if key in ["FILTER_50Hz", "FILTER_60Hz", "FILTER_HPF", "FILTER_LPF"]:
+            if value in ["ON", "OFF"]:
+                return await self._data_ctx.setFilter(key, value)
+
+        if key == "DEBUG_BLE_DATA_PATH":
+            return await self._data_ctx.setDebugCSV(value)
+        return "Not supported"
+
     def setParam(self, key: str, value: str) -> str:
         """
         设置传感器的参数。
@@ -483,9 +500,12 @@ class SensorProfile:
         :return:            str: 设置参数的结果。
 
         """
-        return ""
+        return sync_call(
+            self._setParam(key, value),
+            20,
+        )
 
-    async def AsyncSetParam(self, key: str, value: str) -> str:
+    async def asyncSetParam(self, key: str, value: str) -> str:
         """
         设置传感器的参数。
 
@@ -495,4 +515,7 @@ class SensorProfile:
         :return:            str: 设置参数的结果。
 
         """
-        return ""
+        return await async_call(
+            self._setParam(key, value),
+            20,
+        )

sensor/utils.py

@@ -37,30 +37,6 @@ def Terminate():
             pass
 
 
-async def delay(_time: float, function) -> any:
-    try:
-        await asyncio.sleep(_time)
-    except Exception as e:
-        pass
-
-    try:
-        return await function
-    except Exception as e:
-        pass
-
-
-def timer(_delay: float, function):
-    checkRunLoop()
-    task: asyncio.Future = None
-    try:
-        task = asyncio.run_coroutine_threadsafe(delay(_delay, function), _runloop)
-        running_tasks.add(task)
-        task.add_done_callback(lambda t: running_tasks.remove(t))
-    except Exception as e:
-        print(e)
-        pass
-
-
 def async_exec(function):
     checkRunLoop()
     task: asyncio.Future = None

{sensor_sdk-0.0.10.dist-info → sensor_sdk-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sensor-sdk
-Version: 0.0.10
+Version: 0.0.13
 Summary: Python sdk for Synchroni
 Home-page: https://github.com/oymotion/SynchroniSDKPython
 Author: Martin Ye
@@ -350,8 +350,47 @@ batteryPower = sensorProfile.getBatteryLevel()
 
 Please check console.py in examples directory
 
-## Async methods
+### Async methods
 
 all methods start with async is async methods, they has same params and return result as sync methods.
 
 Please check async_console.py in examples directory
+
+### setParam method
+
+Use `def setParam(self, key: str, value: str) -> str` to set parameter of sensor profile. Please call after device in 'Ready' state.
+
+Below is available key and value:
+
+```python
+result = sensorProfile.setParam("NTF_EMG", "ON")
+# set EMG data to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("NTF_EEG", "ON")
+# set EEG data to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("NTF_ECG", "ON")
+# set ECG data to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("NTF_IMU", "ON")
+# set IMU data to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("NTF_BRTH", "ON")
+# set BRTH data to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("FILTER_50Hz", "ON")
+# set 50Hz notch filter to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("FILTER_60Hz", "ON")
+# set 60Hz notch filter to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("FILTER_HPF", "ON")
+# set 0.5Hz hpf filter to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("FILTER_LPF", "ON")
+# set 80Hz lpf filter to ON or OFF, result is "OK" if succeed
+
+result = sensorProfile.setParam("DEBUG_BLE_DATA_PATH", "d:/temp/test.csv")
+# set debug ble data path, result is "OK" if succeed
+# please give an absolute path and make sure it is valid and writeable by yourself
+```

sensor_sdk-0.0.13.dist-info/RECORD

@@ -0,0 +1,14 @@
+sensor/__init__.py,sha256=L1VyAP0EDEnJIMeMTzp4iXHSRUUHyHScF_GIl3iYKRI,123
+sensor/gforce.py,sha256=QsbHhTieVKMJ_KHn7nEchh7sChZWL-iSP0LmfxhYYXw,25753
+sensor/sensor_controller.py,sha256=qnM9mVwpJn9PCnxlvbgWFeaLktqVnmaquZgasYGEHqs,9451
+sensor/sensor_data.py,sha256=Hu7Ql0LgQ7V24xYZhaLrKPwU4KWZeWE655v8Gy8xphY,3934
+sensor/sensor_data_context.py,sha256=a45WJPE-8-CzzMn9BlCjjC39fiIxkjV-UQw2Rok-OyM,28536
+sensor/sensor_device.py,sha256=eO1vaqjxCc2UCPBoKXqlk6o498uRyWt6IYs7r7wXSD0,3042
+sensor/sensor_profile.py,sha256=qxjTYt7X45rmVcapI93ynB4dTFLLt0M-WKCp-cpqKrM,18167
+sensor/utils.py,sha256=BudTD083ucGxcTXfmksF-2gLRi_i-k7tMytruLUUKJc,6503
+sensor_sdk-0.0.13.dist-info/LICENSE.txt,sha256=8CSivOpub3IuXODTyqBRI91AxouJZk02YrcKuOAkWu8,1111
+sensor_sdk-0.0.13.dist-info/METADATA,sha256=6hKH69-rbvoswQ9LGvPVcGiY0v8qKKN9rfiVXjX0hLs,9825
+sensor_sdk-0.0.13.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+sensor_sdk-0.0.13.dist-info/top_level.txt,sha256=Ftq49B6bH0Ffdc7c8LkcyakHo6lsg_snlBbpEUoILSk,7
+sensor_sdk-0.0.13.dist-info/zip-safe,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
+sensor_sdk-0.0.13.dist-info/RECORD,,

sensor_sdk-0.0.10.dist-info/RECORD

@@ -1,14 +0,0 @@
-sensor/__init__.py,sha256=L1VyAP0EDEnJIMeMTzp4iXHSRUUHyHScF_GIl3iYKRI,123
-sensor/gforce.py,sha256=mTw0VVQSxPiTzZmjK7I9knTlTkEeu4PISQdlAkMU6WE,25232
-sensor/sensor_controller.py,sha256=qnM9mVwpJn9PCnxlvbgWFeaLktqVnmaquZgasYGEHqs,9451
-sensor/sensor_data.py,sha256=Hu7Ql0LgQ7V24xYZhaLrKPwU4KWZeWE655v8Gy8xphY,3934
-sensor/sensor_data_context.py,sha256=9IvXAHhCSQGnkNW5nFNz4Rs6kVq1hEyxxu6ROKjb-Dk,25033
-sensor/sensor_device.py,sha256=eO1vaqjxCc2UCPBoKXqlk6o498uRyWt6IYs7r7wXSD0,3042
-sensor/sensor_profile.py,sha256=64MFNyLJajX6HeGS_tmN9R9wfEW1qjsP_MjH3ZRPq4s,17322
-sensor/utils.py,sha256=FuBEChpJYbXnUaDvlta3aT6dYGAK2fQdGbIB7BYMQp4,7077
-sensor_sdk-0.0.10.dist-info/LICENSE.txt,sha256=8CSivOpub3IuXODTyqBRI91AxouJZk02YrcKuOAkWu8,1111
-sensor_sdk-0.0.10.dist-info/METADATA,sha256=OjhjiuAo0t9cgFCyuIC-5PB_O7KTi8duBZz-H2RCTAw,8375
-sensor_sdk-0.0.10.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-sensor_sdk-0.0.10.dist-info/top_level.txt,sha256=Ftq49B6bH0Ffdc7c8LkcyakHo6lsg_snlBbpEUoILSk,7
-sensor_sdk-0.0.10.dist-info/zip-safe,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
-sensor_sdk-0.0.10.dist-info/RECORD,,