Difficult-Rocket/Difficult_Rocket/gui/widget/button.py

742 lines
23 KiB
Python

# -------------------------------
# Difficult Rocket
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
# All rights reserved
# -------------------------------
from __future__ import annotations
import math
from typing import Optional, Tuple, Type, Sequence
# from libs import pyglet
import pyglet
from pyglet.gl import GL_ONE_MINUS_SRC_ALPHA, GL_SRC_ALPHA
from pyglet.graphics.shader import ShaderProgram
from pyglet.text import Label
from pyglet.gui import widgets
from pyglet.window import mouse
# from pyglet.sprite import Sprite
from pyglet.shapes import Rectangle, BorderedRectangle, ShapeBase, _rotate_point
from pyglet.gui.widgets import WidgetBase
# from pyglet.image import AbstractImage
from pyglet.graphics import Batch, Group
from Difficult_Rocket.api.types import Options, FontData
# from Difficult_Rocket import DR_status
RGBA = Tuple[int, int, int, int]
class 拐角(ShapeBase):
def __init__(
self,
x: float,
y: float,
width: float,
height: float,
thick1: float = 1.0,
thick2: float = 1.0,
color: tuple[int, int, int, int] = (255, 255, 255, 255),
clockwise: bool = True,
blend_src: int = GL_SRC_ALPHA,
blend_dest: int = GL_ONE_MINUS_SRC_ALPHA,
batch: Batch | None = None,
group: Group | None = None,
program: ShaderProgram | None = None,
) -> None:
self._x = x
self._y = y
self._width = width
self._height = height
self._thick1 = thick1
self._thick2 = thick2
self._clockwise = clockwise
self._rgba = color
super().__init__(6, blend_src, blend_dest, batch, group, program)
def __contains__(self, point: tuple[float, float]) -> bool:
assert len(point) == 2
# 先大框
point = _rotate_point((self._x, self._y), point, math.radians(self._rotation))
x, y = self._x - self._anchor_x, self._y - self._anchor_y
return (x < point[0] < x + self._width and y < point[1] < y + self._height) and (
(
(point[1] > y + self._height - self._thick2)
or (point[0] < x + self._thick1)
)
if self._clockwise
else (
(point[1] < y + self._thick1)
or (point[0] > x + self._width - self._thick2)
)
)
def _get_vertices(self) -> Sequence[float]:
if not self.visible:
return (0, 0) * self._num_verts
t1, t2 = self._thick1, self._thick2
left = -self._anchor_x
bottom = -self._anchor_y
right = left + self._width
top = bottom + self._height
x1 = left
x2 = left + t1
x3 = right - t1
x4 = right
y1 = bottom
y2 = bottom + t1
y3 = top - t2
y4 = top
# fmt: off
return ((
x1, y1,
x1, y4,
x4, y4,
x4, y3,
x2, y3,
x2, y1
) if self._clockwise else(
x1, y1,
x4, y1,
x4, y4,
x3, y4,
x3, y2,
x1, y2
))
# fmt: on
def _update_vertices(self) -> None:
self._vertex_list.position[:] = self._get_vertices() # pyright: ignore reportAttributeAccessIssue
def _create_vertex_list(self) -> None:
# 1 2
# 4 3
# 0 5
# or
# 3 2
# 5 4
# 0 1
# fmt: off
groups = [
1, 2, 3,
1, 3, 4,
1, 4, 5,
1, 5, 0,
]
# fmt: on
self._vertex_list = self._program.vertex_list_indexed(
self._num_verts,
self._draw_mode,
groups,
self._batch, # pyright: ignore reportArgumentType
self._group, # pyright: ignore reportArgumentType
position=("f", self._get_vertices()),
colors=("Bn", self._rgba * self._num_verts),
translation=("f", (self._x, self._y) * self._num_verts),
)
@property
def thick1(self) -> float:
return self._thick1
@property
def thick2(self) -> float:
return self.thick2
@property
def thickness(self) -> float:
return self._thick1
@thickness.setter
def thickness(self, value: float):
self._thick1 = value
self._thick2 = value
self._update_vertices()
@property
def width(self) -> float:
return self._width
@property
def height(self) -> float:
return self.height
@property
def clockwise(self) -> bool:
return self._clockwise
@thick1.setter
def thick1(self, value: float):
self._thick1 = value
self._update_vertices()
@thick2.setter
def thick2(self, value: float):
self._thick2 = value
self._update_vertices()
@width.setter
def width(self, value: float):
self._width = value
self._update_vertices()
@height.setter
def height(self, value: float):
self._height = value
self._update_vertices()
@clockwise.setter
def clockwise(self, value: bool):
self._clockwise = bool(value)
self._update_vertices()
class WikiButton(WidgetBase):
# 左上角的普通状态下颜色
upper_normal = (109, 109, 110, 255)
upper_press = (90, 91, 92, 255)
# 右下角的
down_normal = (90, 91, 92, 255)
down_press = (70, 71, 71, 255)
# 左下角方块
corner_normal = (124, 124, 125, 255)
corner_press = (106, 107, 108, 255)
def __init__(
self, x: int, y: int, width: int, height: int, batch: Batch, group: Group
) -> None:
super().__init__(x, y, width, height)
self.enabled = False
pad = 2
# 覆盖式
self.main_batch = batch or Batch()
self.main_group = group or Group()
# 背景的黑框
self.background_group = Group(order=10, parent=self.main_group)
# 左上右下两组
self.border_group = Group(order=20, parent=self.main_group)
# 左下右上两个小方块
self.corner_group = Group(order=32, parent=self.main_group)
self.upper_border = 拐角(
x=self.x,
y=self.y,
width=width,
height=height,
thick1=pad,
thick2=pad,
color=self.upper_normal
)
self.down_border = 拐角(
x=self.x,
y=self.y,
width=width,
height=height,
thick1=pad,
thick2=pad,
clockwise=False,
color=self.down_normal
)
self.left_down = Rectangle(
x=self.x,
y=self.y,
width=pad,
height=pad,
color=self.corner_normal
)
def __contains__(self, pos: tuple[float, float]) -> bool:
box = self.aabb()
return box[0] < pos[0] < box[2] and box[1] < pos[1] < box[3]
def on_mouse_press(self, x: int, y: int, buttons: int, modifiers: int) -> None:
...
class BaseButtonTheme:
"""
按钮的风格
"""
name = "BaseButtonTheme"
def __init__(
self, x: int, y: int, width: int, height: int, batch: Batch, group: Group
):
self.batch = batch
self.group = group
self.x = x
self.y = y
self.width = width
self.height = height
self.enable = False
def on_enable(self, x: int, y: int, button):
"""
当按钮被启用的时候
:param x:
:param y:
:param button:
:return:
"""
def on_disable(self, button):
"""
当按钮被禁用的时候
:param button:
:return:
"""
def on_update(self, button) -> None:
"""
当按钮被更新的时候
:param button:
:return:
"""
class MinecraftWikiButtonTheme(BaseButtonTheme):
"""
直接绘制按钮的风格
"""
name = "MinecraftWikiButtonTheme"
def __init__(
self,
x: int,
y: int,
width: int,
height: int,
batch: Batch,
group: Group,
theme: Optional[dict] = None,
):
super().__init__(x, y, width, height, batch, group)
self.enable = False
self.batch = batch
self.group = group
a = (72, 73, 74, 255)
b = (109, 109, 110, 255)
c = (88, 91, 92, 255)
d = (124, 124, 125, 255) # 左下和右上两个重叠点的颜色.... 画吗?
e = (49, 50, 51, 255)
touch_a = (49, 50, 51, 255)
touch_b = (90, 91, 92, 255)
touch_c = (71, 72, 72, 255)
touch_d = (106, 107, 108, 255) # 同上
pad = 2 # 边框宽度 2 px
list_pad = 4 # 下巴 4px
if theme is None:
theme = {}
pop_out = theme.get("pop_out", False)
if pop_out:
# 主背景
self.back_ground = Rectangle(
x=x + (pad * 2),
y=y + (pad * 2) + list_pad,
width=width - (pad * 4),
height=height - (pad * 4) - list_pad,
color=a,
batch=batch,
group=Group(order=3, parent=group),
)
# 左上方向的覆盖
self.cover_back = Rectangle(
x=x + pad,
y=y + pad + list_pad,
width=width - (pad * 2),
height=height - (pad * 2) - list_pad,
color=b,
batch=batch,
group=Group(order=1, parent=group),
)
# 右下方向的覆盖
self.cover_back2 = Rectangle(
x=x + (pad * 2),
y=y + pad + list_pad,
width=width - (pad * 3),
height=height - (pad * 3) - list_pad,
color=c,
batch=batch,
group=Group(order=2, parent=group),
)
else:
# 主背景
self.back_ground = Rectangle(
x=x + (pad * 2),
y=y + (pad * 2) + list_pad,
width=width - (pad * 4),
height=height - (pad * 4) - list_pad,
color=c,
batch=batch,
group=Group(order=3, parent=group),
)
# 左上方向的覆盖
self.cover_back = Rectangle(
x=x + pad,
y=y + pad + list_pad,
width=width - (pad * 2),
height=height - (pad * 2) - list_pad,
color=a,
batch=batch,
group=Group(order=2, parent=group),
)
# 右下方向的覆盖
self.cover_back2 = Rectangle(
x=x + pad,
y=y + (pad * 2) + list_pad,
width=width - (pad * 3),
height=height - (pad * 3) - list_pad,
color=b,
batch=batch,
group=Group(order=1, parent=group),
)
# 下巴的框
self.list_back = BorderedRectangle(
x=x,
y=y,
width=width,
height=height,
border=pad,
border_color=(0, 0, 0, 255),
color=e,
batch=batch,
group=Group(order=0, parent=group),
)
self.a = a
self.b = b
self.c = c
self.d = d
self.e = e
self.touch_a = touch_a
self.touch_b = touch_b
self.touch_c = touch_c
self.touch_d = touch_d
self.pad = pad
self.list_pad = list_pad
self.pop_out = pop_out
self.drag_list = theme.get("drag_list", False)
def on_enable(self, x: int, y: int, button):
if self.pop_out:
self.back_ground.color = self.touch_a
self.cover_back.color = self.touch_b
self.cover_back2.color = self.touch_c
else:
self.back_ground.color = self.touch_c
self.cover_back.color = self.touch_a
self.cover_back2.color = self.touch_b
if self.drag_list:
button.text_label.y = (
self.y
+ (self.height - button.font_height) // 2
+ (button.font_height * 0.2)
+ (self.list_pad // 2)
)
self.back_ground.y = self.y + (self.pad * 2)
self.back_ground.height = self.height - (self.pad * 4)
self.cover_back.y = self.y + self.pad
self.cover_back.height = self.height - (self.pad * 2)
self.cover_back2.y = self.y + self.pad
self.cover_back2.height = self.height - (self.pad * 3)
else:
button.text_label.y = (
self.y
+ (self.height - button.font_height) // 2
+ (button.font_height * 0.2)
+ self.list_pad
)
self.enable = True
def on_disable(self, button) -> None:
if self.pop_out:
self.back_ground.color = self.a
self.cover_back.color = self.b
self.cover_back2.color = self.c
else:
self.back_ground.color = self.c
self.cover_back.color = self.a
self.cover_back2.color = self.b
if self.drag_list:
self.back_ground.y = self.y + (self.pad * 2) + self.list_pad
self.back_ground.height = self.height - (self.pad * 4) - self.list_pad
self.cover_back.y = self.y + self.pad + self.list_pad
self.cover_back.height = self.height - (self.pad * 2) - self.list_pad
self.cover_back2.y = self.y + self.pad + self.list_pad
self.cover_back2.height = self.height - (self.pad * 3) - self.list_pad
button.text_label.y = (
self.y
+ (self.height - button.font_height) // 2
+ (button.font_height * 0.2)
+ self.list_pad
)
self.enable = False
def on_update(self, button) -> None:
super().on_update(button)
self.list_back.x = self.x
self.list_back.y = self.y
if self.enable and self.drag_list:
button.text_label.y = (
self.y
+ (self.height - button.font_height) // 2
+ (button.font_height * 0.2)
+ self.list_pad // 2
)
self.back_ground.x = self.x + (self.pad * 2)
self.back_ground.y = self.y + (self.pad * 2)
self.back_ground.position = self.x + (self.pad * 2), self.y + (self.pad * 2)
self.back_ground.height = self.height - (self.pad * 4)
self.cover_back.x = self.x + self.pad
self.cover_back.y = self.y + self.pad
self.cover_back.position = self.x + self.pad, self.y + self.pad
self.cover_back.height = self.height - (self.pad * 2)
self.cover_back2.x = self.x + self.pad
self.cover_back2.y = self.y + self.pad
self.cover_back2.position = self.x + self.pad, self.y + self.pad
self.cover_back2.height = self.height - (self.pad * 3)
else:
button.text_label.y = (
self.y
+ (self.height - button.font_height) // 2
+ (button.font_height * 0.2)
+ self.list_pad
)
self.back_ground.position = (
self.x + (self.pad * 2),
self.y + (self.pad * 2) + self.list_pad,
)
self.back_ground.height = self.height - (self.pad * 4) - self.list_pad
self.cover_back.position = (
self.x + self.pad,
self.y + self.pad + self.list_pad,
)
self.cover_back.height = self.height - (self.pad * 2) - self.list_pad
self.cover_back2.position = (
self.x + self.pad,
self.y + self.pad + self.list_pad,
)
self.cover_back2.height = self.height - (self.pad * 3) - self.list_pad
self.back_ground.position = (
self.x + (self.pad * 2),
self.y + (self.pad * 2) + self.list_pad,
)
self.back_ground.height = self.height - (self.pad * 4) - self.list_pad
class ButtonThemeOptions(Options):
"""基于 Options 写的 ButtonTheme"""
name = "ButtonTheme"
untouched_color: RGBA = (39, 73, 114, 255)
touched_color: RGBA = (66, 150, 250, 255)
hit_color: RGBA = (15, 135, 250, 255)
font_theme: FontData = FontData()
def __str__(self):
return self.as_markdown()
class PressTextButton(widgets.WidgetBase):
"""
自带 字符 + 材质 的按钮,就不用单独做材质了
"""
def __init__(
self,
x: int,
y: int,
width: int,
height: int,
text: str,
batch: Optional[Batch] = None,
group: Optional[Group] = None,
theme: Optional[ButtonThemeOptions] = None,
draw_theme: Optional[Type[BaseButtonTheme]] = None,
dict_theme: Optional[dict] = None,
):
super().__init__(x, y, width, height)
self.main_batch = batch or Batch()
self.user_batch = batch is not None
self.front_group = Group(order=10, parent=group)
self.back_ground_group = Group(order=5, parent=group)
self.pressed = False
self.theme = theme or ButtonThemeOptions()
if dict_theme is None:
dict_theme = {}
self.dict_theme = dict_theme
self.untouched_color = self.theme.untouched_color
self.touched_color = self.theme.touched_color
self.hit_color = self.theme.hit_color
# from ImGui
self.text = text
self.text_label = Label(
font_name=self.theme.font_theme.font_name,
font_size=self.theme.font_theme.font_size,
batch=self.main_batch,
group=self.front_group,
x=self._x,
y=self._y + 4,
width=self._width,
height=self._height,
)
self.font = pyglet.font.load(
self.theme.font_theme.font_name,
self.theme.font_theme.font_size,
bold=self.theme.font_theme.bold,
italic=self.theme.font_theme.italic,
stretch=self.theme.font_theme.stretch,
)
self.font_height = self.font.ascent - self.font.descent
if draw_theme is None:
self.back_rec = Rectangle(
x=self._x,
y=self._y,
width=self._width,
height=self._height,
color=self.untouched_color, # ImGui color
batch=self.main_batch,
group=self.back_ground_group,
)
self.draw_theme = False
else:
self.draw_theme = draw_theme(
x=self._x,
y=self._y,
width=self._width,
height=self._height,
batch=self.main_batch,
group=self.back_ground_group,
)
self.draw_theme.on_update(self)
self.value = text # 重新分配一下高度和宽度的位置
@property
def x(self) -> int:
return self._x
@x.setter
def x(self, value: int) -> None:
self._x = value
self.text_label.x = value
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.x = value
self.draw_theme.on_update(self)
@property
def y(self) -> int:
return self._y
@y.setter
def y(self, value: int) -> None:
self._y = value
self.text_label.y = value + 4
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.y = value
self.draw_theme.on_update(self)
@property
def value(self):
return self.text
@value.setter
def value(self, value):
self.text = value
self.text_label.text = value
text_width = self.text_label.content_width
self.text_label.x = self._x + (self.width - text_width) // 2
self.text_label.y = (
self._y + (self.height - self.font_height) // 2 + (self.font_height * 0.2)
) # 修正一下位置
@property
def batch(self) -> Batch:
return self.main_batch
@batch.setter
def batch(self, value: Batch):
assert isinstance(value, Batch), "batch must be a pyglet.graphics.Batch"
self.main_batch = value
self.text_label.batch = value
self.back_rec.batch = value
self.user_batch = True
def __contains__(self, point):
if self.draw_theme:
x, y = self.x, self.y
return x < point[0] < x + self._width and y < point[1] < y + self._height
return point in self.back_rec
def on_draw(self):
if not self.user_batch:
self.main_batch.draw()
def on_mouse_motion(self, x, y, dx, dy):
if (x, y) in self:
if self.draw_theme:
...
else:
self.back_rec.color = self.touched_color
else:
self.pressed = False
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.on_disable(self)
else:
self.back_rec.color = self.untouched_color
def on_mouse_press(self, x, y, buttons, modifiers) -> bool:
if (x, y) in self:
if buttons == mouse.LEFT:
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.on_enable(x, y, self)
else:
self.back_rec.color = self.hit_color
self.dispatch_event("on_press", x, y)
self.pressed = True
return True
else:
self.pressed = False
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.on_disable(self)
else:
self.back_rec.color = self.untouched_color
return False
def on_mouse_release(self, x, y, buttons, modifiers):
if self.pressed and (x, y) in self:
if self.draw_theme:
assert isinstance(self.draw_theme, BaseButtonTheme)
self.draw_theme.on_disable(self)
else:
self.back_rec.color = self.touched_color
self.pressed = False
self.dispatch_event("on_release", self, x, y)
def _update_position(self):
self.text_label.position = self._x, self._y, 0
self.back_rec.position = self._x, self._y
self.back_rec.width = self._width
self.back_rec.height = self._height
PressTextButton.register_event_type("on_press")
PressTextButton.register_event_type("on_release")