314 lines
10 KiB
Python
314 lines
10 KiB
Python
from pyqtgraph.Qt import QtCore, QtGui
|
|
|
|
from .DockDrop import *
|
|
from pyqtgraph.widgets.VerticalLabel import VerticalLabel
|
|
|
|
class Dock(QtGui.QWidget, DockDrop):
|
|
|
|
sigStretchChanged = QtCore.Signal()
|
|
|
|
def __init__(self, name, area=None, size=(10, 10), widget=None, hideTitle=False, autoOrientation=True):
|
|
QtGui.QWidget.__init__(self)
|
|
DockDrop.__init__(self)
|
|
self.area = area
|
|
self.label = DockLabel(name, self)
|
|
self.labelHidden = False
|
|
self.moveLabel = True ## If false, the dock is no longer allowed to move the label.
|
|
self.autoOrient = autoOrientation
|
|
self.orientation = 'horizontal'
|
|
#self.label.setAlignment(QtCore.Qt.AlignHCenter)
|
|
self.topLayout = QtGui.QGridLayout()
|
|
self.topLayout.setContentsMargins(0, 0, 0, 0)
|
|
self.topLayout.setSpacing(0)
|
|
self.setLayout(self.topLayout)
|
|
self.topLayout.addWidget(self.label, 0, 1)
|
|
self.widgetArea = QtGui.QWidget()
|
|
self.topLayout.addWidget(self.widgetArea, 1, 1)
|
|
self.layout = QtGui.QGridLayout()
|
|
self.layout.setContentsMargins(0, 0, 0, 0)
|
|
self.layout.setSpacing(0)
|
|
self.widgetArea.setLayout(self.layout)
|
|
self.widgetArea.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
|
|
self.widgets = []
|
|
self.currentRow = 0
|
|
#self.titlePos = 'top'
|
|
self.raiseOverlay()
|
|
self.hStyle = """
|
|
Dock > QWidget {
|
|
border: 1px solid #000;
|
|
border-radius: 5px;
|
|
border-top-left-radius: 0px;
|
|
border-top-right-radius: 0px;
|
|
border-top-width: 0px;
|
|
}"""
|
|
self.vStyle = """
|
|
Dock > QWidget {
|
|
border: 1px solid #000;
|
|
border-radius: 5px;
|
|
border-top-left-radius: 0px;
|
|
border-bottom-left-radius: 0px;
|
|
border-left-width: 0px;
|
|
}"""
|
|
self.nStyle = """
|
|
Dock > QWidget {
|
|
border: 1px solid #000;
|
|
border-radius: 5px;
|
|
}"""
|
|
self.dragStyle = """
|
|
Dock > QWidget {
|
|
border: 4px solid #00F;
|
|
border-radius: 5px;
|
|
}"""
|
|
self.setAutoFillBackground(False)
|
|
self.widgetArea.setStyleSheet(self.hStyle)
|
|
|
|
self.setStretch(*size)
|
|
|
|
if widget is not None:
|
|
self.addWidget(widget)
|
|
|
|
if hideTitle:
|
|
self.hideTitleBar()
|
|
|
|
def implements(self, name=None):
|
|
if name is None:
|
|
return ['dock']
|
|
else:
|
|
return name == 'dock'
|
|
|
|
def setStretch(self, x=None, y=None):
|
|
"""
|
|
Set the 'target' size for this Dock.
|
|
The actual size will be determined by comparing this Dock's
|
|
stretch value to the rest of the docks it shares space with.
|
|
"""
|
|
#print "setStretch", self, x, y
|
|
#self._stretch = (x, y)
|
|
if x is None:
|
|
x = 0
|
|
if y is None:
|
|
y = 0
|
|
#policy = self.sizePolicy()
|
|
#policy.setHorizontalStretch(x)
|
|
#policy.setVerticalStretch(y)
|
|
#self.setSizePolicy(policy)
|
|
self._stretch = (x, y)
|
|
self.sigStretchChanged.emit()
|
|
#print "setStretch", self, x, y, self.stretch()
|
|
|
|
def stretch(self):
|
|
#policy = self.sizePolicy()
|
|
#return policy.horizontalStretch(), policy.verticalStretch()
|
|
return self._stretch
|
|
|
|
#def stretch(self):
|
|
#return self._stretch
|
|
|
|
def hideTitleBar(self):
|
|
"""
|
|
Hide the title bar for this Dock.
|
|
This will prevent the Dock being moved by the user.
|
|
"""
|
|
self.label.hide()
|
|
self.labelHidden = True
|
|
if 'center' in self.allowedAreas:
|
|
self.allowedAreas.remove('center')
|
|
self.updateStyle()
|
|
|
|
def showTitleBar(self):
|
|
"""
|
|
Show the title bar for this Dock.
|
|
"""
|
|
self.label.show()
|
|
self.labelHidden = False
|
|
self.allowedAreas.add('center')
|
|
self.updateStyle()
|
|
|
|
def setOrientation(self, o='auto', force=False):
|
|
"""
|
|
Sets the orientation of the title bar for this Dock.
|
|
Must be one of 'auto', 'horizontal', or 'vertical'.
|
|
By default ('auto'), the orientation is determined
|
|
based on the aspect ratio of the Dock.
|
|
"""
|
|
#print self.name(), "setOrientation", o, force
|
|
if o == 'auto' and self.autoOrient:
|
|
if self.container().type() == 'tab':
|
|
o = 'horizontal'
|
|
elif self.width() > self.height()*1.5:
|
|
o = 'vertical'
|
|
else:
|
|
o = 'horizontal'
|
|
if force or self.orientation != o:
|
|
self.orientation = o
|
|
self.label.setOrientation(o)
|
|
self.updateStyle()
|
|
|
|
def updateStyle(self):
|
|
## updates orientation and appearance of title bar
|
|
#print self.name(), "update style:", self.orientation, self.moveLabel, self.label.isVisible()
|
|
if self.labelHidden:
|
|
self.widgetArea.setStyleSheet(self.nStyle)
|
|
elif self.orientation == 'vertical':
|
|
self.label.setOrientation('vertical')
|
|
if self.moveLabel:
|
|
#print self.name(), "reclaim label"
|
|
self.topLayout.addWidget(self.label, 1, 0)
|
|
self.widgetArea.setStyleSheet(self.vStyle)
|
|
else:
|
|
self.label.setOrientation('horizontal')
|
|
if self.moveLabel:
|
|
#print self.name(), "reclaim label"
|
|
self.topLayout.addWidget(self.label, 0, 1)
|
|
self.widgetArea.setStyleSheet(self.hStyle)
|
|
|
|
def resizeEvent(self, ev):
|
|
self.setOrientation()
|
|
self.resizeOverlay(self.size())
|
|
|
|
def name(self):
|
|
return str(self.label.text())
|
|
|
|
def container(self):
|
|
return self._container
|
|
|
|
def addWidget(self, widget, row=None, col=0, rowspan=1, colspan=1):
|
|
"""
|
|
Add a new widget to the interior of this Dock.
|
|
Each Dock uses a QGridLayout to arrange widgets within.
|
|
"""
|
|
if row is None:
|
|
row = self.currentRow
|
|
self.currentRow = max(row+1, self.currentRow)
|
|
self.widgets.append(widget)
|
|
self.layout.addWidget(widget, row, col, rowspan, colspan)
|
|
self.raiseOverlay()
|
|
|
|
|
|
def startDrag(self):
|
|
self.drag = QtGui.QDrag(self)
|
|
mime = QtCore.QMimeData()
|
|
#mime.setPlainText("asd")
|
|
self.drag.setMimeData(mime)
|
|
self.widgetArea.setStyleSheet(self.dragStyle)
|
|
self.update()
|
|
action = self.drag.exec_()
|
|
self.updateStyle()
|
|
|
|
def float(self):
|
|
self.area.floatDock(self)
|
|
|
|
def containerChanged(self, c):
|
|
#print self.name(), "container changed"
|
|
self._container = c
|
|
if c.type() != 'tab':
|
|
self.moveLabel = True
|
|
self.label.setDim(False)
|
|
else:
|
|
self.moveLabel = False
|
|
|
|
self.setOrientation(force=True)
|
|
|
|
def __repr__(self):
|
|
return "<Dock %s %s>" % (self.name(), self.stretch())
|
|
|
|
|
|
class DockLabel(VerticalLabel):
|
|
|
|
sigClicked = QtCore.Signal(object, object)
|
|
|
|
def __init__(self, text, dock):
|
|
self.dim = False
|
|
self.fixedWidth = False
|
|
VerticalLabel.__init__(self, text, orientation='horizontal', forceWidth=False)
|
|
self.setAlignment(QtCore.Qt.AlignTop|QtCore.Qt.AlignHCenter)
|
|
self.dock = dock
|
|
self.updateStyle()
|
|
self.setAutoFillBackground(False)
|
|
|
|
#def minimumSizeHint(self):
|
|
##sh = QtGui.QWidget.minimumSizeHint(self)
|
|
#return QtCore.QSize(20, 20)
|
|
|
|
def updateStyle(self):
|
|
r = '3px'
|
|
if self.dim:
|
|
fg = '#aaa'
|
|
bg = '#44a'
|
|
border = '#339'
|
|
else:
|
|
fg = '#fff'
|
|
bg = '#66c'
|
|
border = '#55B'
|
|
|
|
if self.orientation == 'vertical':
|
|
self.vStyle = """DockLabel {
|
|
background-color : %s;
|
|
color : %s;
|
|
border-top-right-radius: 0px;
|
|
border-top-left-radius: %s;
|
|
border-bottom-right-radius: 0px;
|
|
border-bottom-left-radius: %s;
|
|
border-width: 0px;
|
|
border-right: 2px solid %s;
|
|
padding-top: 3px;
|
|
padding-bottom: 3px;
|
|
}""" % (bg, fg, r, r, border)
|
|
self.setStyleSheet(self.vStyle)
|
|
else:
|
|
self.hStyle = """DockLabel {
|
|
background-color : %s;
|
|
color : %s;
|
|
border-top-right-radius: %s;
|
|
border-top-left-radius: %s;
|
|
border-bottom-right-radius: 0px;
|
|
border-bottom-left-radius: 0px;
|
|
border-width: 0px;
|
|
border-bottom: 2px solid %s;
|
|
padding-left: 3px;
|
|
padding-right: 3px;
|
|
}""" % (bg, fg, r, r, border)
|
|
self.setStyleSheet(self.hStyle)
|
|
|
|
def setDim(self, d):
|
|
if self.dim != d:
|
|
self.dim = d
|
|
self.updateStyle()
|
|
|
|
def setOrientation(self, o):
|
|
VerticalLabel.setOrientation(self, o)
|
|
self.updateStyle()
|
|
|
|
def mousePressEvent(self, ev):
|
|
if ev.button() == QtCore.Qt.LeftButton:
|
|
self.pressPos = ev.pos()
|
|
self.startedDrag = False
|
|
ev.accept()
|
|
|
|
def mouseMoveEvent(self, ev):
|
|
if not self.startedDrag and (ev.pos() - self.pressPos).manhattanLength() > QtGui.QApplication.startDragDistance():
|
|
self.dock.startDrag()
|
|
ev.accept()
|
|
#print ev.pos()
|
|
|
|
def mouseReleaseEvent(self, ev):
|
|
if not self.startedDrag:
|
|
#self.emit(QtCore.SIGNAL('clicked'), self, ev)
|
|
self.sigClicked.emit(self, ev)
|
|
ev.accept()
|
|
|
|
def mouseDoubleClickEvent(self, ev):
|
|
if ev.button() == QtCore.Qt.LeftButton:
|
|
self.dock.float()
|
|
|
|
#def paintEvent(self, ev):
|
|
#p = QtGui.QPainter(self)
|
|
##p.setBrush(QtGui.QBrush(QtGui.QColor(100, 100, 200)))
|
|
#p.setPen(QtGui.QPen(QtGui.QColor(50, 50, 100)))
|
|
#p.drawRect(self.rect().adjusted(0, 0, -1, -1))
|
|
|
|
#VerticalLabel.paintEvent(self, ev)
|
|
|
|
|
|
|