mirror of https://github.com/ralsina/xrandroll.git
103 lines
4.2 KiB
Python
103 lines
4.2 KiB
Python
from PySide2.QtCore import Qt, QObject
|
|
from PySide2.QtWidgets import QGraphicsRectItem, QGraphicsTextItem
|
|
from PySide2.QtGui import QBrush, QColor
|
|
|
|
|
|
class MonitorItem(QGraphicsRectItem, QObject):
|
|
z = 0
|
|
|
|
def __init__(self, *a, **kw):
|
|
data = kw.pop("data")
|
|
self.name = kw.pop("name")
|
|
self.window = kw.pop("window")
|
|
super().__init__(0, 0, 0, 0)
|
|
self.setAcceptedMouseButtons(Qt.LeftButton)
|
|
self.label = QGraphicsTextItem("", self)
|
|
self.bottom_edge = QGraphicsRectItem(0, 0, 0, 0, self)
|
|
self.bottom_edge.setBrush(QBrush("red", Qt.SolidPattern))
|
|
self.update_visuals(data)
|
|
|
|
def update_visuals(self, monitor):
|
|
self.setRect(0, 0, monitor.res_x, monitor.res_y)
|
|
self.setPos(monitor.pos_x, monitor.pos_y)
|
|
if monitor.orientation == "normal":
|
|
self.bottom_edge.setRect(0, monitor.res_y - 50, monitor.res_x, 50)
|
|
elif monitor.orientation == "left":
|
|
self.bottom_edge.setRect(monitor.res_x - 50, 0, 50, monitor.res_y)
|
|
elif monitor.orientation == "inverted":
|
|
self.bottom_edge.setRect(0, 0, monitor.res_x, 50)
|
|
elif monitor.orientation == "right":
|
|
self.bottom_edge.setRect(0, 0, 50, monitor.res_y)
|
|
if monitor.replica_of:
|
|
label_text = f"{self.name} [{','.join(monitor.replica_of)}]"
|
|
else:
|
|
label_text = self.name
|
|
self.label.setPlainText(label_text)
|
|
label_scale = min(
|
|
self.rect().width() / self.label.boundingRect().width(),
|
|
self.rect().height() / self.label.boundingRect().height(),
|
|
)
|
|
self.label.setScale(label_scale)
|
|
if monitor.enabled:
|
|
if monitor.primary:
|
|
color = QColor("#eee8d5")
|
|
color.setAlpha(200)
|
|
self.setBrush(QBrush(color, Qt.SolidPattern))
|
|
self.setZValue(1000)
|
|
else:
|
|
color = QColor("#ffffff")
|
|
color.setAlpha(200)
|
|
self.setBrush(QBrush(color, Qt.SolidPattern))
|
|
self.setZValue(self.z)
|
|
self.z -= 1
|
|
self.show()
|
|
else:
|
|
color = QColor("#f1f1f1")
|
|
color.setAlpha(200)
|
|
self.setBrush(QBrush(color, Qt.SolidPattern))
|
|
self.setZValue(-1000)
|
|
self.hide()
|
|
|
|
def mousePressEvent(self, event):
|
|
self.window.pos_label.show()
|
|
self.setCursor(Qt.ClosedHandCursor)
|
|
self.orig_pos = self.pos()
|
|
self.window.ui.screenCombo.setCurrentText(self.name)
|
|
|
|
def mouseReleaseEvent(self, event):
|
|
self.setCursor(Qt.OpenHandCursor)
|
|
self.window.pos_label.hide()
|
|
self.window.monitor_moved()
|
|
|
|
def mouseMoveEvent(self, event):
|
|
snaps_x, snaps_y = self.window.possible_snaps(self.name)
|
|
view = event.widget().parent()
|
|
click_pos = event.buttonDownScreenPos(Qt.LeftButton)
|
|
current_pos = event.screenPos()
|
|
new_pos = view.mapFromScene(self.orig_pos) + current_pos - click_pos
|
|
new_pos = view.mapToScene(new_pos)
|
|
delta = abs(view.mapToScene(0, 25).y())
|
|
if not event.modifiers() & Qt.ControlModifier: # Ctrl was not pressed, so snap
|
|
# This snaps the left and top edges
|
|
x = new_pos.x()
|
|
delta_x = min((abs(x - sx), i) for i, sx in enumerate(snaps_x))
|
|
if delta_x[0] < delta: # snap
|
|
new_pos.setX(int(snaps_x[delta_x[1]]))
|
|
y = new_pos.y()
|
|
delta_y = min((abs(y - sy), i) for i, sy in enumerate(snaps_y))
|
|
if delta_y[0] < delta: # snap
|
|
new_pos.setY(int(snaps_y[delta_y[1]]))
|
|
|
|
# This snaps the right and bottom edges
|
|
x = new_pos.x() + self.rect().width()
|
|
delta_x = min((abs(x - sx), i) for i, sx in enumerate(snaps_x))
|
|
if delta_x[0] < delta: # snap
|
|
new_pos.setX(int(snaps_x[delta_x[1]]) - self.rect().width())
|
|
y = new_pos.y() + self.rect().height()
|
|
delta_y = min((abs(y - sy), i) for i, sy in enumerate(snaps_y))
|
|
if delta_y[0] < delta: # snap
|
|
new_pos.setY(int(snaps_y[delta_y[1]]) - self.rect().height())
|
|
|
|
self.setPos(new_pos)
|
|
self.window.show_pos(int(self.pos().x()), int(self.pos().y()))
|