import pygame
import sys
import traceback
import os
import CXK
import enemy
import bullet
import supply
from pygame.locals import *
from random import *

#游戲主界面
def ui():
    #循環(huán)播放背景音樂
    pygame.mixer.music.play(-1)

    #初始化界面按鍵圖片并獲取圖片的矩形位置
    start_game_image = pygame.image.load("images/start_game.png").convert_alpha()
    start_game_image_rect = start_game_image.get_rect()
    game_rules_image = pygame.image.load("images/game_rules.png").convert_alpha()
    game_rules_image_rect = game_rules_image.get_rect()
    game_quit_image = pygame.image.load("images/game_quit.png").convert_alpha()
    game_quit_image_rect = game_quit_image.get_rect()

    #初始化游戲規(guī)則圖片并獲取圖片的矩形位置
    rules_image = pygame.image.load("images/游戲玩法.png").convert_alpha()
    back_image = pygame.image.load("images/back.png").convert_alpha()
    back_image_rect =  back_image.get_rect()

    #標志是否在主界面
    is_ui = True

    #幀率
    clock = pygame.time.Clock()

    #主界面循環(huán)
    while True:
        #獲取事件信息
        for event in pygame.event.get():
            #如果點擊右上角退出
            if event.type == QUIT:
                #退出程序
                pygame.quit()
                sys.exit()

        #如果是主界面
        if is_ui:
            #繪制背景
            screen.blit(background,(0,0))

            #更改主界面按鍵圖片的矩形位置并繪制主界面按鍵
            start_game_image_rect.left,start_game_image_rect.top = (width - start_game_image_rect.width)//2,height - 500
            screen.blit(start_game_image,start_game_image_rect)

            game_rules_image_rect = game_rules_image.get_rect()
            game_rules_image_rect.left,game_rules_image_rect.top = (width - game_rules_image_rect.width)//2,start_game_image_rect.bottom+50
            screen.blit(game_rules_image,game_rules_image_rect)

            game_quit_image_rect.left,game_quit_image_rect.top = (width - game_quit_image_rect.width)//2, game_rules_image_rect.bottom+50
            screen.blit(game_quit_image,game_quit_image_rect)

            #檢測用戶的鼠標操作
            #如果用戶按下鼠標左鍵
            if pygame.mouse.get_pressed()[0]:
                #獲取鼠標坐標
                pos = pygame.mouse.get_pos()
                #如果用戶點擊”開始游戲“
                if start_game_image_rect.left < pos[0] < start_game_image_rect.right and start_game_image_rect.top < pos[1] < start_game_image_rect.bottom:
                    #調(diào)用主函數(shù)
                    main()
                #如果用戶點擊”退出游戲“
                if game_quit_image_rect.left < pos[0] < game_quit_image_rect.right and game_quit_image_rect.top < pos[1] < game_quit_image_rect.bottom:
                    pygame.quit()
                    sys.exit()
                #如果用戶點擊”游戲規(guī)則“
                if game_rules_image_rect.left < pos[0] < game_rules_image_rect.right and game_rules_image_rect.top < pos[1] < game_rules_image_rect.bottom:
                    #離開主界面
                    is_ui = False