import math
import random
import pygame
from pygame.locals import *
 
running = False
playing = False
screen = None
timer = None
snk1 = None
snk2 = None
foods = None
remainFoods = 7
radiusFood = 8
 
'''
链表节点
'''
class Node:
    def __init__(self, data, prev = None, next = None):
        self.data = data
        self.prev = prev
        self.next = next
 
    def insert_front(self, node):
        if self.prev:
            node.prev = self.prev
            self.prev.next = node
            self.prev = node
            node.next = self
        else:
            self.prev = node
            node.next = self
        return node
 
    def insert_back(self, node):
        if self.next:
            node.next = self.next
            self.next.prev = node
            self.next = node
            node.prev = self
        else:
            self.next = node
            node.prev = self
        return node
    
    def remove(self):
        if self.next:
            self.next.prev = self.prev
        if self.prev:
            self.prev.next = self.next
 
'''
蛇
'''
class Snack:
 
    def __init__(self, surface, color, start_pos, end_pos, face):
        self.color = color
        self.surface = surface
        self.head = Node(start_pos)
        self.tail = Node(end_pos)
        self.head.insert_back(self.tail)
        self.length = self.distanceBetween(start_pos, end_pos)
        self.face = face
        self.speed = 120
        self.eat = 0
        self.grow = 0
        self.mapAngle = [
            [0, math.pi * 3 / 2, math.pi / 2],
            [0, math.pi * 7 / 4, math.pi / 4],
            [math.pi, math.pi * 5 / 4, math.pi * 3 / 4]
        ]
 
    '''坐标取整'''
    def intPos(self, pos):
        return (int(pos[0]), int(pos[1]))
 
    '''坐标转角度'''
    def pos2Angle(self, pos):
        return self.mapAngle[pos[0]][pos[1]]
 
    '''极坐标位移'''
    def polarPos(self, pos, angle, dis):
        xx = pos[0] + dis * math.cos(angle)
        yy = pos[1] + dis * math.sin(angle)
        return (xx, yy)
 
    '''计算两点间距离'''
    def distanceBetween(self, pos1, pos2):
        dx = pos2[0] - pos1[0]
        dy = pos2[1] - pos1[1]
        return math.sqrt(dx*dx + dy*dy)
    
    '''计算两点间角度'''
    def angleBetween(self, pos1, pos2):
        dx = pos2[0] - pos1[0]
        dy = pos2[1] - pos1[1]
        return math.atan2(dy, dx)
 
    '''改变面向'''
    def changeFace(self, newFace):
        if newFace[0] == 0 and newFace[1] == 0:
            return
        if newFace == self.face:
            return
        xx = self.face[0] + newFace[0]
        yy = self.face[1] + newFace[1]
        if xx == 0 and yy == 0:
            return
        self.face = newFace
        self.head = self.head.insert_front(Node(self.head.data))
 
    '''吃到食物'''
    def eatFood(self, grow):
        self.grow = grow
        self.eat += 1
 
    '''绘制蛇身'''
    def draw(self):
        node = self.head
        pygame.draw.circle(self.surface, self.color, self.intPos(node.data), 6, 6)
        while node:
            n2 = node.next 
            if not n2:
                break
            pygame.draw.line(self.surface, self.color, self.intPos(node.data), self.intPos(n2.data), 6)
            node = node.next
    
    '''每帧移动'''
    def walk(self, delta):
        dis = self.speed * delta / 1000
        self.head.data = self.polarPos(self.head.data, self.pos2Angle(self.face), dis)
        if self.grow >= dis:
            self.grow -= dis
        else:
            dis -= self.grow
            self.grow = 0
            self.cutTail(dis)
    
    '''收缩尾巴'''
    def cutTail(self, length):
        node = self.tail
        while length > 0:
            n2 = node.prev
            dis = self.distanceBetween(n2.data, node.data)
            angle = self.angleBetween(node.data, n2.data)
            if dis > length:
                node.data = self.polarPos(node.data, angle, length)
                length = 0
            else:
                self.tail = node.prev
                node.remove()
                length -= dis
 
            node = node.prev
 
'''屏幕指定位置绘制文字'''
def printText(surface, str, pos, size = 24, color = (255, 255, 255)):
    global screen
    font = pygame.font.SysFont("microsoftyaheimicrosoftyaheiui", size)
    text = font.render(str, True, color)
    w = text.get_width()
    h = text.get_height()
    surface.blit(text, (pos[0] - w / 2, pos[1] - h / 2))
 
'''添加食物'''
def addFood():
    global screen, snk1, snk2, foods, remainFoods
    if remainFoods = 0:
        return
    w = screen.get_width()
    h = screen.get_height()
    while True:
        posX = random.randint(5, w - 5)
        posY = random.randint(5, h - 5)
        color = tuple(screen.get_at((posX, posY)))
        if color != snk1.color and color != snk2.color:
            break
    remainFoods -= 1
    if not foods:
        foods = Node((posX, posY))
    else:
        foods = foods.insert_front(Node((posX, posY)))
 
'''删除食物'''
def removeFood(node):
    global foods
    if node == foods:
        foods = foods.next
    else:
        node.remove()
 
'''检测吃到食物'''
def checkEatFood():
    global foods, radiusFood, snk1, snk2
    node = foods
    while node:
        if snk1.distanceBetween(snk1.head.data, node.data)  (radiusFood + 4):
            snk1.eatFood(50)
            removeFood(node)
            addFood()
            break
        elif snk2.distanceBetween(snk2.head.data, node.data)  (radiusFood + 4):
            snk2.eatFood(50)
            removeFood(node)
            addFood()
            break
        else:
            node = node.next
 
'''游戏初始界面'''
def logo():
    global screen, remainFoods
    w = screen.get_width()
    h = screen.get_height()
    printText(screen, "Snack V1.0", (w / 2, h / 3), 48)
    printText(screen, "任意键继续", (w / 2, h / 2), 24, (55, 255, 55))
    printText(screen, str(remainFoods) + "个食物，抢完即止", (w / 2, h * 2 / 3), 32)
    
def quit():
    pygame.font.quit()
 
'''检测游戏结束'''
def checkGameOver():
    global remainFoods, snk1, snk2, foods, playing, screen
    if remainFoods == 0 and foods == None:
        playing = False
        screen.fill((0,0,0))
        w = screen.get_width()
        h = screen.get_height()
        if snk1.eat > snk2.eat:
            printText(screen, "玩家1 胜利", (w / 2, h / 2), 48)
        elif snk1.eat  snk2.eat:
            printText(screen, "玩家2 胜利", (w / 2, h / 2), 48)
        else:
            printText(screen, "平局", (w / 2, h / 2), 48)
 
'''键盘按键转换成面向角度'''
def cmd():
    global snk1, snk2
    keys = pygame.key.get_pressed()
    x1 = x2 = y1 = y2 = 0
    if keys[pygame.K_RIGHT]:
        x2+=1
    if keys[pygame.K_LEFT]:
        x2-=1
    if keys[pygame.K_UP]:
        y2+=1
    if keys[pygame.K_DOWN]:
        y2-=1
    if keys[pygame.K_d]:
        x1+=1
    if keys[pygame.K_a]:
        x1-=1
    if keys[pygame.K_w]:
        y1+=1
    if keys[pygame.K_s]:
        y1-=1
    snk1.changeFace((x1, y1))
    snk2.changeFace((x2, y2))
 
'''游戏每帧更新'''
def play(delta):
    global playing, snk1, snk2
    if not playing:
        return
    cmd()
    snk1.walk(delta)
    snk2.walk(delta)
    checkEatFood()
    checkGameOver()
 
'''绘制'''
def draw():
    global snk1, snk2, playing, screen, radiusFood, remainFoods
    if not playing:
        return
    screen.fill((0,0,0))
    snk1.draw()
    snk2.draw()
    node = foods
    while node:
        color = (255, 255, 255)
        if remainFoods == 0:
            color = (255, 0, 0)
        pygame.draw.circle(screen, color, node.data, radiusFood, radiusFood // 2 + 1)
        node = node.next
 
def start(width = 800, height = 600, fps = 60):
    global running, screen, timer, playing, snk1, snk2
    pygame.init()
    pygame.font.init()
    font = pygame.font.SysFont("microsoftyaheimicrosoftyaheiui", 20)
    pygame.display.set_caption("Snack V1.0")
    screen = pygame.display.set_mode((width, height))
    
    logo()
    snk1 = Snack(screen, (0, 150, 200), (100, 100), (0, 100), (1, 0))
    snk2 = Snack(screen, (255, 100, 0), (width * 5 // 6, height // 2), (width * 5 // 6 + 100, height // 2), (-1, 0))
    for i in range(3):
        addFood()
 
    timer = pygame.time.Clock()
    running = True
    while running:
        delta = timer.tick(fps)
        play(delta)
        draw()
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False
            elif event.type == pygame.KEYDOWN:
                if event.key == pygame.K_SPACE and playing == False:
                    screen.fill((0,0,0))
                    playing = True
 
        pygame.display.flip()
 
    
if __name__ == "__main__":
    start()
    quit()