Atomizer/atomizer_paths.py

"""
Atomizer Path Configuration

Provides intelligent path resolution for Atomizer core modules and directories.
This module can be imported from anywhere in the project hierarchy.
"""

from pathlib import Path
import sys


def get_atomizer_root() -> Path:
    """
    Get the Atomizer project root directory.

    This function intelligently locates the root by looking for marker files
    that uniquely identify the Atomizer project root.

    Returns:
        Path: Absolute path to Atomizer root directory

    Raises:
        RuntimeError: If Atomizer root cannot be found
    """
    # Start from this file's location
    current = Path(__file__).resolve().parent

    # Marker files that uniquely identify Atomizer root
    markers = [
        'optimization_engine',  # Core module directory
        'studies',              # Studies directory
        'README.md'             # Project README
    ]

    # Walk up the directory tree looking for all markers
    max_depth = 10  # Prevent infinite loop
    for _ in range(max_depth):
        # Check if all markers exist at this level
        if all((current / marker).exists() for marker in markers):
            return current

        # Move up one directory
        parent = current.parent
        if parent == current:  # Reached filesystem root
            break
        current = parent

    raise RuntimeError(
        "Could not locate Atomizer root directory. "
        "Make sure you're running from within the Atomizer project."
    )


def setup_python_path():
    """
    Add Atomizer root to Python path if not already present.

    This allows imports like `from optimization_engine.core.runner import ...`
    to work from anywhere in the project.
    """
    root = get_atomizer_root()
    root_str = str(root)

    if root_str not in sys.path:
        sys.path.insert(0, root_str)


# Core directories (lazy-loaded)
_ROOT = None

def root() -> Path:
    """Get Atomizer root directory."""
    global _ROOT
    if _ROOT is None:
        _ROOT = get_atomizer_root()
    return _ROOT


def optimization_engine() -> Path:
    """Get optimization_engine directory."""
    return root() / 'optimization_engine'


def studies() -> Path:
    """Get studies directory."""
    return root() / 'studies'


def tests() -> Path:
    """Get tests directory."""
    return root() / 'tests'


def docs() -> Path:
    """Get docs directory."""
    return root() / 'docs'


def plugins() -> Path:
    """Get plugins directory."""
    return optimization_engine() / 'plugins'


# Common files
def readme() -> Path:
    """Get README.md path."""
    return root() / 'README.md'


def roadmap() -> Path:
    """Get development roadmap path."""
    return root() / 'DEVELOPMENT_ROADMAP.md'


# Convenience function for scripts
def ensure_imports():
    """
    Ensure Atomizer modules can be imported.

    Call this at the start of any script to ensure proper imports:

    ```python
    import atomizer_paths
    atomizer_paths.ensure_imports()

    # Now you can import Atomizer modules
    from optimization_engine.core.runner import OptimizationRunner
    ```
    """
    setup_python_path()


if __name__ == '__main__':
    # Self-test
    print("Atomizer Path Configuration")
    print("=" * 60)
    print(f"Root:                {root()}")
    print(f"Optimization Engine: {optimization_engine()}")
    print(f"Studies:             {studies()}")
    print(f"Tests:               {tests()}")
    print(f"Docs:                {docs()}")
    print(f"Plugins:             {plugins()}")
    print("=" * 60)
    print("\nAll paths resolved successfully!")