109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
import copy
|
|
import math
|
|
import pygame
|
|
from numpy import array as a
|
|
from UI.Objects.Scene import Scene
|
|
|
|
|
|
class Multiscene(Scene):
|
|
def __init__(self, width: float, height: float, scene_size: a):
|
|
super().__init__(width, height, scene_size)
|
|
|
|
self.subscenes = []
|
|
self.subscenes_prepared = False
|
|
|
|
def prepare_subscenes(self):
|
|
|
|
for subscene in self.subscenes:
|
|
subscene.multiscene = self
|
|
subscene.subscene_size = subscene.screen_size
|
|
subscene.resize_screen(self.to_ints((self.matrix @ subscene.subscene_size).tolist()), True)
|
|
|
|
self.subscenes_prepared = True
|
|
|
|
def update(self):
|
|
if self.subscenes_prepared is False:
|
|
self.prepare_subscenes()
|
|
|
|
self.s_.fill(self.bg)
|
|
|
|
for object_ in self.r_objects:
|
|
object_.blit()
|
|
|
|
rs = pygame.transform.scale(self.s_, self.screen_size)
|
|
self.s.blit(rs, [0, 0])
|
|
|
|
for scene in self.subscenes:
|
|
scene.blit()
|
|
|
|
for object_ in self.nr_objects:
|
|
object_.blit()
|
|
|
|
for object_ in self.nrd_objects:
|
|
object_.blit()
|
|
object_.progress()
|
|
|
|
for object_ in self.nrc_objects:
|
|
object_.blit()
|
|
object_.input(self.mouse_pos, self.clicked)
|
|
|
|
pygame.display.update()
|
|
|
|
def resize_screen(self, new_screen_size, subscene=False):
|
|
if self.subscenes_prepared is False:
|
|
self.prepare_subscenes()
|
|
|
|
self.screen_size = a(new_screen_size)
|
|
self.matrix = a([
|
|
[new_screen_size[0] / self.width, 0],
|
|
[0, new_screen_size[1] / self.height]
|
|
])
|
|
self.inverse_matrix = a([
|
|
[1 / self.matrix[0, 0], 0],
|
|
[0, 1 / self.matrix[1, 1]]
|
|
])
|
|
self.pd = abs(self.matrix[0, 0] * self.matrix[1, 1]) # positive determinant
|
|
self.pd_ = math.sqrt(self.pd)
|
|
|
|
if subscene is False:
|
|
self.s = pygame.display.set_mode(new_screen_size)
|
|
else:
|
|
self.s = pygame.Surface(new_screen_size)
|
|
|
|
self.resize_subscenes()
|
|
|
|
def update_mouse_events(self, mouse_pos, clicked):
|
|
mouse_pos = a(mouse_pos)
|
|
self.mouse_pos = self.inverse_matrix @ mouse_pos
|
|
self.clicked = clicked
|
|
|
|
for subscene in self.subscenes:
|
|
subscene.mouse_pos = (subscene.inverse_matrix @ (mouse_pos - (self.matrix @ subscene.position)))
|
|
subscene.clicked = clicked
|
|
|
|
def progress(self):
|
|
self.progress_subscenes()
|
|
|
|
def resize_subscenes(self):
|
|
for subscene in self.subscenes:
|
|
subscene.resize_screen(self.to_ints((self.matrix @ subscene.subscene_size).tolist()), True)
|
|
|
|
def progress_subscenes(self):
|
|
for subscene in self.subscenes:
|
|
subscene.progress()
|
|
|
|
def get_i_objects(self):
|
|
i_objects = copy.copy(self.i_objects)
|
|
|
|
for subscene in self.subscenes:
|
|
i_objects += subscene.get_i_objects()
|
|
|
|
return i_objects
|
|
|
|
@staticmethod
|
|
def to_ints(iterable):
|
|
for i in range(len(iterable)):
|
|
iterable[i] = int(iterable[i])
|
|
|
|
return iterable
|