46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import pygame as pg
|
|
import math
|
|
from settings import *
|
|
vec = pg.math.Vector2
|
|
|
|
class aiPlayer():
|
|
def __init__(self, player, game):
|
|
self.player = player
|
|
self.game = game
|
|
|
|
def rotateAiPlayer(self, d: str):
|
|
if d == 'left':
|
|
print('in left')
|
|
self.direction -= 90
|
|
if d == 'right':
|
|
self.direction += 90
|
|
|
|
def moveAiPlayer(self):
|
|
for i in range(64 * 1):
|
|
self.player.pos += vec(1, 0).rotate(self.player.rot)
|
|
self.player.rect.center = self.player.pos
|
|
# print(f'START COORDS: {x_s, x_bias}; CURRENT AGENT COORDS: {self.player.get_actual_coords()}')
|
|
self.game.update()
|
|
self.player.update()
|
|
self.game.draw()
|
|
# print(self.player.get_actual_coords())
|
|
|
|
|
|
|
|
def turn_left(self):
|
|
self.player.rot -= 90
|
|
|
|
def turn_right(self):
|
|
self.player.rot += 90
|
|
|
|
def startAiController(self, actions):
|
|
for action in actions:
|
|
if action == 'forward':
|
|
self.moveAiPlayer()
|
|
# print(f'ROT IS {self.player.rot}')
|
|
if action == 'right':
|
|
self.turn_right()
|
|
if action == 'left':
|
|
self.turn_left()
|
|
print(f'ROT: {self.player.rot}')
|
|
print("Agent pos: ", math.floor(self.player.pos[0] / TILESIZE), math.floor(self.player.pos[1] / TILESIZE)) |