76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
import random
|
|
import numpy as np
|
|
import pygame
|
|
|
|
from common.constants import ROWS, COLUMNS, GRID_CELL_PADDING, GRID_CELL_WIDTH, GRID_CELL_HEIGHT, BORDER_WIDTH, \
|
|
BORDER_RADIUS, NBR_OF_TREES, NBR_OF_WATER
|
|
from .field import Field
|
|
|
|
|
|
class Grid:
|
|
def __init__(self, textures):
|
|
self.textures = textures
|
|
self.grid = []
|
|
self.free_fields = []
|
|
self.busy_fields = []
|
|
podkladka = np.zeros((ROWS, COLUMNS))
|
|
for drzewa in range(0, NBR_OF_TREES):
|
|
rzad = random.randint(0, ROWS-1)
|
|
kolumna = random.randint(0, COLUMNS-1)
|
|
if podkladka[rzad][kolumna] == 0:
|
|
podkladka[rzad][kolumna] = 1
|
|
else:
|
|
drzewa = drzewa - 1
|
|
for i in range(0, NBR_OF_WATER):
|
|
rzad = random.randint(0, ROWS-1)
|
|
kolumna = random.randint(0, COLUMNS-1)
|
|
if podkladka[rzad][kolumna] == 0:
|
|
podkladka[rzad][kolumna] = 2
|
|
else:
|
|
drzewa = drzewa - 1
|
|
print(podkladka)
|
|
for row in range(ROWS):
|
|
self.grid.append(pygame.sprite.Group())
|
|
for column in range(COLUMNS):
|
|
box_rect = [(GRID_CELL_PADDING + GRID_CELL_WIDTH) * column + GRID_CELL_PADDING + 7,
|
|
(GRID_CELL_PADDING + GRID_CELL_HEIGHT) * row + GRID_CELL_PADDING + 7,
|
|
GRID_CELL_WIDTH,
|
|
GRID_CELL_HEIGHT]
|
|
if podkladka[row][column] == 1:
|
|
texture_path, converted_texture = self.textures[-1]
|
|
elif podkladka[row][column] == 2:
|
|
texture_path, converted_texture = self.textures[-2]
|
|
else:
|
|
texture_path, converted_texture = self.get_random_texture()
|
|
field = Field(
|
|
texture_path,
|
|
converted_texture,
|
|
pygame.transform.scale(converted_texture,
|
|
(GRID_CELL_WIDTH,
|
|
GRID_CELL_HEIGHT)),
|
|
pygame.Rect(box_rect),
|
|
row=row,
|
|
column=column
|
|
)
|
|
if field.busy:
|
|
self.busy_fields.append(field)
|
|
else:
|
|
self.free_fields.append(field)
|
|
|
|
self.grid[row].add(field)
|
|
|
|
def update(self, screen):
|
|
self.draw(screen)
|
|
|
|
def get_random_texture(self):
|
|
texture_index = random.randint(0, len(self.textures) - 3)
|
|
return self.textures[texture_index]
|
|
|
|
def draw(self, screen):
|
|
bg_width = (GRID_CELL_PADDING + GRID_CELL_WIDTH) * COLUMNS + BORDER_WIDTH
|
|
bg_height = (GRID_CELL_PADDING + GRID_CELL_HEIGHT) * ROWS + BORDER_WIDTH
|
|
pygame.draw.rect(screen, (255, 255, 255), pygame.Rect(5, 5, bg_width, bg_height), 0, BORDER_RADIUS)
|
|
|
|
for fields_row in self.grid:
|
|
fields_row.draw(screen)
|