mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
80f2b7a1db
* Fix the context menu * Remove unnecessary overrides * Clean up formatting/naming * Factor out base class for shared behavior
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
from typing import Optional
|
|
|
|
from PySide2.QtGui import QColor, QPainter
|
|
from PySide2.QtWidgets import (
|
|
QAction,
|
|
QGraphicsSceneContextMenuEvent,
|
|
QMenu,
|
|
)
|
|
|
|
import qt_ui.uiconstants as const
|
|
from game import Game
|
|
from qt_ui.windows.basemenu.QBaseMenu2 import QBaseMenu2
|
|
from theater import ControlPoint
|
|
from .QMapObject import QMapObject
|
|
|
|
|
|
class QMapControlPoint(QMapObject):
|
|
|
|
def __init__(self, parent, x: float, y: float, w: float, h: float,
|
|
model: ControlPoint, game: Game) -> None:
|
|
super().__init__(x, y, w, h)
|
|
self.model = model
|
|
self.game = game
|
|
self.parent = parent
|
|
self.setZValue(1)
|
|
self.setToolTip(self.model.name)
|
|
self.base_details_dialog: Optional[QBaseMenu2] = None
|
|
|
|
def paint(self, painter, option, widget=None) -> None:
|
|
if self.parent.get_display_rule("cp"):
|
|
painter.save()
|
|
painter.setRenderHint(QPainter.Antialiasing)
|
|
painter.setBrush(self.brush_color)
|
|
painter.setPen(self.pen_color)
|
|
|
|
if self.model.has_runway():
|
|
if self.isUnderMouse():
|
|
painter.setBrush(const.COLORS["white"])
|
|
painter.setPen(self.pen_color)
|
|
|
|
r = option.rect
|
|
painter.drawEllipse(r.x(), r.y(), r.width(), r.height())
|
|
# TODO: Draw sunk carriers differently.
|
|
# Either don't draw them at all, or perhaps use a sunk ship icon.
|
|
painter.restore()
|
|
|
|
def contextMenuEvent(self, event: QGraphicsSceneContextMenuEvent) -> None:
|
|
if self.model.captured:
|
|
text = "Open base menu"
|
|
else:
|
|
text = "Open intel menu"
|
|
|
|
open_menu = QAction(text)
|
|
open_menu.triggered.connect(self.on_click)
|
|
|
|
menu = QMenu("Menu", self.parent)
|
|
menu.addAction(open_menu)
|
|
menu.exec_(event.screenPos())
|
|
|
|
@property
|
|
def brush_color(self) -> QColor:
|
|
if self.model.captured:
|
|
return const.COLORS["blue"]
|
|
else:
|
|
return const.COLORS["super_red"]
|
|
|
|
@property
|
|
def pen_color(self) -> QColor:
|
|
return const.COLORS["white"]
|
|
|
|
def on_click(self) -> None:
|
|
self.base_details_dialog = QBaseMenu2(
|
|
self.window(),
|
|
self.model,
|
|
self.game
|
|
)
|
|
self.base_details_dialog.show()
|