89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import pygame
|
||
|
|
import engine
|
||
|
|
|
||
|
|
|
||
|
|
class ModelViewer:
|
||
|
|
def __init__(self, window_size: tuple=(1200, 800), fps: int=60):
|
||
|
|
pygame.init()
|
||
|
|
self.screen = pygame.display.set_mode(window_size, flags=pygame.RESIZABLE)
|
||
|
|
|
||
|
|
self.gray = (20, 20, 20)
|
||
|
|
|
||
|
|
self.window_size = window_size
|
||
|
|
self.fps = fps
|
||
|
|
self.rotating = True
|
||
|
|
self.model = engine.Object("models/coffee_cup.png", size_multiplier=20)
|
||
|
|
|
||
|
|
self.clock = pygame.time.Clock()
|
||
|
|
|
||
|
|
self.running = True
|
||
|
|
|
||
|
|
def loop(self):
|
||
|
|
self.screen.fill(self.gray)
|
||
|
|
|
||
|
|
self.get_events()
|
||
|
|
|
||
|
|
if not self.running:
|
||
|
|
return
|
||
|
|
|
||
|
|
if self.rotating:
|
||
|
|
if self.model.rotation < 360:
|
||
|
|
self.model.rotation += 1
|
||
|
|
|
||
|
|
else:
|
||
|
|
self.model.rotation = 0
|
||
|
|
|
||
|
|
self.model.sprites = self.model.rotate_model()
|
||
|
|
self.model.surface = self.model.generate_surface()
|
||
|
|
|
||
|
|
w, h = self.screen.get_size()
|
||
|
|
mw, mh = self.model.surface.get_size()
|
||
|
|
|
||
|
|
self.model.blit(self.screen, (w / 2 - mw / 2, h / 2 - mh / 2))
|
||
|
|
|
||
|
|
pygame.display.update()
|
||
|
|
self.clock.tick(self.fps)
|
||
|
|
|
||
|
|
def get_events(self):
|
||
|
|
for event in pygame.event.get():
|
||
|
|
if event.type == pygame.QUIT:
|
||
|
|
self.quit()
|
||
|
|
|
||
|
|
elif event.type == pygame.KEYDOWN:
|
||
|
|
if event.key == pygame.K_UP:
|
||
|
|
self.model.sprite_spacing += 1
|
||
|
|
self.model.surface = self.model.generate_surface()
|
||
|
|
|
||
|
|
elif event.key == pygame.K_DOWN:
|
||
|
|
self.model.sprite_spacing -= 1
|
||
|
|
self.model.surface = self.model.generate_surface()
|
||
|
|
|
||
|
|
elif event.key == pygame.K_RIGHT:
|
||
|
|
if not self.rotating:
|
||
|
|
self.model.rotation += 10
|
||
|
|
self.model.sprites = self.model.rotate_model()
|
||
|
|
self.model.surface = self.model.generate_surface()
|
||
|
|
|
||
|
|
elif event.key == pygame.K_LEFT:
|
||
|
|
if not self.rotating:
|
||
|
|
self.model.rotation -= 10
|
||
|
|
self.model.sprites = self.model.rotate_model()
|
||
|
|
self.model.surface = self.model.generate_surface()
|
||
|
|
|
||
|
|
elif event.key == pygame.K_r:
|
||
|
|
self.rotating = not self.rotating
|
||
|
|
|
||
|
|
def quit(self):
|
||
|
|
self.running = False
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
model_viewer = ModelViewer()
|
||
|
|
|
||
|
|
while model_viewer.running:
|
||
|
|
model_viewer.loop()
|
||
|
|
|
||
|
|
print("Bye!")
|