From 8c18dd38521d767c6494fef0d725a6157336e24c Mon Sep 17 00:00:00 2001 From: Jeff Stout Date: Fri, 1 Nov 2024 17:51:26 -0400 Subject: [PATCH] Code Cleanup --- nih2mne/GUI/qt_mri_viewer.py | 121 ++++++----------------------------- 1 file changed, 21 insertions(+), 100 deletions(-) mode change 100644 => 100755 nih2mne/GUI/qt_mri_viewer.py diff --git a/nih2mne/GUI/qt_mri_viewer.py b/nih2mne/GUI/qt_mri_viewer.py old mode 100644 new mode 100755 index 0f51768..f8d4421 --- a/nih2mne/GUI/qt_mri_viewer.py +++ b/nih2mne/GUI/qt_mri_viewer.py @@ -8,20 +8,15 @@ import matplotlib.pyplot as plt import numpy as np -from matplotlib.animation import FuncAnimation - import os, os.path as op import nibabel as nib import glob, sys +import pyqtgraph as pg -from PyQt5 import QtWidgets +from PyQt5 import QtWidgets, Qt from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QGridLayout, \ QHBoxLayout, QVBoxLayout, QPushButton, QLabel, QComboBox, QLineEdit from PyQt5.QtWidgets import QSlider - - -import pyqtgraph as pg -from PyQt5.QtGui import QImage, QPixmap, QPalette, QPainter from PyQt5.QtCore import Qt @@ -81,41 +76,45 @@ def __init__(self, img_dict, gridsize_row=4, gridsize_col=6, cut_axis=2): self.ind = self.slices // 2 main_layout = QVBoxLayout() + subject_slider_layout = QHBoxLayout() self.image_grid_layout = self.init_subject_image_layout() - main_layout.addLayout(self.image_grid_layout) - self.b_slice_adjuster = QSlider(Qt.Horizontal) + subject_slider_layout.addLayout(self.image_grid_layout) + self.b_slice_adjuster = QSlider(Qt.Vertical) self.b_slice_adjuster.valueChanged.connect(self.slice_change) self.b_slice_adjuster.setMinimum(0) self.b_slice_adjuster.setMaximum(self.data_dict[self.subject_keys[0]].shape[cut_axis]-1) self.b_slice_adjuster.setValue(self.ind) - main_layout.addWidget(self.b_slice_adjuster) + subject_slider_layout.addWidget(self.b_slice_adjuster) + main_layout.addLayout(subject_slider_layout) + + #Set up final display widget = QWidget() widget.setLayout(main_layout) self.setCentralWidget(widget) def slice_change(self): self.ind = self.b_slice_adjuster.value() - print(self.ind) for i in self.subject_keys[0:self.gridsize_row*self.gridsize_col]: img_new = self.data_dict[i][:,:, self.ind] self.data_setters[i].setImage(img_new, autoRange=False) - - def init_subject_image_layout(self): - self.imgs_layout = QGridLayout() + '''Assemble the Volumetric viewer grid''' + self.imgs_layout = QGridLayout() tile_idxs = np.arange(self.gridsize_row * self.gridsize_col) tile_idxs_grid = tile_idxs.reshape(self.gridsize_row, self.gridsize_col) row_idxs, col_idxs = np.unravel_index(tile_idxs, [self.gridsize_row, self.gridsize_col]) i=0 - self.img_views = {} - self.data_setters = {} + self.img_views = {} #Direct image item for manipulation + self.data_setters = {} #Widget for adding to grid + + #Loop through images and assign GUI outputs for row_idx, col_idx in zip(row_idxs, col_idxs): if i+1 > len(self.subject_keys): self.imgs_layout.addWidget(QLabel(''), row_idx, col_idx) else: - img2 = np.copy(self.data_array[i][:,:,self.ind]) #bids_info = self.bids_project.subjects[self.subject_keys[i]] + img2 = np.copy(self.data_array[i][:,:,self.ind]) key = self.subject_keys[i] imageWidget = pg.GraphicsLayoutWidget() @@ -127,100 +126,22 @@ def init_subject_image_layout(self): img.setColorMap(cm) self.img_views[key] = imageWidget self.data_setters[key] = img - - # h,w = img.shape - # # qimage = QImage(img.copy().data , w, h, 3 * w, QImage.Format_RGB888) - # qimage = QImage(img.copy().data , w, h, img.strides[0], QImage.Format_Grayscale8) - # pixmap = QPixmap(qimage) - # pixmap = pixmap.scaled(400, 300, Qt.KeepAspectRatio) - # self.img_views[key]=QLabel() - # self.img_views[key].setPixmap(pixmap) - self.imgs_layout.addWidget(self.img_views[key], row_idx, col_idx) - # self.subjs_layout.addWidget(Subject_Tile(bids_info), row_idx, col_idx) i+=1 return self.imgs_layout - - # for row_idx, col_idx in zip(row_idxs,col_idxs): - # tmp = self.axes[row_idx, col_idx] - # tmp.xaxis.set_ticks([]) - # tmp.yaxis.set_ticks([]) - # tmp.set_title(self.key_list[i]) #FIX !!!!!! -- will not update on second panel - # tmp.draw() - # plt.tight_layout() - # i+=1 - # self.update() - # self.ani = FuncAnimation(plt.gcf(), self.update, frames=10, interval=None) - - def onscroll(self, event): - print("%s %s" % (event.button, event.step)) - if event.button == 'up': - self.ind = (self.ind + 1) % self.slices - else: - self.ind = (self.ind - 1) % self.slices - self.update() - def update(self): - row_idxs, col_idxs = np.unravel_index(range(self.axes.shape[0]*self.axes.shape[1]), self.axes.shape) - i=0 - for row_idx, col_idx in zip(row_idxs, col_idxs):# in range(self.axes.shape[0]*self.axes.shape[1]): #row_idx, col_idx, in zip (row_idxs, col_idxs): - key = self.subject_keys[i] - # self.img_views[key]. - self.im[i].set_data(self.data_array[i][:, :, self.ind]) - self.im[i].axes.figure.canvas.draw() - i+=1 - # return self.im # tmp = image_plot_grid(img_dict) -test = img_dict[list(img_dict.keys())[0]][:,:,20] -window(img_dict) -#%% -win = pg.GraphicsLayoutWidget(show=True, title="Basic plotting examples") -win.resize(1000,600) -win.setWindowTitle('pyqtgraph example: Plotting') +# test = img_dict[list(img_dict.keys())[0]][:,:,20] +# window(img_dict) -# Enable antialiasing for prettier plots -pg.setConfigOptions(antialias=True) - -p1 = win.addPlot(title="Basic array plotting", y=np.random.normal(size=100)) - - - -#%% - - -# def plot3d(image): -# fig, axes = plt.subplots(3,4) -# plt.subplots_adjust(0,0,1,1) -# plt.tick_params(left = False, right = False , labelleft = False , -# labelbottom = False, bottom = False) -# tracker = IndexTracker(axes, image) - -# fig.canvas.mpl_connect('scroll_event', tracker.onscroll) -# #, interval=10, blit=True) -# plt.show() - - +# Enable antialiasing for prettier plots +# pg.setConfigOptions(antialias=True) if __name__ == "__main__": - window(imgs) - - - -#%% -# from matplotlib.animation import FuncAnimation - -# im = plt.imshow(np.random.randn(10,10)) - -# def update(i): -# A = np.random.randn(10,10) -# im.set_array(A) -# return im, text - -# ani = FuncAnimation(plt.gcf(), update, frames=range(100), interval=50, blit=False) + window(img_dict) -# plt.show()