import numpy as np
import matplotlib.pyplot as plt
import random
import seaborn as sns

def plot_spider_web_and_matrix():
    # ==========================================
    # 1. 核心參數 (延用你的設定)
    # ==========================================
    NUM_RINGS = 4
    NUM_SPOKES = 10
    JITTER = 5.0
    HIGH_RISK = 10.0  # F 矩陣中的高風險懲罰值
    
    # 生成節點
    nodes = [(50, 50)]
    radii = np.linspace(12, 45, NUM_RINGS)
    angles = np.linspace(0, 2*np.pi, NUM_SPOKES, endpoint=False)
    for r in radii:
        for t in angles:
            x = 50 + r * np.cos(t) + np.random.uniform(-JITTER, JITTER)
            y = 50 + r * np.sin(t) + np.random.uniform(-JITTER, JITTER)
            nodes.append((x, y))
    
    N = len(nodes)
    
    # 建立連線清單 (Edges)
    edges = []
    for i in range(NUM_SPOKES): edges.append((0, 1 + i)) # 中心連線
    for ring in range(NUM_RINGS - 1): # 輻射線
        for spoke in range(NUM_SPOKES):
            edges.append((1 + ring * NUM_SPOKES + spoke, 1 + (ring + 1) * NUM_SPOKES + spoke))
    for ring in range(NUM_RINGS): # 同心圓
        for spoke in range(NUM_SPOKES):
            u = 1 + ring * NUM_SPOKES + spoke
            v = 1 + ring * NUM_SPOKES + ((spoke + 1) % NUM_SPOKES)
            edges.append((u, v))

    # ==========================================
    # 2. 核心邏輯：建構擾動矩陣 F
    # ==========================================
    # 初始化：假設所有路徑都是極度危險的 (F=10)
    F = np.full((N, N), HIGH_RISK)
    np.fill_diagonal(F, 0) # 自己到自己沒擾動

    # 只有在「蜘蛛絲」上的路徑，擾動才設為 0
    for u, v in edges:
        F[u, v] = 0.0
        F[v, u] = 0.0

    # ==========================================
    # 3. 繪圖與對比
    # ==========================================
    fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(18, 8))

    # --- 左圖：實體蜘蛛網佈局 ---
    ax1.set_facecolor('#2A2A2A')
    for u, v in edges:
        ax1.plot([nodes[u][0], nodes[v][0]], [nodes[u][1], nodes[v][1]], 
                 color='#E0E0E0', linewidth=1.5, alpha=0.6)
    nx, ny = zip(*nodes)
    ax1.scatter(nx, ny, color='#00FFFF', s=50, zorder=5)
    ax1.scatter(nodes[0][0], nodes[0][1], color='yellow', s=150, marker='*')
    ax1.set_title("1. Physical World (Spider Web Topology)", fontsize=14)
    ax1.axis('off')

    # --- 右圖：擾動矩陣 F 的熱圖 ---
    # 我們畫出 F 矩陣，顏色越深(0)代表越安全，顏色越淺(10)代表越危險
    sns.heatmap(F, cmap="YlGnBu", ax=ax2, cbar_kws={'label': 'Disturbance Value (F)'})
    ax2.set_title(f"2. Algorithm World (Disturbance Matrix F)\nMatrix Size: {N}x{N}", fontsize=14)
    ax2.set_xlabel("City Index")
    ax2.set_ylabel("City Index")

    plt.tight_layout()
    plt.show()

    return F

if __name__ == "__main__":
    F_matrix = plot_spider_web_and_matrix()