class EMInspectorControlScatterPlot(EMInspectorControlShape):
"""
Class to make scatter plot 3D inspector
"""
def __init__(self, name, item3d):
EMInspectorControlShape.__init__(self, name, item3d)
def updateItemControls(self):
""" Updates this item inspector. Function is called by the item it observes"""
super(EMInspectorControlScatterPlot, self).updateItemControls()
self.pointsize.setValue(self.item3d().getPointSize())
def addTabs(self):
""" Add a tab for each 'column' """
super(EMInspectorControlScatterPlot, self).addTabs()
tabwidget = QtGui.QWidget()
gridbox = QtGui.QGridLayout()
EMInspectorControlScatterPlot.addControls(self, gridbox)
tabwidget.setLayout(gridbox)
self.addTab(tabwidget, "scatterplot")
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
scatterframe = QtGui.QFrame()
scatterframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
scattergridbox = QtGui.QGridLayout()
scattergridbox.setAlignment(QtCore.Qt.AlignTop)
# Add widgets to frame
pslabel = QtGui.QLabel("Point Size")
pslabel.setFont(lfont)
pslabel.setAlignment(QtCore.Qt.AlignCenter)
scattergridbox.addWidget(pslabel, 0, 0, 1, 1)
self.pointsize = EMSpinWidget(int(self.item3d().getPointSize()), 1.0, rounding=0)
self.pointsize.setMinimumWidth(120)
scattergridbox.addWidget(self.pointsize, 1, 0, 1, 1)
scatterframe.setLayout(scattergridbox)
gridbox.addWidget(scatterframe, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControlScatterPlot:
self.updateItemControls()
QtCore.QObject.connect(self.pointsize,QtCore.SIGNAL("valueChanged(int)"),self.onPointSizeChanged)
def onPointSizeChanged(self):
self.item3d().setPointSize(self.pointsize.getValue())
if self.inspector: self.inspector().updateSceneGraph()
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
scatterframe = QtGui.QFrame()
scatterframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
scattergridbox = QtGui.QGridLayout()
scattergridbox.setAlignment(QtCore.Qt.AlignTop)
# Add widgets to frame
pslabel = QtGui.QLabel("Point Size")
pslabel.setFont(lfont)
pslabel.setAlignment(QtCore.Qt.AlignCenter)
scattergridbox.addWidget(pslabel, 0, 0, 1, 1)
self.pointsize = EMSpinWidget(int(self.item3d().getPointSize()), 1.0, rounding=0)
self.pointsize.setMinimumWidth(120)
scattergridbox.addWidget(self.pointsize, 1, 0, 1, 1)
scatterframe.setLayout(scattergridbox)
gridbox.addWidget(scatterframe, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControlScatterPlot:
self.updateItemControls()
QtCore.QObject.connect(self.pointsize,QtCore.SIGNAL("valueChanged(int)"),self.onPointSizeChanged)
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
# selection box and label
font = QtGui.QFont()
font.setBold(True)
label = QtGui.QLabel(self.name, self)
label.setFont(font)
label.setAlignment(QtCore.Qt.AlignCenter)
gridbox.addWidget(label, 0, 0, 1, 1)
databox = QtGui.QHBoxLayout()
self.boundingbox = None
if self.item3d().boundingboxsize:
self.boundingbox = QtGui.QLabel(
"Size: " + self.item3d().boundingboxsize, self)
databox.addWidget(self.boundingbox)
gridbox.addLayout(databox, 1, 0, 1, 1)
# angluar controls
xformframe = QtGui.QFrame()
xformframe.setFrameShape(QtGui.QFrame.StyledPanel)
xformbox = QtGui.QGridLayout()
xformlabel = QtGui.QLabel("Transformation", xformframe)
xformlabel.setFont(font)
xformlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(xformlabel, 0, 0, 1, 2)
# Rotations
self.rotcombobox = QtGui.QComboBox()
xformbox.addWidget(self.rotcombobox, 1, 0, 1, 2)
self.rotstackedwidget = QtGui.QStackedWidget()
self.addRotationWidgets()
xformbox.addWidget(self.rotstackedwidget, 2, 0, 1, 2)
#translations
txlabel = QtGui.QLabel("TX", xformframe)
txlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(txlabel, 3, 0, 1, 1)
tylabel = QtGui.QLabel("TY", xformframe)
tylabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tylabel, 3, 1, 1, 1)
self.tx = EMSpinWidget(0.0, 1.0)
self.ty = EMSpinWidget(0.0, 1.0)
xformbox.addWidget(self.tx, 4, 0, 1, 1)
xformbox.addWidget(self.ty, 4, 1, 1, 1)
tzlabel = QtGui.QLabel("TZ", xformframe)
tzlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tzlabel, 5, 0, 1, 1)
zoomlabel = QtGui.QLabel("Zoom", xformframe)
zoomlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(zoomlabel, 5, 1, 1, 1)
self.tz = EMSpinWidget(0.0, 1.0)
self.zoom = EMSpinWidget(1.0, 0.1, postivemode=True, wheelstep=0.1)
xformbox.addWidget(self.tz, 6, 0, 1, 1)
xformbox.addWidget(self.zoom, 6, 1, 1, 1)
self.resetbuttontx = QtGui.QPushButton("Reset Tx")
self.resetbuttonrot = QtGui.QPushButton("Reset Rot")
xformbox.addWidget(self.resetbuttontx, 7, 0, 1, 1)
xformbox.addWidget(self.resetbuttonrot, 7, 1, 1, 1)
xformframe.setLayout(xformbox)
xformframe.setMaximumWidth(350)
xformframe.setMaximumHeight(self.transfromboxmaxheight)
xformframe.setLayout(xformbox)
gridbox.addWidget(xformframe, 2, 0, 1, 1)
# set to default, but run only as a base class
if type(self) == EMItem3DInspector: self.updateItemControls()
QtCore.QObject.connect(self.tx, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.ty, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.tz, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.zoom, QtCore.SIGNAL("valueChanged(int)"),
self._on_scale)
QtCore.QObject.connect(self.resetbuttontx, QtCore.SIGNAL("clicked()"),
self._on_resettx)
QtCore.QObject.connect(self.resetbuttonrot, QtCore.SIGNAL("clicked()"),
self._on_resetrot)
class EMItem3DInspector(QtGui.QTabWidget):
"""
Class to make the EMItem GUI controls
"""
def __init__(self, name, item3d):
QtGui.QTabWidget.__init__(self)
self.item3d = weakref.ref(item3d)
self.name = name
self.inspector = None
self.transfromboxmaxheight = 400 # This might be problematic
self.addTabs()
def setInspector(self, inspector):
""" This is a reference back to the main inspector, which holds all the item inspectors"""
self.inspector = weakref.ref(inspector)
def addTabs(self):
""" Add a tab for each 'column' """
tabwidget = QtGui.QWidget()
gridbox = QtGui.QGridLayout()
EMItem3DInspector.addControls(self, gridbox)
tabwidget.setLayout(gridbox)
self.addTab(tabwidget, "basic")
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
# selection box and label
font = QtGui.QFont()
font.setBold(True)
label = QtGui.QLabel(self.name, self)
label.setFont(font)
label.setAlignment(QtCore.Qt.AlignCenter)
gridbox.addWidget(label, 0, 0, 1, 1)
databox = QtGui.QHBoxLayout()
self.boundingbox = None
if self.item3d().boundingboxsize:
self.boundingbox = QtGui.QLabel(
"Size: " + self.item3d().boundingboxsize, self)
databox.addWidget(self.boundingbox)
gridbox.addLayout(databox, 1, 0, 1, 1)
# angluar controls
xformframe = QtGui.QFrame()
xformframe.setFrameShape(QtGui.QFrame.StyledPanel)
xformbox = QtGui.QGridLayout()
xformlabel = QtGui.QLabel("Transformation", xformframe)
xformlabel.setFont(font)
xformlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(xformlabel, 0, 0, 1, 2)
# Rotations
self.rotcombobox = QtGui.QComboBox()
xformbox.addWidget(self.rotcombobox, 1, 0, 1, 2)
self.rotstackedwidget = QtGui.QStackedWidget()
self.addRotationWidgets()
xformbox.addWidget(self.rotstackedwidget, 2, 0, 1, 2)
#translations
txlabel = QtGui.QLabel("TX", xformframe)
txlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(txlabel, 3, 0, 1, 1)
tylabel = QtGui.QLabel("TY", xformframe)
tylabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tylabel, 3, 1, 1, 1)
self.tx = EMSpinWidget(0.0, 1.0)
self.ty = EMSpinWidget(0.0, 1.0)
xformbox.addWidget(self.tx, 4, 0, 1, 1)
xformbox.addWidget(self.ty, 4, 1, 1, 1)
tzlabel = QtGui.QLabel("TZ", xformframe)
tzlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tzlabel, 5, 0, 1, 1)
zoomlabel = QtGui.QLabel("Zoom", xformframe)
zoomlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(zoomlabel, 5, 1, 1, 1)
self.tz = EMSpinWidget(0.0, 1.0)
self.zoom = EMSpinWidget(1.0, 0.1, postivemode=True, wheelstep=0.1)
xformbox.addWidget(self.tz, 6, 0, 1, 1)
xformbox.addWidget(self.zoom, 6, 1, 1, 1)
self.resetbuttontx = QtGui.QPushButton("Reset Tx")
self.resetbuttonrot = QtGui.QPushButton("Reset Rot")
xformbox.addWidget(self.resetbuttontx, 7, 0, 1, 1)
xformbox.addWidget(self.resetbuttonrot, 7, 1, 1, 1)
xformframe.setLayout(xformbox)
xformframe.setMaximumWidth(350)
xformframe.setMaximumHeight(self.transfromboxmaxheight)
xformframe.setLayout(xformbox)
gridbox.addWidget(xformframe, 2, 0, 1, 1)
# set to default, but run only as a base class
if type(self) == EMItem3DInspector: self.updateItemControls()
QtCore.QObject.connect(self.tx, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.ty, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.tz, QtCore.SIGNAL("valueChanged(int)"),
self._on_translation)
QtCore.QObject.connect(self.zoom, QtCore.SIGNAL("valueChanged(int)"),
self._on_scale)
QtCore.QObject.connect(self.resetbuttontx, QtCore.SIGNAL("clicked()"),
self._on_resettx)
QtCore.QObject.connect(self.resetbuttonrot, QtCore.SIGNAL("clicked()"),
self._on_resetrot)
def _on_translation(self, value):
"""
Need to contain the right coords. And do translation in the correct corrd system
"""
tt = t = Transform({
"tx": self.tx.getValue(),
"ty": self.ty.getValue(),
"tz": self.tz.getValue()
})
tp = self.item3d().getParentMatrixProduct()
if tp: tt = tp.inverse() * t
self.item3d().getTransform().set_trans(tt.get_trans())
self.inspector().updateSceneGraph()
def _on_scale(self, value):
self.item3d().getTransform().set_scale(self.zoom.getValue())
self.inspector().updateSceneGraph()
def _on_resettx(self):
self.item3d().getTransform().set_trans(0.0, 0.0, 0.0)
self.updateItemControls()
self.inspector().updateSceneGraph()
def _on_resetrot(self):
self.item3d().getTransform().set_rotation({
"type": "eman",
"az": 0.0,
"alt": 0.0,
"phi": 0.0
})
self.updateItemControls()
self.inspector().updateSceneGraph()
def _isRotNaN(self, rot1, rot2, rot3):
""" Better check to make sure get_rotation did not return Nan, so to prevent a crash """
if rot1 != rot1: return True
if rot2 != rot2: return True
if rot3 != rot3: return True
return False
def updateItemControls(self):
""" Updates this item inspector. Function is called by the item it observes"""
# Translation update
stdtransfrom = self.item3d().getTransformStdCoord()
translation = stdtransfrom.get_trans()
self.tx.setValue(translation[0])
self.ty.setValue(translation[1])
self.tz.setValue(translation[2])
# Rotation update
rotation = stdtransfrom.get_rotation(
str(self.rotcombobox.currentText()))
is_identity = stdtransfrom.is_rot_identity()
comboboxidx = self.rotcombobox.currentIndex()
if comboboxidx == 0:
if self._isRotNaN(rotation["az"], rotation["alt"],
rotation["phi"]):
return
self.emanazslider.setValue(rotation["az"], quiet=1)
self.emanaltslider.setValue(rotation["alt"], quiet=1)
self.emanphislider.setValue(rotation["phi"], quiet=1)
if comboboxidx == 1:
if self._isRotNaN(rotation["gamma"], rotation["beta"],
rotation["alpha"]):
return
self.imagicgammaslider.setValue(rotation["gamma"], quiet=1)
self.imagicbetaslider.setValue(rotation["beta"], quiet=1)
self.imagicalphaslider.setValue(rotation["alpha"], quiet=1)
if comboboxidx == 2:
if self._isRotNaN(rotation["psi"], rotation["theta"],
rotation["phi"]):
return
self.spiderpsislider.setValue(rotation["psi"], quiet=1)
self.spiderthetaslider.setValue(rotation["theta"], quiet=1)
self.spiderphislider.setValue(rotation["phi"], quiet=1)
if comboboxidx == 3:
if self._isRotNaN(rotation["phi"], rotation["theta"],
rotation["omega"]):
return
self.mrcpsislider.setValue(rotation["phi"], quiet=1)
self.mrcthetaslider.setValue(rotation["theta"], quiet=1)
self.mrcomegaslider.setValue(rotation["omega"], quiet=1)
if comboboxidx == 4:
if self._isRotNaN(rotation["ztilt"], rotation["ytilt"],
rotation["xtilt"]):
return
self.xyzzslider.setValue(rotation["ztilt"], quiet=1)
self.xyzyslider.setValue(rotation["ytilt"], quiet=1)
self.xyzxslider.setValue(rotation["xtilt"], quiet=1)
if comboboxidx == 5:
if self._isRotNaN(rotation["n1"], rotation["n2"], rotation["n3"]):
return
if is_identity and self.spinn1slider.getValue(
) == 0.0 and self.spinn2slider.getValue(
) == 0.0 and self.spinn3slider.getValue() == 0.0:
self.spinomegaslider.setValue(0.0, quiet=1)
self.spinn1slider.setValue(0.0, quiet=1)
self.spinn2slider.setValue(0.0, quiet=1)
self.spinn3slider.setValue(1.0, quiet=1)
else:
self.spinomegaslider.setValue(rotation["omega"], quiet=1)
# Don't change slider if reult is Nan
if rotation["n1"] == rotation["n1"]:
self.spinn1slider.setValue(rotation["n1"], quiet=1)
if rotation["n2"] == rotation["n2"]:
self.spinn2slider.setValue(rotation["n2"], quiet=1)
if rotation["n3"] == rotation["n3"]:
self.spinn3slider.setValue(rotation["n3"], quiet=1)
if comboboxidx == 6:
if self._isRotNaN(rotation["n1"], rotation["n2"], rotation["n3"]):
return
if is_identity and self.spinn1slider.getValue(
) == 0.0 and self.spinn2slider.getValue(
) == 0.0 and self.spinn3slider.getValue() == 0.0:
self.spinomegaslider.setValue(0.0, quiet=1)
self.sgirotn1slider.setValue(0.0, quiet=1)
self.sgirotn2slider.setValue(0.0, quiet=1)
self.sgirotn3slider.setValue(1.0, quiet=1)
else:
self.spinomegaslider.setValue(rotation["q"], quiet=1)
# Don't change slider if reult is Nan
if rotation["n1"] == rotation["n1"]:
self.sgirotn1slider.setValue(rotation["n1"], quiet=1)
if rotation["n2"] == rotation["n2"]:
self.sgirotn2slider.setValue(rotation["n2"], quiet=1)
if rotation["n3"] == rotation["n3"]:
self.sgirotn3slider.setValue(rotation["n3"], quiet=1)
if comboboxidx == 7:
if self._isRotNaN(rotation["e1"], rotation["e2"], rotation["e3"]):
return
if is_identity:
self.quaternione0slider.setValue(1.0, quiet=1)
self.quaternione1slider.setValue(0.0, quiet=1)
self.quaternione2slider.setValue(0.0, quiet=1)
self.quaternione3slider.setValue(0.0, quiet=1)
else:
self.quaternione0slider.setValue(rotation["e0"], quiet=1)
self.quaternione1slider.setValue(rotation["e1"], quiet=1)
self.quaternione2slider.setValue(rotation["e2"], quiet=1)
self.quaternione3slider.setValue(rotation["e3"], quiet=1)
# Scaling update
self.zoom.setValue(self.item3d().getTransform().get_scale())
def updateMetaData(self):
"""
I didn't want to put this in update b/c this data doesn't change very often, and I don't want to waste CPU
Its a judgement call really, less coupling vs. more efficiency
"""
if self.boundingbox:
self.boundingbox.setText("Size: " + self.item3d().boundingboxsize)
def addRotationWidgets(self):
""" Add alll the widgets for the various EMAN2 rotation conventions """
EMANwidget = QtGui.QWidget()
Imagicwidget = QtGui.QWidget()
Spiderwidget = QtGui.QWidget()
MRCwidget = QtGui.QWidget()
XYZwidget = QtGui.QWidget()
spinwidget = QtGui.QWidget()
sgirotwidget = QtGui.QWidget()
quaternionwidget = QtGui.QWidget()
# EMAN
emanbox = QtGui.QVBoxLayout()
self.emanazslider = ValSlider(EMANwidget, (0.0, 360.0),
" Az",
rounding=1)
self.emanaltslider = ValSlider(EMANwidget, (0.0, 180.0),
"Alt",
rounding=1)
self.emanphislider = ValSlider(EMANwidget, (0.0, 360.0),
"Phi",
rounding=1)
emanbox.addWidget(self.emanazslider)
emanbox.addWidget(self.emanaltslider)
emanbox.addWidget(self.emanphislider)
EMANwidget.setLayout(emanbox)
# Imagic
imagicbox = QtGui.QVBoxLayout()
self.imagicgammaslider = ValSlider(Imagicwidget, (0.0, 360.0),
"Gamma",
rounding=1)
self.imagicbetaslider = ValSlider(Imagicwidget, (0.0, 180.0),
" Beta",
rounding=1)
self.imagicalphaslider = ValSlider(Imagicwidget, (0.0, 360.0),
" Alpha",
rounding=1)
imagicbox.addWidget(self.imagicgammaslider)
imagicbox.addWidget(self.imagicbetaslider)
imagicbox.addWidget(self.imagicalphaslider)
Imagicwidget.setLayout(imagicbox)
# Spider
spiderbox = QtGui.QVBoxLayout()
self.spiderpsislider = ValSlider(Spiderwidget, (0.0, 360.0),
" Psi",
rounding=1)
self.spiderthetaslider = ValSlider(Spiderwidget, (0.0, 180.0),
"Theta",
rounding=1)
self.spiderphislider = ValSlider(Spiderwidget, (0.0, 360.0),
" Phi",
rounding=1)
spiderbox.addWidget(self.spiderpsislider)
spiderbox.addWidget(self.spiderthetaslider)
spiderbox.addWidget(self.spiderphislider)
Spiderwidget.setLayout(spiderbox)
# MRC
mrcbox = QtGui.QVBoxLayout()
self.mrcpsislider = ValSlider(MRCwidget, (0.0, 360.0),
" Psi",
rounding=1)
self.mrcthetaslider = ValSlider(MRCwidget, (0.0, 180.0),
" Theta",
rounding=1)
self.mrcomegaslider = ValSlider(MRCwidget, (0.0, 360.0),
"Omega",
rounding=1)
mrcbox.addWidget(self.mrcpsislider)
mrcbox.addWidget(self.mrcthetaslider)
mrcbox.addWidget(self.mrcomegaslider)
MRCwidget.setLayout(mrcbox)
# XYZ
xyzbox = QtGui.QVBoxLayout()
self.xyzzslider = ValSlider(XYZwidget, (0.0, 360.0), "Z", rounding=1)
self.xyzyslider = ValSlider(XYZwidget, (0.0, 180.0), "Y", rounding=1)
self.xyzxslider = ValSlider(XYZwidget, (0.0, 360.0), "X", rounding=1)
xyzbox.addWidget(self.xyzzslider)
xyzbox.addWidget(self.xyzyslider)
xyzbox.addWidget(self.xyzxslider)
XYZwidget.setLayout(xyzbox)
# spin
spinbox = QtGui.QVBoxLayout()
self.spinomegaslider = ValSlider(spinwidget, (0.0, 180.0),
"Omega",
rounding=1)
self.spinn1slider = ValSlider(spinwidget, (0.0, 1.0),
" N1",
rounding=4)
self.spinn2slider = ValSlider(spinwidget, (0.0, 1.0),
" N2",
rounding=4)
self.spinn3slider = ValSlider(spinwidget, (0.0, 1.0),
" N3",
rounding=4)
spinbox.addWidget(self.spinomegaslider)
spinbox.addWidget(self.spinn1slider)
spinbox.addWidget(self.spinn2slider)
spinbox.addWidget(self.spinn3slider)
spinwidget.setLayout(spinbox)
# sgirot
sgirotbox = QtGui.QVBoxLayout()
self.sgirotqslider = ValSlider(sgirotwidget, (0.0, 180.0),
" Q",
rounding=1)
self.sgirotn1slider = ValSlider(sgirotwidget, (0.0, 1.0),
"N1",
rounding=4)
self.sgirotn2slider = ValSlider(sgirotwidget, (0.0, 1.0),
"N2",
rounding=4)
self.sgirotn3slider = ValSlider(sgirotwidget, (0.0, 1.0),
"N3",
rounding=4)
sgirotbox.addWidget(self.sgirotqslider)
sgirotbox.addWidget(self.sgirotn1slider)
sgirotbox.addWidget(self.sgirotn2slider)
sgirotbox.addWidget(self.sgirotn3slider)
sgirotwidget.setLayout(sgirotbox)
# quaternion
quaternionbox = QtGui.QVBoxLayout()
self.quaternione0slider = ValSlider(quaternionwidget, (0.0, 1.0),
"E0",
rounding=4)
self.quaternione1slider = ValSlider(quaternionwidget, (0.0, 1.0),
"E1",
rounding=4)
self.quaternione2slider = ValSlider(quaternionwidget, (0.0, 1.0),
"E2",
rounding=4)
self.quaternione3slider = ValSlider(quaternionwidget, (0.0, 1.0),
"E3",
rounding=4)
quaternionbox.addWidget(self.quaternione0slider)
quaternionbox.addWidget(self.quaternione1slider)
quaternionbox.addWidget(self.quaternione2slider)
quaternionbox.addWidget(self.quaternione3slider)
quaternionwidget.setLayout(quaternionbox)
# Add widgets to the stack
self.rotstackedwidget.addWidget(EMANwidget)
self.rotstackedwidget.addWidget(Imagicwidget)
self.rotstackedwidget.addWidget(Spiderwidget)
self.rotstackedwidget.addWidget(MRCwidget)
self.rotstackedwidget.addWidget(XYZwidget)
self.rotstackedwidget.addWidget(spinwidget)
self.rotstackedwidget.addWidget(sgirotwidget)
self.rotstackedwidget.addWidget(quaternionwidget)
# add choices to combobox
self.rotcombobox.addItem("EMAN")
self.rotcombobox.addItem("Imagic")
self.rotcombobox.addItem("Spider")
self.rotcombobox.addItem("MRC")
self.rotcombobox.addItem("XYZ")
self.rotcombobox.addItem("spin")
self.rotcombobox.addItem("sgirot")
self.rotcombobox.addItem("quaternion")
# Signal for all sliders
QtCore.QObject.connect(self.rotcombobox,
QtCore.SIGNAL("activated(int)"),
self._rotcombobox_changed)
QtCore.QObject.connect(self.emanazslider,
QtCore.SIGNAL("valueChanged"),
self._on_EMAN_rotation)
QtCore.QObject.connect(self.emanaltslider,
QtCore.SIGNAL("valueChanged"),
self._on_EMAN_rotation)
QtCore.QObject.connect(self.emanphislider,
QtCore.SIGNAL("valueChanged"),
self._on_EMAN_rotation)
QtCore.QObject.connect(self.imagicgammaslider,
QtCore.SIGNAL("valueChanged"),
self._on_Imagic_rotation)
QtCore.QObject.connect(self.imagicbetaslider,
QtCore.SIGNAL("valueChanged"),
self._on_Imagic_rotation)
QtCore.QObject.connect(self.imagicalphaslider,
QtCore.SIGNAL("valueChanged"),
self._on_Imagic_rotation)
QtCore.QObject.connect(self.spiderpsislider,
QtCore.SIGNAL("valueChanged"),
self._on_Spider_rotation)
QtCore.QObject.connect(self.spiderthetaslider,
QtCore.SIGNAL("valueChanged"),
self._on_Spider_rotation)
QtCore.QObject.connect(self.spiderphislider,
QtCore.SIGNAL("valueChanged"),
self._on_Spider_rotation)
QtCore.QObject.connect(self.mrcpsislider,
QtCore.SIGNAL("valueChanged"),
self._on_MRC_rotation)
QtCore.QObject.connect(self.mrcthetaslider,
QtCore.SIGNAL("valueChanged"),
self._on_MRC_rotation)
QtCore.QObject.connect(self.mrcomegaslider,
QtCore.SIGNAL("valueChanged"),
self._on_MRC_rotation)
QtCore.QObject.connect(self.xyzzslider, QtCore.SIGNAL("valueChanged"),
self._on_XYZ_rotation)
QtCore.QObject.connect(self.xyzyslider, QtCore.SIGNAL("valueChanged"),
self._on_XYZ_rotation)
QtCore.QObject.connect(self.xyzxslider, QtCore.SIGNAL("valueChanged"),
self._on_XYZ_rotation)
QtCore.QObject.connect(self.spinomegaslider,
QtCore.SIGNAL("valueChanged"),
self._on_spin_rotation)
QtCore.QObject.connect(self.spinn1slider,
QtCore.SIGNAL("valueChanged"),
self._on_spin_rotation)
QtCore.QObject.connect(self.spinn2slider,
QtCore.SIGNAL("valueChanged"),
self._on_spin_rotation)
QtCore.QObject.connect(self.spinn3slider,
QtCore.SIGNAL("valueChanged"),
self._on_spin_rotation)
QtCore.QObject.connect(self.sgirotqslider,
QtCore.SIGNAL("valueChanged"),
self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn1slider,
QtCore.SIGNAL("valueChanged"),
self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn2slider,
QtCore.SIGNAL("valueChanged"),
self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn3slider,
QtCore.SIGNAL("valueChanged"),
self._on_sgirot_rotation)
QtCore.QObject.connect(self.quaternione0slider,
QtCore.SIGNAL("valueChanged"),
self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione1slider,
QtCore.SIGNAL("valueChanged"),
self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione2slider,
QtCore.SIGNAL("valueChanged"),
self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione3slider,
QtCore.SIGNAL("valueChanged"),
self._on_quaternion_rotation)
def _rotcombobox_changed(self, idx):
self.rotstackedwidget.setCurrentIndex(idx)
self.updateItemControls()
def _on_EMAN_rotation(self, value):
self._set_rotation_std_coords(
Transform({
"type": "eman",
"az": self.emanazslider.getValue(),
"alt": self.emanaltslider.getValue(),
"phi": self.emanphislider.getValue()
}))
self.inspector().updateSceneGraph()
def _on_Imagic_rotation(self, value):
self._set_rotation_std_coords(
Transform({
"type": "imagic",
"gamma": self.imagicgammaslider.getValue(),
"beta": self.imagicbetaslider.getValue(),
"alpha": self.imagicalphaslider.getValue()
}))
self.inspector().updateSceneGraph()
def _on_Spider_rotation(self, value):
self._set_rotation_std_coords(
Transform({
"type": "spider",
"psi": self.spiderpsislider.getValue(),
"theta": self.spiderthetaslider.getValue(),
"phi": self.spiderphislider.getValue()
}))
self.inspector().updateSceneGraph()
def _on_MRC_rotation(self, value):
self._set_rotation_std_coords(
Transform({
"type": "mrc",
"phi": self.mrcpsislider.getValue(),
"theta": self.mrcthetaslider.getValue(),
"omega": self.mrcomegaslider.getValue()
}))
self.inspector().updateSceneGraph()
def _on_XYZ_rotation(self, value):
self._set_rotation_std_coords(
Transform({
"type": "xyz",
"ztilt": self.xyzzslider.getValue(),
"ytilt": self.xyzyslider.getValue(),
"xtilt": self.xyzxslider.getValue()
}))
self.inspector().updateSceneGraph()
def _on_spin_rotation(self, value):
v = Vec3f(self.spinn1slider.getValue(), self.spinn2slider.getValue(),
self.spinn3slider.getValue())
v.normalize()
self._set_rotation_std_coords(
Transform({
"type": "spin",
"omega": self.spinomegaslider.getValue(),
"n1": v[0],
"n2": v[1],
"n3": v[2]
}))
self.inspector().updateSceneGraph()
def _on_sgirot_rotation(self, value):
v = Vec3f(self.sgirotn1slider.getValue(),
self.sgirotn2slider.getValue(),
self.sgirotn3slider.getValue())
v.normalize()
self._set_rotation_std_coords(
Transform({
"type": "sgirot",
"q": self.sgirotqslider.getValue(),
"n1": v[0],
"n2": v[1],
"n3": v[2]
}))
self.inspector().updateSceneGraph()
def _on_quaternion_rotation(self, value):
v = Vec4f(self.quaternione0slider.getValue(),
self.quaternione1slider.getValue(),
self.quaternione2slider.getValue(),
self.quaternione3slider.getValue())
v.normalize()
self._set_rotation_std_coords(
Transform({
"type": "quaternion",
"e0": v[0],
"e1": v[1],
"e2": v[2],
"e3": v[3]
}))
self.inspector().updateSceneGraph()
def _set_rotation_std_coords(self, rotation):
""" This function sets the rotation as if there were no preceeding ones, otherwise a rot around Z could be arounf y,x, etc.
Works by transforming local coords into global corrds"""
tt = rotation
tp = self.item3d().getParentMatrixProduct()
if tp: tt = tp.inverse() * rotation
self.item3d().getTransform().set_rotation(tt.get_rotation())
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
# selection box and label
font = QtGui.QFont()
font.setBold(True)
label = QtGui.QLabel(self.name,self)
label.setFont(font)
label.setAlignment(QtCore.Qt.AlignCenter)
gridbox.addWidget(label, 0, 0, 1, 1)
databox = QtGui.QHBoxLayout()
self.boundingbox = None
if self.item3d().boundingboxsize:
self.boundingbox = QtGui.QLabel("Size: "+self.item3d().boundingboxsize,self)
databox.addWidget(self.boundingbox)
gridbox.addLayout(databox, 1, 0, 1, 1)
# angluar controls
xformframe = QtGui.QFrame()
xformframe.setFrameShape(QtGui.QFrame.StyledPanel)
xformbox = QtGui.QGridLayout()
xformlabel = QtGui.QLabel("Transformation", xformframe)
xformlabel.setFont(font)
xformlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(xformlabel, 0, 0, 1, 2)
# Rotations
self.rotcombobox = QtGui.QComboBox()
xformbox.addWidget(self.rotcombobox, 1, 0, 1, 2)
self.rotstackedwidget = QtGui.QStackedWidget()
self.addRotationWidgets()
xformbox.addWidget(self.rotstackedwidget, 2, 0, 1, 2)
#translations
txlabel = QtGui.QLabel("TX",xformframe)
txlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(txlabel, 3, 0, 1, 1)
tylabel = QtGui.QLabel("TY",xformframe)
tylabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tylabel, 3, 1, 1, 1)
self.tx = EMSpinWidget(0.0, 1.0)
self.ty = EMSpinWidget(0.0, 1.0)
xformbox.addWidget(self.tx, 4, 0, 1, 1)
xformbox.addWidget(self.ty, 4, 1, 1, 1)
tzlabel = QtGui.QLabel("TZ",xformframe)
tzlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tzlabel, 5, 0, 1, 1)
zoomlabel = QtGui.QLabel("Zoom",xformframe)
zoomlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(zoomlabel, 5, 1, 1, 1)
self.tz = EMSpinWidget(0.0, 1.0)
self.zoom = EMSpinWidget(1.0, 0.1, postivemode=True, wheelstep=0.1)
xformbox.addWidget(self.tz, 6, 0, 1, 1)
xformbox.addWidget(self.zoom, 6, 1, 1, 1)
self.resetbuttontx = QtGui.QPushButton("Reset Tx")
self.resetbuttonrot = QtGui.QPushButton("Reset Rot")
xformbox.addWidget(self.resetbuttontx, 7, 0, 1, 1)
xformbox.addWidget(self.resetbuttonrot, 7, 1, 1, 1)
xformframe.setLayout(xformbox)
xformframe.setMaximumWidth(350)
xformframe.setMaximumHeight(self.transfromboxmaxheight)
xformframe.setLayout(xformbox)
gridbox.addWidget(xformframe, 2, 0, 1, 1)
# set to default, but run only as a base class
if type(self) == EMItem3DInspector: self.updateItemControls()
QtCore.QObject.connect(self.tx,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.ty,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.tz,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.zoom,QtCore.SIGNAL("valueChanged(int)"),self._on_scale)
QtCore.QObject.connect(self.resetbuttontx,QtCore.SIGNAL("clicked()"),self._on_resettx)
QtCore.QObject.connect(self.resetbuttonrot,QtCore.SIGNAL("clicked()"),self._on_resetrot)
class EMItem3DInspector(QtGui.QTabWidget):
"""
Class to make the EMItem GUI controls
"""
def __init__(self, name, item3d):
QtGui.QTabWidget.__init__(self)
self.item3d = weakref.ref(item3d)
self.name = name
self.inspector = None
self.transfromboxmaxheight = 400 # This might be problematic
self.addTabs()
def setInspector(self, inspector):
""" This is a reference back to the main inspector, which holds all the item inspectors"""
self.inspector = weakref.ref(inspector)
def addTabs(self):
""" Add a tab for each 'column' """
tabwidget = QtGui.QWidget()
gridbox = QtGui.QGridLayout()
EMItem3DInspector.addControls(self, gridbox)
tabwidget.setLayout(gridbox)
self.addTab(tabwidget, "basic")
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
# selection box and label
font = QtGui.QFont()
font.setBold(True)
label = QtGui.QLabel(self.name,self)
label.setFont(font)
label.setAlignment(QtCore.Qt.AlignCenter)
gridbox.addWidget(label, 0, 0, 1, 1)
databox = QtGui.QHBoxLayout()
self.boundingbox = None
if self.item3d().boundingboxsize:
self.boundingbox = QtGui.QLabel("Size: "+self.item3d().boundingboxsize,self)
databox.addWidget(self.boundingbox)
gridbox.addLayout(databox, 1, 0, 1, 1)
# angluar controls
xformframe = QtGui.QFrame()
xformframe.setFrameShape(QtGui.QFrame.StyledPanel)
xformbox = QtGui.QGridLayout()
xformlabel = QtGui.QLabel("Transformation", xformframe)
xformlabel.setFont(font)
xformlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(xformlabel, 0, 0, 1, 2)
# Rotations
self.rotcombobox = QtGui.QComboBox()
xformbox.addWidget(self.rotcombobox, 1, 0, 1, 2)
self.rotstackedwidget = QtGui.QStackedWidget()
self.addRotationWidgets()
xformbox.addWidget(self.rotstackedwidget, 2, 0, 1, 2)
#translations
txlabel = QtGui.QLabel("TX",xformframe)
txlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(txlabel, 3, 0, 1, 1)
tylabel = QtGui.QLabel("TY",xformframe)
tylabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tylabel, 3, 1, 1, 1)
self.tx = EMSpinWidget(0.0, 1.0)
self.ty = EMSpinWidget(0.0, 1.0)
xformbox.addWidget(self.tx, 4, 0, 1, 1)
xformbox.addWidget(self.ty, 4, 1, 1, 1)
tzlabel = QtGui.QLabel("TZ",xformframe)
tzlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(tzlabel, 5, 0, 1, 1)
zoomlabel = QtGui.QLabel("Zoom",xformframe)
zoomlabel.setAlignment(QtCore.Qt.AlignCenter)
xformbox.addWidget(zoomlabel, 5, 1, 1, 1)
self.tz = EMSpinWidget(0.0, 1.0)
self.zoom = EMSpinWidget(1.0, 0.1, postivemode=True, wheelstep=0.1)
xformbox.addWidget(self.tz, 6, 0, 1, 1)
xformbox.addWidget(self.zoom, 6, 1, 1, 1)
self.resetbuttontx = QtGui.QPushButton("Reset Tx")
self.resetbuttonrot = QtGui.QPushButton("Reset Rot")
xformbox.addWidget(self.resetbuttontx, 7, 0, 1, 1)
xformbox.addWidget(self.resetbuttonrot, 7, 1, 1, 1)
xformframe.setLayout(xformbox)
xformframe.setMaximumWidth(350)
xformframe.setMaximumHeight(self.transfromboxmaxheight)
xformframe.setLayout(xformbox)
gridbox.addWidget(xformframe, 2, 0, 1, 1)
# set to default, but run only as a base class
if type(self) == EMItem3DInspector: self.updateItemControls()
QtCore.QObject.connect(self.tx,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.ty,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.tz,QtCore.SIGNAL("valueChanged(int)"),self._on_translation)
QtCore.QObject.connect(self.zoom,QtCore.SIGNAL("valueChanged(int)"),self._on_scale)
QtCore.QObject.connect(self.resetbuttontx,QtCore.SIGNAL("clicked()"),self._on_resettx)
QtCore.QObject.connect(self.resetbuttonrot,QtCore.SIGNAL("clicked()"),self._on_resetrot)
def _on_translation(self, value):
"""
Need to contain the right coords. And do translation in the correct corrd system
"""
tt = t = Transform({"tx":self.tx.getValue(),"ty":self.ty.getValue(),"tz":self.tz.getValue()})
tp = self.item3d().getParentMatrixProduct()
if tp: tt = tp.inverse()*t
self.item3d().getTransform().set_trans(tt.get_trans())
self.inspector().updateSceneGraph()
def _on_scale(self, value):
self.item3d().getTransform().set_scale(self.zoom.getValue())
self.inspector().updateSceneGraph()
def _on_resettx(self):
self.item3d().getTransform().set_trans(0.0, 0.0, 0.0)
self.updateItemControls()
self.inspector().updateSceneGraph()
def _on_resetrot(self):
self.item3d().getTransform().set_rotation({"type":"eman","az":0.0,"alt":0.0,"phi":0.0})
self.updateItemControls()
self.inspector().updateSceneGraph()
def _isRotNaN(self, rot1, rot2, rot3):
""" Better check to make sure get_rotation did not return Nan, so to prevent a crash """
if rot1 != rot1: return True
if rot2 != rot2: return True
if rot3 != rot3: return True
return False
def updateItemControls(self):
""" Updates this item inspector. Function is called by the item it observes"""
# Translation update
stdtransfrom = self.item3d().getTransformStdCoord()
translation = stdtransfrom.get_trans()
self.tx.setValue(translation[0])
self.ty.setValue(translation[1])
self.tz.setValue(translation[2])
# Rotation update
rotation = stdtransfrom.get_rotation(str(self.rotcombobox.currentText()))
is_identity = stdtransfrom.is_rot_identity()
comboboxidx = self.rotcombobox.currentIndex()
if comboboxidx == 0:
if self._isRotNaN(rotation["az"],rotation["alt"],rotation["phi"]): return
self.emanazslider.setValue(rotation["az"], quiet=1)
self.emanaltslider.setValue(rotation["alt"], quiet=1)
self.emanphislider.setValue(rotation["phi"], quiet=1)
if comboboxidx == 1:
if self._isRotNaN(rotation["gamma"],rotation["beta"],rotation["alpha"]): return
self.imagicgammaslider.setValue(rotation["gamma"], quiet=1)
self.imagicbetaslider.setValue(rotation["beta"], quiet=1)
self.imagicalphaslider.setValue(rotation["alpha"], quiet=1)
if comboboxidx == 2:
if self._isRotNaN(rotation["psi"],rotation["theta"],rotation["phi"]): return
self.spiderpsislider.setValue(rotation["psi"], quiet=1)
self.spiderthetaslider.setValue(rotation["theta"], quiet=1)
self.spiderphislider.setValue(rotation["phi"], quiet=1)
if comboboxidx == 3:
if self._isRotNaN(rotation["phi"],rotation["theta"],rotation["omega"]): return
self.mrcpsislider.setValue(rotation["phi"], quiet=1)
self.mrcthetaslider.setValue(rotation["theta"], quiet=1)
self.mrcomegaslider.setValue(rotation["omega"], quiet=1)
if comboboxidx == 4:
if self._isRotNaN(rotation["ztilt"],rotation["ytilt"],rotation["xtilt"]): return
self.xyzzslider.setValue(rotation["ztilt"], quiet=1)
self.xyzyslider.setValue(rotation["ytilt"], quiet=1)
self.xyzxslider.setValue(rotation["xtilt"], quiet=1)
if comboboxidx == 5:
if self._isRotNaN(rotation["n1"],rotation["n2"],rotation["n3"]): return
if is_identity and self.spinn1slider.getValue() == 0.0 and self.spinn2slider.getValue() == 0.0 and self.spinn3slider.getValue() == 0.0:
self.spinomegaslider .setValue(0.0, quiet=1)
self.spinn1slider.setValue(0.0, quiet=1)
self.spinn2slider.setValue(0.0, quiet=1)
self.spinn3slider.setValue(1.0, quiet=1)
else:
self.spinomegaslider .setValue(rotation["omega"], quiet=1)
# Don't change slider if reult is Nan
if rotation["n1"] == rotation["n1"]: self.spinn1slider.setValue(rotation["n1"], quiet=1)
if rotation["n2"] == rotation["n2"]: self.spinn2slider.setValue(rotation["n2"], quiet=1)
if rotation["n3"] == rotation["n3"]: self.spinn3slider.setValue(rotation["n3"], quiet=1)
if comboboxidx == 6:
if self._isRotNaN(rotation["n1"],rotation["n2"],rotation["n3"]): return
if is_identity and self.spinn1slider.getValue() == 0.0 and self.spinn2slider.getValue() == 0.0 and self.spinn3slider.getValue() == 0.0:
self.spinomegaslider.setValue(0.0, quiet=1)
self.sgirotn1slider.setValue(0.0, quiet=1)
self.sgirotn2slider.setValue(0.0, quiet=1)
self.sgirotn3slider.setValue(1.0, quiet=1)
else:
self.spinomegaslider.setValue(rotation["q"], quiet=1)
# Don't change slider if reult is Nan
if rotation["n1"] == rotation["n1"]: self.sgirotn1slider.setValue(rotation["n1"], quiet=1)
if rotation["n2"] == rotation["n2"]: self.sgirotn2slider.setValue(rotation["n2"], quiet=1)
if rotation["n3"] == rotation["n3"]: self.sgirotn3slider.setValue(rotation["n3"], quiet=1)
if comboboxidx == 7:
if self._isRotNaN(rotation["e1"],rotation["e2"],rotation["e3"]): return
if is_identity:
self.quaternione0slider.setValue(1.0, quiet=1)
self.quaternione1slider.setValue(0.0, quiet=1)
self.quaternione2slider.setValue(0.0, quiet=1)
self.quaternione3slider.setValue(0.0, quiet=1)
else:
self.quaternione0slider.setValue(rotation["e0"], quiet=1)
self.quaternione1slider.setValue(rotation["e1"], quiet=1)
self.quaternione2slider.setValue(rotation["e2"], quiet=1)
self.quaternione3slider.setValue(rotation["e3"], quiet=1)
# Scaling update
self.zoom.setValue(self.item3d().getTransform().get_scale())
def updateMetaData(self):
"""
I didn't want to put this in update b/c this data doesn't change very often, and I don't want to waste CPU
Its a judgement call really, less coupling vs. more efficiency
"""
if self.boundingbox: self.boundingbox.setText("Size: "+self.item3d().boundingboxsize)
def addRotationWidgets(self):
""" Add alll the widgets for the various EMAN2 rotation conventions """
EMANwidget = QtGui.QWidget()
Imagicwidget = QtGui.QWidget()
Spiderwidget = QtGui.QWidget()
MRCwidget = QtGui.QWidget()
XYZwidget = QtGui.QWidget()
spinwidget = QtGui.QWidget()
sgirotwidget = QtGui.QWidget()
quaternionwidget = QtGui.QWidget()
# EMAN
emanbox = QtGui.QVBoxLayout()
self.emanazslider = ValSlider(EMANwidget, (0.0, 360.0), " Az", rounding = 1)
self.emanaltslider = ValSlider(EMANwidget, (0.0, 180.0), "Alt", rounding = 1)
self.emanphislider = ValSlider(EMANwidget, (0.0, 360.0), "Phi", rounding = 1)
emanbox.addWidget(self.emanazslider)
emanbox.addWidget(self.emanaltslider)
emanbox.addWidget(self.emanphislider)
EMANwidget.setLayout(emanbox)
# Imagic
imagicbox = QtGui.QVBoxLayout()
self.imagicgammaslider = ValSlider(Imagicwidget, (0.0, 360.0), "Gamma", rounding = 1)
self.imagicbetaslider = ValSlider(Imagicwidget, (0.0, 180.0), " Beta", rounding = 1)
self.imagicalphaslider = ValSlider(Imagicwidget, (0.0, 360.0), " Alpha", rounding = 1)
imagicbox.addWidget(self.imagicgammaslider)
imagicbox.addWidget(self.imagicbetaslider)
imagicbox.addWidget(self.imagicalphaslider)
Imagicwidget.setLayout(imagicbox)
# Spider
spiderbox = QtGui.QVBoxLayout()
self.spiderpsislider = ValSlider(Spiderwidget, (0.0, 360.0), " Psi", rounding = 1)
self.spiderthetaslider = ValSlider(Spiderwidget, (0.0, 180.0), "Theta", rounding = 1)
self.spiderphislider = ValSlider(Spiderwidget, (0.0, 360.0), " Phi", rounding = 1)
spiderbox.addWidget(self.spiderpsislider)
spiderbox.addWidget(self.spiderthetaslider)
spiderbox.addWidget(self.spiderphislider)
Spiderwidget.setLayout(spiderbox)
# MRC
mrcbox = QtGui.QVBoxLayout()
self.mrcpsislider = ValSlider(MRCwidget, (0.0, 360.0), " Psi", rounding = 1)
self.mrcthetaslider = ValSlider(MRCwidget, (0.0, 180.0), " Theta", rounding = 1)
self.mrcomegaslider = ValSlider(MRCwidget, (0.0, 360.0), "Omega", rounding = 1)
mrcbox.addWidget(self.mrcpsislider)
mrcbox.addWidget(self.mrcthetaslider)
mrcbox.addWidget(self.mrcomegaslider)
MRCwidget.setLayout(mrcbox)
# XYZ
xyzbox = QtGui.QVBoxLayout()
self.xyzzslider = ValSlider(XYZwidget, (0.0, 360.0), "Z", rounding = 1)
self.xyzyslider = ValSlider(XYZwidget, (0.0, 180.0), "Y", rounding = 1)
self.xyzxslider = ValSlider(XYZwidget, (0.0, 360.0), "X", rounding = 1)
xyzbox.addWidget(self.xyzzslider)
xyzbox.addWidget(self.xyzyslider)
xyzbox.addWidget(self.xyzxslider)
XYZwidget.setLayout(xyzbox)
# spin
spinbox = QtGui.QVBoxLayout()
self.spinomegaslider = ValSlider(spinwidget, (0.0, 180.0), "Omega", rounding = 1)
self.spinn1slider = ValSlider(spinwidget, (0.0, 1.0), " N1", rounding = 4)
self.spinn2slider = ValSlider(spinwidget, (0.0, 1.0), " N2", rounding = 4)
self.spinn3slider = ValSlider(spinwidget, (0.0, 1.0), " N3", rounding = 4)
spinbox.addWidget(self.spinomegaslider)
spinbox.addWidget(self.spinn1slider)
spinbox.addWidget(self.spinn2slider)
spinbox.addWidget(self.spinn3slider)
spinwidget.setLayout(spinbox)
# sgirot
sgirotbox = QtGui.QVBoxLayout()
self.sgirotqslider = ValSlider(sgirotwidget, (0.0, 180.0), " Q", rounding = 1)
self.sgirotn1slider = ValSlider(sgirotwidget, (0.0, 1.0), "N1", rounding = 4)
self.sgirotn2slider = ValSlider(sgirotwidget, (0.0, 1.0), "N2", rounding = 4)
self.sgirotn3slider = ValSlider(sgirotwidget, (0.0, 1.0), "N3", rounding = 4)
sgirotbox.addWidget(self.sgirotqslider)
sgirotbox.addWidget(self.sgirotn1slider)
sgirotbox.addWidget(self.sgirotn2slider)
sgirotbox.addWidget(self.sgirotn3slider)
sgirotwidget.setLayout(sgirotbox)
# quaternion
quaternionbox = QtGui.QVBoxLayout()
self.quaternione0slider = ValSlider(quaternionwidget, (0.0, 1.0), "E0", rounding = 4)
self.quaternione1slider = ValSlider(quaternionwidget, (0.0, 1.0), "E1", rounding = 4)
self.quaternione2slider = ValSlider(quaternionwidget, (0.0, 1.0), "E2", rounding = 4)
self.quaternione3slider = ValSlider(quaternionwidget, (0.0, 1.0), "E3", rounding = 4)
quaternionbox.addWidget(self.quaternione0slider)
quaternionbox.addWidget(self.quaternione1slider)
quaternionbox.addWidget(self.quaternione2slider)
quaternionbox.addWidget(self.quaternione3slider)
quaternionwidget.setLayout(quaternionbox)
# Add widgets to the stack
self.rotstackedwidget.addWidget(EMANwidget)
self.rotstackedwidget.addWidget(Imagicwidget)
self.rotstackedwidget.addWidget(Spiderwidget)
self.rotstackedwidget.addWidget(MRCwidget)
self.rotstackedwidget.addWidget(XYZwidget)
self.rotstackedwidget.addWidget(spinwidget)
self.rotstackedwidget.addWidget(sgirotwidget)
self.rotstackedwidget.addWidget(quaternionwidget)
# add choices to combobox
self.rotcombobox.addItem("EMAN")
self.rotcombobox.addItem("Imagic")
self.rotcombobox.addItem("Spider")
self.rotcombobox.addItem("MRC")
self.rotcombobox.addItem("XYZ")
self.rotcombobox.addItem("spin")
self.rotcombobox.addItem("sgirot")
self.rotcombobox.addItem("quaternion")
# Signal for all sliders
QtCore.QObject.connect(self.rotcombobox, QtCore.SIGNAL("activated(int)"), self._rotcombobox_changed)
QtCore.QObject.connect(self.emanazslider,QtCore.SIGNAL("valueChanged"),self._on_EMAN_rotation)
QtCore.QObject.connect(self.emanaltslider,QtCore.SIGNAL("valueChanged"),self._on_EMAN_rotation)
QtCore.QObject.connect(self.emanphislider,QtCore.SIGNAL("valueChanged"),self._on_EMAN_rotation)
QtCore.QObject.connect(self.imagicgammaslider,QtCore.SIGNAL("valueChanged"),self._on_Imagic_rotation)
QtCore.QObject.connect(self.imagicbetaslider,QtCore.SIGNAL("valueChanged"),self._on_Imagic_rotation)
QtCore.QObject.connect(self.imagicalphaslider,QtCore.SIGNAL("valueChanged"),self._on_Imagic_rotation)
QtCore.QObject.connect(self.spiderpsislider,QtCore.SIGNAL("valueChanged"),self._on_Spider_rotation)
QtCore.QObject.connect(self.spiderthetaslider,QtCore.SIGNAL("valueChanged"),self._on_Spider_rotation)
QtCore.QObject.connect(self.spiderphislider,QtCore.SIGNAL("valueChanged"),self._on_Spider_rotation)
QtCore.QObject.connect(self.mrcpsislider,QtCore.SIGNAL("valueChanged"),self._on_MRC_rotation)
QtCore.QObject.connect(self.mrcthetaslider,QtCore.SIGNAL("valueChanged"),self._on_MRC_rotation)
QtCore.QObject.connect(self.mrcomegaslider,QtCore.SIGNAL("valueChanged"),self._on_MRC_rotation)
QtCore.QObject.connect(self.xyzzslider,QtCore.SIGNAL("valueChanged"),self._on_XYZ_rotation)
QtCore.QObject.connect(self.xyzyslider,QtCore.SIGNAL("valueChanged"),self._on_XYZ_rotation)
QtCore.QObject.connect(self.xyzxslider,QtCore.SIGNAL("valueChanged"),self._on_XYZ_rotation)
QtCore.QObject.connect(self.spinomegaslider,QtCore.SIGNAL("valueChanged"),self._on_spin_rotation)
QtCore.QObject.connect(self.spinn1slider,QtCore.SIGNAL("valueChanged"),self._on_spin_rotation)
QtCore.QObject.connect(self.spinn2slider,QtCore.SIGNAL("valueChanged"),self._on_spin_rotation)
QtCore.QObject.connect(self.spinn3slider,QtCore.SIGNAL("valueChanged"),self._on_spin_rotation)
QtCore.QObject.connect(self.sgirotqslider,QtCore.SIGNAL("valueChanged"),self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn1slider,QtCore.SIGNAL("valueChanged"),self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn2slider,QtCore.SIGNAL("valueChanged"),self._on_sgirot_rotation)
QtCore.QObject.connect(self.sgirotn3slider,QtCore.SIGNAL("valueChanged"),self._on_sgirot_rotation)
QtCore.QObject.connect(self.quaternione0slider,QtCore.SIGNAL("valueChanged"),self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione1slider,QtCore.SIGNAL("valueChanged"),self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione2slider,QtCore.SIGNAL("valueChanged"),self._on_quaternion_rotation)
QtCore.QObject.connect(self.quaternione3slider,QtCore.SIGNAL("valueChanged"),self._on_quaternion_rotation)
def _rotcombobox_changed(self, idx):
self.rotstackedwidget.setCurrentIndex(idx)
self.updateItemControls()
def _on_EMAN_rotation(self, value):
self._set_rotation_std_coords(Transform({"type":"eman","az":self.emanazslider.getValue(),"alt":self.emanaltslider.getValue(),"phi":self.emanphislider.getValue()}))
self.inspector().updateSceneGraph()
def _on_Imagic_rotation(self, value):
self._set_rotation_std_coords(Transform({"type":"imagic","gamma":self.imagicgammaslider.getValue(),"beta":self.imagicbetaslider.getValue(),"alpha":self.imagicalphaslider.getValue()}))
self.inspector().updateSceneGraph()
def _on_Spider_rotation(self, value):
self._set_rotation_std_coords(Transform({"type":"spider","psi":self.spiderpsislider.getValue(),"theta":self.spiderthetaslider.getValue(),"phi":self.spiderphislider.getValue()}))
self.inspector().updateSceneGraph()
def _on_MRC_rotation(self, value):
self._set_rotation_std_coords(Transform({"type":"mrc","phi":self.mrcpsislider.getValue(),"theta":self.mrcthetaslider.getValue(),"omega":self.mrcomegaslider.getValue()}))
self.inspector().updateSceneGraph()
def _on_XYZ_rotation(self, value):
self._set_rotation_std_coords(Transform({"type":"xyz","ztilt":self.xyzzslider.getValue(),"ytilt":self.xyzyslider.getValue(),"xtilt":self.xyzxslider.getValue()}))
self.inspector().updateSceneGraph()
def _on_spin_rotation(self, value):
v = Vec3f(self.spinn1slider.getValue(), self.spinn2slider.getValue(), self.spinn3slider.getValue())
v.normalize()
self._set_rotation_std_coords(Transform({"type":"spin","omega":self.spinomegaslider.getValue(),"n1":v[0],"n2":v[1],"n3":v[2]}))
self.inspector().updateSceneGraph()
def _on_sgirot_rotation(self, value):
v = Vec3f(self.sgirotn1slider.getValue(), self.sgirotn2slider.getValue(), self.sgirotn3slider.getValue())
v.normalize()
self._set_rotation_std_coords(Transform({"type":"sgirot","q":self.sgirotqslider.getValue(),"n1":v[0],"n2":v[1],"n3":v[2]}))
self.inspector().updateSceneGraph()
def _on_quaternion_rotation(self, value):
v = Vec4f(self.quaternione0slider.getValue(), self.quaternione1slider.getValue(), self.quaternione2slider.getValue(), self.quaternione3slider.getValue())
v.normalize()
self._set_rotation_std_coords(Transform({"type":"quaternion","e0":v[0],"e1":v[1],"e2":v[2],"e3":v[3]}))
self.inspector().updateSceneGraph()
def _set_rotation_std_coords(self, rotation):
""" This function sets the rotation as if there were no preceeding ones, otherwise a rot around Z could be arounf y,x, etc.
Works by transforming local coords into global corrds"""
tt = rotation
tp = self.item3d().getParentMatrixProduct()
if tp: tt = tp.inverse()*rotation
self.item3d().getTransform().set_rotation(tt.get_rotation())
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
#frame to control properties of left/right arrows
lineframe = QtGui.QFrame()
lineframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
linegridbox = QtGui.QGridLayout()
leftlabel = QtGui.QLabel("Left arrow")
leftlabel.setFont(lfont)
leftlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(leftlabel, 0, 1, 1, 1)
sidelabel1 = QtGui.QLabel("Size")
sidelabel1.setFont(lfont)
sidelabel1.setAlignment(QtCore.Qt.AlignVCenter)
linegridbox.addWidget(sidelabel1, 2, 0, 1, 1)
sidelabel2 = QtGui.QLabel("Length")
sidelabel2.setFont(lfont)
sidelabel2.setAlignment(QtCore.Qt.AlignVCenter)
linegridbox.addWidget(sidelabel2, 3, 0, 1, 1)
self.leftShowArrow = QtGui.QCheckBox("Show")
self.leftShowArrow.setChecked(self.item3d().showLeftArrow)
linegridbox.addWidget(self.leftShowArrow, 1, 1, 1, 1)
self.leftArrowSize = EMSpinWidget(int(self.item3d().leftArrowSize), 1.0, rounding=0)
self.leftArrowSize.setMinimumWidth(120)
linegridbox.addWidget(self.leftArrowSize, 2, 1, 1, 1)
self.leftArrowLength = EMSpinWidget(int(self.item3d().leftArrowLength), 1.0, rounding=0)
self.leftArrowLength.setMinimumWidth(120)
linegridbox.addWidget(self.leftArrowLength, 3, 1, 1, 1)
rightlabel = QtGui.QLabel("Right arrow")
rightlabel.setFont(lfont)
rightlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(rightlabel, 0, 2, 1, 1)
self.rightShowArrow = QtGui.QCheckBox("Show")
self.rightShowArrow.setChecked(self.item3d().showRightArrow)
linegridbox.addWidget(self.rightShowArrow, 1, 2, 1, 1)
self.rightArrowSize = EMSpinWidget(int(self.item3d().rightArrowSize), 1.0, rounding=0)
self.rightArrowSize.setMinimumWidth(120)
linegridbox.addWidget(self.rightArrowSize, 2, 2, 1, 1)
self.rightArrowLength = EMSpinWidget(int(self.item3d().rightArrowLength), 1.0, rounding=0)
self.rightArrowLength.setMinimumWidth(120)
linegridbox.addWidget(self.rightArrowLength, 3, 2, 1, 1)
linelengthlabel = QtGui.QLabel("Line Length")
linelengthlabel.setFont(lfont)
linelengthlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(linelengthlabel, 4, 0, 2, 2)
self.linelength = EMSpinWidget(int(self.item3d().length), 1.0, rounding=0)
linegridbox.addWidget(self.linelength, 4, 2, 2, 2)
linewidthlabel = QtGui.QLabel("Line Width")
linewidthlabel.setFont(lfont)
linewidthlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(linewidthlabel, 5, 0, 1, 2)
self.linewidth = EMSpinWidget(int(self.item3d().width), 1.0, rounding=0)
linegridbox.addWidget(self.linewidth, 5, 2, 1, 2)
lineframe.setLayout(linegridbox)
gridbox.addWidget(lineframe, 2, 0)
#frame to control slice/stack of the line
lineframe2 = QtGui.QFrame()
lineframe2.setFrameShape(QtGui.QFrame.StyledPanel)
linehbox = QtGui.QVBoxLayout()
self.slice = ValSlider(lineframe2, (1, 100), "Slice", rounding=0)
self.slice.setValue(self.item3d().slices)
self.stack = ValSlider(lineframe2, (1, 100), "Stack", rounding=0)
self.slice.setValue(self.item3d().stacks)
linehbox.addWidget(self.slice)
linehbox.addWidget(self.stack)
lineframe2.setLayout(linehbox)
gridbox.addWidget(lineframe2, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControl3DText:
self.updateItemControls()
self.updateMetaData()
QtCore.QObject.connect(self.leftShowArrow, QtCore.SIGNAL("stateChanged(int)"), self.redraw)
QtCore.QObject.connect(self.rightShowArrow, QtCore.SIGNAL("stateChanged(int)"), self.redraw)
QtCore.QObject.connect(self.leftArrowSize,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.leftArrowLength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.rightArrowSize,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.rightArrowLength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.linelength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.linewidth,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.slice,QtCore.SIGNAL("valueChanged"),self.redraw)
QtCore.QObject.connect(self.stack,QtCore.SIGNAL("valueChanged"),self.redraw)
class EMInspectorControlLine(EMInspectorControlShape):
"""
Class to make EMItem GUI SHAPE Line Inspector
"""
def __init__(self, name, item3d):
EMInspectorControlShape.__init__(self, name, item3d)
def updateItemControls(self):
""" Updates this item inspector. Function is called by the item it observes"""
super(EMInspectorControlLine, self).updateItemControls()
def updateMetaData(self):
""" Updates the items metadata, such as line length, width. Function is called by the item it observes when the items meta data changes """
super(EMInspectorControlLine, self).updateMetaData()
self.leftArrowSize.setValue(self.item3d().leftArrowSize, quiet=1)
self.leftArrowLength.setValue(self.item3d().leftArrowLength, quiet=1)
self.rightArrowSize.setValue(self.item3d().rightArrowSize, quiet=1)
self.rightArrowLength.setValue(self.item3d().rightArrowLength, quiet=1)
self.slice.setValue(self.item3d().slices, quiet=1)
self.stack.setValue(self.item3d().stacks, quiet=1)
self.linelength.setValue(int(self.item3d().length), quiet=1)
def addTabs(self):
""" Add a tab for each 'column' """
super(EMInspectorControlLine, self).addTabs()
tabwidget = QtGui.QWidget()
gridbox = QtGui.QGridLayout()
EMInspectorControlLine.addControls(self, gridbox)
tabwidget.setLayout(gridbox)
self.addTab(tabwidget, "line")
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
#frame to control properties of left/right arrows
lineframe = QtGui.QFrame()
lineframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
linegridbox = QtGui.QGridLayout()
leftlabel = QtGui.QLabel("Left arrow")
leftlabel.setFont(lfont)
leftlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(leftlabel, 0, 1, 1, 1)
sidelabel1 = QtGui.QLabel("Size")
sidelabel1.setFont(lfont)
sidelabel1.setAlignment(QtCore.Qt.AlignVCenter)
linegridbox.addWidget(sidelabel1, 2, 0, 1, 1)
sidelabel2 = QtGui.QLabel("Length")
sidelabel2.setFont(lfont)
sidelabel2.setAlignment(QtCore.Qt.AlignVCenter)
linegridbox.addWidget(sidelabel2, 3, 0, 1, 1)
self.leftShowArrow = QtGui.QCheckBox("Show")
self.leftShowArrow.setChecked(self.item3d().showLeftArrow)
linegridbox.addWidget(self.leftShowArrow, 1, 1, 1, 1)
self.leftArrowSize = EMSpinWidget(int(self.item3d().leftArrowSize), 1.0, rounding=0)
self.leftArrowSize.setMinimumWidth(120)
linegridbox.addWidget(self.leftArrowSize, 2, 1, 1, 1)
self.leftArrowLength = EMSpinWidget(int(self.item3d().leftArrowLength), 1.0, rounding=0)
self.leftArrowLength.setMinimumWidth(120)
linegridbox.addWidget(self.leftArrowLength, 3, 1, 1, 1)
rightlabel = QtGui.QLabel("Right arrow")
rightlabel.setFont(lfont)
rightlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(rightlabel, 0, 2, 1, 1)
self.rightShowArrow = QtGui.QCheckBox("Show")
self.rightShowArrow.setChecked(self.item3d().showRightArrow)
linegridbox.addWidget(self.rightShowArrow, 1, 2, 1, 1)
self.rightArrowSize = EMSpinWidget(int(self.item3d().rightArrowSize), 1.0, rounding=0)
self.rightArrowSize.setMinimumWidth(120)
linegridbox.addWidget(self.rightArrowSize, 2, 2, 1, 1)
self.rightArrowLength = EMSpinWidget(int(self.item3d().rightArrowLength), 1.0, rounding=0)
self.rightArrowLength.setMinimumWidth(120)
linegridbox.addWidget(self.rightArrowLength, 3, 2, 1, 1)
linelengthlabel = QtGui.QLabel("Line Length")
linelengthlabel.setFont(lfont)
linelengthlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(linelengthlabel, 4, 0, 2, 2)
self.linelength = EMSpinWidget(int(self.item3d().length), 1.0, rounding=0)
linegridbox.addWidget(self.linelength, 4, 2, 2, 2)
linewidthlabel = QtGui.QLabel("Line Width")
linewidthlabel.setFont(lfont)
linewidthlabel.setAlignment(QtCore.Qt.AlignCenter)
linegridbox.addWidget(linewidthlabel, 5, 0, 1, 2)
self.linewidth = EMSpinWidget(int(self.item3d().width), 1.0, rounding=0)
linegridbox.addWidget(self.linewidth, 5, 2, 1, 2)
lineframe.setLayout(linegridbox)
gridbox.addWidget(lineframe, 2, 0)
#frame to control slice/stack of the line
lineframe2 = QtGui.QFrame()
lineframe2.setFrameShape(QtGui.QFrame.StyledPanel)
linehbox = QtGui.QVBoxLayout()
self.slice = ValSlider(lineframe2, (1, 100), "Slice", rounding=0)
self.slice.setValue(self.item3d().slices)
self.stack = ValSlider(lineframe2, (1, 100), "Stack", rounding=0)
self.slice.setValue(self.item3d().stacks)
linehbox.addWidget(self.slice)
linehbox.addWidget(self.stack)
lineframe2.setLayout(linehbox)
gridbox.addWidget(lineframe2, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControl3DText:
self.updateItemControls()
self.updateMetaData()
QtCore.QObject.connect(self.leftShowArrow, QtCore.SIGNAL("stateChanged(int)"), self.redraw)
QtCore.QObject.connect(self.rightShowArrow, QtCore.SIGNAL("stateChanged(int)"), self.redraw)
QtCore.QObject.connect(self.leftArrowSize,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.leftArrowLength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.rightArrowSize,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.rightArrowLength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.linelength,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.linewidth,QtCore.SIGNAL("valueChanged(int)"),self.redraw)
QtCore.QObject.connect(self.slice,QtCore.SIGNAL("valueChanged"),self.redraw)
QtCore.QObject.connect(self.stack,QtCore.SIGNAL("valueChanged"),self.redraw)
def redraw(self):
self.item3d().setShowLeftArrow(self.leftShowArrow.isChecked())
self.item3d().setShowRightArrow(self.rightShowArrow.isChecked())
self.item3d().leftArrowSize = self.leftArrowSize.getValue()
self.item3d().leftArrowLength = self.leftArrowLength.getValue()
self.item3d().rightArrowSize = self.rightArrowSize.getValue()
self.item3d().rightArrowLength = self.rightArrowLength.getValue()
self.item3d().setLength(self.linelength.getValue())
self.item3d().setWidth(self.linewidth.getValue())
self.item3d().setSlices(self.slice.getValue())
self.item3d().setStacks(self.stack.getValue())
if self.inspector:
self.inspector().updateSceneGraph()
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
textframe = QtGui.QFrame()
textframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
textgridbox = QtGui.QGridLayout()
# Add widgets to textframe
textlabel = QtGui.QLabel("3D Font Mode")
textlabel.setFont(lfont)
textlabel.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel, 0, 0, 1, 1)
self.textModeBox = QtGui.QComboBox()
self.textModeBox.addItems(["EXTRUDE", "TEXTURE", "POLYGON", "OUTLINE"])
textgridbox.addWidget(self.textModeBox, 0, 1, 1, 1)
textlabel2 = QtGui.QLabel("3D Font Depth")
textlabel2.setFont(lfont)
textlabel2.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel2, 1, 0, 1, 1)
self.fontDepth = EMSpinWidget(int(self.item3d().getFontDepth()), 1.0, rounding=0)
self.fontDepth.setMinimumWidth(120)
textgridbox.addWidget(self.fontDepth, 1, 1, 1, 1)
textlabel2 = QtGui.QLabel("3D Font Size")
textlabel2.setFont(lfont)
textlabel2.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel2, 2, 0, 1, 1)
self.fontSize = EMSpinWidget(int(self.item3d().getFontSize()), 1.0, rounding=0)
self.fontSize.setMinimumWidth(120)
textgridbox.addWidget(self.fontSize, 2, 1, 1, 1)
textframe.setLayout(textgridbox)
gridbox.addWidget(textframe, 2, 0)
# Add text
text3dframe = QtGui.QFrame()
text3dframe.setFrameShape(QtGui.QFrame.StyledPanel)
text3dgridbox = QtGui.QGridLayout()
textlabel3 = QtGui.QLabel("3D Text")
textlabel3.setFont(lfont)
text3dgridbox.addWidget(textlabel3, 3, 0, 2, 1)
self.text3d = QtGui.QLineEdit(self.item3d().getRenderString())
text3dgridbox.addWidget(self.text3d, 3, 1, 2, 1)
text3dframe.setLayout(text3dgridbox)
gridbox.addWidget(text3dframe, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControl3DText:
self.updateItemControls()
self.updateMetaData()
self.textModeBox.currentIndexChanged.connect(self.on3DTextModeChanged)
QtCore.QObject.connect(self.fontDepth,QtCore.SIGNAL("valueChanged(int)"),self.on3DTextDepthChanged)
QtCore.QObject.connect(self.fontSize,QtCore.SIGNAL("valueChanged(int)"),self.on3DTextFontChanged)
QtCore.QObject.connect(self.text3d,QtCore.SIGNAL("textChanged(const QString&)"),self.on3DTextChanged)
class EMInspectorControl3DText(EMInspectorControlShape):
"""
Class to make EMItem GUI SHAPE 3DText Inspector
"""
def __init__(self, name, item3d):
EMInspectorControlShape.__init__(self, name, item3d)
def updateItemControls(self):
""" Updates this item inspector. Function is called by the item it observes"""
super(EMInspectorControl3DText, self).updateItemControls()
self.fontDepth.setValue(int(self.item3d().getFontDepth()))
self.fontSize.setValue(int(self.item3d().getFontSize()))
def updateMetaData(self):
""" Updates the items metadata, such as line length, width. Function is called by the item it observes when the items meta data changes """
super(EMInspectorControlShape, self).updateMetaData()
if self.item3d().getFontMode() == FTGLFontMode.EXTRUDE:
self.textModeBox.setCurrentIndex(0)
if self.item3d().getFontMode() == FTGLFontMode.TEXTURE:
self.textModeBox.setCurrentIndex(1)
if self.item3d().getFontMode() == FTGLFontMode.POLYGON:
self.textModeBox.setCurrentIndex(2)
if self.item3d().getFontMode() == FTGLFontMode.OUTLINE:
self.textModeBox.setCurrentIndex(3)
def addTabs(self):
""" Add a tab for each 'column' """
super(EMInspectorControl3DText, self).addTabs()
tabwidget = QtGui.QWidget()
gridbox = QtGui.QGridLayout()
EMInspectorControl3DText.addControls(self, gridbox)
tabwidget.setLayout(gridbox)
self.addTab(tabwidget, "text")
def addControls(self, gridbox):
""" Construct all the widgets in this Item Inspector """
textframe = QtGui.QFrame()
textframe.setFrameShape(QtGui.QFrame.StyledPanel)
lfont = QtGui.QFont()
lfont.setBold(True)
textgridbox = QtGui.QGridLayout()
# Add widgets to textframe
textlabel = QtGui.QLabel("3D Font Mode")
textlabel.setFont(lfont)
textlabel.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel, 0, 0, 1, 1)
self.textModeBox = QtGui.QComboBox()
self.textModeBox.addItems(["EXTRUDE", "TEXTURE", "POLYGON", "OUTLINE"])
textgridbox.addWidget(self.textModeBox, 0, 1, 1, 1)
textlabel2 = QtGui.QLabel("3D Font Depth")
textlabel2.setFont(lfont)
textlabel2.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel2, 1, 0, 1, 1)
self.fontDepth = EMSpinWidget(int(self.item3d().getFontDepth()), 1.0, rounding=0)
self.fontDepth.setMinimumWidth(120)
textgridbox.addWidget(self.fontDepth, 1, 1, 1, 1)
textlabel2 = QtGui.QLabel("3D Font Size")
textlabel2.setFont(lfont)
textlabel2.setAlignment(QtCore.Qt.AlignCenter)
textgridbox.addWidget(textlabel2, 2, 0, 1, 1)
self.fontSize = EMSpinWidget(int(self.item3d().getFontSize()), 1.0, rounding=0)
self.fontSize.setMinimumWidth(120)
textgridbox.addWidget(self.fontSize, 2, 1, 1, 1)
textframe.setLayout(textgridbox)
gridbox.addWidget(textframe, 2, 0)
# Add text
text3dframe = QtGui.QFrame()
text3dframe.setFrameShape(QtGui.QFrame.StyledPanel)
text3dgridbox = QtGui.QGridLayout()
textlabel3 = QtGui.QLabel("3D Text")
textlabel3.setFont(lfont)
text3dgridbox.addWidget(textlabel3, 3, 0, 2, 1)
self.text3d = QtGui.QLineEdit(self.item3d().getRenderString())
text3dgridbox.addWidget(self.text3d, 3, 1, 2, 1)
text3dframe.setLayout(text3dgridbox)
gridbox.addWidget(text3dframe, 3, 0)
# set to default, but run only as a base class
if type(self) == EMInspectorControl3DText:
self.updateItemControls()
self.updateMetaData()
self.textModeBox.currentIndexChanged.connect(self.on3DTextModeChanged)
QtCore.QObject.connect(self.fontDepth,QtCore.SIGNAL("valueChanged(int)"),self.on3DTextDepthChanged)
QtCore.QObject.connect(self.fontSize,QtCore.SIGNAL("valueChanged(int)"),self.on3DTextFontChanged)
QtCore.QObject.connect(self.text3d,QtCore.SIGNAL("textChanged(const QString&)"),self.on3DTextChanged)
def on3DTextModeChanged(self):
textMode = str(self.textModeBox.currentText())
if textMode == "EXTRUDE":
self.item3d().setFontMode(FTGLFontMode.EXTRUDE)
elif textMode == "TEXTURE":
self.item3d().setFontMode(FTGLFontMode.TEXTURE)
elif textMode == "POLYGON":
self.item3d().setFontMode(FTGLFontMode.POLYGON)
elif textMode == "OUTLINE":
self.item3d().setFontMode(FTGLFontMode.OUTLINE)
if self.inspector: self.inspector().updateSceneGraph()
def on3DTextDepthChanged(self):
self.item3d().setFontDepth(int(self.fontDepth.getValue()))
if self.inspector: self.inspector().updateSceneGraph()
def on3DTextFontChanged(self):
self.item3d().setRenderString(self.item3d().getRenderString(), int(self.fontSize.getValue()))
if self.inspector: self.inspector().updateSceneGraph()
def on3DTextChanged(self, string):
self.item3d().setRenderString(str(string), self.item3d().getFontSize())
if self.inspector: self.inspector().updateSceneGraph()