import matplotlib

matplotlib.use("webagg")

import matplotlib.pyplot as plt
from matplotlib.widgets import Button
import numpy as np


def test_scroll_capture():
    """Create a test plot for testing scroll functionality with a longer page."""
    # Create a taller figure to make scrolling necessary
    fig, (ax1, ax2, ax3) = plt.subplots(3, 1, figsize=(10, 12))

    # Create some sample data
    x = np.linspace(0, 10, 100)
    y1 = np.sin(x)
    y2 = np.cos(x)
    y3 = np.tan(x / 2)

    # First subplot
    ax1.plot(x, y1, label="sin(x)", linewidth=2, color="blue")
    ax1.set_xlabel("X")
    ax1.set_ylabel("sin(x)")
    ax1.set_title("Test Plot for Scroll Capture Functionality - Part 1")
    ax1.legend()
    ax1.grid(True, alpha=0.3)

    # Second subplot
    ax2.plot(x, y2, label="cos(x)", linewidth=2, color="red")
    ax2.set_xlabel("X")
    ax2.set_ylabel("cos(x)")
    ax2.set_title("Part 2 - This plot should be tall enough to require scrolling")
    ax2.legend()
    ax2.grid(True, alpha=0.3)

    # Third subplot
    ax3.plot(x, y3, label="tan(x/2)", linewidth=2, color="green")
    ax3.set_xlabel("X")
    ax3.set_ylabel("tan(x/2)")
    ax3.set_title("Part 3 - Try scrolling over the plot area!")
    ax3.legend()
    ax3.grid(True, alpha=0.3)
    ax3.set_ylim(-3, 3)  # Limit y range to avoid extreme tan values

    plt.tight_layout()

    fig.canvas.set_capture_scroll(True)

    # Add toggle button
    button_ax = plt.axes([0.4, 0.81, 0.2, 0.04])
    toggle_button = Button(
        button_ax, "Disable Capture", color="lightgreen", hovercolor="green"
    )

    def toggle_scroll_capture(event):
        current_state = fig.canvas.get_capture_scroll()
        new_state = not current_state
        fig.canvas.set_capture_scroll(new_state)

        if new_state:
            toggle_button.label.set_text("Disable Capture")
            toggle_button.color = "lightgreen"
            toggle_button.hovercolor = "green"
        else:
            toggle_button.label.set_text("Enable Capture")
            toggle_button.color = "lightcoral"
            toggle_button.hovercolor = "coral"
        fig.canvas.draw_idle()

    toggle_button.on_clicked(toggle_scroll_capture)

    plt.show()


if __name__ == "__main__":
    test_scroll_capture()