import pygame
from pygame import locals
import os


pygame.init()  # 初始化
# 创建窗口
screen = pygame.display.set_mode((640, 480))
FPS = pygame.time.Clock()  # pygame时钟，控制游戏速度（帧数）
# 载入图片、资源
bg_img = pygame.image.load('background.png')  # 背景图
play_img = pygame.image.load('play.png')  # 播放按钮
stop_img = pygame.image.load('stop.png')  # 暂停按钮
last_img = pygame.image.load('last.png')  # 上一曲按钮
next_img = pygame.image.load('next.png')  # 下一曲按钮
logo_img = pygame.image.load('logo.png')  # 下一曲按钮

path = "C:\\Users\\bellcode\\Desktop\\test"
filelist = os.listdir(path)
music_list = []
for i in filelist:
    if i[-4:] == ".ogg" or i[-4:] == ".wav":
        music_list.append(i)
num = -1
angle = 0

volume = 0.2
pygame.mixer.music.set_volume(volume)  # 初始播放音量
click = 0
play_button = stop_img

while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            exit()
        # 按键，控制声音大小
        if event.type == locals.KEYDOWN:
            if event.key == locals.K_UP:
                volume += 0.1
                if volume > 1:
                    volume = 1
                pygame.mixer.music.set_volume(volume)
            if event.key == locals.K_DOWN:
                volume -= 0.1
                if volume < 0:
                    volume = 0
                pygame.mixer.music.set_volume(volume)

        if event.type == locals.MOUSEBUTTONDOWN:
            if event.button == 1:  # 左击
                x, y = event.pos
                # print(event.pos)
                if x > 270 and x < 370 and y > 350 and y < 450: # 播放按钮
                    click += 1
                    if click % 2 == 0:
                        play_button = stop_img
                        pygame.mixer.music.unpause()
                    else:
                        play_button = play_img
                        pygame.mixer.music.pause()
                if x > 420 and x < 520 and y > 350 and y < 450:
                    pygame.mixer.music.stop()
                    play_button = stop_img
                    if click % 2 == 0:
                        click += 2
                    else:
                        click += 1

                if x > 120 and x < 220 and y > 350 and y < 450:
                    num -= 1
                    if num < 0:
                        num = len(music_list) - 1
                    pygame.mixer.music.load(path + music_list[num])
                    pygame.mixer.music.play()
                    play_button = stop_img
                    if click % 2 == 0:
                        click += 2
                    else:
                        click += 1

    if pygame.mixer.music.get_busy() == False:
        num += 1
        if num > len(music_list)-1:
            num = 0
        pygame.mixer.music.load(path+music_list[num])
        pygame.mixer.music.play()
    # 旋转图片(注意：这里要搞一个新变量，存储旋转后的图片）
    new_logo = pygame.transform.rotate(logo_img, angle)
    # 校正旋转图片的中心点
    newRect = new_logo.get_rect(center=(320, 200))
    # print(newRect)      # rect(170, 65, 300, 270) 左上角坐标、图片大小
    pos = (newRect[0], newRect[1])
    if play_button == stop_img:
        angle += 1
    # 绘制画面
    screen.blit(bg_img, (0, 0))      # 填充背景
    screen.blit(play_button, (270, 330))   # 暂停按钮
    # screen.blit(new_logo, (170, 60))    # 中间logo图
    screen.blit(new_logo, pos)  # 中间logo图
    screen.blit(last_img, (120, 350))   # 上一曲
    screen.blit(next_img, (420, 350))   # 下一曲

    # 刷新画面
    pygame.display.update()
    FPS.tick(60)