Source code for layer_viewer.widgets.gradient_widget

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


# custom gradient widget with custom sizes
[docs]class GradientWidget(pg.GradientWidget):
    def __int__(self, *args, **kwargs):
        super(GradientWidget, self).__init__(*args, **kwargs)
        self.setMaxDim(3)
        self.setMaximumHeight(40)

[docs]    def sizeHint(self):
        sh = super(GradientWidget, self).sizeHint()
        new_sh = QtCore.QSize(10, sh.height())
        return new_sh

[docs]    def loadPreset(self, name):
        self.item.loadPreset(name)