dodany test sprawdzajacy dlugosc sciezki
This commit is contained in:
parent
83c00349d5
commit
42c0a6ccfc
5
astar.py
5
astar.py
@ -12,6 +12,7 @@ def astar(obiekty, start, cel):
|
||||
sasiedzi = [(0, 1), (0, -1), (1, 0), (-1, 0)]
|
||||
close_set = set()
|
||||
came_from = {}
|
||||
hscore = {start: heurystyka(obiekty, start, cel)}
|
||||
gscore = {start: 0}
|
||||
fscore = {start: heurystyka(obiekty, start, cel)}
|
||||
oheap = []
|
||||
@ -20,12 +21,14 @@ def astar(obiekty, start, cel):
|
||||
while oheap:
|
||||
current = heapq.heappop(oheap)[1]
|
||||
if current == cel:
|
||||
obiekty["plansza"][current[0], current[1]].setKolor((255, 0, 0))
|
||||
data = []
|
||||
while current in came_from:
|
||||
data.append(current)
|
||||
current = came_from[current]
|
||||
return data[::-1]
|
||||
close_set.add(current)
|
||||
obiekty["plansza"][current[0], current[1]].setKolor((255, 0, 0))
|
||||
for i, j in sasiedzi:
|
||||
sasiad = current[0] + i, current[1] + j
|
||||
|
||||
@ -36,6 +39,7 @@ def astar(obiekty, start, cel):
|
||||
elif 6 <= sasiad[0] <= 7 and 10 <= sasiad[1] <= 11:
|
||||
continue
|
||||
tentative_g_score = gscore[current] + heurystyka(obiekty, current, sasiad)
|
||||
|
||||
if sasiad in close_set and tentative_g_score >= gscore.get(sasiad, 0):
|
||||
continue
|
||||
if tentative_g_score < gscore.get(sasiad, 0) or sasiad not in [i[1] for i in oheap]:
|
||||
@ -43,4 +47,5 @@ def astar(obiekty, start, cel):
|
||||
gscore[sasiad] = tentative_g_score
|
||||
fscore[sasiad] = tentative_g_score + heurystyka(obiekty, sasiad, cel)
|
||||
heapq.heappush(oheap, (fscore[sasiad], sasiad))
|
||||
obiekty["plansza"][sasiad[0], sasiad[1]].setKolor((0, 255, 0))
|
||||
return False
|
||||
|
@ -57,8 +57,9 @@ class Smieciarka(pygame.sprite.Sprite):
|
||||
self.w_dol()
|
||||
|
||||
def astar_move(self, obiekty):
|
||||
sciezka = astar.astar(obiekty, self.pozycja, (14, 0))
|
||||
sciezka = astar.astar(obiekty, self.pozycja, (random.randrange(15), random.randrange(15)))
|
||||
print(sciezka)
|
||||
if sciezka:
|
||||
for koord in sciezka:
|
||||
if koord[0] == self.x - 1 and koord[1] == self.y:
|
||||
self.w_lewo()
|
||||
|
Loading…
Reference in New Issue
Block a user