71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
import pygame
|
|
from screen import SCREEN
|
|
global BLACK
|
|
|
|
# global SCREEN
|
|
global BLACK
|
|
global gridObjects
|
|
global imgTree
|
|
global imgTree
|
|
imgTree = pygame.image.load('img/tree.png')
|
|
|
|
gridObjects = {} # Store grid-box objects from Grid Class
|
|
|
|
class Grid(object):
|
|
# ta klasa rysuje kratę na ekranie
|
|
def __init__(self, x, y, sx, sy):
|
|
self.x = x
|
|
self.y = y
|
|
self.sx = sx
|
|
self.sy = sy
|
|
self.width = 1
|
|
|
|
def draw(self):
|
|
# global SCREEN
|
|
global BLACK
|
|
# SCREEN = pygame.display.set_mode([600,650])
|
|
BLACK = (0, 0, 0)
|
|
pygame.draw.rect(SCREEN, BLACK, (self.x, self.y, self.sx, self.sy), self.width)
|
|
|
|
class Box(object):
|
|
# global SCREEN
|
|
|
|
def __init__(self, x, y, sx, sy, color):
|
|
self.x = x
|
|
self.y = y
|
|
self.sx = sx
|
|
self.sy = sy
|
|
self.color = color
|
|
|
|
def draw(self):
|
|
# global SCREEN
|
|
# SCREEN = pygame.display.set_mode([600,650])
|
|
# global BLACK
|
|
pygame.draw.rect(SCREEN, self.color, pygame.Rect(self.x, self.y, self.sx, self.sy))
|
|
|
|
class Obstacle(object):
|
|
def __init__(self, mouseObj):
|
|
self.mseX = mouseObj[0]
|
|
self.mseY = mouseObj[1]
|
|
|
|
for grid in gridObjects:
|
|
g = getGridBoxes(grid)
|
|
self.x = g.x
|
|
self.y = g.y
|
|
self.sx = g.sx
|
|
self.sy = g.sy
|
|
if self.mseX > self.x and self.mseX < self.x + self.sx:
|
|
if self.mseY > self.y and self.mseY < self.y + self.sy:
|
|
self.posX = self.x
|
|
self.posY = self.y
|
|
self.gridBox = grid
|
|
|
|
def draw(self):
|
|
# pygame.draw.rect(SCREEN, GREY, pygame.Rect(self.posX, self.posY, self.sx, self.sy))
|
|
global imgTree
|
|
SCREEN.blit(imgTree, (self.posX, self.posY))
|
|
# pygame.display.update()
|
|
|
|
def getGridBoxes(grid_box):
|
|
global gridObjects
|
|
return gridObjects[grid_box] |