rust
/
spacepackets
5
1
Fork 0
Code Issues Pull Requests Projects Releases 24 Wiki Activity
spacepackets/src/time/cds.rs

2372 lines
87 KiB
Rust
Raw Blame History

//! Module to generate or read CCSDS Day Segmented (CDS) timestamps as specified in
//! [CCSDS 301.0-B-4](https://public.ccsds.org/Pubs/301x0b4e1.pdf) section 3.3 .
//!
//! The core data structure to do this is the [CdsTime] struct and the
//! [get_dyn_time_provider_from_bytes] function to retrieve correct instances of the
//! struct from a bytestream.
use crate::private::Sealed;
use crate::ByteConversionError;
use core::cmp::Ordering;
use core::fmt::{Debug, Display, Formatter};
use core::ops::{Add, AddAssign};
use core::time::Duration;
use delegate::delegate;
#[cfg(feature = "std")]
use super::StdTimestampError;
#[cfg(feature = "std")]
use std::error::Error;
#[cfg(feature = "std")]
use std::time::{SystemTime, SystemTimeError};
#[cfg(feature = "chrono")]
use chrono::Datelike;
#[cfg(feature = "alloc")]
use super::ccsds_time_code_from_p_field;
#[cfg(feature = "alloc")]
use alloc::boxed::Box;
#[cfg(feature = "alloc")]
use core::any::Any;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use super::{
ccsds_to_unix_days, unix_to_ccsds_days, CcsdsTimeCode, CcsdsTimeProvider,
DateBeforeCcsdsEpochError, TimeReader, TimeWriter, TimestampError, UnixTime, MS_PER_DAY,
SECONDS_PER_DAY,
};
/// Base value for the preamble field for a time field parser to determine the time field type.
pub const P_FIELD_BASE: u8 = (CcsdsTimeCode::Cds as u8) << 4;
pub const MIN_CDS_FIELD_LEN: usize = 7;
pub const MAX_DAYS_24_BITS: u32 = 2_u32.pow(24) - 1;
/// Generic trait implemented by token structs to specify the length of day field at type
/// level. This trait is only meant to be implemented in this crate and therefore sealed.
pub trait ProvidesDaysLength: Sealed + Clone {
type FieldType: Debug
+ Copy
+ Clone
+ PartialEq
+ Eq
+ TryFrom<i32>
+ TryFrom<u32>
+ From<u16>
+ Into<u32>
+ Into<i64>;
}
/// Type level token to be used as a generic parameter to [CdsTime].
#[derive(Debug, Copy, Clone, PartialEq, Eq, Default)]
pub struct DaysLen16Bits {}
impl Sealed for DaysLen16Bits {}
impl ProvidesDaysLength for DaysLen16Bits {
type FieldType = u16;
}
/// Type level token to be used as a generic parameter to [CdsTime].
#[derive(Debug, Copy, Clone, PartialEq, Eq, Default)]
pub struct DaysLen24Bits {}
impl Sealed for DaysLen24Bits {}
impl ProvidesDaysLength for DaysLen24Bits {
type FieldType = u32;
}
#[derive(Debug, PartialEq, Eq, Copy, Clone)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub enum LengthOfDaySegment {
Short16Bits = 0,
Long24Bits = 1,
}
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub enum SubmillisPrecision {
Absent = 0b00,
Microseconds = 0b01,
Picoseconds = 0b10,
Reserved = 0b11,
}
#[derive(Debug, PartialEq, Eq, Copy, Clone)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub enum CdsError {
/// CCSDS days value exceeds maximum allowed size or is negative
InvalidCcsdsDays(i64),
/// There are distinct constructors depending on the days field width detected in the preamble
/// field. This error will be returned if there is a missmatch.
InvalidCtorForDaysOfLenInPreamble(LengthOfDaySegment),
DateBeforeCcsdsEpoch(DateBeforeCcsdsEpochError),
}
impl Display for CdsError {
fn fmt(&self, f: &mut Formatter<'_>) -> core::fmt::Result {
match self {
CdsError::InvalidCcsdsDays(days) => {
write!(f, "invalid ccsds days {days}")
}
CdsError::InvalidCtorForDaysOfLenInPreamble(length_of_day) => {
write!(
f,
"wrong constructor for length of day {length_of_day:?} detected in preamble",
)
}
CdsError::DateBeforeCcsdsEpoch(e) => write!(f, "date before CCSDS epoch: {e}"),
}
}
}
#[cfg(feature = "std")]
impl Error for CdsError {
fn source(&self) -> Option<&(dyn Error + 'static)> {
match self {
CdsError::DateBeforeCcsdsEpoch(e) => Some(e),
_ => None,
}
}
}
impl From<DateBeforeCcsdsEpochError> for CdsError {
fn from(value: DateBeforeCcsdsEpochError) -> Self {
Self::DateBeforeCcsdsEpoch(value)
}
}
pub fn length_of_day_segment_from_pfield(pfield: u8) -> LengthOfDaySegment {
if (pfield >> 2) & 0b1 == 1 {
return LengthOfDaySegment::Long24Bits;
}
LengthOfDaySegment::Short16Bits
}
pub fn precision_from_pfield(pfield: u8) -> SubmillisPrecision {
match pfield & 0b11 {
0b01 => SubmillisPrecision::Microseconds,
0b10 => SubmillisPrecision::Picoseconds,
0b00 => SubmillisPrecision::Absent,
0b11 => SubmillisPrecision::Reserved,
_ => panic!("pfield to SubmillisPrecision failed"),
}
}
/// This object is the abstraction for the CCSDS Day Segmented Time Code (CDS).
///
/// It has the capability to generate and read timestamps as specified in the CCSDS 301.0-B-4
/// section 3.3 . The width of the days field is configured at compile time via the generic
/// [ProvidesDaysLength] trait which is implemented by [DaysLen16Bits] and [DaysLen24Bits].
///
/// If you do not want to perform a forward check of the days length field with
/// [length_of_day_segment_from_pfield] and you have [alloc] support, you can also
/// use [get_dyn_time_provider_from_bytes] to retrieve the correct instance as a [DynCdsTimeProvider]
/// trait object.
///
/// Custom epochs are not supported yet.
/// Furthermore, the preamble field (p-field) is explicitly conveyed.
/// That means it will always be present when writing the time stamp to a raw buffer, and it
/// must be present when reading a CDS timestamp from a raw buffer.
///
/// # Example
///
/// ```
/// use core::time::Duration;
/// use spacepackets::time::cds::{CdsTime, length_of_day_segment_from_pfield, LengthOfDaySegment};
/// use spacepackets::time::{TimeWriter, CcsdsTimeCode, CcsdsTimeProvider};
///
/// let timestamp_now = CdsTime::from_now_with_u16_days().unwrap();
/// let mut raw_stamp = [0; 7];
/// {
/// let written = timestamp_now.write_to_bytes(&mut raw_stamp).unwrap();
/// assert_eq!((raw_stamp[0] >> 4) & 0b111, CcsdsTimeCode::Cds as u8);
/// assert_eq!(written, 7);
/// }
/// {
/// assert_eq!(length_of_day_segment_from_pfield(raw_stamp[0]), LengthOfDaySegment::Short16Bits);
/// let read_result = CdsTime::from_bytes_with_u16_days(&raw_stamp);
/// assert!(read_result.is_ok());
/// let stamp_deserialized = read_result.unwrap();
/// assert_eq!(stamp_deserialized.len_as_bytes(), 7);
/// }
/// // It is possible to add a Duration offset to a timestamp provider. Add 5 minutes offset here
/// let offset = Duration::from_secs(60 * 5);
/// let former_unix_seconds = timestamp_now.unix_secs();
/// let timestamp_in_5_minutes = timestamp_now + offset;
/// assert_eq!(timestamp_in_5_minutes.unix_secs(), former_unix_seconds + 5 * 60);
/// ```
#[derive(Debug, Copy, Clone, Eq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub struct CdsTime<DaysLen: ProvidesDaysLength = DaysLen16Bits> {
pfield: u8,
ccsds_days: DaysLen::FieldType,
ms_of_day: u32,
submillis: u32,
/// This is not strictly necessary but still cached because it significantly simplifies the
/// calculation of [`DateTime<Utc>`].
unix_time: UnixTime,
}
/// Common properties for all CDS time providers.
///
/// Also exists to encapsulate properties used by private converters.
pub trait CdsCommon {
fn submillis_precision(&self) -> SubmillisPrecision;
fn submillis(&self) -> u32;
fn ms_of_day(&self) -> u32;
fn ccsds_days_as_u32(&self) -> u32;
}
/// Generic properties for all CDS time providers.
pub trait CdsTimestamp: CdsCommon {
fn len_of_day_seg(&self) -> LengthOfDaySegment;
}
/// Private trait which serves as an abstraction for different converters.
trait CdsConverter: CdsCommon {
fn unix_days_seconds(&self) -> i64;
}
struct ConversionFromUnix {
ccsds_days: u32,
ms_of_day: u32,
submilis_prec: SubmillisPrecision,
submillis: u32,
/// This is a side-product of the calculation of the CCSDS days. It is useful for
/// re-calculating the datetime at a later point and therefore supplied as well.
unix_days_seconds: i64,
}
impl ConversionFromUnix {
fn new(
unix_seconds: i64,
subsec_nanos: u32,
precision: SubmillisPrecision,
) -> Result<Self, DateBeforeCcsdsEpochError> {
let (unix_days, secs_of_day) = calc_unix_days_and_secs_of_day(unix_seconds);
let ccsds_days = unix_to_ccsds_days(unix_days);
if ccsds_days == 0 && (secs_of_day > 0 || subsec_nanos > 0) || ccsds_days < 0 {
return Err(DateBeforeCcsdsEpochError(
UnixTime::new_checked(unix_seconds, subsec_nanos)
.expect("unix timestamp creation failed"),
));
}
let ms_of_day = secs_of_day * 1000 + subsec_nanos / 10_u32.pow(6);
let submillis = match precision {
SubmillisPrecision::Microseconds => (subsec_nanos / 1_000) % 1000,
SubmillisPrecision::Picoseconds => (subsec_nanos % 10_u32.pow(6)) * 1000,
_ => 0,
};
Ok(Self {
ccsds_days: unix_to_ccsds_days(unix_days) as u32,
ms_of_day,
unix_days_seconds: unix_days * SECONDS_PER_DAY as i64,
submilis_prec: precision,
submillis,
})
}
}
impl CdsCommon for ConversionFromUnix {
fn submillis_precision(&self) -> SubmillisPrecision {
self.submilis_prec
}
fn ms_of_day(&self) -> u32 {
self.ms_of_day
}
fn ccsds_days_as_u32(&self) -> u32 {
self.ccsds_days
}
fn submillis(&self) -> u32 {
self.submillis
}
}
impl CdsConverter for ConversionFromUnix {
fn unix_days_seconds(&self) -> i64 {
self.unix_days_seconds
}
}
/// Helper struct which generates fields for the CDS time provider from a datetime.
struct ConversionFromChronoDatetime {
unix_conversion: ConversionFromUnix,
submillis_prec: SubmillisPrecision,
submillis: u32,
}
impl CdsCommon for ConversionFromChronoDatetime {
fn submillis_precision(&self) -> SubmillisPrecision {
self.submillis_prec
}
delegate! {
to self.unix_conversion {
fn ms_of_day(&self) -> u32;
fn ccsds_days_as_u32(&self) -> u32;
}
}
fn submillis(&self) -> u32 {
self.submillis
}
}
impl CdsConverter for ConversionFromChronoDatetime {
delegate! {to self.unix_conversion { fn unix_days_seconds(&self) -> i64; }}
}
#[inline]
fn calc_unix_days_and_secs_of_day(unix_seconds: i64) -> (i64, u32) {
let mut secs_of_day = unix_seconds % SECONDS_PER_DAY as i64;
let mut unix_days = (unix_seconds - secs_of_day) / SECONDS_PER_DAY as i64;
// Imagine the CCSDS epoch time minus 5 seconds: We now have the last day in the year
// 1969 (-1 unix days) shortly before midnight (SECONDS_PER_DAY - 5).
if secs_of_day < 0 {
unix_days -= 1;
secs_of_day += SECONDS_PER_DAY as i64
}
(unix_days, secs_of_day as u32)
}
#[cfg(feature = "chrono")]
impl ConversionFromChronoDatetime {
fn new(dt: &chrono::DateTime<chrono::Utc>) -> Result<Self, DateBeforeCcsdsEpochError> {
Self::new_generic(dt, SubmillisPrecision::Absent)
}
fn new_with_submillis_us_prec(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, DateBeforeCcsdsEpochError> {
Self::new_generic(dt, SubmillisPrecision::Microseconds)
}
fn new_with_submillis_ps_prec(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, DateBeforeCcsdsEpochError> {
Self::new_generic(dt, SubmillisPrecision::Picoseconds)
}
#[cfg(feature = "chrono")]
fn new_generic(
dt: &chrono::DateTime<chrono::Utc>,
prec: SubmillisPrecision,
) -> Result<Self, DateBeforeCcsdsEpochError> {
// The CDS timestamp does not support timestamps before the CCSDS epoch.
if dt.year() < 1958 {
return Err(DateBeforeCcsdsEpochError(UnixTime::from(*dt)));
}
// The contained values in the conversion should be all positive now
let unix_conversion =
ConversionFromUnix::new(dt.timestamp(), dt.timestamp_subsec_nanos(), prec)?;
let mut submillis = 0;
match prec {
SubmillisPrecision::Microseconds => {
submillis = dt.timestamp_subsec_micros() % 1000;
}
SubmillisPrecision::Picoseconds => {
submillis = (dt.timestamp_subsec_nanos() % 10_u32.pow(6)) * 1000;
}
_ => (),
}
Ok(Self {
unix_conversion,
submillis_prec: prec,
submillis,
})
}
}
#[cfg(feature = "std")]
struct ConversionFromNow {
unix_conversion: ConversionFromUnix,
submillis_prec: SubmillisPrecision,
submillis: u32,
}
#[cfg(feature = "std")]
impl ConversionFromNow {
fn new() -> Result<Self, SystemTimeError> {
Self::new_generic(SubmillisPrecision::Absent)
}
fn new_with_submillis_us_prec() -> Result<Self, SystemTimeError> {
Self::new_generic(SubmillisPrecision::Microseconds)
}
fn new_with_submillis_ps_prec() -> Result<Self, SystemTimeError> {
Self::new_generic(SubmillisPrecision::Picoseconds)
}
fn new_generic(prec: SubmillisPrecision) -> Result<Self, SystemTimeError> {
let now = SystemTime::now().duration_since(SystemTime::UNIX_EPOCH)?;
let epoch = now.as_secs();
// This should always return a value with valid (non-negative) CCSDS days,
// so it is okay to unwrap
let unix_conversion =
ConversionFromUnix::new(epoch as i64, now.subsec_nanos(), prec).unwrap();
let mut submillis = 0;
match prec {
SubmillisPrecision::Microseconds => {
submillis = now.subsec_micros() % 1000;
}
SubmillisPrecision::Picoseconds => {
submillis = (now.subsec_nanos() % 10_u32.pow(6)) * 1000;
}
_ => (),
}
Ok(Self {
unix_conversion,
submillis_prec: prec,
submillis,
})
}
}
#[cfg(feature = "std")]
impl CdsCommon for ConversionFromNow {
fn submillis_precision(&self) -> SubmillisPrecision {
self.submillis_prec
}
delegate! {
to self.unix_conversion {
fn ms_of_day(&self) -> u32;
fn ccsds_days_as_u32(&self) -> u32;
}
}
fn submillis(&self) -> u32 {
self.submillis
}
}
#[cfg(feature = "std")]
impl CdsConverter for ConversionFromNow {
delegate! {to self.unix_conversion { fn unix_days_seconds(&self) -> i64; }}
}
#[cfg(feature = "alloc")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
pub trait DynCdsTimeProvider: CcsdsTimeProvider + CdsTimestamp + TimeWriter + Any {}
#[cfg(feature = "alloc")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
impl DynCdsTimeProvider for CdsTime<DaysLen16Bits> {}
#[cfg(feature = "alloc")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
impl DynCdsTimeProvider for CdsTime<DaysLen24Bits> {}
/// This function returns the correct [CdsTime] instance from a raw byte array
/// by checking the length of days field. It also checks the CCSDS time code for correctness.
///
/// # Example
///
/// ```
/// use spacepackets::time::cds::{
/// CdsTime, LengthOfDaySegment, get_dyn_time_provider_from_bytes, SubmillisPrecision,
/// };
/// use spacepackets::time::{TimeWriter, CcsdsTimeCode, CcsdsTimeProvider};
///
/// let timestamp_now = CdsTime::new_with_u16_days(24, 24);
/// let mut raw_stamp = [0; 7];
/// {
/// let written = timestamp_now.write_to_bytes(&mut raw_stamp).unwrap();
/// assert_eq!((raw_stamp[0] >> 4) & 0b111, CcsdsTimeCode::Cds as u8);
/// assert_eq!(written, 7);
/// }
/// {
/// let dyn_provider = get_dyn_time_provider_from_bytes(&raw_stamp).unwrap();
/// assert_eq!(dyn_provider.len_of_day_seg(), LengthOfDaySegment::Short16Bits);
/// assert_eq!(dyn_provider.ccsds_days_as_u32(), 24);
/// assert_eq!(dyn_provider.ms_of_day(), 24);
/// assert_eq!(dyn_provider.submillis_precision(), SubmillisPrecision::Absent);
/// }
/// ```
#[cfg(feature = "alloc")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
pub fn get_dyn_time_provider_from_bytes(
buf: &[u8],
) -> Result<Box<dyn DynCdsTimeProvider>, TimestampError> {
let time_code = ccsds_time_code_from_p_field(buf[0]);
if let Err(e) = time_code {
return Err(TimestampError::InvalidTimeCode {
expected: CcsdsTimeCode::Cds,
found: e,
});
}
let time_code = time_code.unwrap();
if time_code != CcsdsTimeCode::Cds {
return Err(TimestampError::InvalidTimeCode {
expected: CcsdsTimeCode::Cds,
found: time_code as u8,
});
}
if length_of_day_segment_from_pfield(buf[0]) == LengthOfDaySegment::Short16Bits {
Ok(Box::new(CdsTime::from_bytes_with_u16_days(buf)?))
} else {
Ok(Box::new(CdsTime::from_bytes_with_u24_days(buf)?))
}
}
impl<ProvidesDaysLen: ProvidesDaysLength> CdsCommon for CdsTime<ProvidesDaysLen> {
fn submillis_precision(&self) -> SubmillisPrecision {
precision_from_pfield(self.pfield)
}
fn ms_of_day(&self) -> u32 {
self.ms_of_day
}
fn ccsds_days_as_u32(&self) -> u32 {
self.ccsds_days.into()
}
fn submillis(&self) -> u32 {
self.submillis
}
}
impl<ProvidesDaysLen: ProvidesDaysLength> CdsTime<ProvidesDaysLen> {
/// Please note that a precision value of 0 will be converted to [None] (no precision).
pub fn set_submillis(&mut self, prec: SubmillisPrecision, value: u32) -> bool {
self.pfield &= !(0b11);
if let SubmillisPrecision::Absent = prec {
// self.submillis_precision = prec;
self.submillis = 0;
return true;
}
// self.submillis_precision = prec;
match prec {
SubmillisPrecision::Microseconds => {
if value > u16::MAX as u32 {
return false;
}
self.pfield |= SubmillisPrecision::Microseconds as u8;
self.submillis = value;
}
SubmillisPrecision::Picoseconds => {
self.pfield |= SubmillisPrecision::Picoseconds as u8;
self.submillis = value;
}
_ => (),
}
true
}
pub fn clear_submillis(&mut self) {
self.pfield &= !(0b11);
self.submillis = 0;
}
pub fn ccsds_days(&self) -> ProvidesDaysLen::FieldType {
self.ccsds_days
}
/// Maps the submillisecond precision to a nanosecond value. This will reduce precision when
/// using picosecond resolution, but significantly simplifies comparison of timestamps.
pub fn precision_as_ns(&self) -> Option<u32> {
match self.submillis_precision() {
SubmillisPrecision::Microseconds => Some(self.submillis * 1000),
SubmillisPrecision::Picoseconds => Some(self.submillis / 1000),
_ => None,
}
}
fn generic_raw_read_checks(
buf: &[u8],
days_len: LengthOfDaySegment,
) -> Result<SubmillisPrecision, TimestampError> {
if buf.len() < MIN_CDS_FIELD_LEN {
return Err(TimestampError::ByteConversion(
ByteConversionError::FromSliceTooSmall {
expected: MIN_CDS_FIELD_LEN,
found: buf.len(),
},
));
}
let pfield = buf[0];
match CcsdsTimeCode::try_from(pfield >> 4 & 0b111) {
Ok(cds_type) => match cds_type {
CcsdsTimeCode::Cds => (),
_ => {
return Err(TimestampError::InvalidTimeCode {
expected: CcsdsTimeCode::Cds,
found: cds_type as u8,
})
}
},
_ => {
return Err(TimestampError::InvalidTimeCode {
expected: CcsdsTimeCode::Cds,
found: pfield >> 4 & 0b111,
});
}
};
if ((pfield >> 3) & 0b1) == 1 {
return Err(TimestampError::CustomEpochNotSupported);
}
let days_len_from_pfield = length_of_day_segment_from_pfield(pfield);
if days_len_from_pfield != days_len {
return Err(CdsError::InvalidCtorForDaysOfLenInPreamble(days_len_from_pfield).into());
}
let stamp_len = Self::calc_stamp_len(pfield);
if buf.len() < stamp_len {
return Err(TimestampError::ByteConversion(
ByteConversionError::FromSliceTooSmall {
expected: stamp_len,
found: buf.len(),
},
));
}
Ok(precision_from_pfield(pfield))
}
fn calc_stamp_len(pfield: u8) -> usize {
let mut init_len = 7;
if length_of_day_segment_from_pfield(pfield) == LengthOfDaySegment::Long24Bits {
init_len += 1
}
match pfield & 0b11 {
0b01 => {
init_len += 2;
}
0b10 => {
init_len += 4;
}
_ => (),
}
init_len
}
fn setup(&mut self, unix_days_seconds: i64, ms_of_day: u32) {
self.calc_unix_seconds(unix_days_seconds, ms_of_day);
}
#[inline]
fn calc_unix_seconds(&mut self, mut unix_days_seconds: i64, ms_of_day: u32) {
let seconds_of_day = (ms_of_day / 1000) as i64;
if unix_days_seconds < 0 {
unix_days_seconds -= seconds_of_day;
} else {
unix_days_seconds += seconds_of_day;
}
let mut subsec_nanos = (ms_of_day % 1000) * 10_u32.pow(6);
if let Some(precision) = self.precision_as_ns() {
subsec_nanos += precision;
}
self.unix_time = UnixTime::new(unix_days_seconds, subsec_nanos);
}
fn length_check(&self, buf: &[u8], len_as_bytes: usize) -> Result<(), TimestampError> {
if buf.len() < len_as_bytes {
return Err(TimestampError::ByteConversion(
ByteConversionError::ToSliceTooSmall {
expected: len_as_bytes,
found: buf.len(),
},
));
}
Ok(())
}
fn generic_new(
days_len: LengthOfDaySegment,
ccsds_days: ProvidesDaysLen::FieldType,
ms_of_day: u32,
) -> Result<Self, CdsError>
where
i64: From<ProvidesDaysLen::FieldType>,
{
let mut provider = Self {
pfield: Self::generate_p_field(days_len, SubmillisPrecision::Absent),
ccsds_days,
ms_of_day,
unix_time: Default::default(),
submillis: 0,
};
let unix_days_seconds = ccsds_to_unix_days(i64::from(ccsds_days)) * SECONDS_PER_DAY as i64;
provider.setup(unix_days_seconds, ms_of_day);
Ok(provider)
}
#[cfg(feature = "chrono")]
fn from_dt_generic(
dt: &chrono::DateTime<chrono::Utc>,
days_len: LengthOfDaySegment,
) -> Result<Self, CdsError> {
let conv_from_dt = ConversionFromChronoDatetime::new(dt)?;
Self::generic_from_conversion(days_len, conv_from_dt)
}
#[cfg(feature = "chrono")]
fn from_dt_generic_us_prec(
dt: &chrono::DateTime<chrono::Utc>,
days_len: LengthOfDaySegment,
) -> Result<Self, CdsError> {
let conv_from_dt = ConversionFromChronoDatetime::new_with_submillis_us_prec(dt)?;
Self::generic_from_conversion(days_len, conv_from_dt)
}
#[cfg(feature = "chrono")]
fn from_dt_generic_ps_prec(
dt: &chrono::DateTime<chrono::Utc>,
days_len: LengthOfDaySegment,
) -> Result<Self, CdsError> {
let conv_from_dt = ConversionFromChronoDatetime::new_with_submillis_ps_prec(dt)?;
Self::generic_from_conversion(days_len, conv_from_dt)
}
fn from_unix_generic(
unix_stamp: &UnixTime,
days_len: LengthOfDaySegment,
submillis_prec: SubmillisPrecision,
) -> Result<Self, CdsError> {
let conv_from_dt =
ConversionFromUnix::new(unix_stamp.secs, unix_stamp.subsec_nanos, submillis_prec)?;
Self::generic_from_conversion(days_len, conv_from_dt)
}
#[cfg(feature = "std")]
fn from_now_generic(days_len: LengthOfDaySegment) -> Result<Self, StdTimestampError> {
let conversion_from_now = ConversionFromNow::new()?;
Self::generic_from_conversion(days_len, conversion_from_now)
.map_err(|e| StdTimestampError::Timestamp(TimestampError::from(e)))
}
#[cfg(feature = "std")]
fn from_now_generic_us_prec(days_len: LengthOfDaySegment) -> Result<Self, StdTimestampError> {
let conversion_from_now = ConversionFromNow::new_with_submillis_us_prec()?;
Self::generic_from_conversion(days_len, conversion_from_now)
.map_err(|e| StdTimestampError::Timestamp(TimestampError::from(e)))
}
#[cfg(feature = "std")]
fn from_now_generic_ps_prec(days_len: LengthOfDaySegment) -> Result<Self, StdTimestampError> {
let conversion_from_now = ConversionFromNow::new_with_submillis_ps_prec()?;
Self::generic_from_conversion(days_len, conversion_from_now)
.map_err(|e| StdTimestampError::Timestamp(TimestampError::from(e)))
}
fn generic_from_conversion<C: CdsConverter>(
days_len: LengthOfDaySegment,
converter: C,
) -> Result<Self, CdsError> {
let ccsds_days: ProvidesDaysLen::FieldType = converter
.ccsds_days_as_u32()
.try_into()
.map_err(|_| CdsError::InvalidCcsdsDays(converter.ccsds_days_as_u32().into()))?;
let mut provider = Self {
pfield: Self::generate_p_field(days_len, converter.submillis_precision()),
ccsds_days,
ms_of_day: converter.ms_of_day(),
unix_time: Default::default(),
submillis: converter.submillis(),
};
provider.setup(converter.unix_days_seconds(), converter.ms_of_day());
Ok(provider)
}
#[cfg(feature = "std")]
fn generic_conversion_from_now(&self) -> Result<ConversionFromNow, SystemTimeError> {
Ok(match self.submillis_precision() {
SubmillisPrecision::Microseconds => ConversionFromNow::new_with_submillis_us_prec()?,
SubmillisPrecision::Picoseconds => ConversionFromNow::new_with_submillis_ps_prec()?,
_ => ConversionFromNow::new()?,
})
}
fn generate_p_field(day_seg_len: LengthOfDaySegment, submillis_prec: SubmillisPrecision) -> u8 {
let mut pfield = P_FIELD_BASE | ((day_seg_len as u8) << 2);
match submillis_prec {
SubmillisPrecision::Microseconds => pfield |= SubmillisPrecision::Microseconds as u8,
SubmillisPrecision::Picoseconds => pfield |= SubmillisPrecision::Picoseconds as u8,
SubmillisPrecision::Reserved => pfield |= SubmillisPrecision::Reserved as u8,
_ => (),
}
pfield
}
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn update_from_now(&mut self) -> Result<(), StdTimestampError> {
let conversion_from_now = self.generic_conversion_from_now()?;
let ccsds_days: ProvidesDaysLen::FieldType = conversion_from_now
.unix_conversion
.ccsds_days
.try_into()
.map_err(|_| {
StdTimestampError::Timestamp(
CdsError::InvalidCcsdsDays(
conversion_from_now.unix_conversion.ccsds_days as i64,
)
.into(),
)
})?;
self.ccsds_days = ccsds_days;
self.ms_of_day = conversion_from_now.unix_conversion.ms_of_day;
self.setup(
conversion_from_now.unix_conversion.unix_days_seconds,
conversion_from_now.unix_conversion.ms_of_day,
);
Ok(())
}
}
impl CdsTime<DaysLen24Bits> {
/// Generate a new timestamp provider with the days field width set to 24 bits
pub fn new_with_u24_days(ccsds_days: u32, ms_of_day: u32) -> Result<Self, CdsError> {
if ccsds_days > MAX_DAYS_24_BITS {
return Err(CdsError::InvalidCcsdsDays(ccsds_days.into()));
}
Self::generic_new(LengthOfDaySegment::Long24Bits, ccsds_days, ms_of_day)
}
/// Generate a time stamp from the current time using the system clock.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u24_days() -> Result<Self, StdTimestampError> {
Self::from_now_generic(LengthOfDaySegment::Long24Bits)
}
/// Create a provider from a [`chrono::DateTime<chrono::Utc>`] struct.
///
/// ## Errors
///
/// This function will return [TimestampError::DateBeforeCcsdsEpoch] or
/// [TimestampError::Cds] if the time is before the CCSDS epoch (1958-01-01T00:00:00+00:00)
/// or the CCSDS days value exceeds the allowed bit width (24 bits).
#[cfg(feature = "chrono")]
pub fn from_dt_with_u24_days(dt: &chrono::DateTime<chrono::Utc>) -> Result<Self, CdsError> {
Self::from_dt_generic(dt, LengthOfDaySegment::Long24Bits)
}
/// Create a provider from a generic UNIX timestamp (seconds since 1970-01-01T00:00:00+00:00).
///
/// ## Errors
///
/// This function will return [TimestampError::DateBeforeCcsdsEpoch] or
/// [TimestampError::Cds] if the time is before the CCSDS epoch (1958-01-01T00:00:00+00:00)
/// or the CCSDS days value exceeds the allowed bit width (24 bits).
pub fn from_unix_stamp_with_u24_days(
unix_stamp: &UnixTime,
submillis_prec: SubmillisPrecision,
) -> Result<Self, CdsError> {
Self::from_unix_generic(unix_stamp, LengthOfDaySegment::Long24Bits, submillis_prec)
}
/// Like [Self::from_dt_with_u24_days] but with microsecond sub-millisecond precision.
#[cfg(feature = "chrono")]
pub fn from_dt_with_u24_days_us_precision(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, CdsError> {
Self::from_dt_generic_us_prec(dt, LengthOfDaySegment::Long24Bits)
}
/// Like [Self::from_dt_with_u24_days] but with picoseconds sub-millisecond precision.
#[cfg(feature = "chrono")]
pub fn from_dt_with_u24_days_ps_precision(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, CdsError> {
Self::from_dt_generic_ps_prec(dt, LengthOfDaySegment::Long24Bits)
}
/// Like [Self::from_now_with_u24_days] but with microsecond sub-millisecond precision.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u24_days_us_precision() -> Result<Self, StdTimestampError> {
Self::from_now_generic_us_prec(LengthOfDaySegment::Long24Bits)
}
/// Like [Self::from_now_with_u24_days] but with picoseconds sub-millisecond precision.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u24_days_ps_precision() -> Result<Self, StdTimestampError> {
Self::from_now_generic_us_prec(LengthOfDaySegment::Long24Bits)
}
pub fn from_bytes_with_u24_days(buf: &[u8]) -> Result<Self, TimestampError> {
let submillis_precision =
Self::generic_raw_read_checks(buf, LengthOfDaySegment::Long24Bits)?;
let mut temp_buf: [u8; 4] = [0; 4];
temp_buf[1..4].copy_from_slice(&buf[1..4]);
let cccsds_days: u32 = u32::from_be_bytes(temp_buf);
let ms_of_day: u32 = u32::from_be_bytes(buf[4..8].try_into().unwrap());
let mut provider = Self::new_with_u24_days(cccsds_days, ms_of_day)?;
match submillis_precision {
SubmillisPrecision::Microseconds => {
provider.set_submillis(
SubmillisPrecision::Microseconds,
u16::from_be_bytes(buf[8..10].try_into().unwrap()) as u32,
);
}
SubmillisPrecision::Picoseconds => {
provider.set_submillis(
SubmillisPrecision::Picoseconds,
u32::from_be_bytes(buf[8..12].try_into().unwrap()),
);
}
_ => (),
}
Ok(provider)
}
}
impl CdsTime<DaysLen16Bits> {
/// Generate a new timestamp provider with the days field width set to 16 bits
pub fn new_with_u16_days(ccsds_days: u16, ms_of_day: u32) -> Self {
// This should never fail, type system ensures CCSDS can not be negative or too large
Self::generic_new(LengthOfDaySegment::Short16Bits, ccsds_days, ms_of_day).unwrap()
}
/// Create a provider from a [`chrono::DateTime<Utc>`] struct.
///
/// This function will return a [TimestampError::DateBeforeCcsdsEpoch] or a
/// [TimestampError::Cds] if the time is before the CCSDS epoch (01-01-1958 00:00:00) or
/// the CCSDS days value exceeds the allowed bit width (16 bits).
#[cfg(feature = "chrono")]
pub fn from_dt_with_u16_days(dt: &chrono::DateTime<chrono::Utc>) -> Result<Self, CdsError> {
Self::from_dt_generic(dt, LengthOfDaySegment::Short16Bits)
}
/// Generate a time stamp from the current time using the system clock.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u16_days() -> Result<Self, StdTimestampError> {
Self::from_now_generic(LengthOfDaySegment::Short16Bits)
}
/// Create a provider from a generic UNIX timestamp (seconds since 1970-01-01T00:00:00+00:00).
///
/// ## Errors
///
/// This function will return [TimestampError::DateBeforeCcsdsEpoch] or
/// [TimestampError::Cds] if the time is before the CCSDS epoch (1958-01-01T00:00:00+00:00)
/// or the CCSDS days value exceeds the allowed bit width (24 bits).
pub fn from_unix_stamp_with_u16_days(
unix_stamp: &UnixTime,
submillis_prec: SubmillisPrecision,
) -> Result<Self, CdsError> {
Self::from_unix_generic(unix_stamp, LengthOfDaySegment::Short16Bits, submillis_prec)
}
/// Like [Self::from_dt_with_u16_days] but with microsecond sub-millisecond precision.
#[cfg(feature = "chrono")]
pub fn from_dt_with_u16_days_us_precision(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, CdsError> {
Self::from_dt_generic_us_prec(dt, LengthOfDaySegment::Short16Bits)
}
/// Like [Self::from_dt_with_u16_days] but with picoseconds sub-millisecond precision.
#[cfg(feature = "chrono")]
pub fn from_dt_with_u16_days_ps_precision(
dt: &chrono::DateTime<chrono::Utc>,
) -> Result<Self, CdsError> {
Self::from_dt_generic_ps_prec(dt, LengthOfDaySegment::Short16Bits)
}
/// Like [Self::from_now_with_u16_days] but with microsecond sub-millisecond precision.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u16_days_us_precision() -> Result<Self, StdTimestampError> {
Self::from_now_generic_us_prec(LengthOfDaySegment::Short16Bits)
}
/// Like [Self::from_now_with_u16_days] but with picosecond sub-millisecond precision.
#[cfg(feature = "std")]
#[cfg_attr(doc_cfg, doc(cfg(feature = "std")))]
pub fn from_now_with_u16_days_ps_precision() -> Result<Self, StdTimestampError> {
Self::from_now_generic_ps_prec(LengthOfDaySegment::Short16Bits)
}
pub fn from_bytes_with_u16_days(buf: &[u8]) -> Result<Self, TimestampError> {
let submillis_precision =
Self::generic_raw_read_checks(buf, LengthOfDaySegment::Short16Bits)?;
let ccsds_days: u16 = u16::from_be_bytes(buf[1..3].try_into().unwrap());
let ms_of_day: u32 = u32::from_be_bytes(buf[3..7].try_into().unwrap());
let mut provider = Self::new_with_u16_days(ccsds_days, ms_of_day);
provider.pfield = buf[0];
match submillis_precision {
SubmillisPrecision::Microseconds => {
provider.set_submillis(
SubmillisPrecision::Microseconds,
u16::from_be_bytes(buf[7..9].try_into().unwrap()) as u32,
);
}
SubmillisPrecision::Picoseconds => {
provider.set_submillis(
SubmillisPrecision::Picoseconds,
u32::from_be_bytes(buf[7..11].try_into().unwrap()),
);
}
_ => (),
}
Ok(provider)
}
}
fn add_for_max_ccsds_days_val<T: ProvidesDaysLength>(
time_provider: &CdsTime<T>,
max_days_val: u32,
duration: Duration,
) -> (u32, u32, u32) {
let mut next_ccsds_days = time_provider.ccsds_days_as_u32();
let mut next_ms_of_day = time_provider.ms_of_day;
// Increment CCSDS days by a certain amount while also accounting for overflow.
let increment_days = |ccsds_days: &mut u32, days_inc: u32| {
let days_addition: u64 = *ccsds_days as u64 + days_inc as u64;
if days_addition > max_days_val as u64 {
*ccsds_days = (days_addition - max_days_val as u64) as u32;
} else {
*ccsds_days += days_inc;
}
};
// Increment MS of day by a certain amount while also accounting for overflow, where
// the new value exceeds the MS of a day.
let increment_ms_of_day = |ms_of_day: &mut u32, ms_inc: u32, ccsds_days: &mut u32| {
*ms_of_day += ms_inc;
if *ms_of_day >= MS_PER_DAY {
*ms_of_day -= MS_PER_DAY;
// Re-use existing closure to always amount for overflow.
increment_days(ccsds_days, 1);
}
};
let mut submillis = time_provider.submillis();
match time_provider.submillis_precision() {
SubmillisPrecision::Microseconds => {
let subsec_micros = duration.subsec_micros();
let subsec_millis = subsec_micros / 1000;
let submilli_micros = subsec_micros % 1000;
submillis += submilli_micros;
if submillis >= 1000 {
let carryover_us = submillis - 1000;
increment_ms_of_day(&mut next_ms_of_day, 1, &mut next_ccsds_days);
submillis = carryover_us;
}
increment_ms_of_day(&mut next_ms_of_day, subsec_millis, &mut next_ccsds_days);
}
SubmillisPrecision::Picoseconds => {
let subsec_nanos = duration.subsec_nanos();
let subsec_millis = subsec_nanos / 10_u32.pow(6);
// 1 ms as ns is 1e6.
let submilli_nanos = subsec_nanos % 10_u32.pow(6);
// No overflow risk: The maximum value of an u32 is ~4.294e9, and one ms as ps
// is 1e9. The amount ps can now have is always less than 2e9.
submillis += submilli_nanos * 1000;
if submillis >= 10_u32.pow(9) {
let carry_over_ps = submillis - 10_u32.pow(9);
increment_ms_of_day(&mut next_ms_of_day, 1, &mut next_ccsds_days);
submillis = carry_over_ps;
}
increment_ms_of_day(&mut next_ms_of_day, subsec_millis, &mut next_ccsds_days);
}
_ => {
increment_ms_of_day(
&mut next_ms_of_day,
duration.subsec_millis(),
&mut next_ccsds_days,
);
}
}
// The subsecond millisecond were already handled.
let full_seconds = duration.as_secs();
let secs_of_day = (full_seconds % SECONDS_PER_DAY as u64) as u32;
let ms_of_day = secs_of_day * 1000;
increment_ms_of_day(&mut next_ms_of_day, ms_of_day, &mut next_ccsds_days);
increment_days(
&mut next_ccsds_days,
(full_seconds as u32 - secs_of_day) / SECONDS_PER_DAY,
);
(next_ccsds_days, next_ms_of_day, submillis)
}
impl CdsTimestamp for CdsTime<DaysLen16Bits> {
fn len_of_day_seg(&self) -> LengthOfDaySegment {
LengthOfDaySegment::Short16Bits
}
}
impl CdsTimestamp for CdsTime<DaysLen24Bits> {
fn len_of_day_seg(&self) -> LengthOfDaySegment {
LengthOfDaySegment::Long24Bits
}
}
/// Allows adding an duration in form of an offset. Please note that the CCSDS days will rollover
/// when they overflow, because addition needs to be infallible. The user needs to check for a
/// days overflow when this is a possibility and might be a problem.
impl Add<Duration> for CdsTime<DaysLen16Bits> {
type Output = Self;
fn add(self, duration: Duration) -> Self::Output {
let (next_ccsds_days, next_ms_of_day, precision) =
add_for_max_ccsds_days_val(&self, u16::MAX as u32, duration);
let mut provider = Self::new_with_u16_days(next_ccsds_days as u16, next_ms_of_day);
provider.set_submillis(self.submillis_precision(), precision);
provider
}
}
impl Add<Duration> for &CdsTime<DaysLen16Bits> {
type Output = CdsTime<DaysLen16Bits>;
fn add(self, duration: Duration) -> Self::Output {
let (next_ccsds_days, next_ms_of_day, precision) =
add_for_max_ccsds_days_val(self, u16::MAX as u32, duration);
let mut provider = Self::Output::new_with_u16_days(next_ccsds_days as u16, next_ms_of_day);
provider.set_submillis(self.submillis_precision(), precision);
provider
}
}
/// Allows adding an duration in form of an offset. Please note that the CCSDS days will rollover
/// when they overflow, because addition needs to be infallible. The user needs to check for a
/// days overflow when this is a possibility and might be a problem.
impl Add<Duration> for CdsTime<DaysLen24Bits> {
type Output = Self;
fn add(self, duration: Duration) -> Self::Output {
let (next_ccsds_days, next_ms_of_day, precision) =
add_for_max_ccsds_days_val(&self, MAX_DAYS_24_BITS, duration);
let mut provider = Self::new_with_u24_days(next_ccsds_days, next_ms_of_day).unwrap();
provider.set_submillis(self.submillis_precision(), precision);
provider
}
}
impl Add<Duration> for &CdsTime<DaysLen24Bits> {
type Output = CdsTime<DaysLen24Bits>;
fn add(self, duration: Duration) -> Self::Output {
let (next_ccsds_days, next_ms_of_day, precision) =
add_for_max_ccsds_days_val(self, MAX_DAYS_24_BITS, duration);
let mut provider =
Self::Output::new_with_u24_days(next_ccsds_days, next_ms_of_day).unwrap();
provider.set_submillis(self.submillis_precision(), precision);
provider
}
}
/// Allows adding an duration in form of an offset. Please note that the CCSDS days will rollover
/// when they overflow, because addition needs to be infallible. The user needs to check for a
/// days overflow when this is a possibility and might be a problem.
impl AddAssign<Duration> for CdsTime<DaysLen16Bits> {
fn add_assign(&mut self, duration: Duration) {
let (next_ccsds_days, next_ms_of_day, submillis) =
add_for_max_ccsds_days_val(self, u16::MAX as u32, duration);
self.ccsds_days = next_ccsds_days as u16;
self.ms_of_day = next_ms_of_day;
self.submillis = submillis;
}
}
/// Allows adding an duration in form of an offset. Please note that the CCSDS days will rollover
/// when they overflow, because addition needs to be infallible. The user needs to check for a
/// days overflow when this is a possibility and might be a problem.
impl AddAssign<Duration> for CdsTime<DaysLen24Bits> {
fn add_assign(&mut self, duration: Duration) {
let (next_ccsds_days, next_ms_of_day, submillis) =
add_for_max_ccsds_days_val(self, MAX_DAYS_24_BITS, duration);
self.ccsds_days = next_ccsds_days;
self.ms_of_day = next_ms_of_day;
self.submillis = submillis;
}
}
#[cfg(feature = "chrono")]
impl TryFrom<chrono::DateTime<chrono::Utc>> for CdsTime<DaysLen16Bits> {
type Error = CdsError;
fn try_from(dt: chrono::DateTime<chrono::Utc>) -> Result<Self, Self::Error> {
let conversion = ConversionFromChronoDatetime::new(&dt)?;
Self::generic_from_conversion(LengthOfDaySegment::Short16Bits, conversion)
}
}
#[cfg(feature = "chrono")]
impl TryFrom<chrono::DateTime<chrono::Utc>> for CdsTime<DaysLen24Bits> {
type Error = CdsError;
fn try_from(dt: chrono::DateTime<chrono::Utc>) -> Result<Self, Self::Error> {
let conversion = ConversionFromChronoDatetime::new(&dt)?;
Self::generic_from_conversion(LengthOfDaySegment::Long24Bits, conversion)
}
}
impl<ProvidesDaysLen: ProvidesDaysLength> CcsdsTimeProvider for CdsTime<ProvidesDaysLen> {
fn len_as_bytes(&self) -> usize {
Self::calc_stamp_len(self.pfield)
}
fn p_field(&self) -> (usize, [u8; 2]) {
(1, [self.pfield, 0])
}
fn ccdsd_time_code(&self) -> CcsdsTimeCode {
CcsdsTimeCode::Cds
}
#[inline]
fn unix_secs(&self) -> i64 {
self.unix_time.secs
}
#[inline]
fn subsec_nanos(&self) -> u32 {
self.unix_time.subsec_nanos
}
#[inline]
fn unix_time(&self) -> UnixTime {
self.unix_time
}
}
impl TimeReader for CdsTime<DaysLen16Bits> {
fn from_bytes(buf: &[u8]) -> Result<Self, TimestampError> {
Self::from_bytes_with_u16_days(buf)
}
}
impl TimeReader for CdsTime<DaysLen24Bits> {
fn from_bytes(buf: &[u8]) -> Result<Self, TimestampError> {
Self::from_bytes_with_u24_days(buf)
}
}
impl TimeWriter for CdsTime<DaysLen16Bits> {
fn write_to_bytes(&self, buf: &mut [u8]) -> Result<usize, TimestampError> {
self.length_check(buf, self.len_as_bytes())?;
buf[0] = self.pfield;
buf[1..3].copy_from_slice(self.ccsds_days.to_be_bytes().as_slice());
buf[3..7].copy_from_slice(self.ms_of_day.to_be_bytes().as_slice());
match self.submillis_precision() {
SubmillisPrecision::Microseconds => {
buf[7..9].copy_from_slice((self.submillis() as u16).to_be_bytes().as_slice());
}
SubmillisPrecision::Picoseconds => {
buf[7..11].copy_from_slice(self.submillis().to_be_bytes().as_slice());
}
_ => (),
}
Ok(self.len_as_bytes())
}
fn len_written(&self) -> usize {
self.len_as_bytes()
}
}
impl TimeWriter for CdsTime<DaysLen24Bits> {
fn write_to_bytes(&self, buf: &mut [u8]) -> Result<usize, TimestampError> {
self.length_check(buf, self.len_as_bytes())?;
buf[0] = self.pfield;
let be_days = self.ccsds_days.to_be_bytes();
buf[1..4].copy_from_slice(&be_days[1..4]);
buf[4..8].copy_from_slice(self.ms_of_day.to_be_bytes().as_slice());
match self.submillis_precision() {
SubmillisPrecision::Microseconds => {
buf[8..10].copy_from_slice((self.submillis() as u16).to_be_bytes().as_slice());
}
SubmillisPrecision::Picoseconds => {
buf[8..12].copy_from_slice(self.submillis().to_be_bytes().as_slice());
}
_ => (),
}
Ok(self.len_as_bytes())
}
fn len_written(&self) -> usize {
self.len_as_bytes()
}
}
impl<DaysLenProvider: ProvidesDaysLength> PartialEq for CdsTime<DaysLenProvider> {
fn eq(&self, other: &Self) -> bool {
if self.ccsds_days == other.ccsds_days
&& self.ms_of_day == other.ms_of_day
&& self.precision_as_ns().unwrap_or(0) == other.precision_as_ns().unwrap_or(0)
{
return true;
}
false
}
}
impl<DaysLenProvider: ProvidesDaysLength> PartialOrd for CdsTime<DaysLenProvider> {
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
if self == other {
return Some(Ordering::Equal);
}
match self.ccsds_days_as_u32().cmp(&other.ccsds_days_as_u32()) {
Ordering::Less => return Some(Ordering::Less),
Ordering::Greater => return Some(Ordering::Greater),
_ => (),
}
match self.ms_of_day().cmp(&other.ms_of_day()) {
Ordering::Less => return Some(Ordering::Less),
Ordering::Greater => return Some(Ordering::Greater),
_ => (),
}
match self
.precision_as_ns()
.unwrap_or(0)
.cmp(&other.precision_as_ns().unwrap_or(0))
{
Ordering::Less => return Some(Ordering::Less),
Ordering::Greater => return Some(Ordering::Greater),
_ => (),
}
Some(Ordering::Equal)
}
}
impl<DaysLenProvider: ProvidesDaysLength + Eq> Ord for CdsTime<DaysLenProvider> {
fn cmp(&self, other: &Self) -> Ordering {
PartialOrd::partial_cmp(self, other).unwrap()
}
}
impl From<CdsTime<DaysLen16Bits>> for CdsTime<DaysLen24Bits> {
fn from(value: CdsTime<DaysLen16Bits>) -> Self {
// This function only fails if the days value exceeds 24 bits, which is not possible here,
// so it is okay to unwrap.
Self::new_with_u24_days(value.ccsds_days_as_u32(), value.ms_of_day()).unwrap()
}
}
/// This conversion can fail if the days value exceeds 16 bits.
impl TryFrom<CdsTime<DaysLen24Bits>> for CdsTime<DaysLen16Bits> {
type Error = CdsError;
fn try_from(value: CdsTime<DaysLen24Bits>) -> Result<Self, CdsError> {
let ccsds_days = value.ccsds_days_as_u32();
if ccsds_days > u16::MAX as u32 {
return Err(CdsError::InvalidCcsdsDays(ccsds_days as i64));
}
Ok(Self::new_with_u16_days(
ccsds_days as u16,
value.ms_of_day(),
))
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::time::TimestampError::{ByteConversion, InvalidTimeCode};
use crate::time::{UnixTime, DAYS_CCSDS_TO_UNIX, MS_PER_DAY};
use crate::ByteConversionError::{FromSliceTooSmall, ToSliceTooSmall};
use alloc::string::ToString;
use chrono::{Datelike, NaiveDate, Timelike};
#[cfg(feature = "serde")]
use postcard::{from_bytes, to_allocvec};
use std::format;
#[test]
fn test_time_stamp_zero_args() {
let time_stamper = CdsTime::new_with_u16_days(0, 0);
let unix_stamp = time_stamper.unix_time();
assert_eq!(
unix_stamp.secs,
(DAYS_CCSDS_TO_UNIX * SECONDS_PER_DAY as i32) as i64
);
let subsecond_millis = unix_stamp.subsec_nanos;
assert_eq!(subsecond_millis, 0);
assert_eq!(
time_stamper.submillis_precision(),
SubmillisPrecision::Absent
);
assert_eq!(time_stamper.subsec_nanos(), 0);
assert_eq!(time_stamper.ccdsd_time_code(), CcsdsTimeCode::Cds);
assert_eq!(
time_stamper.p_field(),
(1, [(CcsdsTimeCode::Cds as u8) << 4, 0])
);
let date_time = time_stamper.chrono_date_time().unwrap();
assert_eq!(date_time.year(), 1958);
assert_eq!(date_time.month(), 1);
assert_eq!(date_time.day(), 1);
assert_eq!(date_time.hour(), 0);
assert_eq!(date_time.minute(), 0);
assert_eq!(date_time.second(), 0);
}
#[test]
fn test_time_stamp_unix_epoch() {
let time_stamper = CdsTime::new_with_u16_days((-DAYS_CCSDS_TO_UNIX) as u16, 0);
assert_eq!(time_stamper.unix_time().secs, 0);
assert_eq!(
time_stamper.submillis_precision(),
SubmillisPrecision::Absent
);
let date_time = time_stamper.chrono_date_time().unwrap();
assert_eq!(date_time.year(), 1970);
assert_eq!(date_time.month(), 1);
assert_eq!(date_time.day(), 1);
assert_eq!(date_time.hour(), 0);
assert_eq!(date_time.minute(), 0);
assert_eq!(date_time.second(), 0);
let time_stamper = CdsTime::new_with_u16_days((-DAYS_CCSDS_TO_UNIX) as u16, 40);
assert_eq!(time_stamper.subsec_nanos(), 40 * 10_u32.pow(6));
assert_eq!(time_stamper.subsec_millis(), 40);
let time_stamper = CdsTime::new_with_u16_days((-DAYS_CCSDS_TO_UNIX) as u16, 1040);
assert_eq!(time_stamper.subsec_nanos(), 40 * 10_u32.pow(6));
assert_eq!(time_stamper.subsec_millis(), 40);
}
#[test]
fn test_large_days_field_write() {
let time_stamper = CdsTime::new_with_u24_days(0x108020_u32, 0x10203040);
assert!(time_stamper.is_ok());
let time_stamper = time_stamper.unwrap();
assert_eq!(time_stamper.len_as_bytes(), 8);
let mut buf = [0; 16];
let written = time_stamper.write_to_bytes(&mut buf);
assert!(written.is_ok());
let written = written.unwrap();
assert_eq!(written, 8);
assert_eq!(buf[1], 0x10);
assert_eq!(buf[2], 0x80);
assert_eq!(buf[3], 0x20);
let ms = u32::from_be_bytes(buf[4..8].try_into().unwrap());
assert_eq!(ms, 0x10203040);
assert_eq!((buf[0] >> 2) & 0b1, 1);
}
#[test]
fn test_large_days_field_read() {
let time_stamper = CdsTime::new_with_u24_days(0x108020_u32, 0);
assert!(time_stamper.is_ok());
let time_stamper = time_stamper.unwrap();
let mut buf = [0; 16];
let written = time_stamper.write_to_bytes(&mut buf);
assert!(written.is_ok());
let provider = CdsTime::<DaysLen24Bits>::from_bytes(&buf);
assert!(provider.is_ok());
let provider = provider.unwrap();
assert_eq!(provider.ccsds_days(), 0x108020);
assert_eq!(provider.ms_of_day(), 0);
}
#[test]
fn test_large_days_field_read_invalid_ctor() {
let time_stamper = CdsTime::new_with_u24_days(0x108020, 0);
assert!(time_stamper.is_ok());
let time_stamper = time_stamper.unwrap();
let mut buf = [0; 16];
let written = time_stamper.write_to_bytes(&mut buf);
assert!(written.is_ok());
let faulty_ctor = CdsTime::<DaysLen16Bits>::from_bytes(&buf);
assert!(faulty_ctor.is_err());
let error = faulty_ctor.unwrap_err();
if let TimestampError::Cds(CdsError::InvalidCtorForDaysOfLenInPreamble(len_of_day)) = error
{
assert_eq!(len_of_day, LengthOfDaySegment::Long24Bits);
} else {
panic!("Wrong error type");
}
}
#[test]
fn test_write() {
let mut buf = [0; 16];
let time_stamper_0 = CdsTime::new_with_u16_days(0, 0);
let unix_stamp = time_stamper_0.unix_time();
assert_eq!(
unix_stamp.secs,
(DAYS_CCSDS_TO_UNIX * SECONDS_PER_DAY as i32).into()
);
let mut res = time_stamper_0.write_to_bytes(&mut buf);
assert!(res.is_ok());
assert_eq!(buf[0], (CcsdsTimeCode::Cds as u8) << 4);
assert_eq!(
u16::from_be_bytes(buf[1..3].try_into().expect("Byte conversion failed")),
0
);
assert_eq!(
u32::from_be_bytes(buf[3..7].try_into().expect("Byte conversion failed")),
0
);
let time_stamper_1 = CdsTime::new_with_u16_days(u16::MAX - 1, u32::MAX - 1);
res = time_stamper_1.write_to_bytes(&mut buf);
assert!(res.is_ok());
assert_eq!(buf[0], (CcsdsTimeCode::Cds as u8) << 4);
assert_eq!(
u16::from_be_bytes(buf[1..3].try_into().expect("Byte conversion failed")),
u16::MAX - 1
);
assert_eq!(
u32::from_be_bytes(buf[3..7].try_into().expect("Byte conversion failed")),
u32::MAX - 1
);
}
#[test]
fn test_faulty_write_buf_too_small() {
let mut buf = [0; 7];
let time_stamper = CdsTime::new_with_u16_days(u16::MAX - 1, u32::MAX - 1);
for i in 0..6 {
let res = time_stamper.write_to_bytes(&mut buf[0..i]);
assert!(res.is_err());
let error = res.unwrap_err();
match error {
ByteConversion(ToSliceTooSmall { found, expected }) => {
assert_eq!(found, i);
assert_eq!(expected, 7);
assert_eq!(
error.to_string(),
format!("time stamp: target slice with size {i} is too small, expected size of at least 7")
);
}
_ => panic!(
"{}",
format!("Invalid error {:?} detected", res.unwrap_err())
),
}
}
}
#[test]
fn test_faulty_read_buf_too_small() {
let buf = [0; 7];
for i in 0..6 {
let res = CdsTime::<DaysLen16Bits>::from_bytes(&buf[0..i]);
assert!(res.is_err());
let err = res.unwrap_err();
match err {
ByteConversion(e) => match e {
FromSliceTooSmall { found, expected } => {
assert_eq!(found, i);
assert_eq!(expected, 7);
}
_ => panic!("{}", format!("Invalid error {:?} detected", e)),
},
_ => {
panic!("Unexpected error {:?}", err);
}
}
}
}
#[test]
fn test_faulty_invalid_pfield() {
let mut buf = [0; 16];
let time_stamper_0 = CdsTime::new_with_u16_days(0, 0);
let res = time_stamper_0.write_to_bytes(&mut buf);
assert!(res.is_ok());
buf[0] = 0;
let res = CdsTime::<DaysLen16Bits>::from_bytes(&buf);
assert!(res.is_err());
let err = res.unwrap_err();
if let InvalidTimeCode { expected, found } = err {
assert_eq!(expected, CcsdsTimeCode::Cds);
assert_eq!(found, 0);
assert_eq!(
err.to_string(),
"invalid raw time code value 0 for time code Cds"
);
}
}
#[test]
fn test_reading() {
let mut buf = [0; 16];
let time_stamper = CdsTime::new_with_u16_days(u16::MAX - 1, u32::MAX - 1);
let res = time_stamper.write_to_bytes(&mut buf);
assert!(res.is_ok());
assert_eq!(buf[0], (CcsdsTimeCode::Cds as u8) << 4);
assert_eq!(
u16::from_be_bytes(buf[1..3].try_into().expect("Byte conversion failed")),
u16::MAX - 1
);
assert_eq!(
u32::from_be_bytes(buf[3..7].try_into().expect("Byte conversion failed")),
u32::MAX - 1
);
let read_stamp: CdsTime<DaysLen16Bits> =
CdsTime::from_bytes(&buf).expect("Reading timestamp failed");
assert_eq!(read_stamp.ccsds_days(), u16::MAX - 1);
assert_eq!(read_stamp.ms_of_day(), u32::MAX - 1);
}
fn generic_now_test<T: ProvidesDaysLength>(
timestamp_now: CdsTime<T>,
compare_stamp: chrono::DateTime<chrono::Utc>,
) {
let dt = timestamp_now.chrono_date_time().unwrap();
if compare_stamp.year() > dt.year() {
assert_eq!(compare_stamp.year() - dt.year(), 1);
} else {
assert_eq!(dt.year(), compare_stamp.year());
}
generic_dt_property_equality_check(dt.month(), compare_stamp.month(), 1, 12);
assert_eq!(dt.day(), compare_stamp.day());
if compare_stamp.day() < dt.day() {
assert!(dt.day() >= 28);
assert_eq!(compare_stamp.day(), 1);
} else if compare_stamp.day() > dt.day() {
assert_eq!(compare_stamp.day() - dt.day(), 1);
} else {
assert_eq!(compare_stamp.day(), dt.day());
}
generic_dt_property_equality_check(dt.hour(), compare_stamp.hour(), 0, 23);
generic_dt_property_equality_check(dt.minute(), compare_stamp.minute(), 0, 59);
}
#[test]
fn test_time_now() {
let timestamp_now = CdsTime::from_now_with_u16_days().unwrap();
let compare_stamp = chrono::Utc::now();
generic_now_test(timestamp_now, compare_stamp);
}
#[test]
fn test_time_now_us_prec() {
let timestamp_now = CdsTime::from_now_with_u16_days_us_precision().unwrap();
let compare_stamp = chrono::Utc::now();
generic_now_test(timestamp_now, compare_stamp);
}
#[test]
fn test_time_now_ps_prec() {
let timestamp_now = CdsTime::from_now_with_u16_days_ps_precision().unwrap();
let compare_stamp = chrono::Utc::now();
generic_now_test(timestamp_now, compare_stamp);
}
#[test]
fn test_time_now_ps_prec_u16_days() {
let timestamp_now = CdsTime::from_now_with_u16_days_ps_precision().unwrap();
let compare_stamp = chrono::Utc::now();
generic_now_test(timestamp_now, compare_stamp);
}
#[test]
fn test_time_now_ps_prec_u24_days() {
let timestamp_now = CdsTime::from_now_with_u24_days_ps_precision().unwrap();
let compare_stamp = chrono::Utc::now();
generic_now_test(timestamp_now, compare_stamp);
}
#[test]
fn test_submillis_precision_micros() {
let mut time_stamper = CdsTime::new_with_u16_days(0, 0);
time_stamper.set_submillis(SubmillisPrecision::Microseconds, 500);
assert_eq!(
time_stamper.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(time_stamper.submillis(), 500);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
assert_eq!(written, 9);
let cross_check: u16 = 500;
assert_eq!(write_buf[7..9], cross_check.to_be_bytes());
}
#[test]
fn test_submillis_precision_picos() {
let mut time_stamper = CdsTime::new_with_u16_days(0, 0);
time_stamper.set_submillis(SubmillisPrecision::Picoseconds, 5e8 as u32);
assert_eq!(
time_stamper.submillis_precision(),
SubmillisPrecision::Picoseconds
);
assert_eq!(time_stamper.submillis(), 5e8 as u32);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
assert_eq!(written, 11);
let cross_check: u32 = 5e8 as u32;
assert_eq!(write_buf[7..11], cross_check.to_be_bytes());
}
#[test]
fn read_stamp_with_ps_submillis_precision() {
let mut time_stamper = CdsTime::new_with_u16_days(0, 0);
time_stamper.set_submillis(SubmillisPrecision::Picoseconds, 5e8 as u32);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
assert_eq!(written, 11);
let stamp_deserialized = CdsTime::<DaysLen16Bits>::from_bytes(&write_buf);
assert!(stamp_deserialized.is_ok());
let stamp_deserialized = stamp_deserialized.unwrap();
assert_eq!(stamp_deserialized.len_as_bytes(), 11);
assert_eq!(
stamp_deserialized.submillis_precision(),
SubmillisPrecision::Picoseconds
);
assert_eq!(stamp_deserialized.submillis(), 5e8 as u32);
}
#[test]
fn read_stamp_with_us_submillis_precision() {
let mut time_stamper = CdsTime::new_with_u16_days(0, 0);
time_stamper.set_submillis(SubmillisPrecision::Microseconds, 500);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
assert_eq!(written, 9);
let stamp_deserialized = CdsTime::<DaysLen16Bits>::from_bytes(&write_buf);
assert!(stamp_deserialized.is_ok());
let stamp_deserialized = stamp_deserialized.unwrap();
assert_eq!(stamp_deserialized.len_as_bytes(), 9);
assert_eq!(
stamp_deserialized.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(stamp_deserialized.submillis(), 500);
}
#[test]
fn read_u24_stamp_with_us_submillis_precision() {
let mut time_stamper = CdsTime::new_with_u24_days(u16::MAX as u32 + 1, 0).unwrap();
time_stamper.set_submillis(SubmillisPrecision::Microseconds, 500);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
// 1 byte pfield + 3 bytes days + 4 bytes ms of day + 2 bytes us precision
assert_eq!(written, 10);
let stamp_deserialized = CdsTime::from_bytes_with_u24_days(&write_buf);
assert!(stamp_deserialized.is_ok());
let stamp_deserialized = stamp_deserialized.unwrap();
assert_eq!(stamp_deserialized.len_as_bytes(), 10);
assert_eq!(stamp_deserialized.ccsds_days(), u16::MAX as u32 + 1);
assert_eq!(
stamp_deserialized.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(stamp_deserialized.submillis(), 500);
}
#[test]
fn read_u24_stamp_with_ps_submillis_precision() {
let mut time_stamper = CdsTime::new_with_u24_days(u16::MAX as u32 + 1, 0).unwrap();
time_stamper.set_submillis(SubmillisPrecision::Picoseconds, 5e8 as u32);
let mut write_buf: [u8; 16] = [0; 16];
let written = time_stamper
.write_to_bytes(&mut write_buf)
.expect("Writing timestamp failed");
// 1 byte pfield + 3 bytes days + 4 bytes ms of day + 4 bytes us precision
assert_eq!(written, 12);
let stamp_deserialized = CdsTime::from_bytes_with_u24_days(&write_buf);
assert!(stamp_deserialized.is_ok());
let stamp_deserialized = stamp_deserialized.unwrap();
assert_eq!(stamp_deserialized.len_as_bytes(), 12);
assert_eq!(stamp_deserialized.ccsds_days(), u16::MAX as u32 + 1);
assert_eq!(
stamp_deserialized.submillis_precision(),
SubmillisPrecision::Picoseconds
);
assert_eq!(stamp_deserialized.submillis(), 5e8 as u32);
}
fn generic_dt_case_0_no_prec(subsec_millis: u32) -> chrono::DateTime<chrono::Utc> {
NaiveDate::from_ymd_opt(2023, 1, 14)
.unwrap()
.and_hms_milli_opt(16, 49, 30, subsec_millis)
.unwrap()
.and_local_timezone(chrono::Utc)
.unwrap()
}
fn generic_check_dt_case_0<DaysLen: ProvidesDaysLength>(
time_provider: &CdsTime<DaysLen>,
subsec_millis: u32,
datetime_utc: chrono::DateTime<chrono::Utc>,
) {
// https://www.timeanddate.com/date/durationresult.html?d1=01&m1=01&y1=1958&d2=14&m2=01&y2=2023
// Leap years need to be accounted for as well.
assert_eq!(time_provider.ccsds_days, 23754.into());
assert_eq!(
time_provider.ms_of_day,
30 * 1000 + 49 * 60 * 1000 + 16 * 60 * 60 * 1000 + subsec_millis
);
assert_eq!(time_provider.chrono_date_time().unwrap(), datetime_utc);
}
#[test]
fn test_creation_from_dt_u16_days() {
let subsec_millis = 250;
let datetime_utc = generic_dt_case_0_no_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u16_days(&datetime_utc).unwrap();
generic_check_dt_case_0(&time_provider, subsec_millis, datetime_utc);
let time_provider_2: CdsTime<DaysLen16Bits> =
datetime_utc.try_into().expect("conversion failed");
// Test the TryInto trait impl
assert_eq!(time_provider, time_provider_2);
}
#[test]
fn test_creation_from_dt_u24_days() {
let subsec_millis = 250;
let datetime_utc = generic_dt_case_0_no_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u24_days(&datetime_utc).unwrap();
generic_check_dt_case_0(&time_provider, subsec_millis, datetime_utc);
let time_provider_2: CdsTime<DaysLen24Bits> =
datetime_utc.try_into().expect("conversion failed");
// Test the TryInto trait impl
assert_eq!(time_provider, time_provider_2);
}
fn generic_dt_case_1_us_prec(subsec_millis: u32) -> chrono::DateTime<chrono::Utc> {
// 250 ms + 500 us
let subsec_micros = subsec_millis * 1000 + 500;
NaiveDate::from_ymd_opt(2023, 1, 14)
.unwrap()
.and_hms_micro_opt(16, 49, 30, subsec_micros)
.unwrap()
.and_local_timezone(chrono::Utc)
.unwrap()
}
fn generic_check_dt_case_1_us_prec<DaysLen: ProvidesDaysLength>(
time_provider: &CdsTime<DaysLen>,
subsec_millis: u32,
datetime_utc: chrono::DateTime<chrono::Utc>,
) {
// https://www.timeanddate.com/date/durationresult.html?d1=01&m1=01&y1=1958&d2=14&m2=01&y2=2023
// Leap years need to be accounted for as well.
assert_eq!(time_provider.ccsds_days, 23754.into());
assert_eq!(
time_provider.ms_of_day,
30 * 1000 + 49 * 60 * 1000 + 16 * 60 * 60 * 1000 + subsec_millis
);
assert_eq!(
time_provider.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(time_provider.submillis(), 500);
assert_eq!(time_provider.chrono_date_time().unwrap(), datetime_utc);
}
#[test]
fn test_creation_from_dt_u16_days_us_prec() {
let subsec_millis = 250;
let datetime_utc = generic_dt_case_1_us_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u16_days_us_precision(&datetime_utc).unwrap();
generic_check_dt_case_1_us_prec(&time_provider, subsec_millis, datetime_utc);
}
#[test]
fn test_creation_from_dt_u24_days_us_prec() {
let subsec_millis = 250;
let datetime_utc = generic_dt_case_1_us_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u24_days_us_precision(&datetime_utc).unwrap();
generic_check_dt_case_1_us_prec(&time_provider, subsec_millis, datetime_utc);
}
fn generic_dt_case_2_ps_prec(subsec_millis: u32) -> (chrono::DateTime<chrono::Utc>, u32) {
// 250 ms + 500 us
let subsec_nanos = subsec_millis * 1000 * 1000 + 500 * 1000;
let submilli_nanos = subsec_nanos % 10_u32.pow(6);
(
NaiveDate::from_ymd_opt(2023, 1, 14)
.unwrap()
.and_hms_nano_opt(16, 49, 30, subsec_nanos)
.unwrap()
.and_local_timezone(chrono::Utc)
.unwrap(),
submilli_nanos,
)
}
fn generic_check_dt_case_2_ps_prec<DaysLen: ProvidesDaysLength>(
time_provider: &CdsTime<DaysLen>,
subsec_millis: u32,
submilli_nanos: u32,
datetime_utc: chrono::DateTime<chrono::Utc>,
) {
// https://www.timeanddate.com/date/durationresult.html?d1=01&m1=01&y1=1958&d2=14&m2=01&y2=2023
// Leap years need to be accounted for as well.
assert_eq!(time_provider.ccsds_days, 23754.into());
assert_eq!(
time_provider.ms_of_day,
30 * 1000 + 49 * 60 * 1000 + 16 * 60 * 60 * 1000 + subsec_millis
);
assert_eq!(
time_provider.submillis_precision(),
SubmillisPrecision::Picoseconds
);
assert_eq!(time_provider.submillis(), submilli_nanos * 1000);
assert_eq!(time_provider.chrono_date_time().unwrap(), datetime_utc);
}
#[test]
fn test_creation_from_dt_u16_days_ps_prec() {
let subsec_millis = 250;
let (datetime_utc, submilli_nanos) = generic_dt_case_2_ps_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u16_days_ps_precision(&datetime_utc).unwrap();
generic_check_dt_case_2_ps_prec(
&time_provider,
subsec_millis,
submilli_nanos,
datetime_utc,
);
}
#[test]
fn test_creation_from_dt_u24_days_ps_prec() {
let subsec_millis = 250;
let (datetime_utc, submilli_nanos) = generic_dt_case_2_ps_prec(subsec_millis);
let time_provider = CdsTime::from_dt_with_u24_days_ps_precision(&datetime_utc).unwrap();
generic_check_dt_case_2_ps_prec(
&time_provider,
subsec_millis,
submilli_nanos,
datetime_utc,
);
}
#[test]
fn test_creation_from_unix_stamp_0_u16_days() {
let unix_secs = 0;
let subsec_millis = 0;
let time_provider = CdsTime::from_unix_stamp_with_u16_days(
&UnixTime::new(unix_secs, subsec_millis),
SubmillisPrecision::Absent,
)
.expect("creating provider from unix stamp failed");
assert_eq!(time_provider.ccsds_days, -DAYS_CCSDS_TO_UNIX as u16)
}
#[test]
fn test_creation_from_unix_stamp_0_u24_days() {
let unix_secs = 0;
let subsec_millis = 0;
let time_provider = CdsTime::from_unix_stamp_with_u24_days(
&UnixTime::new(unix_secs, subsec_millis),
SubmillisPrecision::Absent,
)
.expect("creating provider from unix stamp failed");
assert_eq!(time_provider.ccsds_days, (-DAYS_CCSDS_TO_UNIX) as u32)
}
#[test]
fn test_creation_from_unix_stamp_1() {
let subsec_millis = 250;
let datetime_utc = NaiveDate::from_ymd_opt(2023, 1, 14)
.unwrap()
.and_hms_milli_opt(16, 49, 30, subsec_millis)
.unwrap()
.and_local_timezone(chrono::Utc)
.unwrap();
let time_provider = CdsTime::from_unix_stamp_with_u16_days(
&datetime_utc.into(),
SubmillisPrecision::Absent,
)
.expect("creating provider from unix stamp failed");
// https://www.timeanddate.com/date/durationresult.html?d1=01&m1=01&y1=1958&d2=14&m2=01&y2=2023
// Leap years need to be accounted for as well.
assert_eq!(time_provider.ccsds_days, 23754);
assert_eq!(
time_provider.ms_of_day,
30 * 1000 + 49 * 60 * 1000 + 16 * 60 * 60 * 1000 + subsec_millis
);
let dt_back = time_provider.chrono_date_time().unwrap();
assert_eq!(datetime_utc, dt_back);
}
#[test]
fn test_creation_0_ccsds_days() {
let unix_secs = DAYS_CCSDS_TO_UNIX as i64 * SECONDS_PER_DAY as i64;
let subsec_millis = 0;
let time_provider = CdsTime::from_unix_stamp_with_u16_days(
&UnixTime::new(unix_secs, subsec_millis),
SubmillisPrecision::Absent,
)
.expect("creating provider from unix stamp failed");
assert_eq!(time_provider.ccsds_days, 0)
}
#[test]
fn test_invalid_creation_from_unix_stamp_days_too_large() {
let invalid_unix_secs: i64 = (u16::MAX as i64 + 1) * SECONDS_PER_DAY as i64;
let subsec_millis = 0;
match CdsTime::from_unix_stamp_with_u16_days(
&UnixTime::new(invalid_unix_secs, subsec_millis),
SubmillisPrecision::Absent,
) {
Ok(_) => {
panic!("creation should not succeed")
}
Err(e) => {
if let CdsError::InvalidCcsdsDays(days) = e {
assert_eq!(
days,
unix_to_ccsds_days(invalid_unix_secs / SECONDS_PER_DAY as i64)
);
assert_eq!(e.to_string(), "invalid ccsds days 69919");
} else {
panic!("unexpected error {}", e)
}
}
}
}
#[test]
fn test_invalid_creation_from_unix_stamp_before_ccsds_epoch() {
// This is a unix stamp before the CCSDS epoch (01-01-1958 00:00:00), this should be
// precisely 31-12-1957 23:59:55
let unix_secs = DAYS_CCSDS_TO_UNIX * SECONDS_PER_DAY as i32 - 5;
let subsec_millis = 0;
match CdsTime::from_unix_stamp_with_u16_days(
&UnixTime::new(unix_secs as i64, subsec_millis),
SubmillisPrecision::Absent,
) {
Ok(_) => {
panic!("creation should not succeed")
}
Err(e) => {
if let CdsError::DateBeforeCcsdsEpoch(DateBeforeCcsdsEpochError(unix_dt)) = e {
let dt = unix_dt.chrono_date_time();
if let chrono::LocalResult::Single(dt) = dt {
assert_eq!(dt.year(), 1957);
assert_eq!(dt.month(), 12);
assert_eq!(dt.day(), 31);
assert_eq!(dt.hour(), 23);
assert_eq!(dt.minute(), 59);
assert_eq!(dt.second(), 55);
} else {
panic!("unexpected error {}", e)
}
} else {
panic!("unexpected error {}", e)
}
}
}
}
#[test]
fn test_addition_u16_days_day_increment() {
let mut provider = CdsTime::new_with_u16_days(0, MS_PER_DAY - 5 * 1000);
let seconds_offset = Duration::from_secs(10);
assert_eq!(provider.ccsds_days, 0);
assert_eq!(provider.ms_of_day, MS_PER_DAY - 5 * 1000);
provider += seconds_offset;
assert_eq!(provider.ccsds_days, 1);
assert_eq!(provider.ms_of_day, 5000);
}
#[test]
fn test_addition_u16_days() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
let seconds_offset = Duration::from_secs(5);
assert_eq!(provider.ccsds_days, 0);
assert_eq!(provider.ms_of_day, 0);
provider += seconds_offset;
assert_eq!(provider.ms_of_day, 5000);
// Add one day and test Add operator
let provider2 = provider + Duration::from_secs(60 * 60 * 24);
assert_eq!(provider2.ccsds_days, 1);
assert_eq!(provider2.ms_of_day, 5000);
}
#[test]
fn test_addition_u24_days() {
let mut provider = CdsTime::new_with_u24_days(u16::MAX as u32, 0).unwrap();
let seconds_offset = Duration::from_secs(5);
assert_eq!(provider.ccsds_days, u16::MAX as u32);
assert_eq!(provider.ms_of_day, 0);
provider += seconds_offset;
assert_eq!(provider.ms_of_day, 5000);
// Add one day and test Add operator
let provider2 = provider + Duration::from_secs(60 * 60 * 24);
assert_eq!(provider2.ccsds_days, u16::MAX as u32 + 1);
assert_eq!(provider2.ms_of_day, 5000);
}
#[test]
fn test_dyn_creation_u24_days() {
let stamp = CdsTime::new_with_u24_days(u16::MAX as u32 + 1, 24).unwrap();
let mut buf: [u8; 32] = [0; 32];
stamp.write_to_bytes(&mut buf).unwrap();
let dyn_provider = get_dyn_time_provider_from_bytes(&buf);
assert!(dyn_provider.is_ok());
let dyn_provider = dyn_provider.unwrap();
assert_eq!(dyn_provider.ccdsd_time_code(), CcsdsTimeCode::Cds);
assert_eq!(dyn_provider.ccsds_days_as_u32(), u16::MAX as u32 + 1);
assert_eq!(dyn_provider.ms_of_day(), 24);
assert_eq!(
dyn_provider.submillis_precision(),
SubmillisPrecision::Absent
);
assert_eq!(
dyn_provider.len_of_day_seg(),
LengthOfDaySegment::Long24Bits
);
}
#[test]
fn test_addition_with_us_precision_u16_days() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
provider.set_submillis(SubmillisPrecision::Microseconds, 0);
let duration = Duration::from_micros(500);
provider += duration;
assert_eq!(
provider.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(provider.submillis(), 500);
}
#[test]
fn test_addition_with_us_precision_u16_days_with_subsec_millis() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
provider.set_submillis(SubmillisPrecision::Microseconds, 0);
let duration = Duration::from_micros(1200);
provider += duration;
assert_eq!(
provider.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(provider.submillis(), 200);
assert_eq!(provider.ms_of_day(), 1);
}
#[test]
fn test_addition_with_us_precision_u16_days_carry_over() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
provider.set_submillis(SubmillisPrecision::Microseconds, 800);
let duration = Duration::from_micros(400);
provider += duration;
assert_eq!(
provider.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(provider.submillis(), 200);
assert_eq!(provider.ms_of_day(), 1);
}
#[test]
fn test_addition_with_ps_precision_u16_days() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
provider.set_submillis(SubmillisPrecision::Picoseconds, 0);
// 500 us as ns
let duration = Duration::from_nanos(500 * 10u32.pow(3) as u64);
provider += duration;
assert_eq!(
provider.submillis_precision(),
SubmillisPrecision::Picoseconds
);
assert_eq!(provider.submillis(), 500 * 10u32.pow(6));
}
#[test]
fn test_addition_on_ref() {
// This test case also tests the case where there is no submillis precision but subsecond
// milliseconds.
let provider_ref = &CdsTime::new_with_u16_days(2, 500);
let new_stamp = provider_ref + Duration::from_millis(2 * 24 * 60 * 60 * 1000 + 500);
assert_eq!(new_stamp.ccsds_days_as_u32(), 4);
assert_eq!(new_stamp.ms_of_day, 1000);
}
fn check_ps_and_carryover(prec: SubmillisPrecision, submillis: u32, ms_of_day: u32, val: u32) {
if prec == SubmillisPrecision::Picoseconds {
assert_eq!(submillis, val);
assert_eq!(ms_of_day, 1);
} else {
panic!("invalid precision {:?}", prec)
}
}
#[test]
fn test_addition_with_ps_precision_u16_days_with_subsec_millis() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
provider.set_submillis(SubmillisPrecision::Picoseconds, 0);
// 1200 us as ns
let duration = Duration::from_nanos(1200 * 10u32.pow(3) as u64);
provider += duration;
check_ps_and_carryover(
provider.submillis_precision(),
provider.submillis(),
provider.ms_of_day,
200 * 10_u32.pow(6),
);
}
#[test]
fn test_addition_with_ps_precision_u16_days_carryover() {
let mut provider = CdsTime::new_with_u16_days(0, 0);
// 800 us as ps
provider.set_submillis(SubmillisPrecision::Picoseconds, 800 * 10_u32.pow(6));
// 400 us as ns
let duration = Duration::from_nanos(400 * 10u32.pow(3) as u64);
provider += duration;
check_ps_and_carryover(
provider.submillis_precision(),
provider.submillis(),
provider.ms_of_day,
200 * 10_u32.pow(6),
);
}
#[test]
fn test_dyn_creation_u16_days_with_precision() {
let mut stamp = CdsTime::new_with_u16_days(24, 24);
stamp.set_submillis(SubmillisPrecision::Microseconds, 666);
let mut buf: [u8; 32] = [0; 32];
stamp.write_to_bytes(&mut buf).unwrap();
let dyn_provider = get_dyn_time_provider_from_bytes(&buf);
assert!(dyn_provider.is_ok());
let dyn_provider = dyn_provider.unwrap();
assert_eq!(dyn_provider.ccdsd_time_code(), CcsdsTimeCode::Cds);
assert_eq!(dyn_provider.ccsds_days_as_u32(), 24);
assert_eq!(dyn_provider.ms_of_day(), 24);
assert_eq!(
dyn_provider.len_of_day_seg(),
LengthOfDaySegment::Short16Bits
);
assert_eq!(
dyn_provider.submillis_precision(),
SubmillisPrecision::Microseconds
);
assert_eq!(dyn_provider.submillis(), 666);
}
#[test]
fn test_new_u24_days_too_large() {
let time_provider = CdsTime::new_with_u24_days(2_u32.pow(24), 0);
assert!(time_provider.is_err());
let e = time_provider.unwrap_err();
if let CdsError::InvalidCcsdsDays(days) = e {
assert_eq!(days, 2_u32.pow(24) as i64);
} else {
panic!("unexpected error {}", e)
}
}
#[test]
fn test_from_dt_invalid_time() {
// Date before CCSDS epoch
let datetime_utc = NaiveDate::from_ymd_opt(1957, 12, 31)
.unwrap()
.and_hms_milli_opt(23, 59, 59, 999)
.unwrap()
.and_local_timezone(chrono::Utc)
.unwrap();
let time_provider = CdsTime::from_dt_with_u24_days(&datetime_utc);
assert!(time_provider.is_err());
if let CdsError::DateBeforeCcsdsEpoch(DateBeforeCcsdsEpochError(dt)) =
time_provider.unwrap_err()
{
assert_eq!(dt, datetime_utc.into());
}
}
#[test]
fn test_eq() {
let stamp0 = CdsTime::new_with_u16_days(0, 0);
let mut buf: [u8; 7] = [0; 7];
stamp0.write_to_bytes(&mut buf).unwrap();
let stamp1 = CdsTime::from_bytes_with_u16_days(&buf).unwrap();
assert_eq!(stamp0, stamp1);
assert!(stamp0 >= stamp1);
assert!(stamp1 <= stamp0);
}
#[test]
fn test_ord() {
let stamp0 = CdsTime::new_with_u24_days(0, 0).unwrap();
let stamp1 = CdsTime::new_with_u24_days(0, 50000).unwrap();
let mut stamp2 = CdsTime::new_with_u24_days(0, 50000).unwrap();
stamp2.set_submillis(SubmillisPrecision::Microseconds, 500);
let stamp3 = CdsTime::new_with_u24_days(1, 0).unwrap();
assert!(stamp1 > stamp0);
assert!(stamp2 > stamp0);
assert!(stamp2 > stamp1);
assert!(stamp3 > stamp0);
assert!(stamp3 > stamp1);
assert!(stamp3 > stamp2);
}
#[test]
fn test_conversion() {
let mut stamp_small = CdsTime::new_with_u16_days(u16::MAX, 500);
let stamp_larger: CdsTime<DaysLen24Bits> = stamp_small.into();
assert_eq!(stamp_larger.ccsds_days_as_u32(), u16::MAX as u32);
assert_eq!(stamp_larger.ms_of_day(), 500);
stamp_small = stamp_larger.try_into().unwrap();
assert_eq!(stamp_small.ccsds_days_as_u32(), u16::MAX as u32);
assert_eq!(stamp_small.ms_of_day(), 500);
}
#[test]
fn test_update_from_now() {
let mut stamp = CdsTime::new_with_u16_days(0, 0);
let _ = stamp.update_from_now();
let dt = stamp.unix_time().chrono_date_time().unwrap();
assert!(dt.year() > 2020);
}
#[test]
fn test_setting_submillis_precision() {
let mut provider = CdsTime::new_with_u16_days(0, 15);
provider.set_submillis(SubmillisPrecision::Microseconds, 500);
}
#[test]
#[cfg(feature = "serde")]
fn test_serialization() {
let stamp_now = CdsTime::from_now_with_u16_days().expect("Error retrieving time");
let val = to_allocvec(&stamp_now).expect("Serializing timestamp failed");
assert!(val.len() > 0);
let stamp_deser: CdsTime = from_bytes(&val).expect("Stamp deserialization failed");
assert_eq!(stamp_deser, stamp_now);
}
fn generic_dt_property_equality_check(first: u32, second: u32, start: u32, end: u32) {
if second < first {
assert_eq!(second, start);
assert_eq!(first, end);
} else if second > first {
assert_eq!(second - first, 1);
} else {
assert_eq!(first, second);
}
}
#[test]
fn test_stamp_to_vec_u16() {
let stamp = CdsTime::new_with_u16_days(1, 1);
let stamp_vec = stamp.to_vec().unwrap();
let mut buf: [u8; 7] = [0; 7];
stamp.write_to_bytes(&mut buf).unwrap();
assert_eq!(stamp_vec, buf);
}
#[test]
fn test_stamp_to_vec_u24() {
let stamp = CdsTime::new_with_u24_days(1, 1).unwrap();
let stamp_vec = stamp.to_vec().unwrap();
let mut buf: [u8; 10] = [0; 10];
stamp.write_to_bytes(&mut buf).unwrap();
assert_eq!(stamp_vec, buf[..stamp.len_written()]);
}
#[test]
#[cfg(feature = "timelib")]
fn test_timelib_stamp() {
let stamp = CdsTime::new_with_u16_days(0, 0);
let timelib_dt = stamp.timelib_date_time().unwrap();
assert_eq!(timelib_dt.year(), 1958);
assert_eq!(timelib_dt.month(), time::Month::January);
assert_eq!(timelib_dt.day(), 1);
assert_eq!(timelib_dt.hour(), 0);
assert_eq!(timelib_dt.minute(), 0);
assert_eq!(timelib_dt.second(), 0);
}
}
Reference in New Issue View Git Blame Copy Permalink