Skip to content

Commit

Permalink
Code Cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
@jstout211
jstout211 committed Nov 1, 2024
1 parent 06b9c13 commit 8c18dd3
Showing 1 changed file with 21 additions and 100 deletions.
121 changes: 21 additions & 100 deletions nih2mne/GUI/qt_mri_viewer.py
100644 → 100755
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down Expand Up @@ -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()
Expand All @@ -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()

0 comments on commit 8c18dd3

Please sign in to comment.