50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
|
from numpy import array as a
|
||
|
|
import numpy as np
|
||
|
|
import pygame
|
||
|
|
import copy
|
||
|
|
|
||
|
|
|
||
|
|
class Image:
|
||
|
|
def __init__(self, center, scalar, path, scene):
|
||
|
|
self.center = center
|
||
|
|
self.scalar = scalar
|
||
|
|
self.path = path
|
||
|
|
self.scene = scene
|
||
|
|
self.pd_ = self.scene.pd_
|
||
|
|
|
||
|
|
self.image = pygame.image.load(self.path).convert()
|
||
|
|
self.image = pygame.transform.scale(self.image,
|
||
|
|
self.to_ints((a(self.image.get_size()) * self.scalar).tolist()))
|
||
|
|
|
||
|
|
self.size = a(self.image.get_size(), dtype=np.float64)
|
||
|
|
self.sides_ratio = self.size[0] / self.size[1]
|
||
|
|
|
||
|
|
self.rescaled_size = copy.copy(self.size)
|
||
|
|
if self.scene.matrix[1, 1] - self.scene.matrix[0, 0] / self.sides_ratio >= 0:
|
||
|
|
self.rescaled_size *= self.scene.matrix[0, 0]
|
||
|
|
else:
|
||
|
|
self.rescaled_size *= self.scene.matrix[1, 1]
|
||
|
|
|
||
|
|
self.rescaled_image = pygame.transform.scale(self.image, self.to_ints((self.size * self.pd_).tolist()))
|
||
|
|
|
||
|
|
def blit(self):
|
||
|
|
|
||
|
|
if self.pd_ != self.scene.pd_:
|
||
|
|
self.pd_ = self.scene.pd_
|
||
|
|
self.rescaled_size = copy.copy(self.size)
|
||
|
|
if self.scene.matrix[1, 1] - self.scene.matrix[0, 0] / self.sides_ratio >= 0:
|
||
|
|
self.rescaled_size *= self.scene.matrix[0, 0]
|
||
|
|
else:
|
||
|
|
self.rescaled_size *= self.scene.matrix[1, 1]
|
||
|
|
|
||
|
|
self.rescaled_image = pygame.transform.scale(self.image, self.to_ints((self.size * self.pd_).tolist()))
|
||
|
|
|
||
|
|
self.scene.s.blit(self.rescaled_image, self.scene.matrix @ self.center)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def to_ints(iterable):
|
||
|
|
for i in range(len(iterable)):
|
||
|
|
iterable[i] = int(iterable[i])
|
||
|
|
|
||
|
|
return iterable
|