"""One-shot workflow for capturing a full reference set."""

from __future__ import annotations

import logging

from python_app.models.dataset_model import SweepCollection
from python_app.models.run_config_model import RunConfigModel
from python_app.storage.npz_store import NpzStore
from python_app.workflows.sequential_capture_workflow import (
    DEFAULT_CALIBRATION_MEDIAN_SWEEP_COUNT,
    SequentialCaptureSession,
)

logger = logging.getLogger(__name__)


def capture_reference_set(
    config: RunConfigModel,
    set_name: str,
    store: NpzStore,
    *,
    median_sweep_count: int = DEFAULT_CALIBRATION_MEDIAN_SWEEP_COUNT,
) -> tuple[str, SweepCollection]:
    """Capture all switch combinations and persist them as reference set."""
    logger.info("Starting one-shot reference capture: set=%s", set_name)
    session = SequentialCaptureSession(
        config=config,
        kind="s21_reference",
        set_name=set_name,
        median_sweep_count=median_sweep_count,
    )
    try:
        session.open()
        while not session.is_complete():
            session.capture_current_combo()
        return session.finalize(store)
    finally:
        session.close()