연구실(bwlee42)/251109

import tkinter as tk
from tkinter import messagebox
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

def start_animation():
    f_expr = f_entry.get()
    g_expr = g_entry.get()

    if not f_expr or not g_expr:
        messagebox.showerror("Error", "f(x)와 g(x)를 모두 입력하세요!")
        return

    try:
        f = lambda x: eval(f_expr, {"x": x, "np": np, "__builtins__": {}})
        g = lambda x: eval(g_expr, {"x": x, "np": np, "__builtins__": {}})
    except Exception as e:
        messagebox.showerror("Error", f"함수 입력 오류: {e}")
        return

    # x값 범위
    x_vals = np.linspace(-3, 3, 200)

    # f, g(f(x)) 전체 계산
    f_all, gf_all = [], []
    for x in x_vals:
        y_f = f(x)
        if not isinstance(y_f, (list, tuple, np.ndarray)):
            y_f = [y_f]
        f_all.extend(y_f)
        gf_all.extend([g(y) for y in y_f])
    f_all = np.array(f_all)
    gf_all = np.array(gf_all)

    # ---------------------------
    # 애니메이션 패널
    # ---------------------------
    fig, axes = plt.subplots(1,3,figsize=(15,4))

    # f 패널
    axes[0].plot(x_vals, [np.mean(f(x)) if isinstance(f(x),(list,tuple,np.ndarray)) else f(x) for x in x_vals], color='lightblue')
    axes[0].set_title('f(x)')
    axes[0].set_xlabel('x')
    axes[0].set_ylabel('f(x)')
    dot_f, = axes[0].plot([],[], 'ro')
    trail_f, = axes[0].plot([],[], 'b', lw=2)
    axes[0].set_xlim(min(x_vals), max(x_vals))
    axes[0].set_ylim(min(f_all)-0.5, max(f_all)+0.5)

    # g 패널
    g_all_sorted = np.sort(f_all)
    g_vals = [g(y) for y in g_all_sorted]
    axes[1].plot(g_all_sorted, g_vals, color='lightgreen')
    axes[1].set_title('g(y)')
    axes[1].set_xlabel('f(x)')
    axes[1].set_ylabel('g(f(x))')
    dot_g, = axes[1].plot([],[], 'ro')
    trail_g_scatter = axes[1].scatter([],[], c=[], cmap='Greens', s=50)
    axes[1].set_xlim(min(f_all)-0.5, max(f_all)+0.5)
    axes[1].set_ylim(min(gf_all)-0.5, max(gf_all)+0.5)

    # h 패널
    axes[2].set_title('h(x) = g(f(x))')
    axes[2].set_xlabel('x')
    axes[2].set_ylabel('h(x)')
    dot_h, = axes[2].plot([],[], 'ro')
    trail_h, = axes[2].plot([],[], 'r', lw=2)
    axes[2].set_xlim(min(x_vals), max(x_vals))
    axes[2].set_ylim(min(gf_all)-0.5, max(gf_all)+0.5)

    # ---------------------------
    # 히스토리
    # ---------------------------
    x_hist_f, y_hist_f = [], []
    x_hist_h, y_hist_h = [], []
    trail_g_count = {}

    # ---------------------------
    # 애니메이션 업데이트
    # ---------------------------
    def update(frame):
        x = x_vals[frame]
        y_f_raw = f(x)
        if not isinstance(y_f_raw,(list,tuple,np.ndarray)):
            y_f_vals = [y_f_raw]
        else:
            y_f_vals = y_f_raw
        y_h_vals = [g(y) for y in y_f_vals]

        # f
        dot_f.set_data([x]*len(y_f_vals), y_f_vals)
        x_hist_f.extend([x]*len(y_f_vals))
        y_hist_f.extend(y_f_vals)
        trail_f.set_data(x_hist_f, y_hist_f)

        # g
        dot_g.set_data(y_f_vals, y_h_vals)
        for xf,yf in zip(y_f_vals,y_h_vals):
            key = (xf,yf)
            trail_g_count[key] = trail_g_count.get(key,0)+1

        xs,ys,cs=[],[],[]
        for (xf,yf),count in trail_g_count.items():
            xs.append(xf)
            ys.append(yf)
            cs.append(min(count,10))
        if xs:
            trail_g_scatter.set_offsets(np.c_[xs,ys])
            trail_g_scatter.set_array(np.array(cs))

        # h
        dot_h.set_data([x]*len(y_h_vals), y_h_vals)
        x_hist_h.extend([x]*len(y_h_vals))
        y_hist_h.extend(y_h_vals)
        trail_h.set_data(x_hist_h, y_hist_h)

        return dot_f, trail_f, dot_g, trail_g_scatter, dot_h, trail_h

    ani = FuncAnimation(fig, update, frames=len(x_vals), interval=50, blit=False, repeat=False)
    plt.show()


# ---------------------------
# Tkinter GUI
# ---------------------------
root = tk.Tk()
root.title("합성함수 애니메이션")

tk.Label(root, text="f(x) =").grid(row=0,column=0)
f_entry = tk.Entry(root, width=30)
f_entry.grid(row=0,column=1)
f_entry.insert(0,"x**2")

tk.Label(root, text="g(x) =").grid(row=1,column=0)
g_entry = tk.Entry(root, width=30)
g_entry.grid(row=1,column=1)
g_entry.insert(0,"x+1")

start_btn = tk.Button(root, text="Start Animation", command=start_animation)
start_btn.grid(row=2,column=0,columnspan=2,pady=10)

root.mainloop()