import pygame
import random
import sys

# 窗口设置
WIDTH = 640
HEIGHT = 480
FPS = 60

# 颜色常量
WHITE = (255, 255, 255)
BLACK = (20, 20, 20)
GRAY = (100, 100, 100)
RED_SLIDER = (220, 40, 40)
GREEN_SLIDER = (40, 180, 40)
BLUE_SLIDER = (40, 80, 220)

class ColorPractice:
    def __init__(self):
        pygame.init()
        self.screen = pygame.display.set_mode((WIDTH, HEIGHT))
        pygame.display.set_caption("配色练习器")
        self.clock = pygame.time.Clock()
        self.font = pygame.font.SysFont("SimHei", 18)
        self.big_font = pygame.font.SysFont("SimHei", 24)

        # 目标颜色（随机）
        self.target_color = self.random_color()
        # 用户调节的RGB
        self.r = 128
        self.g = 128
        self.b = 128

        # 滑块位置
        self.slider_y_start = 280
        self.slider_length = 400
        self.slider_height = 12
        self.dragging = None  # r/g/b

    def random_color(self):
        return (random.randint(0, 255),
                random.randint(0, 255),
                random.randint(0, 255))

    def color_distance(self, c1, c2):
        """计算颜色欧式距离"""
        r1, g1, b1 = c1
        r2, g2, b2 = c2
        return ((r1-r2)**2 + (g1-g2)**2 + (b1-b2)**2)**0.5

    def get_score(self):
        dist = self.color_distance((self.r,self.g,self.b), self.target_color)
        max_dist = 441.7  # sqrt(255²*3)
        score = max(0, 100 - (dist / max_dist) * 100)
        return round(score, 1)

    def draw_slider(self, y, color, value):
        x_start = 20
        x_end = x_start + self.slider_length
        # 滑轨
        pygame.draw.rect(self.screen, GRAY, (x_start, y, self.slider_length, self.slider_height))
        # 滑块位置
        pos_x = x_start + (value / 255) * self.slider_length
        pygame.draw.circle(self.screen, color, (int(pos_x), y+self.slider_height//2), 8)
        # 数值文字
        txt = self.font.render(f"{value}", True, WHITE)
        self.screen.blit(txt, (x_end + 15, y - 3))
        return x_start, x_end, y

    def pos_to_value(self, mx, x_start, x_end):
        ratio = (mx - x_start) / (x_end - x_start)
        return max(0, min(255, int(ratio * 255)))

    def new_target(self):
        self.target_color = self.random_color()
        self.r = 128
        self.g = 128
        self.b = 128

    def run(self):
        while True:
            self.screen.fill(BLACK)

            # 色块展示区域
            # 目标颜色
            pygame.draw.rect(self.screen, self.target_color, (40, 30, 240, 160))
            t_text = self.big_font.render("目标颜色", True, WHITE)
            self.screen.blit(t_text, (85, 195))

            # 你调的颜色
            user_color = (self.r, self.g, self.b)
            pygame.draw.rect(self.screen, user_color, (360, 30, 240, 160))
            u_text = self.big_font.render("你的配色", True, WHITE)
            self.screen.blit(u_text, (405, 195))

            # 分数
            score_text = self.big_font.render(f"匹配分数：{self.get_score()} 分", True, WHITE)
            self.screen.blit(score_text, (220, 210))

            # 绘制三个RGB滑块
            r_info = self.draw_slider(self.slider_y_start, RED_SLIDER, self.r)
            g_info = self.draw_slider(self.slider_y_start + 40, GREEN_SLIDER, self.g)
            b_info = self.draw_slider(self.slider_y_start + 80, BLUE_SLIDER, self.b)

            r_label = self.font.render("R", True, RED_SLIDER)
            g_label = self.font.render("G", True, GREEN_SLIDER)
            b_label = self.font.render("B", True, BLUE_SLIDER)
            self.screen.blit(r_label, (5, self.slider_y_start))
            self.screen.blit(g_label, (5, self.slider_y_start+40))
            self.screen.blit(b_label, (5, self.slider_y_start+80))

            # 按钮：下一题
            btn_rect = pygame.Rect(240, 410, 160, 40)
            pygame.draw.rect(self.screen, GRAY, btn_rect)
            btn_text = self.big_font.render("下一组颜色", True, WHITE)
            self.screen.blit(btn_text, (265, 415))

            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    pygame.quit()
                    sys.exit()

                mx, my = pygame.mouse.get_pos()
                if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
                    # 判断点击按钮
                    if btn_rect.collidepoint(mx, my):
                        self.new_target()
                    # 判断拖动滑块
                    sx1, ex1, sy1 = r_info
                    if sx1 <= mx <= ex1 and sy1 <= my <= sy1+self.slider_height:
                        self.dragging = "r"
                    sx2, ex2, sy2 = g_info
                    if sx2 <= mx <= ex2 and sy2 <= my <= sy2+self.slider_height:
                        self.dragging = "g"
                    sx3, ex3, sy3 = b_info
                    if sx3 <= mx <= ex3 and sy3 <= my <= sy3+self.slider_height:
                        self.dragging = "b"

                if event.type == pygame.MOUSEBUTTONUP:
                    self.dragging = None

                if event.type == pygame.MOUSEMOTION and self.dragging is not None:
                    if self.dragging == "r":
                        self.r = self.pos_to_value(mx, *r_info[:2])
                    elif self.dragging == "g":
                        self.g = self.pos_to_value(mx, *g_info[:2])
                    elif self.dragging == "b":
                        self.b = self.pos_to_value(mx, *b_info[:2])

            pygame.display.flip()
            self.clock.tick(FPS)

if __name__ == "__main__":
    app = ColorPractice()
    app.run()