"""Resolve package-relative resources without machine-specific locations."""
import os
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent


def env(name, default=""):
    return os.environ.get(name, default)


def project_path(relative):
    path = Path(relative)
    return str(path if path.is_absolute() else ROOT / path)


def initialize(entrypoint=False):
    for relative in ("eval", "diffusion", "data"):
        location = project_path(relative)
        if location not in sys.path:
            sys.path.insert(0, location)
    if entrypoint:
        os.chdir(ROOT)
    os.environ.setdefault("TORCH_HOME", project_path(".cache/torch"))



def seg_mask_root(root=None):
    """External PhiSeg output root; no masks are distributed with the code."""
    return project_path(root or os.environ.get("SEG_MASK_ROOT", "data/seg_masks"))


def seg_mask_dir(clip_id, root=None):
    """Same layout as infer_phiseg: <root>/<clip_id>/ego/*.png."""
    return str(Path(seg_mask_root(root)) / clip_id / "ego")