import math from numpy import array as a import pygame class Screen: def __init__(self, width: float, height: float, screen_size: a): pygame.init() self.width = width self.height = height self.size = a([float(width), float(height)]) self.screen_size = a(screen_size) self.matrix = a([ [screen_size[0] / width, 0], [0, screen_size[1] / 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) self.s_ = pygame.Surface([width, height]) self.s = pygame.display.set_mode(screen_size) self.center = a([width, height]) / 2 def resize_screen(self, new_screen_size, subscene=False): 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) def save(self, path, size=None): if size is None: pygame.image.save(self.s, path) else: surface = pygame.Surface(size) surface.blit(self.s, [0, 0]) pygame.image.save(surface, path)