Source code for spectral.utils.plotting

"""Plotting utilities for spectral method visualizations."""

from __future__ import annotations

from pathlib import Path




def get_repo_root() -> Path:
    """Get repository root directory, handling both local and sphinx-gallery execution.

    Returns the repository root by detecting the presence of pyproject.toml.
    Works in both local execution (via main.py) and sphinx-gallery contexts.

    Returns
    -------
    Path
        Absolute path to the repository root

    """
    try:
        # Try to get caller's __file__ if available (local execution)
        import inspect

        frame = inspect.currentframe().f_back
        caller_file = frame.f_globals.get("__file__")
        if caller_file:
            current = Path(caller_file).resolve().parent
        else:
            # __file__ not available (sphinx-gallery)
            current = Path.cwd()
    except (AttributeError, KeyError):
        # Fallback to cwd
        current = Path.cwd()

    # Walk up until we find pyproject.toml (marks repo root)
    for parent in [current] + list(current.parents):
        if (parent / "pyproject.toml").exists():
            return parent

    # Fallback: assume 2 levels up from script directory
    # Works for Exercises/exercise_X/script.py structure
    return current.parent.parent if caller_file else current.parent.parent