def __init__(self, parentApplet, topLevelOperatorView, labelingDrawerUiPath=None ):
self.parentApplet = parentApplet
# Tell our base class which slots to monitor
labelSlots = LabelingGui.LabelingSlots()
labelSlots.labelInput = topLevelOperatorView.LabelInputs
labelSlots.labelOutput = topLevelOperatorView.LabelImages
labelSlots.labelEraserValue = topLevelOperatorView.opLabelPipeline.opLabelArray.eraser
labelSlots.labelDelete = topLevelOperatorView.opLabelPipeline.DeleteLabel
labelSlots.labelNames = topLevelOperatorView.LabelNames
self.__cleanup_fns = []
# We provide our own UI file (which adds an extra control for interactive mode)
if labelingDrawerUiPath is None:
labelingDrawerUiPath = os.path.split(__file__)[0] + '/labelingDrawer.ui'
# Base class init
super(PixelClassificationGui, self).__init__( parentApplet, labelSlots, topLevelOperatorView, labelingDrawerUiPath )
self.topLevelOperatorView = topLevelOperatorView
self.interactiveModeActive = False
# Immediately update our interactive state
self.toggleInteractive( not self.topLevelOperatorView.FreezePredictions.value )
self._currentlySavingPredictions = False
self.labelingDrawerUi.labelListView.support_merges = True
self.labelingDrawerUi.liveUpdateButton.setEnabled(False)
self.labelingDrawerUi.liveUpdateButton.setIcon( QIcon(ilastikIcons.Play) )
self.labelingDrawerUi.liveUpdateButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
self.labelingDrawerUi.liveUpdateButton.toggled.connect( self.toggleInteractive )
self.initFeatSelDlg()
self.labelingDrawerUi.suggestFeaturesButton.clicked.connect(self.show_feature_selection_dialog)
self.featSelDlg.accepted.connect(self.update_features_from_dialog)
self.labelingDrawerUi.suggestFeaturesButton.setEnabled(False)
self.topLevelOperatorView.LabelNames.notifyDirty( bind(self.handleLabelSelectionChange) )
self.__cleanup_fns.append( partial( self.topLevelOperatorView.LabelNames.unregisterDirty, bind(self.handleLabelSelectionChange) ) )
self._initShortcuts()
self._bookmarks_window = BookmarksWindow(self, self.topLevelOperatorView)
# FIXME: We MUST NOT enable the render manager by default,
# since it will drastically slow down the app for large volumes.
# For now, we leave it off by default.
# To re-enable rendering, we need to allow the user to render a segmentation
# and then initialize the render manager on-the-fly.
# (We might want to warn the user if her volume is not small.)
self.render = False
self._renderMgr = None
self._renderedLayers = {} # (layer name, label number)
# Always off for now (see note above)
if self.render:
try:
self._renderMgr = RenderingManager( self.editor.view3d )
except:
self.render = False
# toggle interactive mode according to freezePredictions.value
self.toggleInteractive(not self.topLevelOperatorView.FreezePredictions.value)
def FreezePredDirty():
self.toggleInteractive(not self.topLevelOperatorView.FreezePredictions.value)
# listen to freezePrediction changes
self.topLevelOperatorView.FreezePredictions.notifyDirty( bind(FreezePredDirty) )
self.__cleanup_fns.append( partial( self.topLevelOperatorView.FreezePredictions.unregisterDirty, bind(FreezePredDirty) ) )
def __init__(self, parentApplet, topLevelOperatorView, drawerUiPath=None):
self.topLevelOperatorView = topLevelOperatorView
self.isInitialized = (
False
) # Need this flag in carvingApplet where initialization is terminated with label selection
# members
self._doneSegmentationLayer = None
self._showSegmentationIn3D = False
# self._showUncertaintyLayer = False
# end: members
labelingSlots = LabelingGui.LabelingSlots()
labelingSlots.labelInput = topLevelOperatorView.WriteSeeds
labelingSlots.labelOutput = topLevelOperatorView.opLabelArray.Output
labelingSlots.labelEraserValue = topLevelOperatorView.opLabelArray.EraserLabelValue
labelingSlots.labelNames = topLevelOperatorView.LabelNames
labelingSlots.labelDelete = topLevelOperatorView.opLabelArray.DeleteLabel
labelingSlots.maxLabelValue = topLevelOperatorView.opLabelArray.MaxLabelValue
# We provide our own UI file (which adds an extra control for interactive mode)
directory = os.path.split(__file__)[0]
if drawerUiPath is None:
drawerUiPath = os.path.join(directory, "carvingDrawer.ui")
self.dialogdirCOM = os.path.join(directory,
"carvingObjectManagement.ui")
self.dialogdirSAD = os.path.join(directory, "saveAsDialog.ui")
# Add 3DWidget only if the data is 3D
is_3d = self._is_3d()
super(CarvingGui, self).__init__(parentApplet,
labelingSlots,
topLevelOperatorView,
drawerUiPath,
is_3d_widget_visible=is_3d)
self.parentApplet = parentApplet
self.labelingDrawerUi.currentObjectLabel.setText("<not saved yet>")
# Init special base class members
self.minLabelNumber = 2
self.maxLabelNumber = 2
mgr = ShortcutManager()
ActionInfo = ShortcutManager.ActionInfo
# set up keyboard shortcuts
mgr.register(
"3",
ActionInfo(
"Carving",
"Run interactive segmentation",
"Run interactive segmentation",
self.labelingDrawerUi.segment.click,
self.labelingDrawerUi.segment,
self.labelingDrawerUi.segment,
),
)
# Disable 3D view by default
self.render = False
if is_3d:
try:
self._renderMgr = RenderingManager(self.editor.view3d)
self._shownObjects3D = {}
self.render = True
except:
self.render = False
# Segmentation is toggled on by default in _after_init, below.
# (We can't enable it until the layers are all present.)
self._showSegmentationIn3D = False
self._segmentation_3d_label = None
self.labelingDrawerUi.segment.clicked.connect(self.onSegmentButton)
self.labelingDrawerUi.segment.setEnabled(True)
self.topLevelOperatorView.Segmentation.notifyDirty(
bind(self._segmentation_dirty))
self.topLevelOperatorView.HasSegmentation.notifyValueChanged(
bind(self._updateGui))
## uncertainty
# self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(False)
# self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(False)
# def onUncertaintyFGButton():
# logger.debug( "uncertFG button clicked" )
# pos = self.topLevelOperatorView.getMaxUncertaintyPos(label=2)
# self.editor.posModel.slicingPos = (pos[0], pos[1], pos[2])
# self.labelingDrawerUi.pushButtonUncertaintyFG.clicked.connect(onUncertaintyFGButton)
# def onUncertaintyBGButton():
# logger.debug( "uncertBG button clicked" )
# pos = self.topLevelOperatorView.getMaxUncertaintyPos(label=1)
# self.editor.posModel.slicingPos = (pos[0], pos[1], pos[2])
# self.labelingDrawerUi.pushButtonUncertaintyBG.clicked.connect(onUncertaintyBGButton)
# def onUncertaintyCombo(value):
# if value == 0:
# value = "none"
# self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(False)
# self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(False)
# self._showUncertaintyLayer = False
# else:
# if value == 1:
# value = "localMargin"
# elif value == 2:
# value = "exchangeCount"
# elif value == 3:
# value = "gabow"
# else:
# raise RuntimeError("unhandled case '%r'" % value)
# self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(True)
# self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(True)
# self._showUncertaintyLayer = True
# logger.debug( "uncertainty changed to %r" % value )
# self.topLevelOperatorView.UncertaintyType.setValue(value)
# self.updateAllLayers() #make sure that an added/deleted uncertainty layer is recognized
# self.labelingDrawerUi.uncertaintyCombo.currentIndexChanged.connect(onUncertaintyCombo)
self.labelingDrawerUi.objPrefix.setText(self.objectPrefix)
self.labelingDrawerUi.objPrefix.textChanged.connect(
self.setObjectPrefix)
## save
self.labelingDrawerUi.save.clicked.connect(self.onSaveButton)
## clear
self.labelingDrawerUi.clear.clicked.connect(self._onClearAction)
## object names
self.labelingDrawerUi.namesButton.clicked.connect(
self.onShowObjectNames)
if hasattr(self.labelingDrawerUi, "exportAllMeshesButton"):
self.labelingDrawerUi.exportAllMeshesButton.clicked.connect(
self._exportAllObjectMeshes)
self.labelingDrawerUi.labelListView.allowDelete = False
self._labelControlUi.labelListModel.allowRemove(False)
def layerIndexForName(name):
return self.layerstack.findMatchingIndex(lambda x: x.name == name)
def addLayerToggleShortcut(layername, shortcut):
def toggle():
row = layerIndexForName(layername)
self.layerstack.selectRow(row)
layer = self.layerstack[row]
layer.visible = not layer.visible
self.viewerControlWidget().layerWidget.setFocus()
mgr.register(
shortcut,
ActionInfo(
"Carving",
"Toggle layer %s" % layername,
"Toggle layer %s" % layername,
toggle,
self.viewerControlWidget(),
None,
),
)
# TODO
addLayerToggleShortcut("Completed segments (unicolor)", "d")
addLayerToggleShortcut("Segmentation", "s")
addLayerToggleShortcut("Input Data", "r")
def makeColortable():
self._doneSegmentationColortable = [QColor(0, 0, 0, 0).rgba()]
for i in range(254):
r, g, b = numpy.random.randint(0, 255), numpy.random.randint(
0, 255), numpy.random.randint(0, 255)
# ensure colors have sufficient distance to pure red and pure green
while (255 - r) + g + b < 128 or r + (255 - g) + b < 128:
r, g, b = numpy.random.randint(
0, 255), numpy.random.randint(
0, 255), numpy.random.randint(0, 255)
self._doneSegmentationColortable.append(QColor(r, g, b).rgba())
self._doneSegmentationColortable.append(QColor(0, 255, 0).rgba())
makeColortable()
self._updateGui()
def __init__(self,
parentApplet,
topLevelOperatorView,
labelingDrawerUiPath=None):
self.parentApplet = parentApplet
self.isInitialized = (
False
) # need this flag in pixelClassificationApplet where initialization is terminated with label selection
# Tell our base class which slots to monitor
labelSlots = LabelingGui.LabelingSlots()
labelSlots.labelInput = topLevelOperatorView.LabelInputs
labelSlots.labelOutput = topLevelOperatorView.LabelImages
labelSlots.labelEraserValue = topLevelOperatorView.opLabelPipeline.opLabelArray.eraser
labelSlots.labelDelete = topLevelOperatorView.opLabelPipeline.DeleteLabel
labelSlots.labelNames = topLevelOperatorView.LabelNames
self.__cleanup_fns = []
# We provide our own UI file (which adds an extra control for interactive mode)
if labelingDrawerUiPath is None:
labelingDrawerUiPath = os.path.split(
__file__)[0] + "/labelingDrawer.ui"
# Base class init
super(PixelClassificationGui,
self).__init__(parentApplet, labelSlots, topLevelOperatorView,
labelingDrawerUiPath,
topLevelOperatorView.InputImages)
self.topLevelOperatorView = topLevelOperatorView
self._currentlySavingPredictions = False
self.labelingDrawerUi.labelListView.support_merges = True
self.labelingDrawerUi.liveUpdateButton.toggled.connect(
self.setLiveUpdateEnabled)
self.initFeatSelDlg()
self.labelingDrawerUi.suggestFeaturesButton.clicked.connect(
self.show_feature_selection_dialog)
self.featSelDlg.accepted.connect(self.update_features_from_dialog)
self.labelingDrawerUi.suggestFeaturesButton.setEnabled(False)
# Always force at least two labels because it makes no sense to have less here
self.forceAtLeastTwoLabels(True)
self._initShortcuts()
self._bookmarks_window = BookmarksWindow(self,
self.topLevelOperatorView)
# FIXME: We MUST NOT enable the render manager by default,
# since it will drastically slow down the app for large volumes.
# For now, we leave it off by default.
# To re-enable rendering, we need to allow the user to render a segmentation
# and then initialize the render manager on-the-fly.
# (We might want to warn the user if her volume is not small.)
self.render = False
self._renderMgr = None
self._renderedLayers = {} # (layer name, label number)
# Always off for now (see note above)
if self.render:
try:
self._renderMgr = RenderingManager(self.editor.view3d)
except:
self.render = False
# listen to freezePrediction changes
unsub_callback = self.topLevelOperatorView.FreezePredictions.notifyDirty(
lambda *args: self.setLiveUpdateEnabled())
self.__cleanup_fns.append(unsub_callback)
self.setLiveUpdateEnabled()
def __init__(self,
parentApplet,
topLevelOperatorView,
labelingDrawerUiPath=None):
self.parentApplet = parentApplet
# Tell our base class which slots to monitor
labelSlots = LabelingGui.LabelingSlots()
labelSlots.labelInput = topLevelOperatorView.LabelInputs
labelSlots.labelOutput = topLevelOperatorView.LabelImages
labelSlots.labelEraserValue = topLevelOperatorView.opLabelPipeline.opLabelArray.eraser
labelSlots.labelDelete = topLevelOperatorView.opLabelPipeline.DeleteLabel
labelSlots.labelNames = topLevelOperatorView.LabelNames
self.__cleanup_fns = []
# We provide our own UI file (which adds an extra control for interactive mode)
if labelingDrawerUiPath is None:
labelingDrawerUiPath = os.path.dirname(
__file__) + "/labelingDrawer.ui"
# Base class init
super(VoxelSegmentationGui,
self).__init__(parentApplet, labelSlots, topLevelOperatorView,
labelingDrawerUiPath)
self.topLevelOperatorView = topLevelOperatorView
self.interactiveModeActive = False
# Immediately update our interactive state
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
self._currentlySavingPredictions = False
self._showSegmentationIn3D = False
self.labelingDrawerUi.labelListView.support_merges = True
self.labelingDrawerUi.liveUpdateButton.setEnabled(False)
self.labelingDrawerUi.liveUpdateButton.setIcon(QIcon(
ilastikIcons.Play))
self.labelingDrawerUi.liveUpdateButton.setToolButtonStyle(
Qt.ToolButtonTextBesideIcon)
self.labelingDrawerUi.liveUpdateButton.toggled.connect(
self.toggleInteractive)
self.initFeatSelDlg()
self.labelingDrawerUi.suggestFeaturesButton.clicked.connect(
self.show_feature_selection_dialog)
self.featSelDlg.accepted.connect(self.update_features_from_dialog)
self.labelingDrawerUi.suggestFeaturesButton.setEnabled(False)
# Always force at least two labels because it makes no sense to have less here
self.forceAtLeastTwoLabels(True)
self.topLevelOperatorView.LabelNames.notifyDirty(
bind(self.handleLabelSelectionChange))
self.__cleanup_fns.append(
partial(self.topLevelOperatorView.LabelNames.unregisterDirty,
bind(self.handleLabelSelectionChange)))
self._initShortcuts()
self._bookmarks_window = pixelClassificationGui.BookmarksWindow(
self, self.topLevelOperatorView)
# FIXME: We MUST NOT enable the render manager by default,
# since it will drastically slow down the app for large volumes.
# For now, we leave it off by default.
# To re-enable rendering, we need to allow the user to render a segmentation
# and then initialize the render manager on-the-fly.
# (We might want to warn the user if her volume is not small.)
self.render = False
self._renderMgr = None
self._renderedLayers = {} # (layer name, label number)
# Always off for now (see note above)
if self.render:
# try:
self._renderMgr = RenderingManager(self.editor.view3d)
# except:
# self.render = False
# toggle interactive mode according to freezePredictions.value
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
def FreezePredDirty():
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
# listen to freezePrediction changes
# self.topLevelOperatorView.FreezePredictions.notifyDirty(bind(FreezePredDirty))
self.__cleanup_fns.append(
partial(
self.topLevelOperatorView.FreezePredictions.unregisterDirty,
bind(FreezePredDirty)))
def SelectBestAnnotationPlane():
if self.mostUncertainPlanes is None and self.mostUncertainAxis is None:
axis, indices = self.topLevelOperatorView.BestAnnotationPlane.value
# Axes from the classifiation operator are in the order z y x while volumina has x y z
axis = [2, 1, 0][axis]
self.mostUncertainAxis = axis
self.mostUncertainPlanes = indices
# Pop most uncertain plane from list
index, self.mostUncertainPlanes = self.mostUncertainPlanes[
-1], self.mostUncertainPlanes[:-1]
# Display the plane
self.editor.navCtrl.changeSliceAbsolute(index,
self.mostUncertainAxis)
def resetBestPlanes():
logger.info("resetting planes")
self.mostUncertainPlanes = None
self.mostUncertainAxis = None
resetBestPlanes()
self.topLevelOperatorView.FreezePredictions.notifyDirty(
bind(resetBestPlanes))
self.topLevelOperatorView.LabelImages.notifyDirty(
bind(resetBestPlanes))
self.labelingDrawerUi.bestAnnotationPlaneButton.clicked.connect(
SelectBestAnnotationPlane)
def __init__(self, parentApplet, topLevelOperatorView, drawerUiPath=None ):
self.topLevelOperatorView = topLevelOperatorView
#members
self._doneSegmentationLayer = None
self._showSegmentationIn3D = False
self._showUncertaintyLayer = False
#end: members
labelingSlots = LabelingGui.LabelingSlots()
labelingSlots.labelInput = topLevelOperatorView.WriteSeeds
labelingSlots.labelOutput = topLevelOperatorView.opLabelArray.Output
labelingSlots.labelEraserValue = topLevelOperatorView.opLabelArray.EraserLabelValue
labelingSlots.labelNames = topLevelOperatorView.LabelNames
labelingSlots.labelDelete = topLevelOperatorView.opLabelArray.DeleteLabel
labelingSlots.maxLabelValue = topLevelOperatorView.opLabelArray.MaxLabelValue
labelingSlots.labelsAllowed = topLevelOperatorView.LabelsAllowed
# We provide our own UI file (which adds an extra control for interactive mode)
directory = os.path.split(__file__)[0]
if drawerUiPath is None:
drawerUiPath = os.path.join(directory, 'carvingDrawer.ui')
self.dialogdirCOM = os.path.join(directory, 'carvingObjectManagement.ui')
self.dialogdirSAD = os.path.join(directory, 'saveAsDialog.ui')
rawInputSlot = topLevelOperatorView.RawData
super(CarvingGui, self).__init__(parentApplet, labelingSlots, topLevelOperatorView, drawerUiPath, rawInputSlot)
self.labelingDrawerUi.currentObjectLabel.setText("<not saved yet>")
# Init special base class members
self.minLabelNumber = 2
self.maxLabelNumber = 2
mgr = ShortcutManager()
ActionInfo = ShortcutManager.ActionInfo
#set up keyboard shortcuts
mgr.register( "3", ActionInfo( "Carving",
"Run interactive segmentation",
"Run interactive segmentation",
self.labelingDrawerUi.segment.click,
self.labelingDrawerUi.segment,
self.labelingDrawerUi.segment ) )
try:
self.render = True
self._shownObjects3D = {}
self._renderMgr = RenderingManager( self.editor.view3d )
except:
self.render = False
# Segmentation is toggled on by default in _after_init, below.
# (We can't enable it until the layers are all present.)
self._showSegmentationIn3D = False
self._segmentation_3d_label = None
self.labelingDrawerUi.segment.clicked.connect(self.onSegmentButton)
self.labelingDrawerUi.segment.setEnabled(True)
self.topLevelOperatorView.Segmentation.notifyDirty( bind( self._update_rendering ) )
self.topLevelOperatorView.HasSegmentation.notifyValueChanged( bind( self._updateGui ) )
## uncertainty
self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(False)
self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(False)
def onUncertaintyFGButton():
logger.debug( "uncertFG button clicked" )
pos = self.topLevelOperatorView.getMaxUncertaintyPos(label=2)
self.editor.posModel.slicingPos = (pos[0], pos[1], pos[2])
self.labelingDrawerUi.pushButtonUncertaintyFG.clicked.connect(onUncertaintyFGButton)
def onUncertaintyBGButton():
logger.debug( "uncertBG button clicked" )
pos = self.topLevelOperatorView.getMaxUncertaintyPos(label=1)
self.editor.posModel.slicingPos = (pos[0], pos[1], pos[2])
self.labelingDrawerUi.pushButtonUncertaintyBG.clicked.connect(onUncertaintyBGButton)
def onUncertaintyCombo(value):
if value == 0:
value = "none"
self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(False)
self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(False)
self._showUncertaintyLayer = False
else:
if value == 1:
value = "localMargin"
elif value == 2:
value = "exchangeCount"
elif value == 3:
value = "gabow"
else:
raise RuntimeError("unhandled case '%r'" % value)
self.labelingDrawerUi.pushButtonUncertaintyFG.setEnabled(True)
self.labelingDrawerUi.pushButtonUncertaintyBG.setEnabled(True)
self._showUncertaintyLayer = True
logger.debug( "uncertainty changed to %r" % value )
self.topLevelOperatorView.UncertaintyType.setValue(value)
self.updateAllLayers() #make sure that an added/deleted uncertainty layer is recognized
self.labelingDrawerUi.uncertaintyCombo.currentIndexChanged.connect(onUncertaintyCombo)
## background priority
def onBackgroundPrioritySpin(value):
logger.debug( "background priority changed to %f" % value )
self.topLevelOperatorView.BackgroundPriority.setValue(value)
self.labelingDrawerUi.backgroundPrioritySpin.valueChanged.connect(onBackgroundPrioritySpin)
def onBackgroundPriorityDirty(slot, roi):
oldValue = self.labelingDrawerUi.backgroundPrioritySpin.value()
newValue = self.topLevelOperatorView.BackgroundPriority.value
if newValue != oldValue:
self.labelingDrawerUi.backgroundPrioritySpin.setValue(newValue)
self.topLevelOperatorView.BackgroundPriority.notifyDirty(onBackgroundPriorityDirty)
## bias
def onNoBiasBelowDirty(slot, roi):
oldValue = self.labelingDrawerUi.noBiasBelowSpin.value()
newValue = self.topLevelOperatorView.NoBiasBelow.value
if newValue != oldValue:
self.labelingDrawerUi.noBiasBelowSpin.setValue(newValue)
self.topLevelOperatorView.NoBiasBelow.notifyDirty(onNoBiasBelowDirty)
def onNoBiasBelowSpin(value):
logger.debug( "background priority changed to %f" % value )
self.topLevelOperatorView.NoBiasBelow.setValue(value)
self.labelingDrawerUi.noBiasBelowSpin.valueChanged.connect(onNoBiasBelowSpin)
## save
self.labelingDrawerUi.save.clicked.connect(self.onSaveButton)
## clear
self.labelingDrawerUi.clear.clicked.connect(self._onClearAction)
## object names
self.labelingDrawerUi.namesButton.clicked.connect(self.onShowObjectNames)
if hasattr( self.labelingDrawerUi, 'exportAllMeshesButton' ):
self.labelingDrawerUi.exportAllMeshesButton.clicked.connect(self._exportAllObjectMeshes)
def labelBackground():
self.selectLabel(0)
def labelObject():
self.selectLabel(1)
self._labelControlUi.labelListModel.allowRemove(False)
bgToolTipObject = LabelListModel.EntryToolTipAdapter(self._labelControlUi.labelListModel, 0)
mgr.register( "1", ActionInfo( "Carving",
"Select background label",
"Select background label",
labelBackground,
self.viewerControlWidget(),
bgToolTipObject ) )
fgToolTipObject = LabelListModel.EntryToolTipAdapter(self._labelControlUi.labelListModel, 1)
mgr.register( "2", ActionInfo( "Carving",
"Select object label",
"Select object label",
labelObject,
self.viewerControlWidget(),
fgToolTipObject ) )
def layerIndexForName(name):
return self.layerstack.findMatchingIndex(lambda x: x.name == name)
def addLayerToggleShortcut(layername, shortcut):
def toggle():
row = layerIndexForName(layername)
self.layerstack.selectRow(row)
layer = self.layerstack[row]
layer.visible = not layer.visible
self.viewerControlWidget().layerWidget.setFocus()
mgr.register(shortcut, ActionInfo( "Carving",
"Toggle layer %s" % layername,
"Toggle layer %s" % layername,
toggle,
self.viewerControlWidget(),
None ) )
#TODO
addLayerToggleShortcut("Completed segments (unicolor)", "d")
addLayerToggleShortcut("Segmentation", "s")
addLayerToggleShortcut("Input Data", "r")
'''
def updateLayerTimings():
s = "Layer timings:\n"
for l in self.layerstack:
s += "%s: %f sec.\n" % (l.name, l.averageTimePerTile)
self.labelingDrawerUi.layerTimings.setText(s)
t = QTimer(self)
t.setInterval(1*1000) # 10 seconds
t.start()
t.timeout.connect(updateLayerTimings)
'''
def makeColortable():
self._doneSegmentationColortable = [QColor(0,0,0,0).rgba()]
for i in range(254):
r,g,b = numpy.random.randint(0,255), numpy.random.randint(0,255), numpy.random.randint(0,255)
# ensure colors have sufficient distance to pure red and pure green
while (255 - r)+g+b<128 or r+(255-g)+b<128:
r,g,b = numpy.random.randint(0,255), numpy.random.randint(0,255), numpy.random.randint(0,255)
self._doneSegmentationColortable.append(QColor(r,g,b).rgba())
self._doneSegmentationColortable.append(QColor(0,255,0).rgba())
makeColortable()
def onRandomizeColors():
if self._doneSegmentationLayer is not None:
logger.debug( "randomizing colors ..." )
makeColortable()
self._doneSegmentationLayer.colorTable = self._doneSegmentationColortable
if self.render and self._renderMgr.ready:
self._update_rendering()
#self.labelingDrawerUi.randomizeColors.clicked.connect(onRandomizeColors)
self._updateGui()
def __init__(self, topLevelOperatorView, shellRequestSignal,
guiControlSignal, predictionSerializer):
# Tell our base class which slots to monitor
labelSlots = LabelingGui.LabelingSlots()
labelSlots.labelInput = topLevelOperatorView.LabelInputs
labelSlots.labelOutput = topLevelOperatorView.LabelImages
labelSlots.labelEraserValue = topLevelOperatorView.opLabelPipeline.opLabelArray.eraser
labelSlots.labelDelete = topLevelOperatorView.opLabelPipeline.opLabelArray.deleteLabel
labelSlots.maxLabelValue = topLevelOperatorView.MaxLabelValue
labelSlots.labelsAllowed = topLevelOperatorView.LabelsAllowedFlags
labelSlots.LabelNames = topLevelOperatorView.LabelNames
self.__cleanup_fns = []
# We provide our own UI file (which adds an extra control for interactive mode)
labelingDrawerUiPath = os.path.split(
__file__)[0] + '/labelingDrawer.ui'
# Base class init
super(PixelClassificationGui,
self).__init__(labelSlots, topLevelOperatorView,
labelingDrawerUiPath)
self.topLevelOperatorView = topLevelOperatorView
self.shellRequestSignal = shellRequestSignal
self.guiControlSignal = guiControlSignal
self.predictionSerializer = predictionSerializer
self.interactiveModeActive = False
# Immediately update our interactive state
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
self._currentlySavingPredictions = False
self.labelingDrawerUi.liveUpdateButton.setEnabled(False)
self.labelingDrawerUi.liveUpdateButton.setIcon(QIcon(
ilastikIcons.Play))
self.labelingDrawerUi.liveUpdateButton.setToolButtonStyle(
Qt.ToolButtonTextBesideIcon)
self.labelingDrawerUi.liveUpdateButton.toggled.connect(
self.toggleInteractive)
self.topLevelOperatorView.MaxLabelValue.notifyDirty(
bind(self.handleLabelSelectionChange))
self.__cleanup_fns.append(
partial(self.topLevelOperatorView.MaxLabelValue.unregisterDirty,
bind(self.handleLabelSelectionChange)))
self._initShortcuts()
try:
self.render = True
self._renderedLayers = {} # (layer name, label number)
self._renderMgr = RenderingManager(
renderer=self.editor.view3d.qvtk.renderer,
qvtk=self.editor.view3d.qvtk)
except:
self.render = False
# toggle interactive mode according to freezePredictions.value
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
def FreezePredDirty():
self.toggleInteractive(
not self.topLevelOperatorView.FreezePredictions.value)
# listen to freezePrediction changes
self.topLevelOperatorView.FreezePredictions.notifyDirty(
bind(FreezePredDirty))
self.__cleanup_fns.append(
partial(
self.topLevelOperatorView.FreezePredictions.unregisterDirty,
bind(FreezePredDirty)))
