use crate::constants::UNIX_EPOCH_JD;
use crate::julian::JulianDate;
use crate::parsing::parse_iso8601;
use crate::{TimeError, TimeResult};
use celestial_core::constants::SECONDS_PER_DAY_F64;
use std::fmt;
use std::str::FromStr;
#[derive(Debug, Clone, Copy, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct TCG(JulianDate);
impl TCG {
pub fn new(seconds: i64, nanos: u32) -> Self {
let total_seconds =
seconds as f64 + nanos as f64 / celestial_core::constants::NANOSECONDS_PER_SECOND_F64;
let jd = JulianDate::from_f64(UNIX_EPOCH_JD + total_seconds / SECONDS_PER_DAY_F64);
Self(jd)
}
pub fn from_julian_date(jd: JulianDate) -> Self {
Self(jd)
}
pub fn j2000() -> Self {
Self(JulianDate::j2000())
}
pub fn to_julian_date(&self) -> JulianDate {
self.0
}
pub fn add_seconds(&self, seconds: f64) -> Self {
Self(self.0.add_seconds(seconds))
}
pub fn add_days(&self, days: f64) -> Self {
Self(self.0.add_days(days))
}
}
impl fmt::Display for TCG {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "TCG {}", self.0)
}
}
impl From<JulianDate> for TCG {
fn from(jd: JulianDate) -> Self {
Self::from_julian_date(jd)
}
}
impl FromStr for TCG {
type Err = TimeError;
fn from_str(s: &str) -> TimeResult<Self> {
let parsed = parse_iso8601(s)?;
Ok(Self::from_julian_date(parsed.to_julian_date()))
}
}
pub fn tcg_from_calendar(year: i32, month: u8, day: u8, hour: u8, minute: u8, second: f64) -> TCG {
let jd = JulianDate::from_calendar(year, month, day, hour, minute, second);
TCG::from_julian_date(jd)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::constants::UNIX_EPOCH_JD;
use celestial_core::constants::J2000_JD;
#[test]
fn test_tcg_constructors() {
assert_eq!(TCG::new(0, 0).to_julian_date().to_f64(), UNIX_EPOCH_JD);
assert_eq!(TCG::j2000().to_julian_date().to_f64(), J2000_JD);
assert_eq!(
tcg_from_calendar(2000, 1, 1, 12, 0, 0.0)
.to_julian_date()
.to_f64(),
J2000_JD
);
let jd = JulianDate::new(J2000_JD, 0.123456789);
let tcg_direct = TCG::from_julian_date(jd);
let tcg_from_trait: TCG = jd.into();
assert_eq!(
tcg_direct.to_julian_date().jd1(),
tcg_from_trait.to_julian_date().jd1()
);
assert_eq!(
tcg_direct.to_julian_date().jd2(),
tcg_from_trait.to_julian_date().jd2()
);
}
#[test]
fn test_tcg_arithmetic() {
let tcg = TCG::j2000();
assert_eq!(tcg.add_days(1.0).to_julian_date().to_f64(), J2000_JD + 1.0);
assert_eq!(
tcg.add_seconds(3600.0).to_julian_date().to_f64(),
J2000_JD + 1.0 / 24.0
);
}
#[test]
fn test_tcg_string_parsing() {
assert_eq!(
TCG::from_str("2000-01-01T12:00:00")
.unwrap()
.to_julian_date()
.to_f64(),
TCG::j2000().to_julian_date().to_f64()
);
let result = TCG::from_str("2000-01-01T12:00:00.123").unwrap();
let expected_jd = J2000_JD + 0.123 / SECONDS_PER_DAY_F64;
let diff = (result.to_julian_date().to_f64() - expected_jd).abs();
assert!(diff < 1e-14, "fractional seconds diff: {:.2e}", diff);
assert!(TCG::from_str("invalid-date").is_err());
}
#[test]
fn test_tcg_display() {
let display_str = format!("{}", TCG::from_julian_date(JulianDate::new(J2000_JD, 0.5)));
assert!(display_str.starts_with("TCG"));
assert!(display_str.contains("2451545"));
}
#[cfg(feature = "serde")]
#[test]
fn test_tcg_serde_round_trip() {
let test_cases = [
("J2000", TCG::j2000()),
("Unix epoch", TCG::new(0, 0)),
(
"Modern date",
tcg_from_calendar(2024, 6, 15, 14, 30, 45.123),
),
(
"High precision",
tcg_from_calendar(1990, 12, 31, 23, 59, 59.999999999),
),
];
for (name, original) in test_cases {
let json = serde_json::to_string(&original).unwrap();
let deserialized: TCG = serde_json::from_str(&json).unwrap();
let jd1_diff =
(original.to_julian_date().jd1() - deserialized.to_julian_date().jd1()).abs();
let jd2_diff =
(original.to_julian_date().jd2() - deserialized.to_julian_date().jd2()).abs();
let total_diff =
(original.to_julian_date().to_f64() - deserialized.to_julian_date().to_f64()).abs();
assert!(jd1_diff < 1e-14, "{}: jd1 diff {:.2e}", name, jd1_diff);
assert!(jd2_diff < 1e-14, "{}: jd2 diff {:.2e}", name, jd2_diff);
assert!(
total_diff < 1e-14,
"{}: total diff {:.2e}",
name,
total_diff
);
}
}
}