Source code for ess.offspec.conversions

# SPDX-License-Identifier: BSD-3-Clause
# Copyright (c) 2025 Scipp contributors (https://github.com/scipp)
from scippneutron.conversion.graph import beamline, tof

from ..reflectometry.types import (
    CoordTransformationGraph,
    ReferenceRun,
    SampleRun,
)




def coordinate_transformation_graph_sample() -> CoordTransformationGraph[SampleRun]:
    """Build coordinate transformation graph for OFFSPEC sample runs."""
    return {
        **beamline.beamline(scatter=True),
        **tof.elastic_wavelength("tof"),
    }





def coordinate_transformation_graph_reference() -> CoordTransformationGraph[
    ReferenceRun
]:
    """Build coordinate transformation graph for OFFSPEC reference runs."""
    return {
        **beamline.beamline(scatter=False),
        **tof.elastic_wavelength("tof"),
    }



providers = (
    coordinate_transformation_graph_sample,
    coordinate_transformation_graph_reference,
)