from jfjoch_client.models.scan_result import ScanResult
from pydantic import BaseModel

from aaredaqlib.coordinate import SmargonCoordinate


class RotationScanRequest(BaseModel):
    dtz: float | None = None
    transmission: float | None = None
    file_prefix: str | None = None
    exp_time_s: float

    start: SmargonCoordinate | None = None  # None = start where you are
    end: SmargonCoordinate | None = None

    start_omega_deg: float = 0
    incr_omega_deg: float
    steps: int

    screening: bool = False
    wedge_omega_deg: float | None = None

class CompletedRotationScan(BaseModel):
    request: RotationScanRequest
    result: ScanResult