import numpy as np
from PyQt5 import QtGui, QtCore
def make_selection(parent):
""" buttons to draw a square on view """
parent.topbtns = QtGui.QButtonGroup()
ql = QtGui.QLabel("select cells")
ql.setStyleSheet("color: white;")
ql.setFont(QtGui.QFont("Arial", 8, QtGui.QFont.Bold))
parent.l0.addWidget(ql, 0, 2, 1, 2)
pos = [2, 3, 4]
for b in range(3):
btn = TopButton(b, parent)
btn.setFont(QtGui.QFont("Arial", 8))
parent.topbtns.addButton(btn, b)
parent.l0.addWidget(btn, 0, (pos[b]) * 2, 1, 2)
btn.setEnabled(False)
parent.topbtns.setExclusive(True)
parent.isROI = False
parent.ROIplot = 0
ql = QtGui.QLabel("n=")
ql.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
ql.setStyleSheet("color: white;")
ql.setFont(QtGui.QFont("Arial", 8, QtGui.QFont.Bold))
parent.l0.addWidget(ql, 0, 10, 1, 1)
parent.topedit = QtGui.QLineEdit(parent)
parent.topedit.setValidator(QtGui.QIntValidator(0, 500))
parent.topedit.setText("40")
parent.ntop = 40
parent.topedit.setFixedWidth(35)
parent.topedit.setAlignment(QtCore.Qt.AlignRight)
parent.topedit.returnPressed.connect(parent.top_number_chosen)
parent.l0.addWidget(parent.topedit, 0, 11, 1, 1)
# minimize view
def make_cellnotcell(parent):
""" buttons for cell / not cell views at top """
# number of ROIs in each image
parent.lcell0 = QtGui.QLabel("")
parent.lcell0.setStyleSheet("color: white;")
parent.lcell0.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
parent.l0.addWidget(parent.lcell0, 0, 12, 1, 2)
parent.lcell1 = QtGui.QLabel("")
parent.lcell1.setStyleSheet("color: white;")
parent.l0.addWidget(parent.lcell1, 0, 20, 1, 2)
parent.sizebtns = QtGui.QButtonGroup(parent)
b = 0
labels = [" cells", " both", " not cells"]
for l in labels:
btn = SizeButton(b, l, parent)
parent.sizebtns.addButton(btn, b)
parent.l0.addWidget(btn, 0, 14 + 2 * b, 1, 2)
btn.setEnabled(False)
if b == 1:
btn.setEnabled(True)
b += 1
parent.sizebtns.setExclusive(True)
def make_quadrants(parent):
""" make quadrant buttons """
parent.quadbtns = QtGui.QButtonGroup(parent)
for b in range(9):
btn = QuadButton(b, ' '+str(b+1), parent)
parent.quadbtns.addButton(btn, b)
parent.l0.addWidget(btn, 0 + parent.quadbtns.button(b).ypos, 29 + parent.quadbtns.button(b).xpos, 1, 1)
btn.setEnabled(False)
b += 1
parent.quadbtns.setExclusive(True)
class QuadButton(QtGui.QPushButton):
""" custom QPushButton class for quadrant plotting
requires buttons to put into a QButtonGroup (parent.quadbtns)
allows only 1 button to pressed at a time
"""
def __init__(self, bid, Text, parent=None):
super(QuadButton,self).__init__(parent)
self.setText(Text)
self.setCheckable(True)
self.setStyleSheet(parent.styleInactive)
self.setFont(QtGui.QFont("Arial", 8, QtGui.QFont.Bold))
self.resize(self.minimumSizeHint())
self.setMaximumWidth(22)
self.xpos = bid%3
self.ypos = int(np.floor(bid/3))
self.clicked.connect(lambda: self.press(parent, bid))
self.show()
def press(self, parent, bid):
for b in range(9):
if parent.quadbtns.button(b).isEnabled():
parent.quadbtns.button(b).setStyleSheet(parent.styleUnpressed)
self.setStyleSheet(parent.stylePressed)
self.xrange = np.array([self.xpos-.15, self.xpos+1.15]) * parent.ops['Lx']/3
self.yrange = np.array([self.ypos-.15, self.ypos+1.15]) * parent.ops['Ly']/3
# change the zoom
parent.p1.setXRange(self.xrange[0], self.xrange[1])
parent.p1.setYRange(self.yrange[0], self.yrange[1])
parent.p2.setXRange(self.xrange[0], self.xrange[1])
parent.p2.setYRange(self.yrange[0], self.yrange[1])
parent.p2.setXLink("plot1")
parent.p2.setYLink("plot1")
parent.show()
# size of view
class SizeButton(QtGui.QPushButton):
""" buttons to make trace box bigger or smaller """
def __init__(self, bid, Text, parent=None):
super(SizeButton,self).__init__(parent)
self.setText(Text)
self.setCheckable(True)
self.setStyleSheet(parent.styleInactive)
self.setFont(QtGui.QFont("Arial", 8, QtGui.QFont.Bold))
self.resize(self.minimumSizeHint())
self.clicked.connect(lambda: self.press(parent))
self.bid = bid
self.show()
def press(self, parent):
bid = self.bid
for b in parent.sizebtns.buttons():
b.setStyleSheet(parent.styleUnpressed)
self.setStyleSheet(parent.stylePressed)
ts = 100
if bid==0:
parent.p2.linkView(parent.p2.XAxis,view=None)
parent.p2.linkView(parent.p2.YAxis,view=None)
parent.win.ci.layout.setColumnStretchFactor(0,ts)
parent.win.ci.layout.setColumnStretchFactor(1,0)
elif bid==1:
parent.win.ci.layout.setColumnStretchFactor(0,ts)
parent.win.ci.layout.setColumnStretchFactor(1,ts)
parent.p2.setXLink('plot1')
parent.p2.setYLink('plot1')
elif bid==2:
parent.p2.linkView(parent.p2.XAxis,view=None)
parent.p2.linkView(parent.p2.YAxis,view=None)
parent.win.ci.layout.setColumnStretchFactor(0,0)
parent.win.ci.layout.setColumnStretchFactor(1,ts)
# only enable selection buttons when not in 'both' view
if bid!=1:
if parent.ops_plot['color']!=0:
for btn in parent.topbtns.buttons():
btn.setStyleSheet(parent.styleUnpressed)
btn.setEnabled(True)
else:
parent.topbtns.button(0).setStyleSheet(parent.styleUnpressed)
parent.topbtns.button(0).setEnabled(True)
else:
parent.ROI_remove()
for btn in parent.topbtns.buttons():
btn.setEnabled(False)
btn.setStyleSheet(parent.styleInactive)
parent.win.show()
parent.show()
#
class TopButton(QtGui.QPushButton):
""" selection of top neurons"""
def __init__(self, bid, parent=None):
super(TopButton,self).__init__(parent)
text = [' draw selection', ' select top n', ' select bottom n']
self.bid = bid
self.setText(text[bid])
self.setCheckable(True)
self.setStyleSheet(parent.styleInactive)
self.setFont(QtGui.QFont("Arial", 8, QtGui.QFont.Bold))
self.resize(self.minimumSizeHint())
self.clicked.connect(lambda: self.press(parent))
self.show()
def press(self, parent):
bid = self.bid
if not parent.sizebtns.button(1).isChecked():
if parent.ops_plot['color']==0:
for b in [1,2]:
parent.topbtns.button(b).setEnabled(False)
parent.topbtns.button(b).setStyleSheet(parent.styleInactive)
else:
for b in [1,2]:
parent.topbtns.button(b).setEnabled(True)
parent.topbtns.button(b).setStyleSheet(parent.styleUnpressed)
else:
for b in range(3):
parent.topbtns.button(b).setEnabled(False)
parent.topbtns.button(b).setStyleSheet(parent.styleInactive)
self.setStyleSheet(parent.stylePressed)
if bid==0:
parent.ROI_selection()
else:
self.top_selection(parent)
def top_selection(self, parent):
bid = self.bid
parent.ROI_remove()
draw = False
ncells = len(parent.stat)
icells = np.minimum(ncells, parent.ntop)
if bid == 1:
top = True
elif bid == 2:
top = False
if parent.sizebtns.button(0).isChecked():
wplot = 0
draw = True
elif parent.sizebtns.button(2).isChecked():
wplot = 1
draw = True
if draw:
if parent.ops_plot['color'] != 0:
c = parent.ops_plot['color']
istat = parent.colors['istat'][c]
if wplot == 0:
icell = np.array(parent.iscell.nonzero()).flatten()
istat = istat[parent.iscell]
else:
icell = np.array((~parent.iscell).nonzero()).flatten()
istat = istat[~parent.iscell]
inds = istat.argsort()
if top:
inds = inds[-icells:]
parent.ichosen = icell[inds[-1]]
else:
inds = inds[:icells]
parent.ichosen = icell[inds[0]]
parent.imerge = []
for n in inds:
parent.imerge.append(icell[n])
# draw choices
parent.update_plot()
parent.show()
