import math
import random
import tkinter as tk
import matplotlib.pyplot as plt
import csv
from tkinter import Label

# 创建窗口
root = tk.Tk()
root.attributes('-fullscreen', True)  # 全屏显示

label_n = Label(root, text="n: 0", font=("Helvetica", 16))
label_n.pack()

csv_file_path = "mouse_data.csv"

screen_width = root.winfo_screenwidth()
screen_height = root.winfo_screenheight()

# 设置小球的初始位置
ball1_pos = (screen_width/2, screen_height/2)
ball2_pos = (ball1_pos[0] + random.randint(-200, 200),ball1_pos[1] + random.randint(-200, 200))

# 设置小球的半径
ball_radius = 20

# 设置鼠标记录状态
recording = False
mouse_path = []

n=0

# 鼠标移动事件处理函数
def motion(event):
    global recording, mouse_path, n
    if recording:
        mouse_path.append((event.x, event.y))

# 鼠标点击事件处理函数
def mouse_click(event):
    global recording, mouse_path, ball2_pos, n

    if event.x >= ball1_pos[0] - ball_radius and event.x <= ball1_pos[0] + ball_radius and event.y >= ball1_pos[1] - ball_radius and event.y <= ball1_pos[1] + ball_radius:
        recording = True
        mouse_path = [(event.x, event.y)]
    elif event.x >= ball2_pos[0] - ball_radius and event.x <= ball2_pos[0] + ball_radius and event.y >= ball2_pos[1] - ball_radius and event.y <= ball2_pos[1] + ball_radius:
        recording = False
        canvas.delete("ball2")
        #visualize_path(mouse_path)  # 可视化鼠标轨迹
        save_to_csv(mouse_path)
        n = n+1
        if n == 100:
            root.destroy()
        label_n.config(text=f"n: {n}")
        mouse_path = []

        # 重新生成第二个小球的位置
        ball2_pos = (ball1_pos[0] + random.randint(-200, 200), ball1_pos[1] + random.randint(-200, 200))
        
        # 绘制新的第二个小球
        canvas.create_oval(ball2_pos[0]-ball_radius, ball2_pos[1]-ball_radius, ball2_pos[0]+ball_radius, ball2_pos[1]+ball_radius, fill="blue", tags="ball2")


# 键盘事件处理函数
def key(event):
    if event.keysym == "Escape":
        root.destroy()

def save_to_csv(path):
    # 将路径坐标转换为相对于起点的坐标
    x_rel = [px - path[0][0] for px, py in path]
    y_rel = [-(py - path[0][1]) for px, py in path]

    # 计算每个点相对于起点的距离，用于z轴表示
    distances = [math.sqrt((x_rel[0] - px)**2 + (y_rel[0] - py)**2) for px, py in zip(x_rel, y_rel)]

    # 选择10个关键点
    key_points_indices = [int(i) for i in range(0, len(path), max(1, len(path)//10))]
    key_points_x = [x_rel[i] for i in key_points_indices]
    key_points_y = [y_rel[i] for i in key_points_indices]
    key_points_distances = [distances[i] for i in key_points_indices]

    # 打开 CSV 文件进行写操作
    with open(csv_file_path, mode='a', newline='') as csv_file:
        csv_writer = csv.writer(csv_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
        # 写入一行数据
        csv_writer.writerow([f"{key_points_x[-1]},{key_points_y[-1]}"] + [f"{key_points_x[i]},{key_points_y[i]}" for i in range(0,10)])


def visualize_path(path):
    # 将路径坐标转换为相对于起点的坐标
    x_rel = [px - path[0][0] for px, py in path]
    y_rel = [-(py - path[0][1]) for px, py in path]

    # 计算每个点相对于起点的距离，用于z轴表示
    distances = [math.sqrt((x_rel[0] - px)**2 + (y_rel[0] - py)**2) for px, py in zip(x_rel, y_rel)]

    # 选择10个关键点
    key_points_indices = [int(i) for i in range(0, len(path), max(1, len(path)//10))]
    key_points_x = [x_rel[i] for i in key_points_indices]
    key_points_y = [y_rel[i] for i in key_points_indices]
    key_points_distances = [distances[i] for i in key_points_indices]

    # 使用z轴信息，通过颜色表示距离的远近
    plt.scatter(key_points_x, key_points_y, c=key_points_distances, cmap='viridis', marker='o', s=50)

    # 在关键点位置添加文本标签，显示终点到起点的距离
    
    plt.text(key_points_x[-1], key_points_y[-1], f'Distance to Origin: {key_points_distances[-1]:.2f}', ha='right', va='bottom', bbox=dict(facecolor='white', alpha=0.5))

    # 添加颜色条，表示z轴信息
    plt.colorbar(label='Distance to Endpoint')

    plt.show()

# 绘制小球
canvas = tk.Canvas(root, width=root.winfo_screenwidth(), height=root.winfo_screenheight())
canvas.pack()
canvas.create_oval(ball1_pos[0]-ball_radius, ball1_pos[1]-ball_radius, ball1_pos[0]+ball_radius, ball1_pos[1]+ball_radius, fill="red")
canvas.create_oval(ball2_pos[0]-ball_radius, ball2_pos[1]-ball_radius, ball2_pos[0]+ball_radius, ball2_pos[1]+ball_radius, fill="blue", tags="ball2")

# 绑定鼠标事件
canvas.bind('<Motion>', motion)
canvas.bind('<Button-1>', mouse_click)

# 绑定键盘事件
root.bind('<Key>', key)

# 运行窗口
root.mainloop()