luckymouse0 / snake-rl Goto Github PK

from pygame.constants import K_DOWN, K_UP, K_LEFT, K_RIGHT
from pygame_player import PyGamePlayer
import random
from Game.Snake import UP, DOWN, LEFT, RIGHT


class SnakePlayer(PyGamePlayer):
    def __init__(self):
        """  
    Example class for playing Pong  
    """
        super(SnakePlayer, self).__init__(
            force_game_fps=10
        )  # we want to run at 10 frames per second
        self.last_score = 0.0

    def get_keys_pressed(self, screen_array, feedback):
        # The code for running the actual learning agent would go here with the screen_array and feeback as the inputs
        # and an output for each key in the game, activating them if they are over some threshold.
        self._last_action = self._choose_next_action()
        return SnakePlayer._key_presses_from_action(self._last_action)

    def get_feedback(self):
        # import must be done here because otherwise importing would cause the game to start playing
        from Game.Snake import GAME_SCORE

        # get the difference in score between this and the last run
        score_change = GAME_SCORE - self.last_score
        self.last_score = GAME_SCORE
        return score_change, score_change != 0

    def _choose_next_action(self):
        **return random.choice([UP, DOWN, LEFT, RIGHT])**


if __name__ == "__main__":
    player = SnakePlayer()
    player.start()
    # importing will start the game playing
    import Game.Snake