Source code for layer_viewer.layer_viewer_widget

import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui
from pyqtgraph.dockarea import *

from . layer_view_widget import LayerViewWidget 
from . layer_ctrl_widget import LayerCtrlWidget 
from . settings_widget import SettingsWidget


[docs]class LayerViewerWidget(QtGui.QWidget):
    def __init__(self, gui_stlye='splitter', parent=None):
        QtGui.QWidget.__init__(self, parent)


        self.m_hbox = QtGui.QHBoxLayout()
        self.setLayout(self.m_hbox)


        self.settings_widget = SettingsWidget()
        self.m_layer_view_widget = LayerViewWidget(settings_widget=self.settings_widget)
        self.m_layer_ctrl_widget = LayerCtrlWidget()


        if gui_stlye == 'dock':

            self.area = DockArea()
            self.m_hbox.addWidget(self.area)
            d_view = Dock("Viewer", size=(500, 500))
            d_ctrl = Dock("Ctrl",   size=(200, 500))
            d_view.addWidget(self.m_layer_view_widget)
            d_ctrl.addWidget(self.m_layer_ctrl_widget)
            self.area.addDock(d_view)
            self.area.addDock(d_ctrl,'right', d_view)

        elif gui_stlye == 'splitter':
            self.splitter = QtGui.QSplitter()
            self.m_hbox.addWidget(self.splitter)
            self.splitter.addWidget(self.m_layer_view_widget)
            self.splitter.addWidget(self.m_layer_ctrl_widget)
            

        self.m_layers = dict()

        self.m_exlusive_layer = None


    @property
    def view_box(self):
        return self.m_layer_view_widget.view_box
    

[docs]    def addLayer(self, layer, opacity=1.0, visible=True):

        image_item = layer.get_image_item()
        self.m_layer_ctrl_widget.add_layer(layer) 
        self.view_box.addItem(image_item)  
        self.m_layers[layer.name] = layer
        self.setLayerVisibility(layer.name, bool(visible))
        self.setLayerOpacity(layer.name, float(opacity))
        layer.viewer = self

[docs]    def removeAllLayers(self):
        to_rm = list(self.m_layers.keys())
        for l in to_rm:
            self.removeLayer(l)

[docs]    def removeLayer(self, layer_name):
        layer = self.m_layers[layer_name] 
        self.view_box.removeItem(layer.get_image_item())
        self.m_layer_ctrl_widget.remove_layer(layer)
        del self.m_layers[layer_name] 

[docs]    def hasLayer(self, layer_name):
        return layer_name in self.m_layers

[docs]    def layerVisibility(self, layer_name):
        layer = self.m_layers[layer_name] 
        image_item = layer.get_image_item()
        return  image_item.isVisible()

[docs]    def layerOpacity(self, layer_name):
        layer = self.m_layers[layer_name] 
        image_item = layer.get_image_item()
        return  image_item.opacity()

[docs]    def setLayerVisibility(self, layer_name, visible):
        self.m_layers[layer_name].setVisible(visible)
[docs]    def setLayerOpacity(self, layer_name, opacity):
        self.m_layers[layer_name].setOpacity(opacity)

[docs]    def updateData(self, layer_name, **kwargs):
        self.m_layers[layer_name].updateData(**kwargs)

[docs]    def setData(self, layer_name, **kwargs):
        self.m_layers[layer_name].setData(**kwargs)


[docs]    def showAndHideOthers(self, layer_name):

        for ln in self.m_layers.keys():
            layer = self.m_layers[ln]
            if ln == layer_name:
                layer.setVisible(2)
                self.m_exlusive_layer = layer
            else:
                layer.setVisible(False)