"""Format object for XDS files"""


from __future__ import annotations

import sys

from rstbx.cftbx.coordinate_frame_converter import coordinate_frame_converter
from rstbx.cftbx.coordinate_frame_helpers import is_recognized_file
from scitbx import matrix

from dxtbx import IncorrectFormatError
from dxtbx.format.Format import Format
from dxtbx.model import Detector


class FormatXDS(Format):
    """An image reading class for XDS files"""

    @staticmethod
    def understand(image_file):
        """Check to see if this looks like an CBF format image, i.e. we can
        make sense of it."""
        return is_recognized_file(image_file)

    def __init__(self, image_file, **kwargs):
        """Initialise the image structure from the given file."""

        if not self.understand(image_file):
            raise IncorrectFormatError(self, image_file)
        super().__init__(image_file, **kwargs)

    def _start(self):
        """Open the image file as a cbf file handle, and keep this somewhere
        safe."""

        # Convert the parameters to cbf conventions
        self._convert_to_cbf_convention(self._image_file)

    def _convert_to_cbf_convention(self, xparm_filename):
        """Get the parameters from the XPARM file and convert them to CBF
        conventions.

        Params:
            xparm_handle The handle to the xparm file.

        """
        # Create a coordinate frame converter and extract other quantities
        cfc = coordinate_frame_converter(xparm_filename)
        self._detector_origin = cfc.get("detector_origin")
        self._rotation_axis = cfc.get("rotation_axis")
        self._fast_axis = cfc.get("detector_fast")
        self._slow_axis = cfc.get("detector_slow")
        self._wavelength = cfc.get("wavelength")
        self._image_size = cfc.get("detector_size_fast_slow")
        self._pixel_size = cfc.get("detector_pixel_size_fast_slow")
        self._starting_angle = cfc.get("starting_angle")
        self._oscillation_range = cfc.get("oscillation_range")
        self._starting_frame = cfc.get("starting_frame")
        self._data_range = cfc.get("data_range")
        self._divergence = 0.0
        self._sigma_divergence = cfc.get("sigma_divergence")
        sample_vector = cfc.get("sample_to_source")
        self._beam_vector = tuple(matrix.col(sample_vector))
        self._panel_offset = cfc.get("panel_offset")
        self._panel_size = cfc.get("panel_size")
        self._panel_origin = cfc.get("panel_origin")
        self._panel_fast = cfc.get("panel_fast")
        self._panel_slow = cfc.get("panel_slow")

    def _goniometer(self):
        """Return a working goniometer instance."""
        return self._goniometer_factory.known_axis(self._rotation_axis)

    def _detector(self):
        """Return a working detector instance."""
        if self._panel_origin is not None:
            detector = Detector()
            root = detector.hierarchy()
            root.set_frame(self._fast_axis, self._slow_axis, self._detector_origin)

            i_panel = 0
            for p_offset, p_size, origin, fast, slow in zip(
                self._panel_offset,
                self._panel_size,
                self._panel_origin,
                self._panel_fast,
                self._panel_slow,
            ):
                # ensure mutual orthogonality in presence of numerical rounding errors
                normal = fast.cross(slow)
                slow = normal.cross(fast)
                p = root.add_panel()
                p.set_type("unknown")
                p.set_raw_image_offset(p_offset)
                p.set_image_size(p_size)
                p.set_name("Panel%d" % i_panel)
                p.set_pixel_size(self._pixel_size)
                p.set_frame(fast.elems, slow.elems, origin.elems)
                i_panel += 1
            return detector

        return self._detector_factory.complex(
            self._detector_factory.sensor("unknown"),
            self._detector_origin,
            self._fast_axis,
            self._slow_axis,
            self._pixel_size,
            self._image_size,
            (0, 1.0e9),
        )

    def _beam(self):
        """Return a working beam instance."""
        return self._beam_factory.make_beam(
            sample_to_source=self._beam_vector,
            wavelength=self._wavelength,
            divergence=self._divergence,
            sigma_divergence=self._sigma_divergence,
        )

    def _scan(self):
        """Return a working scan instance."""
        # Set the scan parameters
        if self._data_range is not None:
            image_range = tuple(self._data_range)
        else:
            image_range = (self._starting_frame, self._starting_frame)
        oscillation = (self._starting_angle, self._oscillation_range)

        # Create the scan object
        return self._scan_factory.make_scan(
            image_range,
            0.0,
            oscillation,
            [0] * (image_range[-1] - image_range[0] + 1),
            deg=True,
        )

    def get_raw_data(self):
        """Get the raw image data. For GXPARM.XDS file raise am exception."""
        raise OSError("GXPARM.XDS does not support image data!")


if __name__ == "__main__":
    for arg in sys.argv[1:]:
        print(FormatXDS.understand(arg))
