awendt/pyrelacs
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[repos/calibration] fixing plots

Browse Source
This commit is contained in:
wendtalexander 2024-09-30 10:10:12 +02:00
parent 9c80091d16
commit e0491c5917
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyrelacs/ui/mainwindow.py
View File

@ -36,9 +36,15 @@ class PyRelacs(QMainWindow):
Qt.ToolButtonStyle.ToolButtonTextBesideIcon Qt.ToolButtonStyle.ToolButtonTextBesideIcon
) # Ensure icons are displayed with text ) # Ensure icons are displayed with text
self.setWindowTitle("PyRelacs") self.setWindowTitle("PyRelacs")
self.beat_plot = pg.PlotWidget() self.beat_plot = pg.PlotWidget()
self.power_plot = pg.PlotWidget() self.power_plot = pg.PlotWidget()
self.beat_plot.addLegend()
self.power_plot.addLegend()
self.beat_plot.setBackground("w")
self.power_plot.setBackground("w")
self.threadpool = QThreadPool() self.threadpool = QThreadPool()
self.repros = Repro() self.repros = Repro()
@ -60,8 +66,8 @@ class PyRelacs(QMainWindow):
widget.setLayout(layout) widget.setLayout(layout)
self.setCentralWidget(widget) self.setCentralWidget(widget)
filename = path.joinpath(path.cwd(), "data") filename = path.joinpath(path.cwd(), "data.nix")
self.nix_file = nix.File.open(str(filename), nix.FileMode.Overwrite) self.nix_file = nix.File.open(str(filename), nix.FileMode.ReadWrite)
def create_actions(self): def create_actions(self):
self._rlx_exitaction = QAction( self._rlx_exitaction = QAction(
@ -216,15 +222,15 @@ class PyRelacs(QMainWindow):
f_sq, powerspec_sq = welch(beat_squared, fs=40_000.0) f_sq, powerspec_sq = welch(beat_squared, fs=40_000.0)
powerspec_sq = decibel(powerspec_sq) powerspec_sq = decibel(powerspec_sq)
peaks = find_peaks(powerspec_sq, prominence=20)[0] peaks = find_peaks(powerspec_sq, prominence=10)[0]
pen = pg.mkPen(colors[i]) pen = pg.mkPen(colors[i])
self.beat_plot.plot( self.beat_plot.plot(
np.arange(0, len(beat)) / 40_000.0, np.arange(0, len(beat)) / 40_000.0,
beat_squared, beat_squared,
pen=pen, pen=pen,
# name=stim.name, name=stim.name,
) )
self.power_plot.plot(f_sq, powerspec_sq, pen=pen) self.power_plot.plot(f_sq, powerspec_sq, pen=pen, name=stim.name)
self.power_plot.plot(f[peaks], powerspec_sq[peaks], pen=None, symbol="x") self.power_plot.plot(f[peaks], powerspec_sq[peaks], pen=None, symbol="x")
def connect_dac(self): def connect_dac(self):