import random
import time
import pygame
# 30幀
fps = 30
fps_clock = pygame.time.Clock()
screen_width = 1024
screen_height = 768
# 分辨率，標(biāo)題
display = pygame.display.set_mode((screen_width, screen_height), 0, 32)
pygame.display.set_caption('好似滿天星')
tile_size = 60
tile_width = 60
tile_height = 60
x_margin = 400
y_margin = 100
# 列
columns = 10
# 行
rows = 11
# 配色RGB
white = (255, 255, 255)
black = 'black'
bg_color = 'sky blue'
border_color = white
body_color = 'purple'
inter_body_color = 'green'
line_color = white
directions = ['up', 'down', 'left', 'right']
text_color = (144, 59, 28)
my_score = 0
snake = []
# 畫(huà)格子
def draw_chest():
    for i in range(rows + 1):
        pygame.draw.line(display, border_color, (x_margin/2, y_margin/2 + i * tile_size),
                         (x_margin/2 + (columns * tile_size), y_margin/2 + i * tile_size), 2)
    for j in range(columns + 1):
        pygame.draw.line(display, border_color, (x_margin/2 + j * tile_size, y_margin/2),
                         (x_margin/2 + j * tile_size, y_margin/2 + (rows * tile_size)), 2)
if __name__ == '__main__':
    pygame.init()
    bad_touch = pygame.mixer.Sound('badswap.wav')
    match_three = pygame.mixer.Sound('match3.wav')
    match_more_than_three = pygame.mixer.Sound('match4.wav')
    display.fill(bg_color)
    while True:
        draw_chest()
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
        pygame.display.update()
        fps_clock.tick(fps)

def make_stars():
    board = []
    for i in range(rows):
        temp = []
        for j in range(columns):
            star = random.choice(stars)
            temp.append(star)
        board.append(temp)
    return board

def get_left_top_location(x, y):
    left_position = y * tile_size + x_margin/2
    top_position = x * tile_size + y_margin/2
    return left_position, top_position
def draw_stars(board):
    for i in range(len(board)):
        for j in range(len(board[i])):
            path = 'gems/' + board[i][j] + '.png'
            star = pygame.image.load(path)
            star = pygame.transform.scale(star, (tile_size-5, tile_size-5))
            left, top = get_left_top_location(i, j)
            # 微調(diào)一下，這樣圖片比較居中，誰(shuí)讓釘在左上角呢
            display.blit(star, (left+3, top+3))

if __name__ == '__main__':
    pygame.init()
    bad_touch = pygame.mixer.Sound('badswap.wav')
    match_three = pygame.mixer.Sound('match3.wav')
    match_more_than_three = pygame.mixer.Sound('match4.wav')
    display.fill(bg_color)
    main_board = make_stars()
    print(main_board)
    while True:
        draw_chest()
        draw_stars(main_board)
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
        pygame.display.update()
        fps_clock.tick(fps)

# 將游戲中的界面坐標(biāo)轉(zhuǎn)換為main_board中的位置
def get_spot_clicked(board, x, y):
    for i in range(len(board)):
        for j in range(len(board[i])):
            # 注意在二維數(shù)組里我們的查詢順序是比如maze[3][5]代表第四行第六列，但是對(duì)應(yīng)游戲界面的坐標(biāo)則3用來(lái)計(jì)算y坐標(biāo)，5用來(lái)計(jì)算x坐標(biāo)
            left, top = get_left_top_location(i, j)
            tile_rect = pygame.Rect(left, top, tile_width, tile_height)
            if tile_rect.collidepoint(x, y):
                return i, j
    return None, None

            if event.type == pygame.QUIT:
                pygame.quit()
            elif event.type == pygame.MOUSEBUTTONUP:
                # 鼠標(biāo)左鍵點(diǎn)擊后抬起
                if event.button == 1:
                    pos_x, pos_y = event.pos
                    print(get_spot_clicked(main_board, pos_x, pos_y))

# 查找"星星"的相同情況
def check_around(board, star_pos):
    store_stars = set()
    has_three_stars = False
    shape_type = board[star_pos[0]][star_pos[1]]
    # 查找上下兩個(gè)方向是否有與當(dāng)前棋子類型一樣的棋子
    store_stars.add(star_pos)
    # 向上的方向上尋找
    star_i = star_pos[0] - 1
    up_and_down_star = set()
    while star_i >= 0 and board[star_i][star_pos[1]] == shape_type:
        store_stars.add((star_i, star_pos[1]))
        star_i -= 1
    # 向下的方向?qū)ふ?
    star_i = star_pos[0] + 1
    while star_i < rows and board[star_i][star_pos[1]] == shape_type:
        store_stars.add((star_i, star_pos[1]))
        star_i += 1
    # 向左的方向上尋找
    star_j = star_pos[1] - 1
    while star_j >= 0 and board[star_pos[0]][star_j] == shape_type:
        store_stars.add((star_pos[0], star_j))
        star_j -= 1
    # 向右的方向?qū)ふ?
    star_j = star_pos[1] + 1
    while star_j < columns and board[star_pos[0]][star_j] == shape_type:
        store_stars.add((star_pos[0], star_j))
        star_j += 1
    if len(store_stars) >= 3:
        has_three_stars = True
    return store_stars, has_three_stars

def remove_stars(board, removed_stars):
    for star in removed_stars:
        board[star[0]][star[1]] = 'empty'

                if event.button == 1:
                    pos_x, pos_y = event.pos
                    x, y = get_spot_clicked(main_board, pos_x, pos_y)
                    need_to_removed, remove_or_not = check_around(main_board, (x, y))
                    if remove_or_not:
                        remove_stars(main_board, need_to_removed)
                        display.fill(bg_color)
                        draw_chest()
                        draw_stars(main_board)

def draw_stars(board):
    for i in range(len(board)):
        for j in range(len(board[i])):
            if board[i][j] != 'empty':
                path = 'gems/' + board[i][j] + '.png'
                star = pygame.image.load(path)
                star = pygame.transform.scale(star, (tile_size-5, tile_size-5))
                left, top = get_left_top_location(i, j)
                display.blit(star, (left+3, top+3))

# 查找"星星"的相同情況
def check_around(board, star_pos):
    stars_temp = []
    store_stars = set()
    has_three_stars = False
    shape_type = board[star_pos[0]][star_pos[1]]
    # 查找上下兩個(gè)方向是否有與當(dāng)前棋子類型一樣的棋子
    store_stars.add(star_pos)
    stars_temp.append(star_pos)
    while len(stars_temp):
        current_star = stars_temp[0]
        del stars_temp[0]
        # 向上的方向上尋找
        star_i = current_star[0] - 1

        while star_i >= 0 and board[star_i][current_star[1]] == shape_type:
            if (star_i, current_star[1]) not in store_stars:
                store_stars.add((star_i, current_star[1]))
                stars_temp.append((star_i, current_star[1]))
            star_i -= 1
        # 向下的方向?qū)ふ?
        star_i = current_star[0] + 1
        while star_i < rows and board[star_i][current_star[1]] == shape_type:
            if (star_i, current_star[1]) not in store_stars:
                store_stars.add((star_i, current_star[1]))
                stars_temp.append((star_i, current_star[1]))
            star_i += 1
        # 向左的方向上尋找
        star_j = current_star[1] - 1
        while star_j >= 0 and board[current_star[0]][star_j] == shape_type:
            if (current_star[0], star_j) not in store_stars:
                store_stars.add((current_star[0], star_j))
                stars_temp.append((current_star[0], star_j))
            star_j -= 1
        # 向右的方向?qū)ふ?
        star_j = current_star[1] + 1
        while star_j < columns and board[current_star[0]][star_j] == shape_type:
            if (current_star[0], star_j) not in store_stars:
                store_stars.add((current_star[0], star_j))
                stars_temp.append((current_star[0], star_j))
            star_j += 1
        if len(store_stars) >= 2:
            has_three_stars = True
    #print('stars: ', store_stars)
    #input()
    return store_stars, has_three_stars