diff options
Diffstat (limited to 'rust/vendor/time/src/util.rs')
-rw-r--r-- | rust/vendor/time/src/util.rs | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/rust/vendor/time/src/util.rs b/rust/vendor/time/src/util.rs new file mode 100644 index 0000000..3735b0b --- /dev/null +++ b/rust/vendor/time/src/util.rs @@ -0,0 +1,80 @@ +//! Utility functions. + +use crate::Month; + +/// Whether to adjust the date, and in which direction. Useful when implementing arithmetic. +pub(crate) enum DateAdjustment { + /// The previous day should be used. + Previous, + /// The next day should be used. + Next, + /// The date should be used as-is. + None, +} + +/// Get the number of days in the month of a given year. +/// +/// ```rust +/// # use time::{Month, util}; +/// assert_eq!(util::days_in_year_month(2020, Month::February), 29); +/// ``` +pub const fn days_in_year_month(year: i32, month: Month) -> u8 { + use Month::*; + match month { + January | March | May | July | August | October | December => 31, + April | June | September | November => 30, + February if is_leap_year(year) => 29, + February => 28, + } +} + +/// Returns if the provided year is a leap year in the proleptic Gregorian calendar. Uses +/// [astronomical year numbering](https://en.wikipedia.org/wiki/Astronomical_year_numbering). +/// +/// ```rust +/// # use time::util::is_leap_year; +/// assert!(!is_leap_year(1900)); +/// assert!(is_leap_year(2000)); +/// assert!(is_leap_year(2004)); +/// assert!(!is_leap_year(2005)); +/// assert!(!is_leap_year(2100)); +/// ``` +pub const fn is_leap_year(year: i32) -> bool { + year % 4 == 0 && (year % 25 != 0 || year % 16 == 0) +} + +/// Get the number of calendar days in a given year. +/// +/// The returned value will always be either 365 or 366. +/// +/// ```rust +/// # use time::util::days_in_year; +/// assert_eq!(days_in_year(1900), 365); +/// assert_eq!(days_in_year(2000), 366); +/// assert_eq!(days_in_year(2004), 366); +/// assert_eq!(days_in_year(2005), 365); +/// assert_eq!(days_in_year(2100), 365); +/// ``` +pub const fn days_in_year(year: i32) -> u16 { + if is_leap_year(year) { 366 } else { 365 } +} + +/// Get the number of weeks in the ISO year. +/// +/// The returned value will always be either 52 or 53. +/// +/// ```rust +/// # use time::util::weeks_in_year; +/// assert_eq!(weeks_in_year(2019), 52); +/// assert_eq!(weeks_in_year(2020), 53); +/// ``` +pub const fn weeks_in_year(year: i32) -> u8 { + match year.rem_euclid(400) { + 4 | 9 | 15 | 20 | 26 | 32 | 37 | 43 | 48 | 54 | 60 | 65 | 71 | 76 | 82 | 88 | 93 | 99 + | 105 | 111 | 116 | 122 | 128 | 133 | 139 | 144 | 150 | 156 | 161 | 167 | 172 | 178 + | 184 | 189 | 195 | 201 | 207 | 212 | 218 | 224 | 229 | 235 | 240 | 246 | 252 | 257 + | 263 | 268 | 274 | 280 | 285 | 291 | 296 | 303 | 308 | 314 | 320 | 325 | 331 | 336 + | 342 | 348 | 353 | 359 | 364 | 370 | 376 | 381 | 387 | 392 | 398 => 53, + _ => 52, + } +} |