primitives.py

import pygame


class DrawableObject:
    def __init__(self, x, y, w, h):
        self.x = x
        self.y = y
        self.width = w
        self.height = h
        self.object_name = "Drawable"

    def get_type(self):
        return self.object_name

    def draw(self, context: pygame.display):
        pygame.draw.circle(context, self.x, self.y, 100)


class Ellipse(DrawableObject):
    def __init__(self, color, x, y, w, h):
        super().__init__(x, y, w, h)
        self.object_name = "Ellipse"
        self.color = color

    def draw(self, context: pygame.display):
        center = (self.x, self.y)
        circle_rect = pygame.Rect(center[0] - self.width / 2,
                                  center[1] - self.height / 2,
                                  self.width,
                                  self.height)
        pygame.draw.ellipse(context,
                            self.color,
                            circle_rect,
                            self.width)


class Rectangle(DrawableObject):
    def __init__(self, color, x, y, w, h):
        super().__init__(x, y, w, h)
        self.object_name = "Rectangle"
        self.color = color

    def draw(self, context: pygame.display):
        center = (self.x, self.y)
        rectangle_rect = pygame.Rect(center[0] - self.width / 2,
                                     center[1] - self.height / 2,
                                     self.width,
                                     self.height)
        pygame.draw.rect(context, self.color, rectangle_rect, self.width)


class Text(DrawableObject):
    def __init__(self, content, surface, color, x, y, w, h):
        super().__init__(x, y, w, h)
        self.object_name = "Tex"
        self.color = color
        self.contents = content
        self.surface = surface

    def draw(self, context: pygame.display):
        # Set the position of the text
        text_position = self.surface.get_rect()
        text_position.center = (self.x, self.y)  # Centered on the screen
        context.blit(self.surface, text_position)