277 lines
10 KiB
Python
277 lines
10 KiB
Python
import pygame
|
|
from config import *
|
|
from agent import *
|
|
from map_add_ons import *
|
|
from archer_ork import *
|
|
from infantry_ork import *
|
|
from infantry_ork2 import *
|
|
from sauron import *
|
|
from health_flower import *
|
|
#from unknown_mob import * #unknown mob
|
|
|
|
class Game:
|
|
|
|
def __init__(self):
|
|
pygame.init()
|
|
self.state =[-1,-1,-1,-1,-1,-1,-1,-1]
|
|
self.SCREEN = pygame.display.set_mode((WIDTH, HEIGHT))
|
|
self.running = True
|
|
self.clock = pygame.time.Clock()
|
|
|
|
self.BACKGROUND_IMG= pygame.image.load("./zdjecia/podloze.jpg")
|
|
self.BACKGROUND = pygame.transform.scale(self.BACKGROUND_IMG,(64,64))
|
|
|
|
self.LVL_ICON_PNG = pygame.image.load("./zdjecia/lvl_icon.png")
|
|
self.LVL_ICON = pygame.transform.scale(self.LVL_ICON_PNG,(24,24))
|
|
|
|
pygame.display.set_caption('Gra-SI')
|
|
|
|
def new(self): # tworzy się nowa sesja grania
|
|
self.all_sprites = pygame.sprite.LayeredUpdates()
|
|
self.rock_sprites = pygame.sprite.LayeredUpdates()
|
|
self.archer_orks = pygame.sprite.LayeredUpdates()
|
|
self.infantry_orks = pygame.sprite.LayeredUpdates()
|
|
self.infantry_orks2 = pygame.sprite.LayeredUpdates()
|
|
self.sauronL = pygame.sprite.LayeredUpdates()
|
|
self.flowers = pygame.sprite.LayeredUpdates()
|
|
#self.unknown_mobs = pygame.sprite.LayeredUpdates() #unknown mob
|
|
self.agent = Agent(self,1,1)
|
|
self.archer_ork = Archer_ork(self,10,10)
|
|
self.enemy_cells.append(self.get_cell_number(self.archer_ork.x,self.archer_ork.y))
|
|
self.infantry_ork = Infantry_ork(self,10,4)
|
|
self.enemy_cells.append(self.get_cell_number(self.infantry_ork.x,self.infantry_ork.y))
|
|
self.infantry_ork2 = Infantry_ork2(self,6,3)
|
|
self.enemy_cells.append(self.get_cell_number(self.infantry_ork2.x,self.infantry_ork2.y))
|
|
self.sauron = Sauron(self, 1, 10)
|
|
self.enemy_cells.append(self.get_cell_number(self.sauron.x,self.sauron.y))
|
|
self.flower = Health_flower(self, 8,2)
|
|
#self.unknown_mob = Unknown_mob(self,8,8) #unknown mob
|
|
for y in range(5):
|
|
self.rock = Rocks(self,3,y)
|
|
self.wall_cells.append(self.get_cell_number(self.rock.x,self.rock.y))
|
|
|
|
|
|
def update(self):
|
|
self.all_sprites.update()
|
|
|
|
|
|
def events(self):
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
self.running = False
|
|
pygame.quit()
|
|
if event.type == pygame.MOUSEBUTTONDOWN:
|
|
mouse_presses = pygame.mouse.get_pressed()
|
|
if mouse_presses[0]:
|
|
self.bfs()
|
|
|
|
def map(self): # tworzenie mapy
|
|
self.clock.tick(FRAMERATE)
|
|
for x in range(0, WIDTH, TILE_SIZE):
|
|
for y in range(0, 768, TILE_SIZE):
|
|
self.SCREEN.blit(self.BACKGROUND,(x,y))
|
|
self.rect = pygame.Rect(x, y, TILE_SIZE, TILE_SIZE)
|
|
pygame.draw.rect(self.SCREEN, BLACK, self.rect, 1)
|
|
self.all_sprites.draw(self.SCREEN)
|
|
self.rock_sprites.draw(self.SCREEN)
|
|
self.SCREEN.blit(self.LVL_ICON, (340 ,780))
|
|
pygame.display.update()
|
|
|
|
def main(self):
|
|
self.events()
|
|
self.update()
|
|
self.map()
|
|
|
|
# BFS ALGORITHM
|
|
|
|
open_queue = []
|
|
close_queue = []
|
|
wall_cells = []
|
|
enemy_cells = []
|
|
|
|
def bfs(self):
|
|
print("x: ", self.agent.x, "y: ", self.agent.y)
|
|
|
|
self.open_queue.append(self.get_cell_number(self.agent.x,self.agent.y))
|
|
# tutaj dodaje się cel agenta
|
|
goal_cell = self.get_cell_number(448,128)
|
|
|
|
path = []
|
|
processing = True
|
|
find_path = False
|
|
while processing: # główna pętla
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
exit()
|
|
|
|
if len(self.open_queue) > 0 :
|
|
current_node_cell = self.open_queue.pop(0)
|
|
|
|
if(current_node_cell in self.close_queue):
|
|
continue
|
|
|
|
print("Aktualna kratka: ", current_node_cell)
|
|
print("Cel znajduje się na kratce: ", goal_cell)
|
|
|
|
if (current_node_cell == goal_cell):
|
|
self.close_queue.append(current_node_cell)
|
|
found_goal_cell = current_node_cell
|
|
print("Znaleziono cel, szukanie drogi z odwiedzonych węzłów, kolejka odwiedzonych:", self.close_queue)
|
|
processing = False
|
|
find_path = True
|
|
self.clock.tick(2)
|
|
else:
|
|
child_node_cells = self.get_child_nodes(current_node_cell)
|
|
self.close_queue.append(current_node_cell)
|
|
print("Sąsiedzi: ", child_node_cells)
|
|
for child_node in child_node_cells:
|
|
if child_node not in self.open_queue and child_node not in self.close_queue:
|
|
self.open_queue.append(child_node)
|
|
print("Kolejka: ", self.open_queue, "\n")
|
|
else:
|
|
print("Brak nowych węzłów, kolejka: ",self. open_queue)
|
|
print("Odwiedzone : ", self.close_queue)
|
|
return self.close_queue
|
|
|
|
dead_end_nodes = []
|
|
while find_path:
|
|
path.append(self.close_queue[0])
|
|
|
|
for i in range(len(self.close_queue) -1):
|
|
from_cell = path[-1]
|
|
to_cell = self.close_queue[i+1]
|
|
|
|
if to_cell in dead_end_nodes:
|
|
continue
|
|
|
|
if self.verify_move(from_cell, to_cell):
|
|
path.append(to_cell)
|
|
|
|
if path[-1] == found_goal_cell:
|
|
find_path = False
|
|
else:
|
|
dead_end_nodes.append(path[-1])
|
|
path = []
|
|
|
|
print("Droga: ", path)
|
|
self.move_agent(path)
|
|
|
|
|
|
def get_cell_number(self,x, y): #zamienia koordynaty na numer kratki
|
|
cell_number = None
|
|
cell_number =(x // TILE_SIZE) + (NUM_ROWS * (( y// TILE_SIZE)))
|
|
return cell_number
|
|
|
|
def get_child_nodes(self,cell_number):
|
|
children = []
|
|
up = self.get_up_cell(cell_number)
|
|
if up is not None and up not in self.wall_cells and up not in self.enemy_cells:
|
|
children.append(up)
|
|
|
|
right = self.get_right_cell(cell_number)
|
|
if right is not None and right not in self.wall_cells and up not in self.enemy_cells:
|
|
children.append(right)
|
|
|
|
down = self.get_down_cell(cell_number)
|
|
if down is not None and down not in self.wall_cells and up not in self.enemy_cells:
|
|
children.append(down)
|
|
|
|
left = self.get_left_cell(cell_number)
|
|
if left is not None and left not in self.wall_cells and up not in self.enemy_cells:
|
|
children.append(left)
|
|
|
|
return children
|
|
|
|
|
|
def get_up_cell(self,cell_number):
|
|
cell_row_number = cell_number // NUM_ROWS
|
|
if (cell_row_number - 1 < 0):
|
|
return None
|
|
else:
|
|
return (cell_number - NUM_ROWS)
|
|
|
|
def get_right_cell(self,cell_number):
|
|
cell_column_number = cell_number % NUM_ROWS
|
|
if (cell_column_number + 1 >= NUM_ROWS):
|
|
return None
|
|
else:
|
|
return (cell_number + 1)
|
|
|
|
def get_down_cell(self,cell_number):
|
|
cell_row_number = cell_number // NUM_ROWS
|
|
if (cell_row_number + 1 >= NUM_ROWS):
|
|
return None
|
|
else:
|
|
return (cell_number + NUM_ROWS)
|
|
|
|
def get_left_cell(self,cell_number):
|
|
cell_column_number = cell_number % NUM_ROWS
|
|
if (cell_column_number - 1 < 0):
|
|
return None
|
|
else:
|
|
return (cell_number - 1)
|
|
|
|
def verify_move(self,from_cell, to_cell): #sprawdzenie czy ruch jest poprawny czyt. czy następna kratka to przeszkoda lub mob
|
|
if (to_cell in self.wall_cells or to_cell in self.enemy_cells):
|
|
return False
|
|
|
|
if(from_cell + 1 == to_cell):
|
|
return True
|
|
|
|
if(from_cell - 1 == to_cell):
|
|
return True
|
|
|
|
if(from_cell - NUM_ROWS == to_cell):
|
|
return True
|
|
|
|
if(from_cell + NUM_ROWS == to_cell):
|
|
return True
|
|
|
|
return False
|
|
|
|
def move_agent(self,path):
|
|
for cell_to_move in path:
|
|
x, y = self.get_coordinates(cell_to_move)
|
|
print("Ruch do kratki : ", cell_to_move, " z x: ", x, ", y: ", y, ", agent.x: ", self.agent.rect.x, ", agent.y: ", self.agent.rect.y)
|
|
if(self.get_cell_number(self.agent.x,self.agent.y)!=cell_to_move):
|
|
if x > self.agent.rect.x:
|
|
self.agent.direction = 0
|
|
elif y > self.agent.rect.y:
|
|
self.agent.direction = 1
|
|
elif x < self.agent.rect.x:
|
|
self.agent.direction = 2
|
|
elif y < self.agent.rect.y:
|
|
self.agent.direction = 3
|
|
if self.agent.direction==0:
|
|
print("DIRECTION: "+self.agent.AGENT_IMAGES[self.agent.direction])
|
|
self.agent.x_change += TILE_SIZE
|
|
elif self.agent.direction==1:
|
|
print("DIRECTION: "+self.agent.AGENT_IMAGES[self.agent.direction])
|
|
self.agent.y_change += TILE_SIZE
|
|
elif self.agent.direction==2:
|
|
print("DIRECTION: "+self.agent.AGENT_IMAGES[self.agent.direction])
|
|
self.agent.x_change -= TILE_SIZE
|
|
elif self.agent.direction==3:
|
|
print("DIRECTION: "+self.agent.AGENT_IMAGES[self.agent.direction])
|
|
self.agent.y_change -= TILE_SIZE
|
|
|
|
self.update()
|
|
self.map()
|
|
|
|
print("Położenie agenta: agent.x: ", self.agent.rect.x, ", agent.y: ", self.agent.rect.y)
|
|
self.clock.tick(2)
|
|
|
|
def get_coordinates(self,cell_to_move): #zamienia numer kratki na koordynaty
|
|
cell_row_number = cell_to_move // NUM_ROWS
|
|
cell_column_number = cell_to_move % NUM_ROWS
|
|
|
|
y = cell_row_number * TILE_SIZE
|
|
x = cell_column_number * TILE_SIZE
|
|
return x, y
|
|
|
|
|
|
g = Game()
|
|
g.new()
|
|
while g.running:
|
|
g.main()
|