resources.py

from PIL import Image
from abc import ABC, abstractmethod


class ResourceIcon:
    labelImgSpacing = 5
    ImgSpacing = 28
    def __init__(self,tile, app):
        self.tile = tile
        self.app = app
        self.imgSize = self.tile.getResources()[0].getType().getImage().size
        app.resourceIcons.add(self)

    def getResources(self):
        return self.tile.resources

    def getImageLocs(self):
        result = []
        resources = self.getResources()
        numOfResources = len(resources)
        tileWidth, tileHeight = self.tile.getSize()

        location = self.tile.getRelativeLoc(self.tile.row,self.tile.col, self.app.map)
        if location != None:
            relativeRow, relativeCol = location
        else: return None
        
        centerX,y = self.tile.mapToScreenCords((relativeRow, relativeCol), self.tile.getSize(), 
                                                    (self.app.width, self.app.height), self.app.map.getRenderedMap(), 
                                                    self.app.mapRenderer)
        centerX = centerX + tileWidth//2 - self.imgSize[0]//2
        y = y + tileHeight//2 - 1*self.imgSize[1]//4

        for i in range(numOfResources):
            correction = i * ResourceIcon.ImgSpacing - (ResourceIcon.ImgSpacing * (numOfResources))//4
            result.append((centerX + correction, y))

        return result
    
    def getLabelLocs(self, imageLocs):
        result = []
        
        for i in range(len(imageLocs)):
            x = imageLocs[i][0] - ResourceIcon.labelImgSpacing
            y = imageLocs[i][1] + self.imgSize[i]//2
            result.append((x,y))
        return result
    
    def deleteIcon(self):
        self.app.resourceIcons.remove(self)
        self.tile.resoruceIcon = None
        
class ResourceType(ABC):
    @abstractmethod
    def getName(self):
        pass
    @abstractmethod
    def getImagePath(self):
        pass
    @abstractmethod
    def getImage(self):
        pass

class Production(ResourceType):
    def __init__(self):
        self.name = 'Production'
        self.imagePath = 'sprites/small_gear.png'
        self.image = Image.open(self.imagePath)

    def __repr__(self):
        return self.name
    
    def __hash__(self):
        return hash(str(self))
    
    def __eq__(self,other):
        return isinstance(other, Production)

    def getName(self):
        return self.name
    
    def getImagePath(self):
        return self.imagePath
    
    def getImage(self):
        return self.image
    
class Food(ResourceType):
    def __init__(self):
        self.name = 'Food'
        self.imagePath = 'sprites/small_apple.png'
        self.image = Image.open(self.imagePath)

    def __repr__(self):
        return self.name
    
    def __hash__(self):
        return hash(str(self))
    
    def __eq__(self,other):
        return isinstance(other, Food)

    def getName(self):
        return self.name
    
    def getImagePath(self):
        return self.imagePath
    
    def getImage(self):
        return self.image

class ResourceStack:
    ResourceTypes = {'production' : Production(), 'food' : Food()}
    def __init__(self, type, amount):
        self.type = type
        self.amount = amount

    def __repr__(self):
        return f'({self.type.getName()}, {self.amount})'
    
    def __hash__(self):
        return hash(str(self))
    
    def __eq__(self,other):
        return isinstance(other, ResourceStack) and self.getType() == other.getType() and self.getAmount == other.getAmount
    
    def getType(self):
        return self.type
    
    def getAmount(self):
        return self.amount
    
    def setAmount(self, amount):
        self.amount = amount

    def getResource(resourceStr):
        try:
            return ResourceStack.ResourceTypes[resourceStr]
        except:
            if resourceStr not in ResourceStack.ResourceTypes: print("NOT A VALID RESOURCE")
            assert(False == True)

    @staticmethod
    def addStackToCivilization(stack, civilization):
        type = stack.getType()
        _yield = civilization.yieldsByType[type]
        _yield += stack.getAmount()
        civilization.yieldsByType[type] = _yield
    
    @staticmethod
    def addStackToSettlement(stack, settlement):
        ResourceStack.addStackToCivilization(stack, settlement)