1150 lines
36 KiB
Python
1150 lines
36 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 dataclasses import dataclass
|
|
from enum import Enum
|
|
|
|
# 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.shapes import Rectangle, BorderedRectangle, ShapeBase, _rotate_point
|
|
from pyglet.gui.widgets import WidgetBase
|
|
|
|
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]
|
|
|
|
|
|
@dataclass
|
|
class WikiShapeColors:
|
|
# 这里都是 未按下的颜色
|
|
# 外面一圈高光
|
|
highlight: RGBA = (255, 255, 255, 255)
|
|
# 边框
|
|
border: RGBA = (0, 0, 0, 255)
|
|
# 下巴
|
|
down_pad: RGBA = (49, 50, 51, 255)
|
|
# 左下角和右上角的重叠点
|
|
corner: RGBA = (124, 124, 125, 255)
|
|
# 左上拐角
|
|
left_up: RGBA = (109, 109, 110, 255)
|
|
# 右下拐角
|
|
right_down: RGBA = (90, 91, 92, 255)
|
|
# 内部填充
|
|
inner: RGBA = (72, 73, 74, 255)
|
|
|
|
|
|
class WikiButtonStyles(Enum):
|
|
...
|
|
|
|
|
|
class WikiButtonShape(ShapeBase):
|
|
def __init__(
|
|
self,
|
|
x: float,
|
|
y: float,
|
|
width: float,
|
|
height: float,
|
|
pad: float = 2,
|
|
down_pad: float = 5.0,
|
|
pop_out: bool = True,
|
|
highlight: bool = False,
|
|
colors: WikiShapeColors | None = None,
|
|
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,
|
|
):
|
|
self._x = x
|
|
self._y = y
|
|
self._width = width
|
|
self._height = height
|
|
self._pad = pad
|
|
self._down_pad = down_pad
|
|
self._pop_out = pop_out
|
|
self._highlight = highlight
|
|
self._colors = colors or WikiShapeColors()
|
|
|
|
vertex = 32
|
|
if pop_out:
|
|
vertex += 4
|
|
super().__init__(vertex, blend_src, blend_dest, batch, group, program)
|
|
|
|
@property
|
|
def pop_out(self) -> bool:
|
|
return self._pop_out
|
|
|
|
@property
|
|
def pad(self) -> float:
|
|
return self._pad
|
|
|
|
@property
|
|
def down_pad(self) -> float:
|
|
return self._down_pad
|
|
|
|
@property
|
|
def colors(self) -> WikiShapeColors:
|
|
return self._colors
|
|
|
|
@property
|
|
def highlight(self) -> bool:
|
|
return self._highlight
|
|
|
|
@pop_out.setter
|
|
def pop_out(self, value: bool) -> None:
|
|
self._pop_out = value
|
|
self._create_vertex_list()
|
|
|
|
@pad.setter
|
|
def pad(self, value: float) -> None:
|
|
self._pad = value
|
|
self._update_vertices()
|
|
|
|
@down_pad.setter
|
|
def down_pad(self, value: float) -> None:
|
|
self._down_pad = value
|
|
self._update_vertices()
|
|
|
|
@colors.setter
|
|
def colors(self, value: WikiShapeColors) -> None:
|
|
self._colors = value
|
|
self._update_color()
|
|
|
|
@highlight.setter
|
|
def highlight(self, value: bool) -> None:
|
|
self._highlight = value
|
|
self._create_vertex_list()
|
|
|
|
def _update_color(self) -> None:
|
|
colors = (
|
|
self._colors.down_pad * 4
|
|
+ self._colors.corner * 8
|
|
+ self._colors.right_down * 6
|
|
+ self._colors.left_up * 6
|
|
+ self._colors.inner * 4
|
|
+ self._colors.highlight * 4
|
|
)
|
|
if self._pop_out:
|
|
colors = self._colors.border * 4 + colors
|
|
# if self._highlight:
|
|
# colors += self._colors.highlight * 4
|
|
self._vertex_list.colors[:] = colors
|
|
|
|
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
|
|
|
|
def _get_vertices(self) -> Sequence[float]:
|
|
if not self._visible:
|
|
return (0, 0) * self._num_verts
|
|
|
|
left = -self._anchor_x
|
|
right = left + self._width
|
|
bottom = -self.anchor_y
|
|
top = bottom + self._height
|
|
|
|
pad = self._pad
|
|
down_pad = self._down_pad
|
|
|
|
in_left = left + pad
|
|
i_right = right - pad
|
|
in_bottom = bottom + pad
|
|
inner_top = top - pad
|
|
"""
|
|
pop out ( 默认的弹出状态 )
|
|
20 顶点 (还有外面一圈高光)
|
|
3 2
|
|
18 15 14
|
|
19 16 17
|
|
|
|
11 10 13
|
|
7 9 6
|
|
4 5
|
|
0 1
|
|
unpop
|
|
16 顶点
|
|
3 2
|
|
7 15 6
|
|
13 12 14
|
|
|
|
10 9 11
|
|
4 8 5
|
|
0 1
|
|
"""
|
|
# fmt: off
|
|
out_border = [
|
|
left, bottom, # 0
|
|
right, bottom, # 1
|
|
right, top, # 2
|
|
left, top, # 3
|
|
]
|
|
# if self._highlight:
|
|
highlight = [
|
|
left - pad, bottom - pad, # max+1
|
|
right + pad, bottom - pad, # max+2
|
|
right + pad, top + pad, # max+3
|
|
left - pad, top + pad, # max+4
|
|
]
|
|
if self._pop_out:
|
|
down_top = in_bottom + down_pad
|
|
# 底下那个下巴
|
|
down_part = [
|
|
in_left, in_bottom, # 4
|
|
i_right, in_bottom, # 5
|
|
i_right, down_top, # 6
|
|
in_left, down_top, # 7
|
|
]
|
|
# 左下角的小方块
|
|
left_down = [
|
|
in_left, down_top, # 8
|
|
in_left + pad, down_top, # 9
|
|
in_left + pad, down_top + pad, # 10
|
|
in_left, down_top + pad, # 11
|
|
]
|
|
# 右上角的小方块
|
|
right_up = [
|
|
i_right - pad, inner_top - pad, # 12
|
|
i_right, inner_top - pad, # 13
|
|
i_right, inner_top, # 14
|
|
i_right - pad, inner_top, # 15
|
|
]
|
|
# 左上的拐弯条
|
|
# 1 2
|
|
# 4 3
|
|
# 0 5
|
|
left_up = [
|
|
in_left, down_top + pad, # 16
|
|
in_left, inner_top, # 17
|
|
i_right - pad, inner_top, # 18
|
|
i_right - pad, inner_top - pad, # 19
|
|
in_left + pad, inner_top - pad, # 20
|
|
in_left + pad, down_top + pad, # 21
|
|
]
|
|
# 右下的拐弯条
|
|
# 3 2
|
|
# 5 4
|
|
# 0 1
|
|
right_down = [
|
|
in_left + pad, down_top, # 22
|
|
i_right, down_top, # 23
|
|
i_right, inner_top - pad, # 24
|
|
i_right - pad, inner_top - pad, # 25
|
|
i_right - pad, down_top + pad, # 26
|
|
in_left + pad, down_top + pad, # 27
|
|
]
|
|
# 中间的方块
|
|
inner_box = [
|
|
in_left + pad, down_top + pad, # 28
|
|
i_right - pad, down_top + pad, # 29
|
|
i_right - pad, inner_top - pad, # 30
|
|
in_left + pad, inner_top - pad, # 31
|
|
]
|
|
return (out_border +
|
|
down_part +
|
|
left_down + right_up +
|
|
left_up + right_down +
|
|
inner_box +
|
|
highlight)
|
|
else:
|
|
# 左下角的小方块
|
|
left_down = [
|
|
in_left, in_bottom, # 4
|
|
in_left + pad, in_bottom, # 5
|
|
in_left + pad, in_bottom + pad, # 6
|
|
in_left, in_bottom + pad, # 7
|
|
]
|
|
# 右上角的小方块
|
|
right_up = [
|
|
i_right - pad, inner_top - pad, # 8
|
|
i_right, inner_top - pad, # 9
|
|
i_right, inner_top, # 10
|
|
i_right - pad, inner_top, # 11
|
|
]
|
|
# 左上的拐弯条
|
|
# 1 2
|
|
# 4 3
|
|
# 0 5
|
|
left_up = [
|
|
in_left, in_bottom + pad, # 12
|
|
in_left, inner_top, # 13
|
|
i_right - pad, inner_top, # 14
|
|
i_right - pad, inner_top - pad, # 15
|
|
in_left + pad, inner_top - pad, # 16
|
|
in_left + pad, in_bottom + pad, # 17
|
|
]
|
|
# 右下的拐弯条
|
|
# 3 2
|
|
# 5 4
|
|
# 0 1
|
|
right_down = [
|
|
in_left + pad, in_bottom, # 18
|
|
i_right, in_bottom, # 19
|
|
i_right, inner_top - pad, # 20
|
|
i_right - pad, inner_top - pad, # 21
|
|
i_right - pad, in_bottom + pad, # 22
|
|
in_left + pad, in_bottom + pad, # 23
|
|
]
|
|
# 中间的方块
|
|
inner_box = [
|
|
in_left + pad, in_bottom + pad, # 24
|
|
i_right - pad, in_bottom + pad, # 25
|
|
i_right - pad, inner_top - pad, # 26
|
|
in_left + pad, inner_top - pad, # 27
|
|
]
|
|
return (out_border +
|
|
left_down + right_up +
|
|
left_up + right_down +
|
|
inner_box +
|
|
highlight)
|
|
# fmt: on
|
|
|
|
def _create_vertex_list(self) -> None:
|
|
colors = self._colors.border * 4
|
|
# fmt: off
|
|
indices = [
|
|
0, 1, 2, # 最基本的两个三角形
|
|
0, 2, 3, # 用来画黑色边框
|
|
]
|
|
if self._pop_out:
|
|
indices += [4, 5, 6, 4, 6, 7] # 下巴
|
|
indices += [8, 9, 10, 8, 10, 11] # 左下角
|
|
indices += [12, 13, 14, 12, 14, 15] # 右上角
|
|
indices += [16, 17, 20, 16, 20, 21,
|
|
18, 19, 20, 18, 17, 20] # 左上拐弯
|
|
indices += [22, 23, 26, 22, 26, 27,
|
|
23, 24, 26, 24, 25, 26] # 右下拐弯
|
|
indices += [28, 29, 30, 28, 30, 31] # 中间的方块
|
|
# if self._highlight:
|
|
# indices = [32, 33, 34, 32, 34, 35] + indices # 高光
|
|
# else:
|
|
# indices = [32, 3, 34, 32, 5, 2] + indices # 高光
|
|
indices = [32, 33, 34, 32, 34, 35] + indices # 高光
|
|
colors += (
|
|
self._colors.down_pad * 4
|
|
+ self._colors.corner * 8
|
|
+ self._colors.right_down * 6
|
|
+ self._colors.left_up * 6
|
|
+ self._colors.inner * 4
|
|
+ ((self._colors.highlight * 4) if self._highlight else (0, 0, 0, 0) * 4)
|
|
)
|
|
self._num_verts = 36
|
|
else:
|
|
indices += [4, 5, 6, 4, 6, 7] # 左下角
|
|
indices += [8, 9, 10, 8, 10, 11] # 右上角
|
|
indices += [12, 16, 17, 12, 13, 16,
|
|
14, 15, 16, 14, 13, 16] # 左上拐弯
|
|
indices += [18, 22, 23, 18, 19, 22,
|
|
20, 21, 22, 20, 19, 22] # 右下拐弯
|
|
indices += [24, 25, 26, 24, 26, 27]
|
|
if self._highlight:
|
|
indices = [28, 29, 30, 28, 30, 31] + indices # 高光
|
|
colors += (
|
|
self._colors.corner * 8
|
|
+ self._colors.right_down * 6
|
|
+ self._colors.left_up * 6
|
|
+ self._colors.inner * 4
|
|
+ self._colors.highlight * 4
|
|
)
|
|
self._num_verts = 32
|
|
# fmt: on
|
|
self._vertex_list = self._program.vertex_list_indexed(
|
|
self._num_verts,
|
|
self._draw_mode,
|
|
indices,
|
|
self._batch,
|
|
self._group,
|
|
position=("f", self._get_vertices()),
|
|
colors=("Bn", colors),
|
|
translation=("f", (self._x, self._y) * self._num_verts),
|
|
)
|
|
|
|
def _update_vertices(self) -> None:
|
|
self._vertex_list.position[:] = self._get_vertices()
|
|
|
|
|
|
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_color(self) -> None:
|
|
self._vertex_list.colors[:] = self._rgba * self._num_verts
|
|
|
|
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):
|
|
# 背景的颜色
|
|
backgroud_color = (0, 0, 0, 255)
|
|
# 内部填充的颜色
|
|
inner_normal = (72, 73, 74, 255)
|
|
inner_press = (49, 50, 51, 255)
|
|
# 左上角的普通状态下颜色
|
|
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)
|
|
# 下巴的颜色
|
|
down_pad_color = (49, 50, 51, 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
|
|
down_pad = 5
|
|
self.pad = pad
|
|
self.down_pad = down_pad
|
|
# 覆盖式
|
|
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=30, parent=self.main_group)
|
|
# 内部填充
|
|
self.inner_group = Group(order=40, parent=self.main_group)
|
|
|
|
self.backgroud = Rectangle(
|
|
x=self.x,
|
|
y=self.y,
|
|
width=width,
|
|
height=height,
|
|
color=self.backgroud_color,
|
|
batch=self.main_batch,
|
|
group=self.background_group,
|
|
)
|
|
self.upper_border = 拐角(
|
|
x=self.x + pad,
|
|
y=self.y + pad + down_pad,
|
|
width=width - (pad * 2),
|
|
height=height - (pad * 2) - down_pad,
|
|
thick1=pad,
|
|
thick2=pad,
|
|
color=self.upper_normal,
|
|
batch=self.main_batch,
|
|
group=self.border_group,
|
|
)
|
|
self.down_border = 拐角(
|
|
x=self.x + pad,
|
|
y=self.y + pad + down_pad,
|
|
width=width - (pad * 2),
|
|
height=height - (pad * 2) - down_pad,
|
|
thick1=pad,
|
|
thick2=pad,
|
|
clockwise=False,
|
|
color=self.down_normal,
|
|
batch=self.main_batch,
|
|
group=self.border_group,
|
|
)
|
|
self.left_down = Rectangle(
|
|
x=self.x + pad,
|
|
y=self.y + pad + down_pad,
|
|
width=pad,
|
|
height=pad,
|
|
color=self.corner_normal,
|
|
batch=self.main_batch,
|
|
group=self.corner_group,
|
|
)
|
|
self.right_up = Rectangle(
|
|
x=self.x + self.width - (pad * 2),
|
|
y=self.y + self.height - (pad * 2),
|
|
width=pad,
|
|
height=pad,
|
|
color=self.corner_normal,
|
|
batch=self.main_batch,
|
|
group=self.corner_group,
|
|
)
|
|
self.inner_fill = Rectangle(
|
|
x=self.x + pad + pad,
|
|
y=self.y + pad + pad + down_pad,
|
|
width=self.width - (pad * 4),
|
|
height=self.height - (pad * 4) - down_pad,
|
|
color=self.inner_normal,
|
|
batch=self.main_batch,
|
|
group=self.inner_group,
|
|
)
|
|
self.down_fill = Rectangle(
|
|
x=self.x + pad,
|
|
y=self.y + pad,
|
|
width=width - (pad * 2),
|
|
height=down_pad,
|
|
color=self.down_pad_color,
|
|
batch=self.main_batch,
|
|
group=self.border_group,
|
|
)
|
|
|
|
def __contains__(self, pos: tuple[float, float]) -> bool:
|
|
return self._check_hit()
|
|
|
|
def on_mouse_press(self, x: int, y: int, buttons: int, modifiers: int) -> None:
|
|
if (x, y) in self:
|
|
...
|
|
|
|
|
|
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")
|