Source code for layer_viewer.layer_view_widget

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


_nameToPattern = {
    'SolidPattern': QtCore.Qt.SolidPattern,
    'LinearGradientPattern': QtCore.Qt.LinearGradientPattern,
    'Dense1Pattern': QtCore.Qt.Dense1Pattern,
    'Dense2Pattern': QtCore.Qt.Dense2Pattern,
    'Dense3Pattern': QtCore.Qt.Dense3Pattern,
    'Dense4Pattern': QtCore.Qt.Dense4Pattern,
    'Dense5Pattern': QtCore.Qt.Dense5Pattern,
    'Dense6Pattern': QtCore.Qt.Dense6Pattern,
    'Dense7Pattern': QtCore.Qt.Dense7Pattern,
    'NoBrush': QtCore.Qt.NoBrush,
    'HorPattern': QtCore.Qt.HorPattern,
    'VerPattern': QtCore.Qt.VerPattern,
    'CrossPattern': QtCore.Qt.CrossPattern,
    'BDiagPattern': QtCore.Qt.BDiagPattern,
    'FDiagPattern': QtCore.Qt.FDiagPattern,
    'DiagCrossPattern': QtCore.Qt.DiagCrossPattern
}


[docs]def getQtPattern(name):
    return _nameToPattern[str(name)]




[docs]class MyViewBox(pg.ViewBox):
    def __init__(self):
        super(MyViewBox, self).__init__()


[docs]    def keyPressEvent(self, ev):
        pass

        #ev.ignore()


[docs]class LayerViewWidget(QtGui.QWidget):
    def __init__(self,settings_widget, parent=None):
        QtGui.QWidget.__init__(self, parent)


        self.graphView       = pg.GraphicsView()
        self.graphViewLayout = QtGui.QGraphicsGridLayout()
        self.graphView.centralWidget.setLayout(self.graphViewLayout)

        #self.setPolicy(self.graphView,QtGui.QSizePolicy.Expanding)


        # view box
        self.view_box = MyViewBox()#parent=self)
        self.view_box.setAspectLocked(True)
        # add view box to graph view layout
        self.graphViewLayout.addItem(self.view_box,0,0)
        self.hbox = QtGui.QHBoxLayout()
        self.setLayout(self.hbox)
        self.hbox.addWidget(self.graphView)

        # flip the view box
        self.view_box.invertY(True)


        self.settings_widget = settings_widget

        def bg_change(*args,**kwargs):
            #print("waerawe")
            self.setBackground()


        bg_params = settings_widget.p.param('ViewBox Options','ViewBox Background')
        
        # Too lazy for recursion:
        for child in bg_params.children():
            child.sigValueChanged.connect(bg_change)
            for ch2 in child.children():
                ch2.sigValueChanged.connect(bg_change)
                

        self.setBackground()
                

 

        s = self.view_box.menu.addAction('Settings')
        s.triggered.connect(self.showSettings)



[docs]    def showSettings(self):
        print("settings")
        self.settings_widget.show()



[docs]    def setBackground(self):        

        self.bgType  = self.settings_widget.p[('ViewBox Options','ViewBox Background','bg-type')]
        self.bgColor1 = self.settings_widget.p[('ViewBox Options','ViewBox Background','bg-color 1')]
        self.bgColor2 = self.settings_widget.p[('ViewBox Options','ViewBox Background','bg-color 2')]

        bg = self.view_box.background
        self.view_box.background.show()
        bg.setVisible(True)
        if self.bgType == 'LinearGradientPattern':
            g =  QtGui.QLinearGradient(
                                       QtCore.QRectF(self.rect()).topLeft(),
                                       QtCore.QRectF(self.rect()).bottomLeft()
            )
            g.setColorAt(0, self.bgColor1);
            g.setColorAt(1, self.bgColor2);
            brush = QtGui.QBrush(g)
        else:
            brush = QtGui.QBrush()
            brush.setStyle(getQtPattern(self.bgType))
            brush.setColor(self.bgColor1)

        bg.setBrush(brush)