150 lines
5.1 KiB
Python
150 lines
5.1 KiB
Python
# Vector Graphics Library
|
|
# vgllib.py
|
|
import pygame
|
|
import uuid
|
|
import time
|
|
import threading
|
|
|
|
class Graph:
|
|
@staticmethod
|
|
def rect(frame, pos: tuple = (0, 0), size: tuple = (1, 1), color: tuple = (255, 255, 255), width: int = 0):
|
|
pygame.draw.rect(frame.surface, color=color, rect=(pos[0], pos[1], size[0], size[1]), width=width)
|
|
|
|
@staticmethod
|
|
def line(frame, start_pos: tuple, end_pos: tuple, color: tuple = (255, 255, 255)):
|
|
pygame.draw.aaline(frame.surface, color, start_pos, end_pos)
|
|
|
|
@staticmethod
|
|
def circle(frame, center: tuple, radius: int, color: tuple = (255, 255, 255), width: int = 0):
|
|
pygame.draw.circle(frame.surface, color, center, radius, width)
|
|
|
|
@staticmethod
|
|
def ellipse(frame, pos: tuple = (0, 0), size: tuple = (1, 1), color: tuple = (255, 255, 255), width: int = 0):
|
|
rect=(pos[0], pos[1], size[0], size[1])
|
|
pygame.draw.ellipse(frame.surface, color, rect, width)
|
|
|
|
@staticmethod
|
|
def polygon(frame, pointlist: list, color: tuple = (255, 255, 255), width: int = 0):
|
|
pygame.draw.polygon(frame.surface, color, pointlist, width)
|
|
|
|
@staticmethod
|
|
def arc(frame, pos: tuple = (0, 0), size: tuple = (1, 1), color: tuple = (255, 255, 255), start_angle: float = 0, stop_angle: float = 3.14, width: int = 1):
|
|
rect=(pos[0], pos[1], size[0], size[1])
|
|
pygame.draw.arc(frame.surface, color, rect, start_angle, stop_angle, width)
|
|
|
|
@staticmethod
|
|
def point(frame, pos: tuple, color: tuple = (255, 255, 255)):
|
|
pygame.draw.point(frame.surface, color, pos)
|
|
|
|
@staticmethod
|
|
def lines(frame, pointlist: list, color: tuple = (255, 255, 255)):
|
|
pygame.draw.aalines(frame.surface, color, closed=False, points=pointlist)
|
|
|
|
@staticmethod
|
|
def call(frame, method, **kwargs):
|
|
if hasattr(Graph, method):
|
|
getattr(Graph, method)(frame, **kwargs)
|
|
else:
|
|
print(f"方法 {method} 不存在")
|
|
|
|
class Frame(object):
|
|
components = dict()
|
|
components_stat = dict()
|
|
thread = None
|
|
motion_queue = None
|
|
def __init__(self, name: str, size: tuple):
|
|
self.name = name
|
|
self.size = size
|
|
self.surface = pygame.Surface(size, flags=pygame.HWSURFACE)
|
|
self.is_hide = False
|
|
print("初始化子模块")
|
|
self.thread = threading.Thread(target=self.render)
|
|
|
|
def move(self, subname, direction, duration, effect="linear"): # our powerful move!
|
|
# direction: 使用角度制, 以直角笛卡尔坐标系的x正半轴方向为0度, 逆时针为加, 接受负数
|
|
# duration: "动画"时间, 为0则即时
|
|
# effect: "动画"效果, linear为线性移动, 或许会在未来增加贝塞尔曲线
|
|
# TODO: 增加 bezier 曲线
|
|
pass
|
|
|
|
def render(self):
|
|
while 1:
|
|
self.draw_all()
|
|
|
|
def show(self, window, position: tuple):
|
|
if not self.is_hide:
|
|
window.blit(self.surface, position)
|
|
|
|
def set_visible(self, newstat=True):
|
|
self.is_hide = newstat
|
|
|
|
def set_position(self, newposition):
|
|
self.position = newposition
|
|
|
|
def register(self, subname="", attr=None):
|
|
if subname == "":
|
|
subname = uuid.uuid4()
|
|
# use percent of frame size instead of pixels :)
|
|
attr['pos'] = list(attr['pos'])
|
|
attr['pos'][0] = round(attr['pos'][0] / 100 * self.size[0])
|
|
attr['pos'][1] = round(attr['pos'][1] / 100 * self.size[1])
|
|
attr['size'] = list(attr['size'])
|
|
attr['size'][0] = round(attr['size'][0] / 100 * self.size[0])
|
|
attr['size'][1] = round(attr['size'][1] / 100 * self.size[1])
|
|
self.components[subname] = attr
|
|
self.components_stat[subname] = 1 # by default, not hiding
|
|
|
|
def draw(self, attr):
|
|
Graph.call(self, **attr)
|
|
|
|
def set_component_visible(self, subname, newstat):
|
|
self.components_stat[subname] = newstat
|
|
|
|
def draw_all(self):
|
|
for i in self.components.keys():
|
|
if self.components_stat[i]:
|
|
self.draw(self.components[i])
|
|
|
|
def refresh(self, color=(0,0,0)):
|
|
self.surface.fill(color)
|
|
|
|
def clear(self):
|
|
components = dict()
|
|
components_stat = dict()
|
|
|
|
def loads(self, grap_str):
|
|
# TODO: 将会重写 以替代不安全的 eval
|
|
self.register(subname="", attr=(eval(grap_str)))
|
|
|
|
def load(self, file="default.vgld", mode="a"):
|
|
# a: 增量加载 (默认)
|
|
# w: 覆盖式加载
|
|
# 文件扩展名: vgld (矢量图形层描述文件)
|
|
if mode == 'w':
|
|
self.clear()
|
|
with open(file=file, mode="r+", encoding="UTF-8") as f:
|
|
for i in f.readlines():
|
|
#print(i)
|
|
self.loads(i)
|
|
|
|
|
|
# 示例
|
|
if __name__ == "__main__":
|
|
pygame.init()
|
|
window = pygame.display.set_mode((800, 600))
|
|
frame = Frame("Test", (800, 600))
|
|
input()
|
|
frame.load()
|
|
running = True
|
|
while running:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
running = False
|
|
|
|
window.fill((0, 0, 0))
|
|
frame.draw_all()
|
|
frame.show(window, (0, 0))
|
|
pygame.display.flip()
|
|
pygame.time.delay(10)
|
|
pygame.quit()
|