#![allow(clippy::exhaustive_structs, clippy::exhaustive_enums)]
pub mod chinese_based;
pub mod hijri;
pub use chinese_based::{CalendarChineseV1, CalendarDangiV1};
pub use hijri::{CalendarHijriObservationalV1, CalendarHijriUmmalquraV1};
use crate::types::Weekday;
use icu_provider::fallback::{LocaleFallbackConfig, LocaleFallbackPriority};
use icu_provider::prelude::*;
use tinystr::TinyStr16;
use zerovec::ZeroVec;
#[cfg(feature = "compiled_data")]
#[derive(Debug)]
pub struct Baked;
#[cfg(feature = "compiled_data")]
#[allow(unused_imports)]
const _: () = {
use icu_calendar_data::*;
pub mod icu {
pub use crate as calendar;
pub use icu_calendar_data::icu_locale as locale;
}
make_provider!(Baked);
impl_calendar_chinese_v1!(Baked);
impl_calendar_dangi_v1!(Baked);
impl_calendar_hijri_observational_v1!(Baked);
impl_calendar_hijri_ummalqura_v1!(Baked);
impl_calendar_japanese_modern_v1!(Baked);
impl_calendar_japanese_extended_v1!(Baked);
impl_calendar_week_v1!(Baked);
};
icu_provider::data_marker!(
CalendarJapaneseModernV1,
"calendar/japanese/modern/v1",
JapaneseEras<'static>,
is_singleton = true
);
icu_provider::data_marker!(
CalendarJapaneseExtendedV1,
"calendar/japanese/extended/v1",
JapaneseEras<'static>,
is_singleton = true
);
icu_provider::data_marker!(
CalendarWeekV1,
"calendar/week/v1",
WeekData,
fallback_config = {
let mut config = LocaleFallbackConfig::default();
config.priority = LocaleFallbackPriority::Region;
config
},
);
#[cfg(feature = "datagen")]
pub const MARKERS: &[DataMarkerInfo] = &[
CalendarChineseV1::INFO,
CalendarDangiV1::INFO,
CalendarHijriObservationalV1::INFO,
CalendarHijriUmmalquraV1::INFO,
CalendarJapaneseModernV1::INFO,
CalendarJapaneseExtendedV1::INFO,
CalendarWeekV1::INFO,
];
#[zerovec::make_ule(EraStartDateULE)]
#[derive(
Copy, Clone, PartialEq, PartialOrd, Eq, Ord, Hash, Debug, yoke::Yokeable, zerofrom::ZeroFrom,
)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct EraStartDate {
pub year: i32,
pub month: u8,
pub day: u8,
}
#[derive(Debug, PartialEq, Clone, Default, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct JapaneseEras<'data> {
#[cfg_attr(feature = "serde", serde(borrow))]
pub dates_to_eras: ZeroVec<'data, (EraStartDate, TinyStr16)>,
}
icu_provider::data_struct!(
JapaneseEras<'_>,
#[cfg(feature = "datagen")]
);
#[derive(Clone, Copy, Debug, PartialEq, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
#[allow(clippy::exhaustive_structs)] pub struct WeekData {
pub first_weekday: Weekday,
pub min_week_days: u8,
pub weekend: WeekdaySet,
}
icu_provider::data_struct!(
WeekData,
#[cfg(feature = "datagen")]
);
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct WeekdaySet(u8);
impl WeekdaySet {
pub const fn contains(&self, day: Weekday) -> bool {
self.0 & day.bit_value() != 0
}
}
impl WeekdaySet {
pub const fn new(days: &[Weekday]) -> Self {
let mut i = 0;
let mut w = 0;
#[allow(clippy::indexing_slicing)]
while i < days.len() {
w |= days[i].bit_value();
i += 1;
}
Self(w)
}
}
impl Weekday {
const fn bit_value(self) -> u8 {
match self {
Weekday::Monday => 1 << 6,
Weekday::Tuesday => 1 << 5,
Weekday::Wednesday => 1 << 4,
Weekday::Thursday => 1 << 3,
Weekday::Friday => 1 << 2,
Weekday::Saturday => 1 << 1,
Weekday::Sunday => 1 << 0,
}
}
}
#[cfg(feature = "datagen")]
impl databake::Bake for WeekdaySet {
fn bake(&self, ctx: &databake::CrateEnv) -> databake::TokenStream {
ctx.insert("icu_calendar");
let days =
crate::week_of::WeekdaySetIterator::new(Weekday::Monday, *self).map(|d| d.bake(ctx));
databake::quote! {
icu_calendar::provider::WeekdaySet::new(&[#(#days),*])
}
}
}
#[cfg(feature = "datagen")]
impl databake::BakeSize for WeekdaySet {
fn borrows_size(&self) -> usize {
0
}
}
#[cfg(feature = "datagen")]
impl serde::Serialize for WeekdaySet {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
if serializer.is_human_readable() {
use serde::ser::SerializeSeq;
let mut seq = serializer.serialize_seq(None)?;
for day in crate::week_of::WeekdaySetIterator::new(Weekday::Monday, *self) {
seq.serialize_element(&day)?;
}
seq.end()
} else {
self.0.serialize(serializer)
}
}
}
#[cfg(feature = "serde")]
impl<'de> serde::Deserialize<'de> for WeekdaySet {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
if deserializer.is_human_readable() {
use core::marker::PhantomData;
struct Visitor<'de>(PhantomData<&'de ()>);
impl<'de> serde::de::Visitor<'de> for Visitor<'de> {
type Value = WeekdaySet;
fn expecting(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
core::write!(f, "a sequence of Weekdays")
}
fn visit_seq<A: serde::de::SeqAccess<'de>>(
self,
mut seq: A,
) -> Result<Self::Value, A::Error> {
let mut set = WeekdaySet::new(&[]);
while let Some(day) = seq.next_element::<Weekday>()? {
set.0 |= day.bit_value();
}
Ok(set)
}
}
deserializer.deserialize_seq(Visitor(PhantomData))
} else {
u8::deserialize(deserializer).map(Self)
}
}
}
#[test]
fn test_weekdayset_bake() {
databake::test_bake!(
WeekdaySet,
const,
crate::provider::WeekdaySet::new(&[
crate::types::Weekday::Monday,
crate::types::Weekday::Wednesday,
crate::types::Weekday::Friday
]),
icu_calendar
);
}
#[test]
fn test_weekdayset_new() {
use Weekday::*;
let sat_sun_bitmap = Saturday.bit_value() | Sunday.bit_value();
let sat_sun_weekend = WeekdaySet::new(&[Saturday, Sunday]);
assert_eq!(sat_sun_bitmap, sat_sun_weekend.0);
let fri_sat_bitmap = Friday.bit_value() | Saturday.bit_value();
let fri_sat_weekend = WeekdaySet::new(&[Friday, Saturday]);
assert_eq!(fri_sat_bitmap, fri_sat_weekend.0);
let fri_sun_bitmap = Friday.bit_value() | Sunday.bit_value();
let fri_sun_weekend = WeekdaySet::new(&[Friday, Sunday]);
assert_eq!(fri_sun_bitmap, fri_sun_weekend.0);
let fri_bitmap = Friday.bit_value();
let fri_weekend = WeekdaySet::new(&[Friday, Friday]);
assert_eq!(fri_bitmap, fri_weekend.0);
let sun_mon_bitmap = Sunday.bit_value() | Monday.bit_value();
let sun_mon_weekend = WeekdaySet::new(&[Sunday, Monday]);
assert_eq!(sun_mon_bitmap, sun_mon_weekend.0);
let mon_sun_bitmap = Monday.bit_value() | Sunday.bit_value();
let mon_sun_weekend = WeekdaySet::new(&[Monday, Sunday]);
assert_eq!(mon_sun_bitmap, mon_sun_weekend.0);
let mon_bitmap = Monday.bit_value();
let mon_weekend = WeekdaySet::new(&[Monday]);
assert_eq!(mon_bitmap, mon_weekend.0);
}