import math from numpy import array as a import pygame class Screen: def __init__(self, width: float, height: float, font_size: int, 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.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.font_size = font_size self.center = a([width, height]) / 2 def save(self, path): pygame.image.save(self.s, path)