DSZI_Survival/src/game/Map.py

import json

import pygame
from game.TerrainTile import TerrainTile
from game.Screen import Locations

from src.entities.Entity import Entity
from src.entities.Pickupable import Pickupable
from src.entities.Statistics import Statistics


class Map:
    def __init__(self, filename, screen):
        self.screen = screen
        self.terrain = []
        self.collidableTerrain = []
        self.collidables = pygame.sprite.Group()

        with open(filename, 'rt') as f:
            for line in f:
                self.terrain.append(line)

        self.tileSize = int(self.screen.mapSize/20)

        self.tileWidth = len(self.terrain[0])
        self.tileHeight = len(self.terrain)
        self.width = self.tileWidth * self.tileSize
        self.height = self.tileHeight * self.tileSize

        self.terrainDraw()

    # Returns a list of entities loaded from mapfile
    def loadEntities(self, mapFileName):
        mapFile = mapFileName.split('.')[0]
        entitiesFile = mapFile + "Entities.json"
        entityListJson = json.loads(entitiesFile)
        actualEntities = []
        for entity in entityListJson:
            try:
                if entity["isPickupable"]:
                    actualEntities.append(Pickupable(entity["name"],
                                                     (entity["position"]["x"], entity["position"]["y"]),
                                                     self.tileSize,
                                                     Statistics(entity["effect"]["hp"],
                                                                entity["effect"]["hunger"],
                                                                entity["effect"]["thirst"],
                                                                entity["effect"]["stamina"])))
                else:
                    actualEntities.append(Entity(entity["name"],
                                                    (entity["position"]["x"], entity["position"]["y"]),
                                                    self.tileSize))
            except KeyError:
                print("Failed to load entity " + entity)
        return actualEntities

    def terrainDraw(self):
        for row, tiles in enumerate(self.terrain):
            for col, tile in enumerate(tiles):
                if tile == 'w':
                    object = TerrainTile(col, row, 'wall.png', self.tileSize)
                    self.screen.draw(object, Locations.MAP, 0, 0)
                    self.collidables.add(object)
                elif tile == ',':
                    self.screen.draw(TerrainTile(col, row, 'floor.png', self.tileSize), Locations.MAP, 0, 0)
                elif tile == '.':
                    self.screen.draw(TerrainTile(col, row, 'grass.png', self.tileSize), Locations.MAP, 0, 0)
                elif tile == 'x':
                    object = TerrainTile(col, row, 'water.jpg', self.tileSize)
                    self.screen.draw(object, Locations.MAP, 0, 0)
                    self.collidables.add(object)

    def addEntity(self, entity):
        self.screen.draw(entity, Locations.MAP, 0, 0)
        self.collidables.add(entity)

    def removeSpriteFromMap(self, entity):
        if self.collidables.has(entity):
            self.collidables.remove(entity)
        self.screen.removeSprite(entity)

    # add object to map.collidables list to be collidable
    def collision(self, x, y):
        for b in self.collidables:
            if b.rect.x == x and b.rect.y == y:
                return True
        return False