lox-space 0.1.0-alpha.48

// SPDX-FileCopyrightText: 2025 Helge Eichhorn <git@helgeeichhorn.de>
//
// SPDX-License-Identifier: MPL-2.0

use std::path::PathBuf;

use lox_earth::eop::{self, EopParser, EopProvider};
use pyo3::exceptions::PyException;
use pyo3::types::{PyAnyMethods, PyTuple};
use pyo3::{Bound, PyErr, PyResult, create_exception, pyclass, pymethods};

create_exception!(
    lox_space,
    EopParserError,
    PyException,
    "Python exception raised when an EOP data file cannot be parsed."
);

/// PyO3 error wrapper for [`lox_earth::eop::EopParserError`].
pub struct PyEopParserError(pub eop::EopParserError);

impl From<PyEopParserError> for PyErr {
    fn from(err: PyEopParserError) -> Self {
        EopParserError::new_err(err.0.to_string())
    }
}

create_exception!(
    lox_space,
    EopProviderError,
    PyException,
    "Python exception raised when EOP data cannot be retrieved for a given time."
);

/// PyO3 error wrapper for [`lox_earth::eop::EopProviderError`].
pub struct PyEopProviderError(pub lox_earth::eop::EopProviderError);

impl From<PyEopProviderError> for PyErr {
    fn from(err: PyEopProviderError) -> Self {
        EopProviderError::new_err(err.0.to_string())
    }
}

/// Earth Orientation Parameters (EOP) data provider.
///
/// EOP data is required for accurate transformations involving UT1 and
/// polar motion corrections. The data can be loaded from IERS finals2000A
/// files (CSV format).
///
/// EOP data files can be obtained from:
///
/// - IERS: <https://www.iers.org/IERS/EN/DataProducts/EarthOrientationData/eop.html>
/// - Celestrak: <https://celestrak.org/SpaceData/>
///
/// Args:
///     path: Path to the EOP data file (CSV format).
///     path2: Optional second path for separate polar motion and UT1 files.
///
/// Raises:
///     EopParserError: If the file cannot be parsed.
///     OSError: If the file cannot be read.
#[pyclass(name = "EOPProvider", module = "lox_space", frozen)]
#[derive(Debug)]
pub struct PyEopProvider(pub EopProvider);

#[pymethods]
impl PyEopProvider {
    #[pyo3(signature = (*args))]
    #[new]
    /// Construct an `EOPProvider` from one or two EOP data file paths.
    pub fn new(args: &Bound<'_, PyTuple>) -> PyResult<PyEopProvider> {
        let (path1, path2) = if let Ok((path1, path2)) = args.extract::<(PathBuf, PathBuf)>() {
            (path1, path2)
        } else if let Ok((path,)) = args.extract::<(PathBuf,)>() {
            (path.clone(), path)
        } else {
            return Err(PyEopParserError(eop::EopParserError::NoFiles).into());
        };
        Ok(PyEopProvider(
            EopParser::new()
                .from_paths(path1, path2)
                .parse()
                .map_err(PyEopParserError)?,
        ))
    }
}