mirror of
https://github.com/m-labs/artiq.git
synced 2024-12-25 11:18:27 +08:00
dndwidgets: add drag drop helper widgets
This commit is contained in:
parent
8a9b6a449b
commit
096664c1ba
98
artiq/gui/dndwidgets.py
Normal file
98
artiq/gui/dndwidgets.py
Normal file
@ -0,0 +1,98 @@
|
||||
from PyQt5 import QtCore, QtWidgets
|
||||
|
||||
|
||||
class VDragDropSplitter(QtWidgets.QSplitter):
|
||||
def __init__(self, parent):
|
||||
QtWidgets.QSplitter.__init__(self, parent=parent)
|
||||
self.setAcceptDrops(True)
|
||||
self.setContentsMargins(0, 0, 0, 0)
|
||||
self.setOrientation(QtCore.Qt.Vertical)
|
||||
self.setChildrenCollapsible(False)
|
||||
|
||||
def resetSizes(self):
|
||||
self.setSizes(self.count() * [1])
|
||||
|
||||
def dragEnterEvent(self, e):
|
||||
e.accept()
|
||||
|
||||
def dragLeaveEvent(self, e):
|
||||
self.setRubberBand(-1)
|
||||
e.accept()
|
||||
|
||||
def dragMoveEvent(self, e):
|
||||
pos = e.pos()
|
||||
src = e.source()
|
||||
src_i = self.indexOf(src)
|
||||
self.setRubberBand(self.height())
|
||||
# case 0: smaller than source widget
|
||||
if pos.y() < src.y():
|
||||
for n in range(src_i):
|
||||
w = self.widget(n)
|
||||
if pos.y() < w.y() + w.size().height():
|
||||
self.setRubberBand(w.y())
|
||||
break
|
||||
# case 2: greater than source widget
|
||||
elif pos.y() > src.y() + src.size().height():
|
||||
for n in range(src_i + 1, self.count()):
|
||||
w = self.widget(n)
|
||||
if pos.y() < w.y():
|
||||
self.setRubberBand(w.y())
|
||||
break
|
||||
else:
|
||||
self.setRubberBand(-1)
|
||||
e.accept()
|
||||
|
||||
def dropEvent(self, e):
|
||||
self.setRubberBand(-1)
|
||||
pos = e.pos()
|
||||
src = e.source()
|
||||
src_i = self.indexOf(src)
|
||||
for n in range(self.count()):
|
||||
w = self.widget(n)
|
||||
if pos.y() < w.y() + w.size().height():
|
||||
self.insertWidget(n, src)
|
||||
break
|
||||
e.accept()
|
||||
|
||||
|
||||
# Scroll area with auto-scroll on vertical drag
|
||||
class VDragScrollArea(QtWidgets.QScrollArea):
|
||||
def __init__(self, parent):
|
||||
QtWidgets.QScrollArea.__init__(self, parent)
|
||||
self.installEventFilter(self)
|
||||
self._margin = 40
|
||||
self._timer = QtCore.QTimer(self)
|
||||
self._timer.setInterval(20)
|
||||
self._timer.timeout.connect(self._on_auto_scroll)
|
||||
self._direction = 0
|
||||
self._speed = 10
|
||||
|
||||
def setAutoScrollMargin(self, margin):
|
||||
self._margin = margin
|
||||
|
||||
def setAutoScrollSpeed(self, speed):
|
||||
self._speed = speed
|
||||
|
||||
def eventFilter(self, obj, e):
|
||||
if e.type() == QtCore.QEvent.DragMove:
|
||||
val = self.verticalScrollBar().value()
|
||||
height = self.viewport().height()
|
||||
y = e.pos().y()
|
||||
self._direction = 0
|
||||
if y < val + self._margin:
|
||||
self._direction = -1
|
||||
elif y > height + val - self._margin:
|
||||
self._direction = 1
|
||||
if not self._timer.isActive():
|
||||
self._timer.start()
|
||||
elif e.type() in (QtCore.QEvent.Drop, QtCore.QEvent.DragLeave):
|
||||
self._timer.stop()
|
||||
return False
|
||||
|
||||
def _on_auto_scroll(self):
|
||||
val = self.verticalScrollBar().value()
|
||||
min_ = self.verticalScrollBar().minimum()
|
||||
max_ = self.verticalScrollBar().maximum()
|
||||
dy = self._direction * self._speed
|
||||
new_val = min(max_, max(min_, val + dy))
|
||||
self.verticalScrollBar().setValue(new_val)
|
Loading…
Reference in New Issue
Block a user