from __future__ import annotations

import importlib.util
import json
import sys
import tempfile
import unittest
from pathlib import Path


SCRIPT = Path(__file__).parents[1] / "prepare_odx_reconstruction_inputs.py"
SPEC = importlib.util.spec_from_file_location("prepare_odx_reconstruction_inputs", SCRIPT)
MODULE = importlib.util.module_from_spec(SPEC)
assert SPEC.loader is not None
sys.modules[SPEC.name] = MODULE
SPEC.loader.exec_module(MODULE)


class OdxInputPreparationTests(unittest.TestCase):
    def test_dji_orientation_matches_referenced_odx_normalization(self) -> None:
        self.assertEqual(MODULE.odx_orientation_from_dji(-25.9, -37.7, 0.0), (-25.9, 52.3, 0.0))

    def test_geo_txt_uses_webodm_epsg4326_field_order(self) -> None:
        record = MODULE.OdxPhoto(
            image="DJI_0001.JPG", longitude=113.317951157, latitude=25.660699129, altitude_m=1201.372,
            yaw_deg=-25.9, pitch_deg=52.3, roll_deg=0.0, horizontal_accuracy_m=0.0046, vertical_accuracy_m=0.0112,
            width=4000, height=3000, sha256="a" * 64, rtk_std_lat_raw=0.0023, rtk_std_lon_raw=0.00201, rtk_std_hgt_raw=0.0056,
        )
        with tempfile.TemporaryDirectory() as temporary:
            path = Path(temporary) / "geo.txt"
            MODULE.write_geo_txt([record], path)
            lines = path.read_text(encoding="utf-8").splitlines()
        self.assertEqual(lines[0], "EPSG:4326")
        fields = lines[1].split()
        self.assertEqual(fields[0], "DJI_0001.JPG")
        self.assertEqual(len(fields), 9)
        self.assertEqual(float(fields[1]), record.longitude)
        self.assertEqual(float(fields[2]), record.latitude)
        self.assertEqual(float(fields[5]), record.pitch_deg)

    def test_camera_positions_are_valid_geojson(self) -> None:
        record = MODULE.OdxPhoto(
            image="DJI_0001.JPG", longitude=113.0, latitude=25.0, altitude_m=100.0,
            yaw_deg=0.0, pitch_deg=90.0, roll_deg=0.0, horizontal_accuracy_m=0.02, vertical_accuracy_m=0.03,
            width=1, height=1, sha256="b" * 64, rtk_std_lat_raw=0.01, rtk_std_lon_raw=0.01, rtk_std_hgt_raw=0.015,
        )
        with tempfile.TemporaryDirectory() as temporary:
            path = Path(temporary) / "positions.geojson"
            MODULE.write_camera_positions_geojson([record], path)
            payload = json.loads(path.read_text(encoding="utf-8"))
        self.assertEqual(payload["type"], "FeatureCollection")
        self.assertEqual(payload["features"][0]["geometry"]["coordinates"], [113.0, 25.0, 100.0])

    def test_recommendations_do_not_override_per_photo_accuracy(self) -> None:
        record = MODULE.OdxPhoto(
            image="DJI_0001.JPG", longitude=113.0, latitude=25.0, altitude_m=100.0,
            yaw_deg=0.0, pitch_deg=90.0, roll_deg=0.0, horizontal_accuracy_m=0.02, vertical_accuracy_m=0.03,
            width=1, height=1, sha256="c" * 64, rtk_std_lat_raw=0.01, rtk_std_lon_raw=0.01, rtk_std_hgt_raw=0.015,
        )
        options = MODULE.odx_option_recommendations([record])
        self.assertNotIn("gps-accuracy-m", options["recommendations"])
        self.assertIn("Leave unset", options["conditional_options"]["gps-accuracy"])


if __name__ == "__main__":
    unittest.main()