import pygame
from image import Image

class Element():
    def __init__(self,type):
        self.choice = 0
        self.image_list = []
        pass
        for i in range(1,4):
            path = f"images/{type}{i}.png"
            loaded_image = Image(path)
            self.image_list.append(loaded_image)
        pass

    def nextChoice(self):
        self.choice += 1
        if self.choice > 2:
            self.choice = 0
        pass

    def getChosenImage(self):
        return self.image_list[self.choice].image
    
class Head(Element):
    def __init__(self):
        super().__init__("head")