91 lines
4.1 KiB
Python
Executable File
91 lines
4.1 KiB
Python
Executable File
import random
|
||
import pygame as pg
|
||
from .. import tool
|
||
from .. import constants as c
|
||
from copy import deepcopy
|
||
|
||
class Map():
|
||
def __init__(self, background_type):
|
||
self.background_type = background_type
|
||
if self.background_type in {c.BACKGROUND_POOL, c.BACKGROUND_FOG}:
|
||
self.width = c.GRID_POOL_X_LEN
|
||
self.height = c.GRID_POOL_Y_LEN
|
||
self.map = [[(deepcopy(c.MAP_STATE_EMPTY), deepcopy(c.MAP_STATE_WATER))[y in {2, 3}] for x in range(self.width)] for y in range(self.height)]
|
||
elif self.background_type in {c.BACKGROUND_ROOF, c.BACKGROUND_ROOFNIGHT}:
|
||
self.width = c.GRID_ROOF_X_LEN
|
||
self.height = c.GRID_ROOF_Y_LEN
|
||
self.map = [[deepcopy(c.MAP_STATE_TILE) for x in range(self.width)] for y in range(self.height)]
|
||
else:
|
||
self.width = c.GRID_X_LEN
|
||
self.height = c.GRID_Y_LEN
|
||
self.map = [[deepcopy(c.MAP_STATE_EMPTY) for x in range(self.width)] for y in range(self.height)]
|
||
|
||
def isValid(self, map_x, map_y):
|
||
if (map_x < 0 or map_x >= self.width or
|
||
map_y < 0 or map_y >= self.height):
|
||
return False
|
||
return True
|
||
|
||
# 判断能否种植
|
||
def isMovable(self, map_x, map_y):
|
||
# 目前没有南瓜头，所以用是否为空判断
|
||
# 可将南瓜头新定义一个状态（如：2），基于此进一步判断
|
||
# 应当改成元组，保存南瓜头、花盆、睡莲等状态（字典也可，还更方便）
|
||
# 当然，不用元组的话字符串也行，但是得把判断植物写在母函数中，并且需要更多参数
|
||
# 这样返回的就是一个具体信息，而非bool值了
|
||
# 到时候还要改一下变量名，还叫isMovable不合适
|
||
#if self.map[map_y][map_x][c.]
|
||
return (self.map[map_y][map_x] == c.MAP_STATE_EMPTY)
|
||
|
||
def getMapIndex(self, x, y):
|
||
# 引入新地图后需要增加这里的内容
|
||
if self.background_type in {c.BACKGROUND_POOL, c.BACKGROUND_FOG}:
|
||
x -= c.MAP_POOL_OFFSET_X
|
||
y -= c.MAP_POOL_OFFSET_Y
|
||
return (x // c.GIRD_POOL_X_SIZE, y // c.GRID_POOL_Y_SIZE)
|
||
elif self.background_type in {c.BACKGROUND_ROOF, c.BACKGROUND_ROOFNIGHT}:
|
||
x -= c.MAP_ROOF_OFFSET_X
|
||
y -= c.MAP_ROOF_OFFSET_X
|
||
girdX = x // c.GRID_ROOF_X_SIZE
|
||
if girdX >= 5:
|
||
gridY = y // c.GRID_ROOF_Y_SIZE
|
||
else:
|
||
gridY = (y - 20*(6 - girdX)) // 85
|
||
return (girdX, gridY)
|
||
else:
|
||
x -= c.MAP_OFFSET_X
|
||
y -= c.MAP_OFFSET_Y
|
||
return (x // c.GRID_X_SIZE, y // c.GRID_Y_SIZE)
|
||
|
||
def getMapGridPos(self, map_x, map_y):
|
||
if self.background_type in {c.BACKGROUND_POOL, c.BACKGROUND_FOG}:
|
||
return (map_x * c.GRID_ROOF_X_SIZE + c.GRID_X_SIZE//2 + c.MAP_ROOF_OFFSET_X,
|
||
map_y * c.GRID_ROOF_Y_SIZE + c.GRID_Y_SIZE//5 * 3 + c.MAP_ROOF_OFFSET_Y)
|
||
elif self.background_type in {c.BACKGROUND_ROOF, c.BACKGROUND_ROOFNIGHT}:
|
||
return (map_x * c.GRID_POOL_X_SIZE + c.GRID_X_SIZE//2 + c.MAP_POOL_OFFSET_X,
|
||
map_y * c.GRID_POOL_Y_SIZE + 20 * max(0, (6 - map_y)) + c.GRID_Y_SIZE//5 * 3 + c.MAP_POOL_OFFSET_Y)
|
||
else:
|
||
return (map_x * c.GRID_X_SIZE + c.GRID_X_SIZE//2 + c.MAP_OFFSET_X,
|
||
map_y * c.GRID_Y_SIZE + c.GRID_Y_SIZE//5 * 3 + c.MAP_OFFSET_Y)
|
||
|
||
def setMapGridType(self, map_x, map_y, type):
|
||
self.map[map_y][map_x] = type
|
||
|
||
def addMapPlant(self, map_x, map_y, plantName):
|
||
self.map[map_y][map_x][c.MAP_PLANT].add(plantName)
|
||
|
||
def removeMapPlant(self, map_x, map_y, plantName):
|
||
self.map[map_y][map_x][c.MAP_PLANT].remove(plantName)
|
||
|
||
def getRandomMapIndex(self):
|
||
map_x = random.randint(0, self.width-1)
|
||
map_y = random.randint(0, self.height-1)
|
||
return (map_x, map_y)
|
||
|
||
def showPlant(self, x, y):
|
||
pos = None
|
||
map_x, map_y = self.getMapIndex(x, y)
|
||
if self.isValid(map_x, map_y) and self.isMovable(map_x, map_y):
|
||
pos = self.getMapGridPos(map_x, map_y)
|
||
return pos
|