def _with_glwidget(data):
app = QtWidgets.QApplication(sys.argv)
win = GLWidget()
d = DataModel(NumpyData(data))
t = time.time()
win.setModel(d)
print("time to set model in glwidget: ", time.time() - t)
win.show()
win.raise_()
app.win = win
QtCore.QTimer.singleShot(200, app.quit)
app.exec_()
def _with_glwidget(data):
app = QtWidgets.QApplication(sys.argv)
win = GLWidget()
d = DataModel(NumpyData(data))
t = time.time()
win.setModel(d)
print("time to set model in glwidget: ", time.time()-t)
win.show()
win.raise_()
app.win = win
QtCore.QTimer.singleShot(200, app.quit)
app.exec_()
def __init__(self, parent=None, interpolation="linear", **kwargs):
super(QtWidgets.QWidget, self).__init__(parent)
self.myparent = parent
self.isCloseFlag = False
self.setWindowTitle('spimagine')
self.resize(900, 700)
self.transform = TransformModel()
self.initActions()
self.glWidget = GLWidget(self, interpolation=interpolation)
self.glWidget.setTransform(self.transform)
self.sliceWidget = SliceWidget(self, interpolation=interpolation)
self.sliceWidget.hide()
# self.jack = JackPlugin(self.transform)
# self.jack.start()
self.sliceWidget.setTransform(self.transform)
self.fwdButton = createStandardButton(
self,
fName=absPath("images/icon_forward.png"),
method=self.forward,
width=18,
tooltip="forward")
self.bwdButton = createStandardButton(
self,
fName=absPath("images/icon_backward.png"),
method=self.backward,
width=18,
tooltip="backward")
self.startButton = createStandardButton(
self,
fName=absPath("images/icon_start.png"),
method=self.startPlay,
tooltip="play")
self.centerButton = createStandardButton(
self,
fName=absPath("images/icon_center.png"),
method=self.center,
tooltip="center view")
self.rotateButton = createStandardButton(
self,
fName=absPath("images/icon_rotate.png"),
method=self.rotate,
tooltip="spin current view")
self.screenshotButton = createStandardButton(
self,
fName=absPath("images/icon_camera.png"),
method=self.screenShot,
tooltip="save as png")
self.fileOpenButton = createStandardButton(
self,
fName=absPath("images/icon_open.png"),
method=self.openFile,
tooltip="open file")
self.fileSaveButton = createStandardButton(
self,
fName=absPath("images/icon_filesave.png"),
method=self.saveFile,
tooltip="save file as tif")
self.checkVolSettings = createImageCheckbox(
self,
absPath("images/icon_volsettings_active.png"),
absPath("images/icon_volsettings_inactive.png"),
tooltip="volume settings")
self.checkSettings = createImageCheckbox(
self,
absPath("images/icon_mainsettings_active.png"),
absPath("images/icon_mainsettings_inactive.png"),
tooltip="general settings")
self.checkKey = createImageCheckbox(
self,
absPath("images/video.png"),
absPath("images/video_inactive.png"),
tooltip="keyframe editor")
self.checkIsoView = createImageCheckbox(
self,
absPath("images/icon_method_vol.png"),
absPath("images/icon_method_iso.png"),
tooltip="iso surface")
self.checkProcView = createImageCheckbox(
self,
absPath("images/icon_process_active.png"),
absPath("images/icon_process_inactive.png"),
tooltip="image processors")
self.checkSliceView = createImageCheckbox(
self,
absPath("images/icon_slice.png"),
absPath("images/icon_slice_inactive.png"),
tooltip="slice view")
self.sliderTime = QtWidgets.QSlider(QtCore.Qt.Horizontal)
self.sliderTime.setTickPosition(QtWidgets.QSlider.TicksBothSides)
self.sliderTime.setTickInterval(1)
self.sliderTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.sliderTime.setTracking(False)
self.spinTime = QtWidgets.QSpinBox()
self.spinTime.setStyleSheet(
"color:white;border:0px solid black;background-color:black;")
self.spinTime.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
self.spinTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.spinTime.setAttribute(QtCore.Qt.WA_MacShowFocusRect, 0)
self.sliderTime.valueChanged.connect(self.spinTime.setValue)
self.spinTime.valueChanged.connect(self.sliderTime.setValue)
self.maxSlider = FloatSlider(QtCore.Qt.Vertical)
self.maxSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP,
1000)
self.maxSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.maxSlider.setToolTip("max")
self.minSlider = FloatSlider(QtCore.Qt.Vertical)
self.minSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP,
1000)
self.minSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.minSlider.setToolTip("min")
self.gammaSlider = FloatSlider(QtCore.Qt.Vertical)
self.gammaSlider.setRange(.01, 2., 200)
self.gammaSlider.setToolTip("gamma")
self.gammaSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.gammaSlider.setValue(1.)
# def func_from_n(n):
# return 2**(self.N_SCALE_MIN_EXP+(self.N_SCALE_MAX_EXP-self.N_SCALE_MIN_EXP)*(n-1.)/(self.N_SCALE_SLIDER-1))
# def func_to_n(x):
# if x<2**self.N_SCALE_MIN_EXP:
# print "gg", x
# return 1
# elif x>2**self.N_SCALE_MAX_EXP:
# return self.N_SCALE_SLIDER
# return int(round(1.+(self.N_SCALE_SLIDER-1.)*(np.log2(x)-self.N_SCALE_MIN_EXP)/(self.N_SCALE_MAX_EXP-1.*self.N_SCALE_MIN_EXP)))
# self.scaleSlider.valueChanged.connect(lambda x: self.transform.setValueScale(0,func_from_n(x)))
# self.transform._maxChanged.connect(lambda x: self.scaleSlider.setValue(func_to_n(x)))
def func1(x):
#print("func2: %.2f \t->\t %.2f" % (x, 2**x))
return 2**x
def func2(x):
#print("func2: %.2f \t->\t %.2f"%(x,np.log2(x) if x>0 else -1.e20))
return np.log2(x) if x > 0 else -1.e20
self.maxSlider.floatValueChanged.connect(
lambda x: self.transform.setMax(func1(x)))
self.transform._maxChanged.connect(
lambda x: self.maxSlider.setValue(func2(x)))
self.minSlider.floatValueChanged.connect(
lambda x: self.transform.setMin(func1(x)))
self.transform._minChanged.connect(
lambda x: self.minSlider.setValue(func2(x)))
self.gammaSlider.floatValueChanged.connect(self.transform.setGamma)
self.transform._gammaChanged.connect(self.gammaSlider.setValue)
# self.keyframes = KeyFrameList()
self.keyPanel = KeyFramePanel(self.glWidget)
self.keyPanel.hide()
self.keyPanel.setStyleSheet("color:white")
self.impListView = ImageProcessorListView([
BlurProcessor(),
BlurXYZProcessor(),
NoiseProcessor(),
FFTProcessor(),
#LucyRichProcessor()
])
self.impListView.hide()
self.volSettingsView = VolumeSettingsPanel()
self.volSettingsView.hide()
self.settingsView = MainSettingsPanel()
self.settingsView.hide()
hbox0 = QtWidgets.QHBoxLayout()
hbox0.addWidget(self.minSlider)
hbox0.addWidget(self.maxSlider)
hbox0.addWidget(self.gammaSlider)
hbox0.addWidget(self.glWidget, stretch=3)
hbox0.addWidget(self.sliceWidget, stretch=2)
hbox0.addWidget(self.impListView)
hbox0.addWidget(self.volSettingsView)
hbox0.addWidget(self.settingsView)
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(self.bwdButton)
hbox.addWidget(self.startButton)
hbox.addWidget(self.fwdButton)
hbox.addWidget(self.sliderTime)
hbox.addWidget(self.spinTime)
hbox.addWidget(self.centerButton)
hbox.addWidget(self.rotateButton)
hbox.addWidget(self.checkKey)
hbox.addWidget(self.screenshotButton)
hbox.addWidget(self.fileOpenButton)
hbox.addWidget(self.fileSaveButton)
hbox.addSpacing(50)
hbox.addWidget(self.checkIsoView)
hbox.addWidget(self.checkSliceView)
hbox.addSpacing(10)
hbox.addWidget(self.checkProcView)
hbox.addSpacing(10)
hbox.addWidget(self.checkVolSettings)
hbox.addSpacing(5)
hbox.addWidget(self.checkSettings)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(hbox0)
vbox.addLayout(hbox)
vbox.addWidget(self.keyPanel)
for box in [hbox, vbox, hbox0]:
box.setContentsMargins(0, 0, 0, 0)
vbox.setSpacing(1)
hbox.setSpacing(11)
hbox0.setSpacing(5)
self.egg3d = spimagine.utils.egg3d.Egg3dController()
# widget = QtWidgets.QWidget()
self.setLayout(vbox)
self.rotateTimer = QtCore.QTimer(self)
self.rotateTimer.setInterval(70)
self.rotateTimer.timeout.connect(self.onRotateTimer)
self.playTimer = QtCore.QTimer(self)
self.playTimer.setInterval(100)
self.playTimer.timeout.connect(self.onPlayTimer)
self.settingsView._playIntervalChanged.connect(
self.playIntervalChanged)
self.setLoopBounce(True)
self.playDir = 1
# a decorator as checkboxe state is 2 if checked
def stateToBool(objFunc, invert=False):
def _foo(x):
if invert:
return objFunc(x != 2)
else:
return objFunc(x == 2)
return _foo
self.volSettingsView.checkBox.stateChanged.connect(
self.glWidget.transform.setBox)
self.volSettingsView.checkInterpolate.stateChanged.connect(
self.glWidget.transform.setInterpolate)
self.transform._interpChanged.connect(
self.volSettingsView.checkInterpolate.setChecked)
self.transform._interpChanged.connect(self.glWidget.set_interpolation)
self.volSettingsView.checkInvert.stateChanged.connect(
stateToBool(self.glWidget.set_background_mode_black, invert=True))
self.settingsView._substepsChanged.connect(self.substepsChanged)
self.checkIsoView.stateChanged.connect(
stateToBool(self.glWidget.transform.setIso))
self.transform._isoChanged.connect(self.checkIsoView.setChecked)
self.settingsView.checkEgg.stateChanged.connect(self.onCheckEgg)
self.volSettingsView._boundsChanged.connect(
self.glWidget.transform.setBounds)
self.volSettingsView.sliderAlphaPow.floatValueChanged.connect(
self.glWidget.transform.setAlphaPow)
self.volSettingsView.sliderOcc.floatValueChanged.connect(
self.glWidget.transform.setOccStrength)
self.volSettingsView.sliderOccRadius.floatValueChanged.connect(
self.glWidget.transform.setOccRadius)
self.volSettingsView.sliderOccNPoints.floatValueChanged.connect(
self.glWidget.transform.setOccNPoints)
self.glWidget.transform._alphaPowChanged.connect(
self.volSettingsView.sliderAlphaPow.setValue)
self.glWidget.transform._boundsChanged.connect(
self.volSettingsView.setBounds)
self.transform._boxChanged.connect(
self.volSettingsView.checkBox.setChecked)
self.volSettingsView.checkProj.stateChanged.connect(
self.transform.setPerspective)
self.transform._perspectiveChanged.connect(
self.volSettingsView.checkProj.setChecked)
self.checkKey.stateChanged.connect(self.keyPanel.setVisible)
self.checkVolSettings.stateChanged.connect(
self.volSettingsView.setVisible)
self.checkSettings.stateChanged.connect(self.settingsView.setVisible)
self.checkSliceView.stateChanged.connect(self.sliceWidget.setVisible)
self.checkSliceView.stateChanged.connect(self.transform.setShowSlice)
self.checkProcView.stateChanged.connect(self.impListView.setVisible)
self.settingsView.checkLoopBounce.stateChanged.connect(
self.setLoopBounce)
self.volSettingsView._stackUnitsChanged.connect(
self.transform.setStackUnits)
self.transform._stackUnitsChanged.connect(
self.volSettingsView.setStackUnits)
self.settingsView._frameNumberChanged.connect(
self.keyPanel.setFrameNumber)
self.settingsView._recordDelayChanged.connect(
self.keyPanel.setRecordDelay)
self.volSettingsView.colorCombo.currentIndexChanged.connect(
self.onColormapChanged)
self.volSettingsView._rgbColorChanged.connect(self.onRgbColorChanged)
self.impListView._stateChanged.connect(self.impStateChanged)
self.settingsView._dirNameChanged.connect(self.keyPanel.setDirName)
# dataModel._dataSourceChanged.connect(self.dataSourceChanged)
# dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.glWidget._dataModelChanged.connect(self.dataModelChanged)
self.onColormapChanged(0)
self.checkSliceView.setChecked(False)
self.hiddableControls = [
self.checkSettings, self.checkVolSettings, self.startButton,
self.sliderTime, self.spinTime, self.checkKey,
self.screenshotButton
]
# self.keyPanel.keyView.setModel(self.keyframes)
self.setStyleSheet("""
background-color:black;
color:white;""")
class MainWidget(QtWidgets.QWidget):
N_SCALE_MIN_EXP = -16
N_SCALE_MAX_EXP = 16
N_SCALE_SLIDER = 500
def __init__(self, parent=None, interpolation="linear", **kwargs):
super(QtWidgets.QWidget, self).__init__(parent)
self.myparent = parent
self.isCloseFlag = False
self.setWindowTitle('spimagine')
self.resize(900, 700)
self.transform = TransformModel()
self.initActions()
self.glWidget = GLWidget(self, interpolation=interpolation)
self.glWidget.setTransform(self.transform)
self.sliceWidget = SliceWidget(self, interpolation=interpolation)
self.sliceWidget.hide()
# self.jack = JackPlugin(self.transform)
# self.jack.start()
self.sliceWidget.setTransform(self.transform)
self.fwdButton = createStandardButton(
self,
fName=absPath("images/icon_forward.png"),
method=self.forward,
width=18,
tooltip="forward")
self.bwdButton = createStandardButton(
self,
fName=absPath("images/icon_backward.png"),
method=self.backward,
width=18,
tooltip="backward")
self.startButton = createStandardButton(
self,
fName=absPath("images/icon_start.png"),
method=self.startPlay,
tooltip="play")
self.centerButton = createStandardButton(
self,
fName=absPath("images/icon_center.png"),
method=self.center,
tooltip="center view")
self.rotateButton = createStandardButton(
self,
fName=absPath("images/icon_rotate.png"),
method=self.rotate,
tooltip="spin current view")
self.screenshotButton = createStandardButton(
self,
fName=absPath("images/icon_camera.png"),
method=self.screenShot,
tooltip="save as png")
self.fileOpenButton = createStandardButton(
self,
fName=absPath("images/icon_open.png"),
method=self.openFile,
tooltip="open file")
self.fileSaveButton = createStandardButton(
self,
fName=absPath("images/icon_filesave.png"),
method=self.saveFile,
tooltip="save file as tif")
self.checkVolSettings = createImageCheckbox(
self,
absPath("images/icon_volsettings_active.png"),
absPath("images/icon_volsettings_inactive.png"),
tooltip="volume settings")
self.checkSettings = createImageCheckbox(
self,
absPath("images/icon_mainsettings_active.png"),
absPath("images/icon_mainsettings_inactive.png"),
tooltip="general settings")
self.checkKey = createImageCheckbox(
self,
absPath("images/video.png"),
absPath("images/video_inactive.png"),
tooltip="keyframe editor")
self.checkIsoView = createImageCheckbox(
self,
absPath("images/icon_method_vol.png"),
absPath("images/icon_method_iso.png"),
tooltip="iso surface")
self.checkProcView = createImageCheckbox(
self,
absPath("images/icon_process_active.png"),
absPath("images/icon_process_inactive.png"),
tooltip="image processors")
self.checkSliceView = createImageCheckbox(
self,
absPath("images/icon_slice.png"),
absPath("images/icon_slice_inactive.png"),
tooltip="slice view")
self.sliderTime = QtWidgets.QSlider(QtCore.Qt.Horizontal)
self.sliderTime.setTickPosition(QtWidgets.QSlider.TicksBothSides)
self.sliderTime.setTickInterval(1)
self.sliderTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.sliderTime.setTracking(False)
self.spinTime = QtWidgets.QSpinBox()
self.spinTime.setStyleSheet(
"color:white;border:0px solid black;background-color:black;")
self.spinTime.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
self.spinTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.spinTime.setAttribute(QtCore.Qt.WA_MacShowFocusRect, 0)
self.sliderTime.valueChanged.connect(self.spinTime.setValue)
self.spinTime.valueChanged.connect(self.sliderTime.setValue)
self.maxSlider = FloatSlider(QtCore.Qt.Vertical)
self.maxSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP,
1000)
self.maxSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.maxSlider.setToolTip("max")
self.minSlider = FloatSlider(QtCore.Qt.Vertical)
self.minSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP,
1000)
self.minSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.minSlider.setToolTip("min")
self.gammaSlider = FloatSlider(QtCore.Qt.Vertical)
self.gammaSlider.setRange(.01, 2., 200)
self.gammaSlider.setToolTip("gamma")
self.gammaSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.gammaSlider.setValue(1.)
# def func_from_n(n):
# return 2**(self.N_SCALE_MIN_EXP+(self.N_SCALE_MAX_EXP-self.N_SCALE_MIN_EXP)*(n-1.)/(self.N_SCALE_SLIDER-1))
# def func_to_n(x):
# if x<2**self.N_SCALE_MIN_EXP:
# print "gg", x
# return 1
# elif x>2**self.N_SCALE_MAX_EXP:
# return self.N_SCALE_SLIDER
# return int(round(1.+(self.N_SCALE_SLIDER-1.)*(np.log2(x)-self.N_SCALE_MIN_EXP)/(self.N_SCALE_MAX_EXP-1.*self.N_SCALE_MIN_EXP)))
# self.scaleSlider.valueChanged.connect(lambda x: self.transform.setValueScale(0,func_from_n(x)))
# self.transform._maxChanged.connect(lambda x: self.scaleSlider.setValue(func_to_n(x)))
def func1(x):
#print("func2: %.2f \t->\t %.2f" % (x, 2**x))
return 2**x
def func2(x):
#print("func2: %.2f \t->\t %.2f"%(x,np.log2(x) if x>0 else -1.e20))
return np.log2(x) if x > 0 else -1.e20
self.maxSlider.floatValueChanged.connect(
lambda x: self.transform.setMax(func1(x)))
self.transform._maxChanged.connect(
lambda x: self.maxSlider.setValue(func2(x)))
self.minSlider.floatValueChanged.connect(
lambda x: self.transform.setMin(func1(x)))
self.transform._minChanged.connect(
lambda x: self.minSlider.setValue(func2(x)))
self.gammaSlider.floatValueChanged.connect(self.transform.setGamma)
self.transform._gammaChanged.connect(self.gammaSlider.setValue)
# self.keyframes = KeyFrameList()
self.keyPanel = KeyFramePanel(self.glWidget)
self.keyPanel.hide()
self.keyPanel.setStyleSheet("color:white")
self.impListView = ImageProcessorListView([
BlurProcessor(),
BlurXYZProcessor(),
NoiseProcessor(),
FFTProcessor(),
#LucyRichProcessor()
])
self.impListView.hide()
self.volSettingsView = VolumeSettingsPanel()
self.volSettingsView.hide()
self.settingsView = MainSettingsPanel()
self.settingsView.hide()
hbox0 = QtWidgets.QHBoxLayout()
hbox0.addWidget(self.minSlider)
hbox0.addWidget(self.maxSlider)
hbox0.addWidget(self.gammaSlider)
hbox0.addWidget(self.glWidget, stretch=3)
hbox0.addWidget(self.sliceWidget, stretch=2)
hbox0.addWidget(self.impListView)
hbox0.addWidget(self.volSettingsView)
hbox0.addWidget(self.settingsView)
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(self.bwdButton)
hbox.addWidget(self.startButton)
hbox.addWidget(self.fwdButton)
hbox.addWidget(self.sliderTime)
hbox.addWidget(self.spinTime)
hbox.addWidget(self.centerButton)
hbox.addWidget(self.rotateButton)
hbox.addWidget(self.checkKey)
hbox.addWidget(self.screenshotButton)
hbox.addWidget(self.fileOpenButton)
hbox.addWidget(self.fileSaveButton)
hbox.addSpacing(50)
hbox.addWidget(self.checkIsoView)
hbox.addWidget(self.checkSliceView)
hbox.addSpacing(10)
hbox.addWidget(self.checkProcView)
hbox.addSpacing(10)
hbox.addWidget(self.checkVolSettings)
hbox.addSpacing(5)
hbox.addWidget(self.checkSettings)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(hbox0)
vbox.addLayout(hbox)
vbox.addWidget(self.keyPanel)
for box in [hbox, vbox, hbox0]:
box.setContentsMargins(0, 0, 0, 0)
vbox.setSpacing(1)
hbox.setSpacing(11)
hbox0.setSpacing(5)
self.egg3d = spimagine.utils.egg3d.Egg3dController()
# widget = QtWidgets.QWidget()
self.setLayout(vbox)
self.rotateTimer = QtCore.QTimer(self)
self.rotateTimer.setInterval(70)
self.rotateTimer.timeout.connect(self.onRotateTimer)
self.playTimer = QtCore.QTimer(self)
self.playTimer.setInterval(100)
self.playTimer.timeout.connect(self.onPlayTimer)
self.settingsView._playIntervalChanged.connect(
self.playIntervalChanged)
self.setLoopBounce(True)
self.playDir = 1
# a decorator as checkboxe state is 2 if checked
def stateToBool(objFunc, invert=False):
def _foo(x):
if invert:
return objFunc(x != 2)
else:
return objFunc(x == 2)
return _foo
self.volSettingsView.checkBox.stateChanged.connect(
self.glWidget.transform.setBox)
self.volSettingsView.checkInterpolate.stateChanged.connect(
self.glWidget.transform.setInterpolate)
self.transform._interpChanged.connect(
self.volSettingsView.checkInterpolate.setChecked)
self.transform._interpChanged.connect(self.glWidget.set_interpolation)
self.volSettingsView.checkInvert.stateChanged.connect(
stateToBool(self.glWidget.set_background_mode_black, invert=True))
self.settingsView._substepsChanged.connect(self.substepsChanged)
self.checkIsoView.stateChanged.connect(
stateToBool(self.glWidget.transform.setIso))
self.transform._isoChanged.connect(self.checkIsoView.setChecked)
self.settingsView.checkEgg.stateChanged.connect(self.onCheckEgg)
self.volSettingsView._boundsChanged.connect(
self.glWidget.transform.setBounds)
self.volSettingsView.sliderAlphaPow.floatValueChanged.connect(
self.glWidget.transform.setAlphaPow)
self.volSettingsView.sliderOcc.floatValueChanged.connect(
self.glWidget.transform.setOccStrength)
self.volSettingsView.sliderOccRadius.floatValueChanged.connect(
self.glWidget.transform.setOccRadius)
self.volSettingsView.sliderOccNPoints.floatValueChanged.connect(
self.glWidget.transform.setOccNPoints)
self.glWidget.transform._alphaPowChanged.connect(
self.volSettingsView.sliderAlphaPow.setValue)
self.glWidget.transform._boundsChanged.connect(
self.volSettingsView.setBounds)
self.transform._boxChanged.connect(
self.volSettingsView.checkBox.setChecked)
self.volSettingsView.checkProj.stateChanged.connect(
self.transform.setPerspective)
self.transform._perspectiveChanged.connect(
self.volSettingsView.checkProj.setChecked)
self.checkKey.stateChanged.connect(self.keyPanel.setVisible)
self.checkVolSettings.stateChanged.connect(
self.volSettingsView.setVisible)
self.checkSettings.stateChanged.connect(self.settingsView.setVisible)
self.checkSliceView.stateChanged.connect(self.sliceWidget.setVisible)
self.checkSliceView.stateChanged.connect(self.transform.setShowSlice)
self.checkProcView.stateChanged.connect(self.impListView.setVisible)
self.settingsView.checkLoopBounce.stateChanged.connect(
self.setLoopBounce)
self.volSettingsView._stackUnitsChanged.connect(
self.transform.setStackUnits)
self.transform._stackUnitsChanged.connect(
self.volSettingsView.setStackUnits)
self.settingsView._frameNumberChanged.connect(
self.keyPanel.setFrameNumber)
self.settingsView._recordDelayChanged.connect(
self.keyPanel.setRecordDelay)
self.volSettingsView.colorCombo.currentIndexChanged.connect(
self.onColormapChanged)
self.volSettingsView._rgbColorChanged.connect(self.onRgbColorChanged)
self.impListView._stateChanged.connect(self.impStateChanged)
self.settingsView._dirNameChanged.connect(self.keyPanel.setDirName)
# dataModel._dataSourceChanged.connect(self.dataSourceChanged)
# dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.glWidget._dataModelChanged.connect(self.dataModelChanged)
self.onColormapChanged(0)
self.checkSliceView.setChecked(False)
self.hiddableControls = [
self.checkSettings, self.checkVolSettings, self.startButton,
self.sliderTime, self.spinTime, self.checkKey,
self.screenshotButton
]
# self.keyPanel.keyView.setModel(self.keyframes)
self.setStyleSheet("""
background-color:black;
color:white;""")
def impStateChanged(self):
data = self.transform.dataModel[self.transform.dataPos]
for imp in self.impListView.impViews:
if imp.is_active():
print("active: ", imp.proc.name, imp.proc.kwargs)
data = imp.proc.apply(data)
self.glWidget.renderer.update_data(data)
self.glWidget.refresh()
def onColormapChanged(self, index):
self.glWidget.set_colormap(self.volSettingsView.colormaps[index])
self.glWidget.refresh()
self.sliceWidget.glSliceWidget.set_colormap(
self.volSettingsView.colormaps[index])
self.sliceWidget.glSliceWidget.refresh()
def onRgbColorChanged(self, r, g, b):
self.glWidget.set_colormap_rgb([r, g, b])
self.glWidget.refresh()
self.sliceWidget.glSliceWidget.set_colormap_rgb([r, g, b])
self.sliceWidget.glSliceWidget.refresh()
def onCheckEgg(self, state):
if state == QtCore.Qt.Checked:
self.connectEgg3d()
else:
self.egg3d.stop()
def connectEgg3d(self):
try:
self.egg3d.listener._quaternionChanged.connect(
self.egg3dQuaternion)
self.egg3d.listener._zoomChanged.connect(self.egg3dZoom)
N = 45
self._quatHist = [Quaternion() for i in range(N)]
self._quatWeights = np.exp(-2. * np.linspace(0, 1, N))
self._quatWeights *= 1. / sum(self._quatWeights)
self.egg3d.start()
except Exception as e:
print(e)
self.settingsView.checkEgg.setCheckState(QtCore.Qt.Unchecked)
def egg3dQuaternion(self, a, b, c, d):
self._quatHist = np.roll(self._quatHist, 1)
self._quatHist[0] = Quaternion(a, c, b, d)
q0 = Quaternion(0, 0, 0, 0)
for q, w in zip(self._quatHist, self._quatWeights):
q0 = q0 + q * w
self.transform.setQuaternion(q0)
def egg3dZoom(self, zoom):
if zoom > 0:
newZoom = self.transform.zoom * 1.01**zoom
else:
newZoom = self.transform.zoom * 1.1**zoom
newZoom = np.clip(newZoom, .7, 3)
self.transform.setZoom(newZoom)
def initUI(self):
pass
def raise_window(self):
#self.setWindowState(QtCore.Qt.WindowActive)
self.raise_()
def keyPressEvent(self, event):
if type(event) == QtGui.QKeyEvent:
if event.modifiers() == QtCore.Qt.ControlModifier and event.key(
) == QtCore.Qt.Key_Q:
return
# super(MainWidget,self).keyPressEvent(event)
def initActions(self):
QtWidgets.QShortcut(QtGui.QKeySequence("Ctrl+W"), self, self.closeMe)
# self.exitAction = QtWidgets.QAction('Quit', self)
# self.exitAction.setShortcut('Ctrl+Q')
# self.exitAction.setStatusTip('Exit application')
# self.exitAction.triggered.connect(self.foo)
def setModel(self, dataModel):
self.glWidget.setModel(dataModel)
self.sliceWidget.setModel(dataModel)
def dataModelChanged(self):
logger.debug("data Model changed")
dataModel = self.glWidget.dataModel
dataModel._dataSourceChanged.connect(self.dataSourceChanged)
dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.sliderTime.valueChanged.connect(self.transform.setPos)
kvList = KeyFrameList()
kvList.addItem(KeyFrame(0.))
kvList.addItem(KeyFrame(1.))
self.keyPanel.connect_to_transform(self.transform)
self.keyPanel.setModel(kvList)
self.dataSourceChanged()
def dataSourceChanged(self):
self.sliderTime.setRange(0, self.glWidget.dataModel.sizeT() - 1)
self.sliderTime.setValue(0)
self.spinTime.setRange(0, self.glWidget.dataModel.sizeT() - 1)
self.volSettingsView.dimensionLabel.setText(
"Dim: %s" % str(tuple(self.glWidget.dataModel.size()[::-1])))
if self.myparent:
self.myparent.setWindowTitle(self.glWidget.dataModel.name())
else:
self.setWindowTitle(self.glWidget.dataModel.name())
self.keyPanel.connect_to_transform(self.transform)
d = self.glWidget.dataModel[self.glWidget.dataModel.pos]
# minMaxMean = (np.amin(d),np.amax(d),np.mean(d))
# self.volSettingsView.statsLabel.setText("Min:\t%.2f\nMax:\t%.2f \nMean:\t%.2f"%minMaxMean)
def forward(self, event):
if self.glWidget.dataModel:
newpos = (self.glWidget.dataModel.pos +
1) % self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def backward(self, event):
if self.glWidget.dataModel:
newpos = (self.glWidget.dataModel.pos -
1) % self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def startPlay(self, event):
if self.playTimer.isActive():
self.playTimer.stop()
self.startButton.setIcon(
QtGui.QIcon(absPath("images/icon_start.png")))
else:
self.playTimer.start()
self.startButton.setIcon(
QtGui.QIcon(absPath("images/icon_pause.png")))
def screenShot(self):
fileName, _ = QtWidgets.QFileDialog.getSaveFileName(
self, 'Save screenshot as', '.', ('*.png'))
if fileName:
self.glWidget.saveFrame(str(fileName))
def set_colormap(self, name):
self.glWidget.set_colormap(name)
def set_colormap_rgb(self, color):
self.glWidget.set_colormap_rgb(color)
def saveFrame(self, fName, with_alpha=False):
self.glWidget.saveFrame(fName, with_alpha=with_alpha)
def setLoopBounce(self, loopBounce):
#if loopBounce = True, then while playing it should loop back and forth
self.loopBounce = loopBounce
self.settingsView.checkLoopBounce.setChecked(loopBounce)
def playIntervalChanged(self, val):
if self.playTimer.isActive():
self.playTimer.stop()
self.playTimer.setInterval(val)
def substepsChanged(self, val):
self.glWidget.NSubrenderSteps = val
self.glWidget.refresh()
def openFile(self, e):
# path = QtWidgets.QFileDialog.getOpenFileName(self, 'Open Path (File or Folder)',
# '.', selectedFilter='*.tif')
f = QtWidgets.QFileDialog()
f.setWindowTitle('Open Path (File or Folder)')
f.setFileMode(QtWidgets.QFileDialog.ExistingFile
& QtWidgets.QFileDialog.Directory)
f.exec_()
path = f.selectedFiles()
if len(path) == 0:
return
path = str(path[0])
if path:
try:
if self.glWidget.dataModel:
self.glWidget.dataModel.loadFromPath(
path, prefetchSize=self.glWidget.N_PREFETCH)
else:
self.glWidget.setModel(
DataModel.fromPath(
path, prefetchSize=self.glWidget.N_PREFETCH))
except Exception as e:
mbox = QtWidgets.QMessageBox()
mbox.setText(str(e))
mbox.setIcon(QtWidgets.QMessageBox.Warning)
mbox.exec_()
def saveFile(self, e):
path, _ = QtWidgets.QFileDialog.getSaveFileName(
self, 'Save as Tif File', '.', ('*.tif'))
if path:
if self.glWidget.dataModel:
write3dTiff(
self.glWidget.dataModel[self.transform.dataPos].astype(
np.float32), path)
def onPlayTimer(self):
if self.glWidget.dataModel:
if self.glWidget.dataModel.pos == self.glWidget.dataModel.sizeT(
) - 1:
self.playDir = 1 - 2 * self.loopBounce
if self.glWidget.dataModel.pos == 0:
self.playDir = 1
newpos = (self.glWidget.dataModel.pos +
self.playDir) % self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def contextMenuEvent(self, event):
# create context menu
popMenu = QtWidgets.QMenu(self)
action = QtWidgets.QAction('toggle controls', self)
action.triggered.connect(self.toggleControls)
popMenu.addAction(action)
popMenu.setStyleSheet("background-color: white")
# popMenu.exec_(QtGui.QCursor.pos())
def toggleControls(self):
for c in self.hiddableControls:
c.setVisible(not c.isVisible())
def closeMe(self):
#little workaround as on MAC ctrl-q cannot be overwritten
self.isCloseFlag = True
self.close()
def closeEvent(self, event):
logger.debug("closeevent")
if not event.spontaneous() and not self.isCloseFlag:
event.ignore()
else:
if self.playTimer.isActive():
self.playTimer.stop()
if self.keyPanel.recordTimer.isActive():
self.keyPanel.recordTimer.stop()
if self.keyPanel.playTimer.isActive():
self.keyPanel.playTimer.stop()
if self.rotateTimer.isActive():
self.rotateTimer.stop()
if hasattr(self, "glWidget"):
logger.debug("deleting the renderer")
try:
self.glWidget.renderTimer.stop()
del self.glWidget.renderer
self.glWidget.setParent(None)
del self.glWidget
except:
pass
logger.debug("....finished deleting the renderer")
event.accept()
def center(self):
self.transform.center()
def rotate(self):
if self.rotateTimer.isActive():
self.rotateTimer.stop()
self.rotateButton.setIcon(
QtGui.QIcon(absPath("images/icon_rotate.png")))
else:
self.rotateTimer.start()
self.rotateButton.setIcon(
QtGui.QIcon(absPath("images/icon_rotate_active.png")))
def onRotateTimer(self):
axis = [0, 0, 0]
axis[spimagine.config.__DEFAULT_SPIN_AXIS__] = 1
self.transform.addRotation(-.02, *axis)
self.glWidget.render()
self.glWidget.updateGL()
def _show_fullscreen(self):
self.show()
self.showFullScreen()
def _show_normal(self):
self.show()
self.showNormal()
def toggle_fullscreen(self):
if self.isFullScreen():
QtCore.QTimer.singleShot(200, self._show_normal)
else:
QtCore.QTimer.singleShot(200, self._show_fullscreen)
def mouseDoubleClickEvent(self, event):
super(MainWidget, self).mouseDoubleClickEvent(event)
self.toggle_fullscreen()
self.glWidget.resized = True
# there's a bug in Qt that disables drop after fullscreen, so reset it...
self.setAcceptDrops(True)
def moveEvent(self, evt):
if hasattr(self, "glWidget"):
self.glWidget.moveEvent(evt)
def __init__(self, parent = None, interpolation = "linear", **kwargs):
super(QtWidgets.QWidget,self).__init__(parent)
self.myparent = parent
self.isCloseFlag = False
self.setWindowTitle('spimagine')
self.resize(900, 700)
self.transform = TransformModel()
self.initActions()
self.glWidget = GLWidget(self,interpolation = interpolation)
self.glWidget.setTransform(self.transform)
self.sliceWidget = SliceWidget(self, interpolation = interpolation)
self.sliceWidget.hide()
# self.jack = JackPlugin(self.transform)
# self.jack.start()
self.sliceWidget.setTransform(self.transform)
self.fwdButton = createStandardButton(self,
fName = absPath("images/icon_forward.png"),
method = self.forward, width = 18, tooltip="forward")
self.bwdButton = createStandardButton(self,
fName = absPath("images/icon_backward.png"),
method = self.backward, width = 18, tooltip="backward")
self.startButton = createStandardButton(self, fName = absPath("images/icon_start.png"),
method = self.startPlay, tooltip="play")
self.centerButton = createStandardButton(self,
fName = absPath("images/icon_center.png"),
method = self.center, tooltip = "center view")
self.rotateButton = createStandardButton(self,
fName = absPath("images/icon_rotate.png"),
method = self.rotate, tooltip = "spin current view")
self.screenshotButton = createStandardButton(self,
fName = absPath("images/icon_camera.png"),
method = self.screenShot,
tooltip = "save as png")
self.fileOpenButton = createStandardButton(self,
fName = absPath("images/icon_open.png"),
method = self.openFile, tooltip = "open file")
self.fileSaveButton = createStandardButton(self,
fName = absPath("images/icon_filesave.png"),
method = self.saveFile, tooltip = "save file as tif")
self.checkVolSettings = createImageCheckbox(self,
absPath("images/icon_volsettings_active.png"),
absPath("images/icon_volsettings_inactive.png"),
tooltip="volume settings")
self.checkSettings = createImageCheckbox(self,
absPath("images/icon_mainsettings_active.png"),
absPath("images/icon_mainsettings_inactive.png"),
tooltip="general settings")
self.checkKey = createImageCheckbox(self, absPath("images/video.png"), absPath("images/video_inactive.png"), tooltip="keyframe editor")
self.checkIsoView = createImageCheckbox(self,
absPath("images/icon_method_vol.png"),
absPath("images/icon_method_iso.png"),
tooltip="iso surface")
self.checkProcView = createImageCheckbox(self,
absPath("images/icon_process_active.png"),
absPath("images/icon_process_inactive.png"),
tooltip="image processors")
self.checkSliceView = createImageCheckbox(
self,
absPath("images/icon_slice.png"),
absPath("images/icon_slice_inactive.png"), tooltip="slice view")
self.sliderTime = QtWidgets.QSlider(QtCore.Qt.Horizontal)
self.sliderTime.setTickPosition(QtWidgets.QSlider.TicksBothSides)
self.sliderTime.setTickInterval(1)
self.sliderTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.sliderTime.setTracking(False)
self.spinTime = QtWidgets.QSpinBox()
self.spinTime.setStyleSheet("color:white;border:0px solid black;background-color:black;")
self.spinTime.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
self.spinTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.spinTime.setAttribute(QtCore.Qt.WA_MacShowFocusRect, 0)
self.sliderTime.valueChanged.connect(self.spinTime.setValue)
self.spinTime.valueChanged.connect(self.sliderTime.setValue)
self.maxSlider = FloatSlider(QtCore.Qt.Vertical)
self.maxSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP, 1000)
self.maxSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.maxSlider.setToolTip("max")
self.minSlider = FloatSlider(QtCore.Qt.Vertical)
self.minSlider.setRange(self.N_SCALE_MIN_EXP,self.N_SCALE_MAX_EXP,1000)
self.minSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.minSlider.setToolTip("min")
self.gammaSlider = FloatSlider(QtCore.Qt.Vertical)
self.gammaSlider.setRange(.01,2.,200)
self.gammaSlider.setToolTip("gamma")
self.gammaSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.gammaSlider.setValue(1.)
# def func_from_n(n):
# return 2**(self.N_SCALE_MIN_EXP+(self.N_SCALE_MAX_EXP-self.N_SCALE_MIN_EXP)*(n-1.)/(self.N_SCALE_SLIDER-1))
# def func_to_n(x):
# if x<2**self.N_SCALE_MIN_EXP:
# print "gg", x
# return 1
# elif x>2**self.N_SCALE_MAX_EXP:
# return self.N_SCALE_SLIDER
# return int(round(1.+(self.N_SCALE_SLIDER-1.)*(np.log2(x)-self.N_SCALE_MIN_EXP)/(self.N_SCALE_MAX_EXP-1.*self.N_SCALE_MIN_EXP)))
# self.scaleSlider.valueChanged.connect(lambda x: self.transform.setValueScale(0,func_from_n(x)))
# self.transform._maxChanged.connect(lambda x: self.scaleSlider.setValue(func_to_n(x)))
def func1(x):
#print("func2: %.2f \t->\t %.2f" % (x, 2**x))
return 2**x
def func2(x):
#print("func2: %.2f \t->\t %.2f"%(x,np.log2(x) if x>0 else -1.e20))
return np.log2(x) if x>0 else -1.e20
self.maxSlider.floatValueChanged.connect(lambda x: self.transform.setMax(func1(x)))
self.transform._maxChanged.connect(lambda x:self.maxSlider.setValue(func2(x)))
self.minSlider.floatValueChanged.connect(lambda x: self.transform.setMin(func1(x)))
self.transform._minChanged.connect(lambda x:self.minSlider.setValue(func2(x)))
self.gammaSlider.floatValueChanged.connect(self.transform.setGamma)
self.transform._gammaChanged.connect(self.gammaSlider.setValue)
# self.keyframes = KeyFrameList()
self.keyPanel = KeyFramePanel(self.glWidget)
self.keyPanel.hide()
self.keyPanel.setStyleSheet("color:white")
self.impListView = ImageProcessorListView([BlurProcessor(),
BlurXYZProcessor(),
NoiseProcessor(),
FFTProcessor(),
#LucyRichProcessor()
])
self.impListView.hide()
self.volSettingsView = VolumeSettingsPanel()
self.volSettingsView.hide()
self.settingsView = MainSettingsPanel()
self.settingsView.hide()
hbox0 = QtWidgets.QHBoxLayout()
hbox0.addWidget(self.minSlider)
hbox0.addWidget(self.maxSlider)
hbox0.addWidget(self.gammaSlider)
hbox0.addWidget(self.glWidget,stretch = 3)
hbox0.addWidget(self.sliceWidget,stretch =2)
hbox0.addWidget(self.impListView)
hbox0.addWidget(self.volSettingsView)
hbox0.addWidget(self.settingsView)
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(self.bwdButton)
hbox.addWidget(self.startButton)
hbox.addWidget(self.fwdButton)
hbox.addWidget(self.sliderTime)
hbox.addWidget(self.spinTime)
hbox.addWidget(self.centerButton)
hbox.addWidget(self.rotateButton)
hbox.addWidget(self.checkKey)
hbox.addWidget(self.screenshotButton)
hbox.addWidget(self.fileOpenButton)
hbox.addWidget(self.fileSaveButton)
hbox.addSpacing(50)
hbox.addWidget(self.checkIsoView)
hbox.addWidget(self.checkSliceView)
hbox.addSpacing(10)
hbox.addWidget(self.checkProcView)
hbox.addSpacing(10)
hbox.addWidget(self.checkVolSettings)
hbox.addSpacing(5)
hbox.addWidget(self.checkSettings)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(hbox0)
vbox.addLayout(hbox)
vbox.addWidget(self.keyPanel)
for box in [hbox,vbox,hbox0]:
box.setContentsMargins(0,0,0,0)
vbox.setSpacing(1)
hbox.setSpacing(11)
hbox0.setSpacing(5)
self.egg3d = spimagine.utils.egg3d.Egg3dController()
# widget = QtWidgets.QWidget()
self.setLayout(vbox)
self.rotateTimer = QtCore.QTimer(self)
self.rotateTimer.setInterval(70)
self.rotateTimer.timeout.connect(self.onRotateTimer)
self.playTimer = QtCore.QTimer(self)
self.playTimer.setInterval(100)
self.playTimer.timeout.connect(self.onPlayTimer)
self.settingsView._playIntervalChanged.connect(self.playIntervalChanged)
self.setLoopBounce(True)
self.playDir = 1
# a decorator as checkboxe state is 2 if checked
def stateToBool(objFunc, invert = False):
def _foo(x):
if invert:
return objFunc(x!=2)
else:
return objFunc(x==2)
return _foo
self.volSettingsView.checkBox.stateChanged.connect(
self.glWidget.transform.setBox)
self.volSettingsView.checkInterpolate.stateChanged.connect(
self.glWidget.transform.setInterpolate)
self.transform._interpChanged.connect(self.volSettingsView.checkInterpolate.setChecked)
self.transform._interpChanged.connect(self.glWidget.set_interpolation)
self.volSettingsView.checkInvert.stateChanged.connect(
stateToBool(self.glWidget.set_background_mode_black,invert = True))
self.settingsView._substepsChanged.connect(self.substepsChanged)
self.checkIsoView.stateChanged.connect(
stateToBool(self.glWidget.transform.setIso))
self.transform._isoChanged.connect(self.checkIsoView.setChecked)
self.settingsView.checkEgg.stateChanged.connect(self.onCheckEgg)
self.volSettingsView._boundsChanged.connect(self.glWidget.transform.setBounds)
self.volSettingsView.sliderAlphaPow.floatValueChanged.connect(self.glWidget.transform.setAlphaPow)
self.volSettingsView.sliderOcc.floatValueChanged.connect(
self.glWidget.transform.setOccStrength)
self.volSettingsView.sliderOccRadius.floatValueChanged.connect(
self.glWidget.transform.setOccRadius)
self.volSettingsView.sliderOccNPoints.floatValueChanged.connect(
self.glWidget.transform.setOccNPoints)
self.glWidget.transform._alphaPowChanged.connect(self.volSettingsView.sliderAlphaPow.setValue)
self.glWidget.transform._boundsChanged.connect(self.volSettingsView.setBounds)
self.transform._boxChanged.connect(self.volSettingsView.checkBox.setChecked)
self.volSettingsView.checkProj.stateChanged.connect(self.transform.setPerspective)
self.transform._perspectiveChanged.connect(self.volSettingsView.checkProj.setChecked)
self.checkKey.stateChanged.connect(self.keyPanel.setVisible)
self.checkVolSettings.stateChanged.connect(self.volSettingsView.setVisible)
self.checkSettings.stateChanged.connect(self.settingsView.setVisible)
self.checkSliceView.stateChanged.connect(self.sliceWidget.setVisible)
self.checkSliceView.stateChanged.connect(self.transform.setShowSlice)
self.checkProcView.stateChanged.connect(self.impListView.setVisible)
self.settingsView.checkLoopBounce.stateChanged.connect(self.setLoopBounce)
self.volSettingsView._stackUnitsChanged.connect(self.transform.setStackUnits)
self.transform._stackUnitsChanged.connect(self.volSettingsView.setStackUnits)
self.settingsView._frameNumberChanged.connect(self.keyPanel.setFrameNumber)
self.settingsView._recordDelayChanged.connect(self.keyPanel.setRecordDelay)
self.volSettingsView.colorCombo.currentIndexChanged.connect(self.onColormapChanged)
self.volSettingsView._rgbColorChanged.connect(self.onRgbColorChanged)
self.impListView._stateChanged.connect(self.impStateChanged)
self.settingsView._dirNameChanged.connect(self.keyPanel.setDirName)
# dataModel._dataSourceChanged.connect(self.dataSourceChanged)
# dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.glWidget._dataModelChanged.connect(self.dataModelChanged)
self.onColormapChanged(0)
self.checkSliceView.setChecked(False)
self.hiddableControls = [self.checkSettings,self.checkVolSettings,
self.startButton,self.sliderTime,self.spinTime,
self.checkKey,self.screenshotButton ]
# self.keyPanel.keyView.setModel(self.keyframes)
self.setStyleSheet("""
background-color:black;
color:white;""")
class MainWidget(QtWidgets.QWidget):
N_SCALE_MIN_EXP = -16
N_SCALE_MAX_EXP = 16
N_SCALE_SLIDER = 500
def __init__(self, parent = None, interpolation = "linear", **kwargs):
super(QtWidgets.QWidget,self).__init__(parent)
self.myparent = parent
self.isCloseFlag = False
self.setWindowTitle('spimagine')
self.resize(900, 700)
self.transform = TransformModel()
self.initActions()
self.glWidget = GLWidget(self,interpolation = interpolation)
self.glWidget.setTransform(self.transform)
self.sliceWidget = SliceWidget(self, interpolation = interpolation)
self.sliceWidget.hide()
# self.jack = JackPlugin(self.transform)
# self.jack.start()
self.sliceWidget.setTransform(self.transform)
self.fwdButton = createStandardButton(self,
fName = absPath("images/icon_forward.png"),
method = self.forward, width = 18, tooltip="forward")
self.bwdButton = createStandardButton(self,
fName = absPath("images/icon_backward.png"),
method = self.backward, width = 18, tooltip="backward")
self.startButton = createStandardButton(self, fName = absPath("images/icon_start.png"),
method = self.startPlay, tooltip="play")
self.centerButton = createStandardButton(self,
fName = absPath("images/icon_center.png"),
method = self.center, tooltip = "center view")
self.rotateButton = createStandardButton(self,
fName = absPath("images/icon_rotate.png"),
method = self.rotate, tooltip = "spin current view")
self.screenshotButton = createStandardButton(self,
fName = absPath("images/icon_camera.png"),
method = self.screenShot,
tooltip = "save as png")
self.fileOpenButton = createStandardButton(self,
fName = absPath("images/icon_open.png"),
method = self.openFile, tooltip = "open file")
self.fileSaveButton = createStandardButton(self,
fName = absPath("images/icon_filesave.png"),
method = self.saveFile, tooltip = "save file as tif")
self.checkVolSettings = createImageCheckbox(self,
absPath("images/icon_volsettings_active.png"),
absPath("images/icon_volsettings_inactive.png"),
tooltip="volume settings")
self.checkSettings = createImageCheckbox(self,
absPath("images/icon_mainsettings_active.png"),
absPath("images/icon_mainsettings_inactive.png"),
tooltip="general settings")
self.checkKey = createImageCheckbox(self, absPath("images/video.png"), absPath("images/video_inactive.png"), tooltip="keyframe editor")
self.checkIsoView = createImageCheckbox(self,
absPath("images/icon_method_vol.png"),
absPath("images/icon_method_iso.png"),
tooltip="iso surface")
self.checkProcView = createImageCheckbox(self,
absPath("images/icon_process_active.png"),
absPath("images/icon_process_inactive.png"),
tooltip="image processors")
self.checkSliceView = createImageCheckbox(
self,
absPath("images/icon_slice.png"),
absPath("images/icon_slice_inactive.png"), tooltip="slice view")
self.sliderTime = QtWidgets.QSlider(QtCore.Qt.Horizontal)
self.sliderTime.setTickPosition(QtWidgets.QSlider.TicksBothSides)
self.sliderTime.setTickInterval(1)
self.sliderTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.sliderTime.setTracking(False)
self.spinTime = QtWidgets.QSpinBox()
self.spinTime.setStyleSheet("color:white;border:0px solid black;background-color:black;")
self.spinTime.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
self.spinTime.setFocusPolicy(QtCore.Qt.ClickFocus)
self.spinTime.setAttribute(QtCore.Qt.WA_MacShowFocusRect, 0)
self.sliderTime.valueChanged.connect(self.spinTime.setValue)
self.spinTime.valueChanged.connect(self.sliderTime.setValue)
self.maxSlider = FloatSlider(QtCore.Qt.Vertical)
self.maxSlider.setRange(self.N_SCALE_MIN_EXP, self.N_SCALE_MAX_EXP, 1000)
self.maxSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.maxSlider.setToolTip("max")
self.minSlider = FloatSlider(QtCore.Qt.Vertical)
self.minSlider.setRange(self.N_SCALE_MIN_EXP,self.N_SCALE_MAX_EXP,1000)
self.minSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.minSlider.setToolTip("min")
self.gammaSlider = FloatSlider(QtCore.Qt.Vertical)
self.gammaSlider.setRange(.01,2.,200)
self.gammaSlider.setToolTip("gamma")
self.gammaSlider.setFocusPolicy(QtCore.Qt.ClickFocus)
self.gammaSlider.setValue(1.)
# def func_from_n(n):
# return 2**(self.N_SCALE_MIN_EXP+(self.N_SCALE_MAX_EXP-self.N_SCALE_MIN_EXP)*(n-1.)/(self.N_SCALE_SLIDER-1))
# def func_to_n(x):
# if x<2**self.N_SCALE_MIN_EXP:
# print "gg", x
# return 1
# elif x>2**self.N_SCALE_MAX_EXP:
# return self.N_SCALE_SLIDER
# return int(round(1.+(self.N_SCALE_SLIDER-1.)*(np.log2(x)-self.N_SCALE_MIN_EXP)/(self.N_SCALE_MAX_EXP-1.*self.N_SCALE_MIN_EXP)))
# self.scaleSlider.valueChanged.connect(lambda x: self.transform.setValueScale(0,func_from_n(x)))
# self.transform._maxChanged.connect(lambda x: self.scaleSlider.setValue(func_to_n(x)))
def func1(x):
#print("func2: %.2f \t->\t %.2f" % (x, 2**x))
return 2**x
def func2(x):
#print("func2: %.2f \t->\t %.2f"%(x,np.log2(x) if x>0 else -1.e20))
return np.log2(x) if x>0 else -1.e20
self.maxSlider.floatValueChanged.connect(lambda x: self.transform.setMax(func1(x)))
self.transform._maxChanged.connect(lambda x:self.maxSlider.setValue(func2(x)))
self.minSlider.floatValueChanged.connect(lambda x: self.transform.setMin(func1(x)))
self.transform._minChanged.connect(lambda x:self.minSlider.setValue(func2(x)))
self.gammaSlider.floatValueChanged.connect(self.transform.setGamma)
self.transform._gammaChanged.connect(self.gammaSlider.setValue)
# self.keyframes = KeyFrameList()
self.keyPanel = KeyFramePanel(self.glWidget)
self.keyPanel.hide()
self.keyPanel.setStyleSheet("color:white")
self.impListView = ImageProcessorListView([BlurProcessor(),
BlurXYZProcessor(),
NoiseProcessor(),
FFTProcessor(),
#LucyRichProcessor()
])
self.impListView.hide()
self.volSettingsView = VolumeSettingsPanel()
self.volSettingsView.hide()
self.settingsView = MainSettingsPanel()
self.settingsView.hide()
hbox0 = QtWidgets.QHBoxLayout()
hbox0.addWidget(self.minSlider)
hbox0.addWidget(self.maxSlider)
hbox0.addWidget(self.gammaSlider)
hbox0.addWidget(self.glWidget,stretch = 3)
hbox0.addWidget(self.sliceWidget,stretch =2)
hbox0.addWidget(self.impListView)
hbox0.addWidget(self.volSettingsView)
hbox0.addWidget(self.settingsView)
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(self.bwdButton)
hbox.addWidget(self.startButton)
hbox.addWidget(self.fwdButton)
hbox.addWidget(self.sliderTime)
hbox.addWidget(self.spinTime)
hbox.addWidget(self.centerButton)
hbox.addWidget(self.rotateButton)
hbox.addWidget(self.checkKey)
hbox.addWidget(self.screenshotButton)
hbox.addWidget(self.fileOpenButton)
hbox.addWidget(self.fileSaveButton)
hbox.addSpacing(50)
hbox.addWidget(self.checkIsoView)
hbox.addWidget(self.checkSliceView)
hbox.addSpacing(10)
hbox.addWidget(self.checkProcView)
hbox.addSpacing(10)
hbox.addWidget(self.checkVolSettings)
hbox.addSpacing(5)
hbox.addWidget(self.checkSettings)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(hbox0)
vbox.addLayout(hbox)
vbox.addWidget(self.keyPanel)
for box in [hbox,vbox,hbox0]:
box.setContentsMargins(0,0,0,0)
vbox.setSpacing(1)
hbox.setSpacing(11)
hbox0.setSpacing(5)
self.egg3d = spimagine.utils.egg3d.Egg3dController()
# widget = QtWidgets.QWidget()
self.setLayout(vbox)
self.rotateTimer = QtCore.QTimer(self)
self.rotateTimer.setInterval(70)
self.rotateTimer.timeout.connect(self.onRotateTimer)
self.playTimer = QtCore.QTimer(self)
self.playTimer.setInterval(100)
self.playTimer.timeout.connect(self.onPlayTimer)
self.settingsView._playIntervalChanged.connect(self.playIntervalChanged)
self.setLoopBounce(True)
self.playDir = 1
# a decorator as checkboxe state is 2 if checked
def stateToBool(objFunc, invert = False):
def _foo(x):
if invert:
return objFunc(x!=2)
else:
return objFunc(x==2)
return _foo
self.volSettingsView.checkBox.stateChanged.connect(
self.glWidget.transform.setBox)
self.volSettingsView.checkInterpolate.stateChanged.connect(
self.glWidget.transform.setInterpolate)
self.transform._interpChanged.connect(self.volSettingsView.checkInterpolate.setChecked)
self.transform._interpChanged.connect(self.glWidget.set_interpolation)
self.volSettingsView.checkInvert.stateChanged.connect(
stateToBool(self.glWidget.set_background_mode_black,invert = True))
self.settingsView._substepsChanged.connect(self.substepsChanged)
self.checkIsoView.stateChanged.connect(
stateToBool(self.glWidget.transform.setIso))
self.transform._isoChanged.connect(self.checkIsoView.setChecked)
self.settingsView.checkEgg.stateChanged.connect(self.onCheckEgg)
self.volSettingsView._boundsChanged.connect(self.glWidget.transform.setBounds)
self.volSettingsView.sliderAlphaPow.floatValueChanged.connect(self.glWidget.transform.setAlphaPow)
self.volSettingsView.sliderOcc.floatValueChanged.connect(
self.glWidget.transform.setOccStrength)
self.volSettingsView.sliderOccRadius.floatValueChanged.connect(
self.glWidget.transform.setOccRadius)
self.volSettingsView.sliderOccNPoints.floatValueChanged.connect(
self.glWidget.transform.setOccNPoints)
self.glWidget.transform._alphaPowChanged.connect(self.volSettingsView.sliderAlphaPow.setValue)
self.glWidget.transform._boundsChanged.connect(self.volSettingsView.setBounds)
self.transform._boxChanged.connect(self.volSettingsView.checkBox.setChecked)
self.volSettingsView.checkProj.stateChanged.connect(self.transform.setPerspective)
self.transform._perspectiveChanged.connect(self.volSettingsView.checkProj.setChecked)
self.checkKey.stateChanged.connect(self.keyPanel.setVisible)
self.checkVolSettings.stateChanged.connect(self.volSettingsView.setVisible)
self.checkSettings.stateChanged.connect(self.settingsView.setVisible)
self.checkSliceView.stateChanged.connect(self.sliceWidget.setVisible)
self.checkSliceView.stateChanged.connect(self.transform.setShowSlice)
self.checkProcView.stateChanged.connect(self.impListView.setVisible)
self.settingsView.checkLoopBounce.stateChanged.connect(self.setLoopBounce)
self.volSettingsView._stackUnitsChanged.connect(self.transform.setStackUnits)
self.transform._stackUnitsChanged.connect(self.volSettingsView.setStackUnits)
self.settingsView._frameNumberChanged.connect(self.keyPanel.setFrameNumber)
self.settingsView._recordDelayChanged.connect(self.keyPanel.setRecordDelay)
self.volSettingsView.colorCombo.currentIndexChanged.connect(self.onColormapChanged)
self.volSettingsView._rgbColorChanged.connect(self.onRgbColorChanged)
self.impListView._stateChanged.connect(self.impStateChanged)
self.settingsView._dirNameChanged.connect(self.keyPanel.setDirName)
# dataModel._dataSourceChanged.connect(self.dataSourceChanged)
# dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.glWidget._dataModelChanged.connect(self.dataModelChanged)
self.onColormapChanged(0)
self.checkSliceView.setChecked(False)
self.hiddableControls = [self.checkSettings,self.checkVolSettings,
self.startButton,self.sliderTime,self.spinTime,
self.checkKey,self.screenshotButton ]
# self.keyPanel.keyView.setModel(self.keyframes)
self.setStyleSheet("""
background-color:black;
color:white;""")
def impStateChanged(self):
data = self.transform.dataModel[self.transform.dataPos]
for imp in self.impListView.impViews:
if imp.is_active():
print("active: ", imp.proc.name, imp.proc.kwargs)
data = imp.proc.apply(data)
self.glWidget.renderer.update_data(data)
self.glWidget.refresh()
def onColormapChanged(self,index):
self.glWidget.set_colormap(self.volSettingsView.colormaps[index])
self.glWidget.refresh()
self.sliceWidget.glSliceWidget.set_colormap(self.volSettingsView.colormaps[index])
self.sliceWidget.glSliceWidget.refresh()
def onRgbColorChanged(self,r,g,b):
self.glWidget.set_colormap_rgb([r,g,b])
self.glWidget.refresh()
self.sliceWidget.glSliceWidget.set_colormap_rgb([r,g,b])
self.sliceWidget.glSliceWidget.refresh()
def onCheckEgg(self,state):
if state == QtCore.Qt.Checked:
self.connectEgg3d()
else:
self.egg3d.stop()
def connectEgg3d(self):
try:
self.egg3d.listener._quaternionChanged.connect(self.egg3dQuaternion)
self.egg3d.listener._zoomChanged.connect(self.egg3dZoom)
N = 45
self._quatHist = [Quaternion() for i in range(N)]
self._quatWeights = np.exp(-2.*np.linspace(0,1,N))
self._quatWeights *= 1./sum(self._quatWeights)
self.egg3d.start()
except Exception as e:
print(e)
self.settingsView.checkEgg.setCheckState(QtCore.Qt.Unchecked)
def egg3dQuaternion(self,a,b,c,d):
self._quatHist = np.roll(self._quatHist,1)
self._quatHist[0] = Quaternion(a,c,b,d)
q0 = Quaternion(0,0,0,0)
for q,w in zip(self._quatHist,self._quatWeights):
q0 = q0+q*w
self.transform.setQuaternion(q0)
def egg3dZoom(self,zoom):
if zoom>0:
newZoom = self.transform.zoom * 1.01**zoom
else:
newZoom = self.transform.zoom * 1.1**zoom
newZoom = np.clip(newZoom,.7,3)
self.transform.setZoom(newZoom)
def initUI(self):
pass
def raise_window(self):
#self.setWindowState(QtCore.Qt.WindowActive)
self.raise_()
def keyPressEvent(self, event):
if type(event) == QtGui.QKeyEvent:
if event.modifiers()== QtCore.Qt.ControlModifier and event.key() == QtCore.Qt.Key_Q:
return
# super(MainWidget,self).keyPressEvent(event)
def initActions(self):
QtWidgets.QShortcut(QtGui.QKeySequence("Ctrl+W"), self, self.closeMe)
# self.exitAction = QtWidgets.QAction('Quit', self)
# self.exitAction.setShortcut('Ctrl+Q')
# self.exitAction.setStatusTip('Exit application')
# self.exitAction.triggered.connect(self.foo)
def setModel(self,dataModel):
self.glWidget.setModel(dataModel)
self.sliceWidget.setModel(dataModel)
def dataModelChanged(self):
logger.debug("data Model changed")
dataModel = self.glWidget.dataModel
dataModel._dataSourceChanged.connect(self.dataSourceChanged)
dataModel._dataPosChanged.connect(self.sliderTime.setValue)
self.sliderTime.valueChanged.connect(self.transform.setPos)
kvList = KeyFrameList()
kvList.addItem(KeyFrame(0.))
kvList.addItem(KeyFrame(1.))
self.keyPanel.connect_to_transform(self.transform)
self.keyPanel.setModel(kvList)
self.dataSourceChanged()
def dataSourceChanged(self):
self.sliderTime.setRange(0,self.glWidget.dataModel.sizeT()-1)
self.sliderTime.setValue(0)
self.spinTime.setRange(0,self.glWidget.dataModel.sizeT()-1)
self.volSettingsView.dimensionLabel.setText("Dim: %s"%str(tuple(self.glWidget.dataModel.size()[::-1])))
if self.myparent:
self.myparent.setWindowTitle(self.glWidget.dataModel.name())
else:
self.setWindowTitle(self.glWidget.dataModel.name())
self.keyPanel.connect_to_transform(self.transform)
d = self.glWidget.dataModel[self.glWidget.dataModel.pos]
# minMaxMean = (np.amin(d),np.amax(d),np.mean(d))
# self.volSettingsView.statsLabel.setText("Min:\t%.2f\nMax:\t%.2f \nMean:\t%.2f"%minMaxMean)
def forward(self,event):
if self.glWidget.dataModel:
newpos = (self.glWidget.dataModel.pos+1)%self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def backward(self,event):
if self.glWidget.dataModel:
newpos = (self.glWidget.dataModel.pos-1)%self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def startPlay(self,event):
if self.playTimer.isActive():
self.playTimer.stop()
self.startButton.setIcon(QtGui.QIcon(absPath("images/icon_start.png")))
else:
self.playTimer.start()
self.startButton.setIcon(QtGui.QIcon(absPath("images/icon_pause.png")))
def screenShot(self):
fileName, _ = QtWidgets.QFileDialog.getSaveFileName(self, 'Save screenshot as',
'.', ('*.png'))
if fileName:
self.glWidget.saveFrame(str(fileName))
def set_colormap(self, name):
self.glWidget.set_colormap(name)
def set_colormap_rgb(self, color):
self.glWidget.set_colormap_rgb(color)
def saveFrame(self, fName, with_alpha = False):
self.glWidget.saveFrame(fName, with_alpha = with_alpha)
def setLoopBounce(self,loopBounce):
#if loopBounce = True, then while playing it should loop back and forth
self.loopBounce = loopBounce
self.settingsView.checkLoopBounce.setChecked(loopBounce)
def playIntervalChanged(self,val):
if self.playTimer.isActive():
self.playTimer.stop()
self.playTimer.setInterval(val)
def substepsChanged(self,val):
self.glWidget.NSubrenderSteps = val
self.glWidget.refresh()
def openFile(self,e):
# path = QtWidgets.QFileDialog.getOpenFileName(self, 'Open Path (File or Folder)',
# '.', selectedFilter='*.tif')
f = QtWidgets.QFileDialog()
f.setWindowTitle('Open Path (File or Folder)')
f.setFileMode(QtWidgets.QFileDialog.ExistingFile & QtWidgets.QFileDialog.Directory)
f.exec_()
path = f.selectedFiles()
if len(path)==0:
return
path = str(path[0])
if path:
try:
if self.glWidget.dataModel:
self.glWidget.dataModel.loadFromPath(path,
prefetchSize = self.glWidget.N_PREFETCH)
else:
self.glWidget.setModel(DataModel.fromPath(path,
prefetchSize = self.glWidget.N_PREFETCH))
except Exception as e:
mbox = QtWidgets.QMessageBox()
mbox.setText(str(e))
mbox.setIcon(QtWidgets.QMessageBox.Warning)
mbox.exec_()
def saveFile(self,e):
path, _ = QtWidgets.QFileDialog.getSaveFileName(self, 'Save as Tif File',
'.', ('*.tif'))
if path:
if self.glWidget.dataModel:
write3dTiff(self.glWidget.dataModel[self.transform.dataPos].astype(np.float32),path)
def onPlayTimer(self):
if self.glWidget.dataModel:
if self.glWidget.dataModel.pos == self.glWidget.dataModel.sizeT()-1:
self.playDir = 1-2*self.loopBounce
if self.glWidget.dataModel.pos == 0:
self.playDir = 1
newpos = (self.glWidget.dataModel.pos+self.playDir)%self.glWidget.dataModel.sizeT()
self.transform.setPos(newpos)
def contextMenuEvent(self,event):
# create context menu
popMenu = QtWidgets.QMenu(self)
action = QtWidgets.QAction('toggle controls', self)
action.triggered.connect(self.toggleControls)
popMenu.addAction(action)
popMenu.setStyleSheet("background-color: white")
# popMenu.exec_(QtGui.QCursor.pos())
def toggleControls(self):
for c in self.hiddableControls:
c.setVisible(not c.isVisible())
def closeMe(self):
#little workaround as on MAC ctrl-q cannot be overwritten
self.isCloseFlag = True
self.close()
def closeEvent(self,event):
logger.debug("closeevent")
if not event.spontaneous() and not self.isCloseFlag:
event.ignore()
else:
if self.playTimer.isActive():
self.playTimer.stop()
if self.keyPanel.recordTimer.isActive():
self.keyPanel.recordTimer.stop()
if self.keyPanel.playTimer.isActive():
self.keyPanel.playTimer.stop()
if self.rotateTimer.isActive():
self.rotateTimer.stop()
if hasattr(self, "glWidget"):
logger.debug("deleting the renderer")
try:
self.glWidget.renderTimer.stop()
del self.glWidget.renderer
self.glWidget.setParent(None)
del self.glWidget
except:
pass
logger.debug("....finished deleting the renderer")
event.accept()
def center(self):
self.transform.center()
def rotate(self):
if self.rotateTimer.isActive():
self.rotateTimer.stop()
self.rotateButton.setIcon(QtGui.QIcon(absPath("images/icon_rotate.png")))
else:
self.rotateTimer.start()
self.rotateButton.setIcon(QtGui.QIcon(absPath("images/icon_rotate_active.png")))
def onRotateTimer(self):
axis = [0,0,0]
axis[spimagine.config.__DEFAULT_SPIN_AXIS__] = 1
self.transform.addRotation(-.02,*axis)
self.glWidget.render()
self.glWidget.updateGL()
def _show_fullscreen(self):
self.show()
self.showFullScreen()
def _show_normal(self):
self.show()
self.showNormal()
def toggle_fullscreen(self):
if self.isFullScreen():
QtCore.QTimer.singleShot(200, self._show_normal)
else:
QtCore.QTimer.singleShot(200, self._show_fullscreen)
def mouseDoubleClickEvent(self,event):
super(MainWidget,self).mouseDoubleClickEvent(event)
self.toggle_fullscreen()
self.glWidget.resized = True
# there's a bug in Qt that disables drop after fullscreen, so reset it...
self.setAcceptDrops(True)
def moveEvent(self, evt):
if hasattr(self, "glWidget"):
self.glWidget.moveEvent(evt)