forked from s464965/WMICraft
92 lines
1.7 KiB
Python
92 lines
1.7 KiB
Python
from enum import Enum
|
|
import torchvision.transforms as transforms
|
|
import torch
|
|
|
|
GAME_TITLE = 'WMICraft'
|
|
WINDOW_HEIGHT = 800
|
|
WINDOW_WIDTH = 1360
|
|
FPS_COUNT = 60
|
|
TURN_INTERVAL = 300
|
|
|
|
GRID_CELL_PADDING = 5
|
|
GRID_CELL_SIZE = 36
|
|
ROWS = 19
|
|
COLUMNS = 24
|
|
|
|
BORDER_WIDTH = 10
|
|
BORDER_RADIUS = 5
|
|
|
|
KNIGHTS_SPAWN_WIDTH = 4
|
|
KNIGHTS_SPAWN_HEIGHT = 7
|
|
LEFT_KNIGHTS_SPAWN_FIRST_ROW = 6
|
|
LEFT_KNIGHTS_SPAWN_FIRST_COL = 0
|
|
RIGHT_KNIGHTS_SPAWN_FIRST_ROW = 6
|
|
RIGHT_KNIGHTS_SPAWN_FIRST_COL = 20
|
|
|
|
CASTLE_SPAWN_WIDTH = 6
|
|
CASTLE_SPAWN_HEIGHT = 5
|
|
CASTLE_SPAWN_FIRST_ROW = 7
|
|
CASTLE_SPAWN_FIRST_COL = 9
|
|
|
|
NBR_OF_WATER = 16
|
|
NBR_OF_TREES = 20
|
|
NBR_OF_MONSTERS = 2
|
|
NBR_OF_SANDS = 35
|
|
|
|
TILES = [
|
|
'grass1.png',
|
|
'grass2.png',
|
|
'grass3.png',
|
|
'grass4.png',
|
|
'sand.png',
|
|
'water.png',
|
|
'grass_with_tree.jpg',
|
|
]
|
|
|
|
|
|
class Direction(Enum):
|
|
UP = 0
|
|
RIGHT = 1
|
|
DOWN = 2
|
|
LEFT = 3
|
|
|
|
def right(self):
|
|
v = (self.value + 1) % 4
|
|
return Direction(v)
|
|
|
|
def left(self):
|
|
v = (self.value - 1) % 4
|
|
return Direction(v)
|
|
|
|
|
|
ACTION = {
|
|
"rotate_left": -1,
|
|
"rotate_right": 1,
|
|
"go": 0,
|
|
}
|
|
|
|
# HEALTH_BAR
|
|
BAR_ANIMATION_SPEED = 1
|
|
BAR_WIDTH_MULTIPLIER = 0.9 # (0;1>
|
|
BAR_HEIGHT_MULTIPLIER = 0.1
|
|
|
|
|
|
#NEURAL_NETWORK
|
|
learning_rate = 0.001
|
|
batch_size = 7
|
|
num_epochs = 10
|
|
|
|
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
|
classes = ['grass', 'sand', 'tree', 'water']
|
|
|
|
setup_photos = transforms.Compose([
|
|
transforms.Resize(36),
|
|
transforms.CenterCrop(36),
|
|
transforms.ToPILImage(),
|
|
transforms.ToTensor(),
|
|
transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
|
|
])
|
|
|
|
id_to_class = {i: j for i, j in enumerate(classes)}
|
|
class_to_id = {value: key for key, value in id_to_class.items()}
|