psychopy-neuroplaypro-neurolab 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

psychopy_neuroplaypro_neurolab/__init__.py

@@ -1 +1,7 @@
-# Инициализация пакета — можно оставить пустым
+# -*- coding: utf-8 -*-
+"""Init file for psychopy-neuroplaypro-neurolab plugin"""
+
+__version__ = '0.1.1'
+
+from .neuroplay_record import NeuroplayRecordingComponent
+from .neuroplay import NeuroPlayProClient

psychopy_neuroplaypro_neurolab/neuroplay.py

@@ -0,0 +1,46 @@
+# neuroplay.py (в корне плагина)
+
+__all__ = ['NeuroPlayProClient']
+
+import websocket
+import threading
+import json
+import time
+
+
+class NeuroPlayProClient:
+    def __init__(self, url="ws://localhost:11234"):
+        self.url = url
+        self.ws = None
+        self.running = False
+        self.listen_thread = None
+        self.messages = []
+
+    def connect(self):
+        self.ws = websocket.create_connection(self.url)
+
+    def send_marker(self, label="TRIGGER"):
+        self.ws.send(json.dumps({"marker": label}))
+
+    def close(self):
+        if self.ws:
+            self.ws.close()
+
+    def listen(self):
+        self.running = True
+
+        def loop():
+            while self.running:
+                try:
+                    msg = self.ws.recv()
+                    self.messages.append(msg)
+                except:
+                    self.running = False
+
+        self.listen_thread = threading.Thread(target=loop)
+        self.listen_thread.start()
+
+    def stop(self):
+        self.running = False
+        if self.listen_thread:
+            self.listen_thread.join()

psychopy_neuroplaypro_neurolab/neuroplay_record/__init__.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+"""
+NeuroplayRecordingComponent
+Инициализация и управление записью NeuroPlayPro через WebSocket из PsychoPy Builder
+"""
+
+__all__ = ['NeuroplayRecordingComponent']
+
+from pathlib import Path
+from psychopy.experiment.components import BaseComponent, Param, getInitVals
+
+WS_OBJ = 'ws_neuro'
+
+class NeuroplayRecordingComponent(BaseComponent):
+    categories = ['EEG']
+    targets = ['PsychoPy']
+    iconFile = Path(__file__).parent / 'neuroplay_record.png'
+    tooltip = 'Start/Stop EEG recording via NeuroPlayPro'
+    plugin = "psychopy-neuroplaypro-neurolab"
+
+    def __init__(self, exp, parentName, name='neuroplay_rec'):
+        super().__init__(
+            exp, parentName, name=name,
+            startType='time (s)', startVal=0,
+            stopType='duration (s)', stopVal=1.0,
+            startEstim='', durationEstim='',
+            saveStartStop=False
+        )
+
+        self.type = 'NeuroplayRecording'
+        self.exp.requireImport(importName='websocket')
+        self.exp.requireImport(importName='json')
+
+        self.params['wsURL'] = Param("ws://localhost:1336", valType='str', inputType='editable',
+                                     hint="WebSocket адрес NeuroPlayPro")
+
+    def writeInitCode(self, buff):
+        inits = getInitVals(self.params, 'PsychoPy')
+        code = (f'{inits["name"]} = visual.BaseVisualStim(' +
+                'win=win, name="{}")\n'.format(inits['name']))
+        buff.writeIndentedLines(code)
+
+        ws_url = inits['wsURL'].val
+        code = f"{WS_OBJ} = websocket.create_connection(\"{ws_url}\")"
+        buff.writeIndentedLines(code)
+
+    def writeRoutineStartCode(self, buff):
+        inits = getInitVals(self.params, 'PsychoPy')
+        buff.writeIndentedLines("# === Start EEG Recording ===")
+        buff.writeIndentedLines("try:")
+        buff.setIndentLevel(1, relative=True)
+
+        buff.writeIndentedLines('start_record_cmd = json.dumps({"command": "startRecord"})')
+        buff.writeIndentedLines(f"{WS_OBJ}.send(start_record_cmd)")
+        buff.writeIndentedLines(f"response = {WS_OBJ}.recv()")
+        buff.writeIndentedLines("print(f'Response from NeuroPlay: {response}')")
+
+        buff.setIndentLevel(-1, relative=True)
+        buff.writeIndentedLines("except Exception as e:")
+        buff.setIndentLevel(1, relative=True)
+        buff.writeIndentedLines('print(f"Ошибка при старте записи NeuroPlay: {e}")')
+        buff.setIndentLevel(-1, relative=True)
+
+    def writeRoutineEndCode(self, buff):
+        buff.writeIndentedLines("# === Stop EEG Recording ===")
+        buff.writeIndentedLines('print("Stopping NeuroPlay...")')
+        buff.writeIndentedLines("try:")
+        buff.setIndentLevel(1, relative=True)
+        buff.writeIndentedLines('stop_record_cmd = json.dumps({"command": "stopRecord"})')
+        buff.writeIndentedLines(f"{WS_OBJ}.send(stop_record_cmd)")
+        buff.writeIndentedLines(f"response = {WS_OBJ}.recv()")
+        buff.writeIndentedLines("print(f'Response from NeuroPlay: {response}')")
+        buff.setIndentLevel(-1, relative=True)
+        buff.writeIndentedLines("except Exception as e:")
+        buff.setIndentLevel(1, relative=True)
+        buff.writeIndentedLines('print(f"Ошибка при остановке записи NeuroPlay: {e}")')
+        buff.setIndentLevel(-1, relative=True)
+
+    def writeExperimentEndCode(self, buff):
+        buff.writeIndentedLines("# === Close WebSocket connection ===")
+        buff.writeIndentedLines("try:")
+        buff.setIndentLevel(1, relative=True)
+        buff.writeIndentedLines(f"{WS_OBJ}.close()")
+        buff.setIndentLevel(-1, relative=True)
+        buff.writeIndentedLines("except Exception as e:")
+        buff.setIndentLevel(1, relative=True)
+        buff.writeIndentedLines("print(f'Ошибка при закрытии WebSocket: {e}')")
+        buff.setIndentLevel(-1, relative=True)
+
+    def writeFrameCode(self, buff):
+        pass
+
+    def writeFrameCodeJS(self, buff):
+        pass
+
+    def writeExperimentEndCodeJS(self, buff):
+        pass
+
+    def writeInitCodeJS(self, buff):
+        pass

psychopy_neuroplaypro_neurolab-0.1.2.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: psychopy-neuroplaypro-neurolab
+Version: 0.1.2
+Summary: PsychoPy plugin to support NeuroPlayPro EEG triggers.
+Author-email: Enicast <enicaster@gmail.com>
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: websocket-client
+Dynamic: license-file
+
+# PsychoPy NeuroPlay Plugin
+
+Component for controlling NeuroPlayPro EEG recording from PsychoPy Builder experiments.
+- Start recording when the component starts
+- Stop recording when the component ends
+- Choose folder and filename prefix
+- Auto-add timestamp if needed

psychopy_neuroplaypro_neurolab-0.1.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+psychopy_neuroplaypro_neurolab/__init__.py,sha256=y5JhTBeswmpSsJIh_qUlL96qqxsp8Pq4AEnQbte75t8,205
+psychopy_neuroplaypro_neurolab/neuroplay.py,sha256=nEAE8PtCwWNU_0Uf7ADCfIEE7iTWE_UlLhHhs9V5A7A,1093
+psychopy_neuroplaypro_neurolab/neuroplay_record/__init__.py,sha256=DWfe0Rrjx5JhjdKHy7KSAw3P57imfjlaFinC1MupRPA,4145
+psychopy_neuroplaypro_neurolab/neuroplay_record/classic/neuroplay_record.png,sha256=yNpZBfil-_BuELl8Qs873-ywIn8mqe7TLzGNuUcC8cQ,5247
+psychopy_neuroplaypro_neurolab/neuroplay_record/classic/neuroplay_record@2x.png,sha256=SllL2cBmRscQ3GG2d-oIhf3tcJNloLIJ2br5hUk32QE,8826
+psychopy_neuroplaypro_neurolab/neuroplay_record/dark/neuroplay_record.png,sha256=yNpZBfil-_BuELl8Qs873-ywIn8mqe7TLzGNuUcC8cQ,5247
+psychopy_neuroplaypro_neurolab/neuroplay_record/dark/neuroplay_record@2x.png,sha256=SllL2cBmRscQ3GG2d-oIhf3tcJNloLIJ2br5hUk32QE,8826
+psychopy_neuroplaypro_neurolab/neuroplay_record/light/neuroplay_record.png,sha256=yNpZBfil-_BuELl8Qs873-ywIn8mqe7TLzGNuUcC8cQ,5247
+psychopy_neuroplaypro_neurolab/neuroplay_record/light/neuroplay_record@2x.png,sha256=SllL2cBmRscQ3GG2d-oIhf3tcJNloLIJ2br5hUk32QE,8826
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/METADATA,sha256=V5uLSin4wi_eKyPclz3_3ctvrKzlfT7PQ8-eqKLlj3c,590
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/entry_points.txt,sha256=yGls5zvDZfmLCDLrSkNMIZgCINfFcqyAioWrlAEuof4,122
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/top_level.txt,sha256=mbHktuA5YpOE3G2X-JWuUqqCKZtXNJ5CY4aPqRxJMRI,31
+psychopy_neuroplaypro_neurolab-0.1.2.dist-info/RECORD,,

{psychopy_neuroplaypro_neurolab-0.1.0.dist-info → psychopy_neuroplaypro_neurolab-0.1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

psychopy_neuroplaypro_neurolab-0.1.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[psychopy.experiment.components]
+NeuroplayRecordingComponent = psychopy_neuroplaypro_neurolab:NeuroplayRecordingComponent

psychopy_neuroplaypro_neurolab/plugin.py

@@ -1,8 +0,0 @@
-from psychopy.plugins import Plugin
-from psychopy.experiment.components import registerComponent
-from psychopy_neuroplaypro_neurolab.components.neuroplay.neuroplay import NeuroPlayComponent
-
-class NeuroPlayPlugin(Plugin):
-    def onLoad(self):
-        registerComponent("neuroplay", NeuroPlayComponent)
-        print("NeuroPlay plugin loaded")

psychopy_neuroplaypro_neurolab-0.1.0.dist-info/METADATA

@@ -1,9 +0,0 @@
-Metadata-Version: 2.4
-Name: psychopy-neuroplaypro-neurolab
-Version: 0.1.0
-Summary: Plugin for integrating NeuroPlayPro EEG recording in PsychoPy Builder.
-Author-email: Enicast <enicaster@gmail.com>
-Requires-Python: >=3.8
-License-File: LICENSE
-Requires-Dist: psychopy
-Dynamic: license-file

psychopy_neuroplaypro_neurolab-0.1.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-psychopy_neuroplaypro_neurolab/__init__.py,sha256=fd3Mi-jIseuqbdbmJqsB86t7nxhkg2fq0l3P4OiboA0,87
-psychopy_neuroplaypro_neurolab/plugin.py,sha256=eAhiWFukVgq1XRN3f0NGR3M74yesw0YakTnw351KqO4,344
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/METADATA,sha256=AhiUEjJpbLdNbg1bw-snsQAcOSXLIRHgcmwjlsxQIro,289
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/entry_points.txt,sha256=7YPHFYA-HZjQ-gizsCcvi704WKx3GrvbojxmQmpq7uo,212
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/top_level.txt,sha256=mbHktuA5YpOE3G2X-JWuUqqCKZtXNJ5CY4aPqRxJMRI,31
-psychopy_neuroplaypro_neurolab-0.1.0.dist-info/RECORD,,

psychopy_neuroplaypro_neurolab-0.1.0.dist-info/entry_points.txt

@@ -1,5 +0,0 @@
-[psychopy.experiment.components]
-neuroplay = psychopy_neuroplaypro_neurolab.components.neuroplay.neuroplay:NeuroPlayComponent
-
-[psychopy.plugins]
-neuroplay = psychopy_neuroplaypro_neurolab.plugin:NeuroPlayPlugin