97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import pygame
|
|
import random
|
|
|
|
class GameObject:
|
|
x = 0
|
|
y = 0
|
|
def __init__(self, image_path, x=0, y=0):
|
|
self.image = pygame.image.load(image_path)
|
|
self.rect = self.image.get_rect(topleft=(x,y))
|
|
self.set_position(x, y)
|
|
def draw(self, surface):
|
|
surface.blit(self.image, self.rect)
|
|
def set_position(self, x, y): # TODO: FIX CENTERING
|
|
self.rect.x = x
|
|
self.rect.y = y
|
|
self.x = x
|
|
self.y = y
|
|
def move(self, dx, dy):
|
|
self.set_position(self.rect.x + dx, self.rect.y + dy)
|
|
def set_size(self, width, height):
|
|
self.image = pygame.transform.scale(self.image, (width,height))
|
|
self.rect = self.image.get_rect(topleft=(self.x,self.y))
|
|
def scale_by(self, scale):
|
|
self.image = pygame.transform.scale(self.image, (self.image.get_width() * scale, self.image.get_height() * scale))
|
|
self.rect = self.image.get_rect(topleft=(self.x,self.y))
|
|
def get_position(self):
|
|
return (self.x, self.y)
|
|
|
|
|
|
|
|
class Horse(GameObject):
|
|
default_vacceleration = 0.5
|
|
vspeed = 0
|
|
vacceleration = 0
|
|
stopped = False
|
|
color = ()
|
|
def __init__(self, image_path, x, y):
|
|
self.vspeed = 0
|
|
self.vacceleration = 0
|
|
self.stopped = False
|
|
self.color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
|
|
super().__init__(image_path, x, y)
|
|
self.set_size(75, 75)
|
|
def apply_vspeed(self):
|
|
pos = self.get_position()
|
|
self.set_position(pos[0], pos[1] + self.vspeed)
|
|
def apply_vacceleration(self):
|
|
self.vspeed += self.vacceleration
|
|
def set_vspeed(self, speed):
|
|
self.vspeed = speed
|
|
def add_vspeed(self, delta):
|
|
self.vspeed += delta
|
|
def set_vacceleration(self, vacceleration):
|
|
self.vacceleration = vacceleration
|
|
def up(self):
|
|
self.set_vacceleration(-self.default_vacceleration)
|
|
def down(self):
|
|
self.set_vacceleration(self.default_vacceleration)
|
|
def stay(self):
|
|
self.set_vacceleration(0)
|
|
def stop(self):
|
|
self.stopped = True
|
|
self.set_vacceleration(0)
|
|
self.set_vspeed(0)
|
|
|
|
|
|
|
|
|
|
class Background(GameObject):
|
|
def __init__(self, image_path, x, y):
|
|
super().__init__(image_path, x, y)
|
|
|
|
class Barrier(GameObject):
|
|
def __init__(self, image_path, x, y):
|
|
super().__init__(image_path, x, y)
|
|
|
|
class UI:
|
|
children = []
|
|
horizontal_offset = 25
|
|
mark_size = 5
|
|
def __init__(self):
|
|
self.children = []
|
|
def add_children(self, children):
|
|
new_children = [child for child in children if child not in self.children]
|
|
self.children.extend(new_children)
|
|
def draw_marks(self, screen):
|
|
x0 = 50
|
|
|
|
y0 = 300
|
|
|
|
for child in self.children:
|
|
if isinstance(child, Horse):
|
|
x0 += self.horizontal_offset
|
|
x1 = x0 + self.mark_size
|
|
y1 = y0 - self.mark_size
|
|
pygame.draw.rect(screen, child.color, (x0, y0, x1, y1))
|
|
|