# Stoicheia environment. Source this before running anything.
#   export STOICHEIA_DATA=/path/to/writable/data+checkpoints
#   source env.sh
# Everything else derives from those two roots.

# This repository's checkout.
export STOICHEIA_ROOT="${STOICHEIA_ROOT:-$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)}"
# Writable data root: corpora, shards, checkpoints, eval outputs.
export STOICHEIA_DATA="${STOICHEIA_DATA:?set STOICHEIA_DATA to a writable data+checkpoint root}"

# Per-task subtrees under the data root. Configs and scripts reference these by name.
export INS_DATA="$STOICHEIA_DATA/insc_data"                            # documentary corpora + runs
export SYN_DATA="$STOICHEIA_DATA/parser_data"                          # treebank shards + parser runs
export METER_DATA="$STOICHEIA_DATA/meter"                              # encoded meter training data
export AGD_DATA="${AGD_DATA:-$STOICHEIA_DATA/agd}"                     # normalized source corpora
# Checkout of the released meter-silver dataset (verse lines, scanner corpus, Norma).
# Only needed to retrain the meter model or to run its data-dependent tests.
export MACRONIZER_SRC="${MACRONIZER_SRC:-$STOICHEIA_DATA/macron_data}"

# Sub-packages resolve their code root through these; all are this checkout.
export INS_ROOT="$STOICHEIA_ROOT"
export SYN_ROOT="$STOICHEIA_ROOT"
export TAGGER_ROOT="$STOICHEIA_ROOT"
export METER_ROOT="$STOICHEIA_ROOT"

# Pretrained torso used by the downstream configs that do not name a checkpoint inline.
export STOICHEIA_CKPT="${STOICHEIA_CKPT:-$STOICHEIA_DATA/runs/stoicheia_doc_clean/best.pt}"

export PYTHONPATH="$STOICHEIA_ROOT:${PYTHONPATH:-}"

# Caches inside the data root, so nothing lands in a read-only or shared HOME.
export HF_HOME="$STOICHEIA_DATA/cache/huggingface"
export XDG_CACHE_HOME="$STOICHEIA_DATA/cache/xdg"
export TRITON_CACHE_DIR="$STOICHEIA_DATA/cache/triton"
export TORCHINDUCTOR_CACHE_DIR="$STOICHEIA_DATA/cache/inductor"
mkdir -p "$XDG_CACHE_HOME" "$TRITON_CACHE_DIR" "$TORCHINDUCTOR_CACHE_DIR" 2>/dev/null

# Optional containerized execution (the SLURM templates use it; leave unset to run in a venv).
#   export STOICHEIA_SIF=/path/to/container.sif
#   export APPTAINER_BINDS="-B $STOICHEIA_DATA"
export SIF="${STOICHEIA_SIF:-}"