33 lines
980 B
Python
33 lines
980 B
Python
#!/usr/bin/env python3.11
|
|
from pygame.image import load
|
|
from pygame import Surface
|
|
from dataclasses import dataclass
|
|
import os
|
|
|
|
@dataclass
|
|
class Colors:
|
|
White: tuple = (255, 255, 255)
|
|
Green: tuple = (0, 255, 0)
|
|
Yellow: tuple = (230, 193, 32)
|
|
Black: tuple = (0, 0, 0)
|
|
Gray: tuple = (50, 50, 50)
|
|
Red: tuple = (199, 54, 54)
|
|
Purple: tuple = (150, 15, 252)
|
|
|
|
colors = Colors()
|
|
|
|
def load_image(filename: str) -> Surface:
|
|
if '.' in filename and filename.split('.')[1] not in ['png', 'jpg', 'jpeg']:
|
|
raise Exception(f'Invalid filename: must be of filetype \'.png\' \'.jpg\' or \'.jpeg\' not of .{filename.split(".")[1]}')
|
|
|
|
cwd = os.getcwd()
|
|
dirName = '' if cwd.endswith('formula2D') else '/formula2D'
|
|
path = f'{cwd}{dirName}/resources/{filename}'
|
|
|
|
if not os.path.isfile(path):
|
|
raise Exception(f'Filepath not found: \'{path}\' cannot be found')
|
|
|
|
return load(path)
|
|
|
|
def limit(num, minimum=1, maximum=255): return max(min(num, maximum), minimum)
|