class RenderParameterWidget(QWidget):
"""
RenderParameterWidget is a widget that is shown in the render property
widget. It holds a combo box with which different visualizations can be
chosen. Beneath the combo box it displays a widget in a scroll view that
contains widgets with which parameters of the visualization can be adjusted.
"""
def __init__(self, renderController, parent=None):
super(RenderParameterWidget, self).__init__(parent=parent)
self.renderController = renderController
self.renderController.visualizationChanged.connect(self.visualizationLoaded)
self.paramWidget = None
self.visTypeComboBox = QComboBox()
for visualizationType in self.renderController.visualizationTypes:
self.visTypeComboBox.addItem(visualizationType)
layout = QGridLayout()
layout.setAlignment(Qt.AlignTop)
layout.setSpacing(10)
layout.setContentsMargins(10, 0, 10, 0)
if len(self.renderController.visualizationTypes) > 1:
layout.addWidget(QLabel("Visualization type:"), 0, 0)
layout.addWidget(self.visTypeComboBox, 0, 1)
self.setLayout(layout)
self.scrollArea = QScrollArea()
self.scrollArea.setFrameShape(QFrame.NoFrame)
self.scrollArea.setAutoFillBackground(False)
self.scrollArea.setAttribute(Qt.WA_TranslucentBackground)
self.scrollArea.setWidgetResizable(True)
self.visTypeComboBox.currentIndexChanged.connect(self.visTypeComboBoxChanged)
def UpdateWidgetFromRenderWidget(self):
"""
Update the parameter widget with a widget from the render widget.
"""
# Add the scroll area for the parameter widget if it is not there yet
layout = self.layout()
if layout.indexOf(self.scrollArea) == -1:
layout.addWidget(self.scrollArea, 1, 0, 1, 2)
self.setLayout(layout)
# Clear the previous parameter widget
if self.paramWidget is not None:
self.paramWidget.setParent(None)
if self.renderController.visualization is not None:
self.renderController.visualization.disconnect(SIGNAL("updatedTransferFunction"), self.transferFunctionChanged)
# Get a new parameter widget from the render widget
self.paramWidget = self.renderController.getParameterWidget()
Style.styleWidgetForTab(self.paramWidget)
self.scrollArea.setWidget(self.paramWidget)
if self.renderController.visualization is not None:
self.renderController.visualization.updatedTransferFunction.connect(self.transferFunctionChanged)
self.visTypeComboBox.setCurrentIndex(self.visTypeComboBox.findText(self.renderController.visualizationType))
@Slot(int)
def visTypeComboBoxChanged(self, index):
"""
Slot that changes the render type. Also updates parameters and makes
sure that the renderWidget renders with the new visualizationType.
:type index: any
"""
self.renderController.setVisualizationType(self.visTypeComboBox.currentText())
self.UpdateWidgetFromRenderWidget()
self.renderController.updateVisualization()
def visualizationLoaded(self, visualization):
self.UpdateWidgetFromRenderWidget()
@Slot()
def transferFunctionChanged(self):
"""
Slot that can be used when a transfer function has changed so that
the render will be updated afterwards.
Should be called on valueChanged by the widgets from the parameter widget.
"""
self.renderController.updateVisualization()
class PointsWidget(QWidget): """ PointsWidget """ activeLandmarkChanged = Signal(int) landmarkDeleted = Signal(int) def __init__(self): super(PointsWidget, self).__init__() self.landmarkWidgets = [] self.activeIndex = 0 self.scrollArea = QScrollArea(self) self.scrollArea.setFrameShape(QFrame.NoFrame) self.scrollArea.setAutoFillBackground(False) self.scrollArea.setAttribute(Qt.WA_TranslucentBackground) self.scrollArea.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff) self.scrollArea.setWidgetResizable(True) landmarkLocationsLayout = QGridLayout() landmarkLocationsLayout.setSpacing(0) landmarkLocationsLayout.setContentsMargins(0, 0, 0, 0) landmarkLocationsLayout.setAlignment(Qt.AlignTop) self.landmarkLocationsWidget = QWidget() Style.styleWidgetForTab(self.landmarkLocationsWidget) self.landmarkLocationsWidget.setLayout(landmarkLocationsLayout) self.scrollArea.setWidget(self.landmarkLocationsWidget) layout = QGridLayout() layout.setContentsMargins(0, 0, 0, 0) layout.addWidget(self.scrollArea) self.setLayout(layout) @Slot(list) def setPoints(self, points): self._clearLandmarkWidgets() layout = self.landmarkLocationsWidget.layout() for index in range(len(points)): landmarkWidget = LandmarkLocationWidget() landmarkWidget.setIndex(index) landmarkWidget.active = (index == self.activeIndex) landmarkWidget.activated.connect(self.activateLandmark) landmarkWidget.deleted.connect(self.deleteLandmark) landmarkWidget.setLandmarkSet(points[index]) layout.addWidget(landmarkWidget, index, 0) self.landmarkWidgets.append(landmarkWidget) def _clearLandmarkWidgets(self): layout = self.landmarkLocationsWidget.layout() for widget in self.landmarkWidgets: widget.activated.disconnect() layout.removeWidget(widget) widget.deleteLater() self.landmarkWidgets = [] @Slot(int, object) def activateLandmark(self, index, state): if not state: self.activeIndex = len(self.landmarkWidgets) else: self.activeIndex = index self.activeLandmarkChanged.emit(self.activeIndex) @Slot(int) def deleteLandmark(self, index): self.activateLandmark(index, False) self.landmarkDeleted.emit(index)
class RenderInfoWidget(QWidget):
"""
RenderInfoWidget shows information about the loaded dataset. Things like
filenames, range of data values, size of data, etc.
"""
def __init__(self):
super(RenderInfoWidget, self).__init__()
self.scrollArea = QScrollArea()
self.scrollArea.setFrameShape(QFrame.NoFrame)
self.scrollArea.setAutoFillBackground(False)
self.scrollArea.setAttribute(Qt.WA_TranslucentBackground)
self.scrollArea.setWidgetResizable(True)
Style.styleWidgetForTab(self)
Style.styleWidgetForTab(self.scrollArea)
@Slot(basestring)
def setFile(self, fileName):
"""
Slot that reads properties of the dataset and displays them in a few widgets.
"""
if fileName is None:
return
self.fileName = fileName
# Read info from dataset
# TODO: read out the real world dimensions in inch or cm
# TODO: scalar type (int, float, short, etc.)
imageReader = DataReader()
imageData = imageReader.GetImageData(fileName)
directory, name = os.path.split(fileName)
dimensions = imageData.GetDimensions()
minimum, maximum = imageData.GetScalarRange()
scalarType = imageData.GetScalarTypeAsString()
bins = DataAnalyzer.histogramForData(imageData, 256)
self.histogram = Histogram()
self.histogram.bins = bins
self.histogram.enabled = True
self.histogramWidget = HistogramWidget()
self.histogramWidget.setMinimumHeight(100)
self.histogramWidget.setHistogram(self.histogram)
self.histogramWidget.setAxeMode(bottom=HistogramWidget.AxeClear,
left=HistogramWidget.AxeLog)
Style.styleWidgetForTab(self.histogramWidget)
nameText = name
dimsText = "(" + str(dimensions[0]) + ", " + str(dimensions[1]) + ", " + str(dimensions[2]) + ")"
voxsText = str(dimensions[0] * dimensions[1] * dimensions[2])
rangText = "[" + str(minimum) + " : " + str(maximum) + "]"
typeText = scalarType
layout = self.layout()
if not layout:
# Create a new layout
layout = QGridLayout()
layout.setAlignment(Qt.AlignTop)
# Create string representations
nameLabels = []
nameLabels.append(QLabel("File name:"))
nameLabels.append(QLabel("Dimensions:"))
nameLabels.append(QLabel("Voxels:"))
nameLabels.append(QLabel("Range:"))
nameLabels.append(QLabel("Data type:"))
for label in nameLabels:
label.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
# Create 'dynamic' labels
self.labelTitle = QLabel(nameText)
self.labelDimensions = QLabel(dimsText)
self.labelVoxels = QLabel(voxsText)
self.labelRange = QLabel(rangText)
self.labelType = QLabel(typeText)
index = 0
for label in nameLabels:
layout.addWidget(label, index, 0)
index += 1
layout.addWidget(self.labelTitle, 0, 1)
layout.addWidget(self.labelDimensions, 1, 1)
layout.addWidget(self.labelVoxels, 2, 1)
layout.addWidget(self.labelRange, 3, 1)
layout.addWidget(self.labelType, 4, 1)
layout.addWidget(self.histogramWidget, 5, 0, 1, 2)
widget = QWidget()
widget.setLayout(layout)
Style.styleWidgetForTab(widget)
self.scrollArea.setWidget(widget)
scrollLayout = QGridLayout()
scrollLayout.setSpacing(0)
scrollLayout.setContentsMargins(0, 0, 0, 0)
scrollLayout.addWidget(self.scrollArea)
self.setLayout(scrollLayout)
else:
# Just update the text for the 'dynamic' labels
self.labelTitle.setText(nameText)
self.labelDimensions.setText(dimsText)
self.labelVoxels.setText(voxsText)
self.labelRange.setText(rangText)
self.labelType.setText(typeText)