jgrewe/minipyrelacs
2
0
Fork 0
forked from awendt/pyrelacs
Code Pull Requests Activity

[dataio/test] adding a sin producer for testing

Browse Source
This commit is contained in:
wendtalexander
2024-10-07 13:51:46 +02:00
parent 52a0821601
commit aaa42db2ae
3 changed files with 92 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
pyrelacs/ui/mainwindow.py
View File

@@ -24,6 +24,7 @@ import numpy as np
from pyrelacs.dataio.daq_producer import DaqProducer
from pyrelacs.dataio.sin_producer import SinProducer
from pyrelacs.worker import Worker
from pyrelacs.repros.repros import Repro
from pyrelacs.dataio.circbuffer import CircBuffer
@@ -54,7 +55,7 @@ class PyRelacs(QMainWindow):
if filename.exists():
self.nix_file = nix.File.open(str(filename), nix.FileMode.ReadOnly)
filename = path.joinpath(path.cwd(), "calibration.nix")
self.nix_calibration = nix.File.open(str(filename), nix.FileMode.Overwrite)
self.nix_file = nix.File.open(str(filename), nix.FileMode.Overwrite)
self.calibration_plot = CalibrationPlot(self.figure, self.nix_file)
@@ -77,10 +78,12 @@ class PyRelacs(QMainWindow):
widget.setLayout(layout)
self.setCentralWidget(widget)
self.buffer = CircBuffer(size=1_000_000, samplerate=40_000)
self.connect_dac()
SAMPLERATE = 10000
self.buffer = CircBuffer(size=1_000_000, samplerate=SAMPLERATE)
# self.connect_dac()
self.daq_producer = DaqProducer(self.buffer, self.daq_device, [1, 1])
# self.daq_producer = DaqProducer(self.buffer, self.daq_device, [1, 1])
self.sin_producer = SinProducer(self.buffer)
self.continously_plot = Continously(self.figure, self.buffer)
def create_actions(self):
@@ -118,7 +121,10 @@ class PyRelacs(QMainWindow):
self._run_action = QAction(QIcon(":/icons/record.png"), "Run", self)
self._run_action.triggered.connect(self.run_daq)
self._stop_recording = QAction(QIcon(":/icons/record.png"), "Stop", self)
self._run_sinus_action = QAction(QIcon(":/icons/record.png"), "Sinus", self)
self._run_sinus_action.triggered.connect(self.run_sinus)
self._stop_recording = QAction("Stop", self)
self._stop_recording.triggered.connect(self.stop_recording)
self.create_menu()
@@ -161,6 +167,7 @@ class PyRelacs(QMainWindow):
daq_toolbar.addAction(self._daq_disconnectaction)
daq_toolbar.addAction(self._daq_calibaction)
daq_toolbar.addAction(self._run_action)
daq_toolbar.addAction(self._run_sinus_action)
daq_toolbar.addAction(self._stop_recording)
self.addToolBar(Qt.ToolBarArea.TopToolBarArea, daq_toolbar)
@@ -196,6 +203,16 @@ class PyRelacs(QMainWindow):
self.continously_plot.plot_daq()
# self.threadpool.start(plot_daq)
def run_sinus(self):
sinus_pro = Worker(self.sin_producer.produce_sin)
sinus_pro.signals.result.connect(self.print_output)
sinus_pro.signals.finished.connect(self.thread_complete)
sinus_pro.signals.progress.connect(self.progress_fn)
self.threadpool.start(sinus_pro)
time.sleep(0.05)
self.continously_plot.plot_daq()
def stop_recording(self):
self.add_to_textfield("pressed")
self._stop_recording.setEnabled(False)

26
pyrelacs/ui/plots/continously.py
View File

@@ -1,4 +1,6 @@
import pyqtgraph as pg
from IPython import embed
import numpy as np
from PyQt6.QtCore import QTimer
from pyrelacs.dataio.circbuffer import CircBuffer
@@ -15,10 +17,15 @@ class Continously:
def plot_daq(self, *args, **kwargs):
self.figure.setBackground("w")
self.daq_plot = self.figure.addPlot(row=0, col=0)
self.time = list(range(10))
pen = pg.mkPen("red")
self.data = list(range(10))
self.line = self.daq_plot.plot(self.time, self.data, pen=pen)
self.time = np.arange(self.buffer.size) / self.buffer.samplerate
self.data = np.ones(self.buffer.size)
log.debug(self.data.size)
log.debug(self.time.size)
self.line = self.daq_plot.plot(
self.time, self.data, pen=pen, setCliptoView=True
)
# self.line.setXRrange(np.arange(0, 10, 1 / self.buffer.samplerate))
self.item = 0
self.timer = QTimer()
@@ -29,12 +36,13 @@ class Continously:
def update_plot(self):
if self.buffer.totalcount() > 100:
if self.buffer.write_index() > self.item:
self.time = self.time[1:]
self.time.append(self.time[-1] + 1)
self.data = self.data[1:]
item = self.buffer.get(self.item)
self.data.append(item)
self.line.setData(self.time, self.data)
# self.time = self.time[1:]
# self.time.append(self.time[-1] + 1)
# self.data = self.data[1:]
item = self.buffer.get_all()
t = np.arange(item.shape[0]) / self.buffer.samplerate
# self.data.append(item)
self.line.setData(t, item)
self.item += 1
else:
pass