37줄 코드로 간단한 게임 AI 구현

17926 단어 프로그래밍 인생
쓸데없는 소리 하지 말고 직접 코드를 올리는 것은 신경 네트워크와 아무런 관계가 없다. 이 37줄 코드는 컴퓨터의 대적 희생률이 1:10 정도인 것을 보장할 수 있을 뿐이다. 수동으로 조종하려면 autopilot 뒤에 있는 코드를 주석해 버리면 된다.
어느 신이 흥미가 있으면 tensorflow나serpentai로 이 게임을 해보세요.
# coding=utf8

from random import randint as rint

import pygame
from pygame.locals import *


class MySprite(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.master_image = None
        self.frame = 0
        self.old_frame = -1
        self.frame_width = 1
        self.frame_height = 1
        self.first_frame = 0
        self.last_frame = 0
        self.columns = 1
        self.last_time = 0
        self.direction = 0
        self.velocity = Point(0.0, 0.0)

    # X property
    def _getx(self):
        return self.rect.x

    def _setx(self, value):
        self.rect.x = value

    X = property(_getx, _setx)

    # Y property
    def _gety(self):
        return self.rect.y

    def _sety(self, value):
        self.rect.y = value

    Y = property(_gety, _sety)

    # position property
    def _getpos(self):
        return self.rect.topleft

    def _setpos(self, pos):
        self.rect.topleft = pos

    position = property(_getpos, _setpos)

    def load(self, filename, width, height, columns):
        self.master_image = pygame.image.load(filename).convert_alpha()
        self.frame_width = width
        self.frame_height = height
        self.rect = Rect(0, 0, width, height)
        self.columns = columns
        # try to auto-calculate total frames
        rect = self.master_image.get_rect()
        self.last_frame = (rect.width // width) * (rect.height // height) - 1

    def update(self, current_time, rate=30):
        # update animation frame number
        if current_time > self.last_time + rate:
            self.frame += 1
            if self.frame > self.last_frame:
                self.frame = self.first_frame
            self.last_time = current_time

        # build current frame only if it changed
        if self.frame != self.old_frame:
            frame_x = (self.frame % self.columns) * self.frame_width
            frame_y = (self.frame // self.columns) * self.frame_height
            rect = Rect(frame_x, frame_y, self.frame_width, self.frame_height)
            self.image = self.master_image.subsurface(rect)
            self.old_frame = self.frame

    def __str__(self):
        return str(self.frame) + "," + str(self.first_frame) + \
               "," + str(self.last_frame) + "," + str(self.frame_width) + \
               "," + str(self.frame_height) + "," + str(self.columns) + \
               "," + str(self.rect)


# Point class


class Point(object):
    def __init__(self, x, y):
        self.__x = x
        self.__y = y

    # X property
    def getx(self): return self.__x

    def setx(self, x): self.__x = x

    x = property(getx, setx)

    # Y property
    def gety(self): return self.__y

    def sety(self, y): self.__y = y

    y = property(gety, sety)

    def __str__(self):
        return "{X:" + "{:.0f}".format(self.__x) + \
               ",Y:" + "{:.0f}".format(self.__y) + "}"


pygame.init()
playerHealth = 100
keys = [False, False, False, False]
skybg = pygame.image.load("resources/images/skybg.jpg")
missile = pygame.image.load("resources/images/bullet2.png")
screen = pygame.display.set_mode((1024, 600))
pygame.display.set_caption("SkyFight")
timer = pygame.time.Clock()
# warship sprite
warship_group = pygame.sprite.Group()
# enemy sprite
chopper_group = pygame.sprite.Group()
# player sprite
player_group = pygame.sprite.Group()
player = MySprite()
player.load("resources/images/playerchopper.png", 173, 58, 4)
player.position = 0, 350
player_group.add(player)
# player missile sprite
missile_group = pygame.sprite.Group()
# enemy missile sprite
enemy_missile_group = pygame.sprite.Group()
# warship missile group
warship_missile_group = pygame.sprite.Group()
# flame sprite
flame_group = pygame.sprite.Group()


def addNewFlame(position):
    flame = MySprite()
    flame.load("resources/images/flame.png", 64, 64, 4)
    flame.position = position
    flame_group.add(flame)


# warship flame
warship_flame_group = pygame.sprite.Group()


def addNewWarshipFlame(position):
    warship_flame = MySprite()
    warship_flame.load("resources/images/warship_flame.png", 64, 64, 4)
    warship_flame.position = position
    warship_flame_group.add(warship_flame)


badtimer = 100
badtimer1 = 0
missile_launched = 0
missile_hit = 0
missile_hit_warship = 0
total_enemy_showedup = 0
enemy_missile_launched = 0
warship_health = 10
# to make autopilot missile launch interval
tickstart = []
tickstart.append(pygame.time.get_ticks())
# player move speed
player_move_speed = 5
while True:
    # start game ticks
    timer.tick(60)
    ticks = pygame.time.get_ticks()
    screen.fill(0)
    # draw background sky
    screen.blit(skybg, (0, 0))
    # add warship sprite (only one warship on the screen)
    if len(warship_group) == 0:
        warship = MySprite()
        warship.load("resources/images/warship.png", 350, 70, 1)
        warship.position = 1024, 500
        warship_group.add(warship)
    # adjust warship to come close to player
    if warship.X < player.X + 200:
        warship.X += 1
    elif warship.X > player.X + 200:
        warship.X -= 1
    # draw warship
    warship_group.update(ticks)
    warship_group.draw(screen)

    # warship launch missile
    for warship in warship_group:
        if len(warship_missile_group) == 0:
            if warship.X < 800:
                warship_missile = MySprite()
                warship_missile.load("resources/images/warship_missile.png", 32, 121, 1)
                warship_missile.position = warship.X + 160, warship.Y - 60
                warship_missile_group.add(warship_missile)

    # guide missile & change missile position
    for warship_missilex in warship_missile_group:
        warship_missilex.Y -= 1
        if warship_missilex.Y < -121:
            warship_missile_group.remove(warship_missilex)
        # warship missile & enemy collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(
            warship_missilex, chopper_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(warship_missilex, collision):
                chopper_group.remove(collision)
                flameposition = warship_missilex.position
                addNewFlame(flameposition)
        # warship missile & enemy missile collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(
            warship_missilex, enemy_missile_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(warship_missilex, collision):
                enemy_missile_group.remove(collision)
                flameposition = warship_missilex.position
                addNewFlame(flameposition)
        # warship missile & player missile collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(
            warship_missilex, missile_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(warship_missilex, collision):
                missile_group.remove(collision)
                flameposition = warship_missilex.position
                addNewFlame(flameposition)
        # warship missile & player collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(
            warship_missilex, player_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(warship_missilex, collision):
                playerHealth -= 10
                warship_missile_group.remove(warship_missilex)
                flameposition = warship_missilex.position
                addNewFlame(flameposition)

    # add random enenmy chopper
    badtimer -= 1
    if badtimer == 0:
        chopper = MySprite()
        chopper.load("resources/images/enemychopper.png", 173, 60, 4)
        chopper.position = 1024, rint(0, 400)
        chopper_group.add(chopper)
        total_enemy_showedup += 1
        badtimer = 150 - (badtimer1 * 2)
        if badtimer1 >= 35:
            badtimer1 = 35
        else:
            badtimer1 += 5
    # change enemy chopper position
    for chopper in chopper_group:
        chopper.X -= 2
        if chopper.X < -173:
            chopper_group.remove(chopper)
        # enemy chopper launch missile
        if chopper.X % 500 == 0:
            enemy_missile_launched += 1
            enemy_missile = MySprite()
            enemy_missile.load("resources/images/missile.png", 64, 16, 1)
            enemy_missile.position = chopper.X, chopper.Y + 25
            enemy_missile_group.add(enemy_missile)
    # change enemy missile position
    for enemy_missilex in enemy_missile_group:
        enemy_missilex.X -= 4
        if enemy_missilex.X < -64:
            enemy_missile_group.remove(enemy_missilex)
        # enemy missile & player collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(
            enemy_missilex, player_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(enemy_missilex, collision):
                enemy_missile_group.remove(enemy_missilex)
                playerHealth -= 1
                flameposition = player.X + 70, player.Y
                addNewFlame(flameposition)
        # enemy missile & player missile collision detect (to remove enemy/player missile & add flame)
        collision = None
        collision = pygame.sprite.spritecollideany(
            enemy_missilex, missile_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(enemy_missilex, collision):
                missile_group.remove(collision)
                enemy_missile_group.remove(enemy_missilex)
                flameposition = enemy_missilex.X, enemy_missilex.Y - 30
                addNewFlame(flameposition)

    # draw enemy chopper spritew
    chopper_group.update(ticks)
    chopper_group.draw(screen)
    # draw player
    player_group.update(ticks)
    player_group.draw(screen)

    # change player missile position
    for missilex in missile_group:
        missilex.velocity.x += 0.5
        missilex.X += missilex.velocity.x
        if missilex.X > 1056:
            missile_group.remove(missilex)
        # player missile & enemy chopper collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(missilex, chopper_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(1)(missilex, collision):
                missile_hit += 1
                chopper_group.remove(collision)
                missile_group.remove(missilex)
                flameposition = missilex.X + \
                                80, missilex.Y - 30
                addNewFlame(flameposition)
        # player missile & enemy warship collision detect
        collision = None
        collision = pygame.sprite.spritecollideany(missilex, warship_group)
        if collision != None:
            if pygame.sprite.collide_circle_ratio(0.65)(missilex, collision):
                missile_hit_warship += 1
                warship_health -= 1
                missile_group.remove(missilex)
                warshipflameposition = missilex.X + \
                                       rint(1, 100), missilex.Y - 20
                addNewWarshipFlame(warshipflameposition)
                # random flame after warship is destroyed
                if warship_health == 0:
                    for i in xrange(20):
                        position = warship.X + \
                                   rint(-10, 300), warship.Y + \
                                   rint(-20, 100)
                        addNewFlame(position)
                    warship_health = 10
                    warship_group.remove(collision)

    # enemy collision with player
    collision = None
    collision = pygame.sprite.spritecollideany(player, chopper_group)
    if collision != None:
        if pygame.sprite.collide_circle_ratio(0.65)(player, collision):
            chopper_group.remove(collision)
            playerHealth -= 1
            flameposition = player.X + 70, player.Y
            addNewFlame(flameposition)

    # draw player missiles
    missile_group.update(ticks)
    missile_group.draw(screen)

    # draw enemy missiles
    enemy_missile_group.update(ticks)
    enemy_missile_group.draw(screen)

    # draw warship missile
    warship_missile_group.update(ticks)
    warship_missile_group.draw(screen)

    # draw flame
    for x in flame_group:
        if x.frame == x.last_frame:
            flame_group.remove(x)
    flame_group.update(ticks)
    flame_group.draw(screen)

    # draw warship flame
    for x in warship_flame_group:
        if x.frame == x.last_frame:
            warship_flame_group.remove(x)
    warship_flame_group.update(ticks)
    warship_flame_group.draw(screen)

    # game statistics
    font = pygame.font.SysFont(None, 25)
    text1 = font.render("Player Health: " +
                        str(playerHealth), True, (255, 0, 0))
    screen.blit(text1, (0, 480))
    text2 = font.render("Missile Launched: " +
                        str(missile_launched), True, (255, 0, 0))
    screen.blit(text2, (0, 500))
    text3 = font.render("Missile Hit: " +
                        str(missile_hit), True, (255, 0, 0))
    screen.blit(text3, (0, 520))
    if not missile_launched == 0:
        hitrate = float(missile_hit) / float(missile_launched) * 100
        text4 = font.render("Hit Rate: " +
                            str(hitrate) + "%", True, (255, 0, 0))
    else:
        text4 = font.render("Hit Rate: " +
                            "0%", True, (255, 0, 0))
    screen.blit(text4, (0, 540))
    text5 = font.render("Total Enemies: " +
                        str(total_enemy_showedup), True, (255, 0, 0))
    screen.blit(text5, (0, 560))
    text6 = font.render("Enemy Missile Launched: " +
                        str(enemy_missile_launched), True, (255, 0, 0))
    screen.blit(text6, (0, 580))

    pygame.display.update()

    key = pygame.key.get_pressed()
    if key[pygame.K_ESCAPE]:
        exit()
    for event in pygame.event.get():
        if event.type == pygame.KEYDOWN:
            if event.key == K_w:
                keys[0] = True
            elif event.key == K_a:
                keys[1] = True
            elif event.key == K_s:
                keys[2] = True
            elif event.key == K_d:
                keys[3] = True
        if event.type == pygame.KEYUP:
            if event.key == pygame.K_w:
                keys[0] = False
            elif event.key == pygame.K_a:
                keys[1] = False
            elif event.key == pygame.K_s:
                keys[2] = False
            elif event.key == pygame.K_d:
                keys[3] = False
        # add missile sprite
        if event.type == pygame.MOUSEBUTTONDOWN:
            pressed_array = pygame.mouse.get_pressed()
            for index in range(len(pressed_array)):
                if pressed_array[index]:
                    if index == 0:
                        missile_launched += 1
                        missile = MySprite()
                        missile.load("resources/images/bullet2.png", 32, 16, 1)
                        missile.position = player.X + \
                                           120, player.Y + 35
                        missile_group.add(missile)
        if event.type == pygame.QUIT:
            pygame.quit()
            exit()

    # move player sprite
    if keys[0]:
        if not player.Y < 0:
            player.Y -= player_move_speed
    elif keys[1]:
        if not player.X < 0:
            player.X -= player_move_speed
    if keys[2]:
        if not player.Y > 500:
            player.Y += player_move_speed
    elif keys[3]:
        if not player.X > 1024:
            player.X += player_move_speed

    # print player.position

    # auto pilot
    if not len(chopper_group) == 0:
        enemy = chopper_group.sprites()[0]
        posdiffy = enemy.Y - player.Y
        # print posdiffy
        if posdiffy < -10:
            for em in enemy_missile_group:
                if abs(em.X - player.X) < 100 and -100 < (em.Y - player.Y) < 0:
                    keys[0] = False
                    keys[2] = False
                    break
                else:
                    keys[0] = True
                    keys[2] = False
        elif posdiffy > 10:
            for em in enemy_missile_group:
                if abs(em.X - player.X) < 100 and 0 < (em.Y - player.Y) < 100:
                    keys[0] = False
                    keys[2] = False
                    break
                else:
                    keys[0] = False
                    keys[2] = True
        elif -10 <= posdiffy <= 10:
            keys[0] = False
            keys[2] = False
            missile_launched += 1
            if len(missile_group) < 2:
                if ticks - tickstart[0] > 200:
                    tickstart.pop(0)
                    tickstart.append(ticks)
                    missile = MySprite()
                    missile.load("resources/images/bullet2.png", 32, 16, 1)
                    missile.position = player.X + \
                                       120, player.Y + 35
                    missile_group.add(missile)

자원 다운로드 주소:
http://download.csdn.net/download/jailman/10160613

좋은 웹페이지 즐겨찾기