#![allow(clippy::exhaustive_structs, clippy::exhaustive_enums)]
use icu_collections::char16trie::Char16TrieIterator;
use icu_collections::codepointtrie::CodePointTrie;
use icu_provider::prelude::*;
use zerovec::ule::{AsULE, RawBytesULE};
use zerovec::ZeroVec;
use zerovec::{zeroslice, ZeroSlice};
use crate::elements::CollationElement;
use crate::elements::CollationElement32;
use crate::elements::Tag;
use crate::elements::EMPTY_U16;
use crate::elements::FFFD_CE;
use crate::elements::FFFD_CE32;
use crate::elements::FFFD_CE32_VALUE;
use crate::elements::FFFD_CE_VALUE;
use crate::elements::NO_CE_PRIMARY;
use crate::preferences::CollationCaseFirst;
use crate::options::MaxVariable;
#[cfg(feature = "compiled_data")]
#[derive(Debug)]
pub struct Baked;
#[cfg(feature = "compiled_data")]
#[allow(unused_imports)]
const _: () = {
use icu_collator_data::*;
pub mod icu {
pub use crate as collator;
pub use icu_collections as collections;
pub use icu_locale as locale;
}
make_provider!(Baked);
impl_collation_root_v1!(Baked);
impl_collation_tailoring_v1!(Baked);
impl_collation_diacritics_v1!(Baked);
impl_collation_jamo_v1!(Baked);
impl_collation_metadata_v1!(Baked);
impl_collation_special_primaries_v1!(Baked);
impl_collation_reordering_v1!(Baked);
};
const SCRIPT_FALLBACK: icu_provider::fallback::LocaleFallbackConfig = {
let mut c = icu_provider::fallback::LocaleFallbackConfig::default();
c.priority = icu_provider::fallback::LocaleFallbackPriority::Script;
c
};
icu_provider::data_marker!(
CollationRootV1,
"collation/root/v1",
CollationData<'static>,
is_singleton = true,
);
icu_provider::data_marker!(
CollationTailoringV1,
"collation/tailoring/v1",
CollationData<'static>,
fallback_config = SCRIPT_FALLBACK,
#[cfg(feature = "datagen")]
attributes_domain = "collator",
);
icu_provider::data_marker!(
CollationDiacriticsV1,
"collation/diacritics/v1",
CollationDiacritics<'static>,
fallback_config = SCRIPT_FALLBACK,
#[cfg(feature = "datagen")]
attributes_domain = "collator",
);
icu_provider::data_marker!(
CollationJamoV1,
"collation/jamo/v1",
CollationJamo<'static>,
is_singleton = true,
);
icu_provider::data_marker!(
CollationReorderingV1,
"collation/reordering/v1",
CollationReordering<'static>,
fallback_config = SCRIPT_FALLBACK,
#[cfg(feature = "datagen")]
attributes_domain = "collator",
);
icu_provider::data_marker!(
CollationMetadataV1,
"collation/metadata/v1",
CollationMetadata,
fallback_config = SCRIPT_FALLBACK,
#[cfg(feature = "datagen")]
attributes_domain = "collator",
);
icu_provider::data_marker!(
CollationSpecialPrimariesV1,
"collation/special/primaries/v1",
CollationSpecialPrimaries<'static>,
is_singleton = true,
);
#[cfg(feature = "datagen")]
pub const MARKERS: &[DataMarkerInfo] = &[
CollationRootV1::INFO,
CollationTailoringV1::INFO,
CollationDiacriticsV1::INFO,
CollationJamoV1::INFO,
CollationMetadataV1::INFO,
CollationReorderingV1::INFO,
CollationSpecialPrimariesV1::INFO,
];
const SINGLE_U32: &ZeroSlice<u32> =
zeroslice!(u32; <u32 as AsULE>::ULE::from_unsigned; [FFFD_CE32_VALUE]);
const SINGLE_U64: &ZeroSlice<u64> =
zeroslice!(u64; <u64 as AsULE>::ULE::from_unsigned; [FFFD_CE_VALUE]);
fn data_ce_to_primary(data_ce: u64, c: char) -> u32 {
let p = (data_ce >> 32) as u32; let lower32 = data_ce as u32 as i32; let mut offset = ((u32::from(c) as i32) - (lower32 >> 8)) * (lower32 & 0x7F); let is_compressible = (lower32 & 0x80) != 0;
offset += (((p >> 8) & 0xFF) as i32) - 2;
let mut primary = (((offset % 254) + 2) as u32) << 8;
offset /= 254;
if is_compressible {
offset += (((p >> 16) & 0xFF) as i32) - 4;
primary |= (((offset % 251) + 4) as u32) << 16;
offset /= 251;
} else {
offset += (((p >> 16) & 0xFF) as i32) - 2;
primary |= (((offset % 254) + 2) as u32) << 16;
offset /= 254;
}
primary | ((p & 0xFF000000) + ((offset as u32) << 24))
}
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationData<'data> {
#[cfg_attr(feature = "serde", serde(borrow))]
pub trie: CodePointTrie<'data, u32>,
#[cfg_attr(feature = "serde", serde(borrow))]
pub ces: ZeroVec<'data, u64>,
#[cfg_attr(feature = "serde", serde(borrow))]
pub ce32s: ZeroVec<'data, u32>,
#[cfg_attr(feature = "serde", serde(borrow))]
pub contexts: ZeroVec<'data, u16>,
}
icu_provider::data_struct!(
CollationData<'_>,
#[cfg(feature = "datagen")]
);
impl<'data> CollationData<'data> {
pub(crate) fn ce32_for_char(&self, c: char) -> CollationElement32 {
CollationElement32::new(self.trie.get(c))
}
pub(crate) fn get_ce32(&'data self, index: usize) -> CollationElement32 {
CollationElement32::new(if let Some(u) = self.ce32s.get(index) {
u
} else {
debug_assert!(false);
FFFD_CE32_VALUE
})
}
pub(crate) fn get_ce32s(&'data self, index: usize, len: usize) -> &'data ZeroSlice<u32> {
if len > 0 {
if let Some(slice) = self.ce32s.get_subslice(index..index + len) {
return slice;
}
}
debug_assert!(false);
SINGLE_U32
}
pub(crate) fn get_ces(&'data self, index: usize, len: usize) -> &'data ZeroSlice<u64> {
if len > 0 {
if let Some(slice) = self.ces.get_subslice(index..index + len) {
return slice;
}
}
debug_assert!(false);
SINGLE_U64
}
fn get_default_and_trie_impl(
&'data self,
index: usize,
) -> (CollationElement32, &'data ZeroSlice<u16>) {
if let Some(slice) = self.contexts.get_subslice(index..self.contexts.len()) {
#[expect(clippy::unwrap_used)]
if slice.len() >= 2 {
let first = slice.get(0).unwrap();
let second = slice.get(1).unwrap();
let trie = slice.get_subslice(2..slice.len()).unwrap();
return (
CollationElement32::new((u32::from(first) << 16) | u32::from(second)),
trie,
);
}
}
debug_assert!(false);
(FFFD_CE32, EMPTY_U16)
}
pub(crate) fn get_default_and_trie(
&'data self,
index: usize,
) -> (CollationElement32, Char16TrieIterator<'data>) {
let (ce32, trie) = self.get_default_and_trie_impl(index);
(ce32, Char16TrieIterator::new(trie))
}
pub(crate) fn get_default(&'data self, index: usize) -> CollationElement32 {
let (ce32, _) = self.get_default_and_trie_impl(index);
ce32
}
pub(crate) fn ce_from_offset_ce32(
&self,
c: char,
ce32: CollationElement32,
) -> CollationElement {
debug_assert!(ce32.tag() == Tag::Offset);
if let Some(data_ce) = self.ces.get(ce32.index()) {
CollationElement::new_from_primary(data_ce_to_primary(data_ce, c))
} else {
debug_assert!(false);
FFFD_CE
}
}
}
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationDiacritics<'data> {
#[cfg_attr(feature = "serde", serde(borrow))]
pub secondaries: ZeroVec<'data, u16>,
}
icu_provider::data_struct!(
CollationDiacritics<'_>,
#[cfg(feature = "datagen")]
);
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationJamo<'data> {
#[cfg_attr(feature = "serde", serde(borrow))]
pub ce32s: ZeroVec<'data, u32>,
}
icu_provider::data_struct!(
CollationJamo<'_>,
#[cfg(feature = "datagen")]
);
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationReordering<'data> {
pub min_high_no_reorder: u32,
#[cfg_attr(feature = "serde", serde(borrow))]
pub reorder_table: ZeroVec<'data, u8>, #[cfg_attr(feature = "serde", serde(borrow))]
pub reorder_ranges: ZeroVec<'data, u32>,
}
icu_provider::data_struct!(
CollationReordering<'_>,
#[cfg(feature = "datagen")]
);
impl CollationReordering<'_> {
pub(crate) fn reorder(&self, primary: u32) -> u32 {
if let Some(b) = self.reorder_table.get((primary >> 24) as usize) {
if b != 0 || primary <= NO_CE_PRIMARY {
(u32::from(b) << 24) | (primary & 0x00FFFFFF)
} else {
self.reorder_ex(primary)
}
} else {
debug_assert!(false);
primary
}
}
fn reorder_ex(&self, primary: u32) -> u32 {
if primary >= self.min_high_no_reorder {
return primary;
}
let q = primary | 0xFFFF;
for &range in self.reorder_ranges.as_ule_slice().iter() {
let r = u32::from_unaligned(range);
if q < r {
return primary.wrapping_add(r << 24);
}
}
debug_assert!(false);
primary
}
}
#[derive(Debug, PartialEq, Clone, Copy, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationMetadata {
pub bits: u32,
}
icu_provider::data_struct!(
CollationMetadata,
#[cfg(feature = "datagen")]
);
impl CollationMetadata {
const MAX_VARIABLE_MASK: u32 = 0b11;
const TAILORED_MASK: u32 = 1 << 3;
const TAILORED_DIACRITICS_MASK: u32 = 1 << 4;
const REORDERING_MASK: u32 = 1 << 5;
const LITHUANIAN_DOT_ABOVE_MASK: u32 = 1 << 6;
const BACWARD_SECOND_LEVEL_MASK: u32 = 1 << 7;
const ALTERNATE_SHIFTED_MASK: u32 = 1 << 8;
const CASE_FIRST_MASK: u32 = 1 << 9;
const UPPER_FIRST_MASK: u32 = 1 << 10;
#[inline(always)]
pub(crate) fn max_variable(self) -> MaxVariable {
unsafe { core::mem::transmute((self.bits & CollationMetadata::MAX_VARIABLE_MASK) as u8) }
}
#[inline(always)]
pub(crate) fn tailored(self) -> bool {
self.bits & CollationMetadata::TAILORED_MASK != 0
}
#[inline(always)]
pub(crate) fn tailored_diacritics(self) -> bool {
self.bits & CollationMetadata::TAILORED_DIACRITICS_MASK != 0
}
#[inline(always)]
pub(crate) fn lithuanian_dot_above(self) -> bool {
self.bits & CollationMetadata::LITHUANIAN_DOT_ABOVE_MASK != 0
}
#[inline(always)]
pub(crate) fn backward_second_level(self) -> bool {
self.bits & CollationMetadata::BACWARD_SECOND_LEVEL_MASK != 0
}
#[inline(always)]
pub(crate) fn reordering(self) -> bool {
self.bits & CollationMetadata::REORDERING_MASK != 0
}
#[inline(always)]
pub(crate) fn alternate_shifted(self) -> bool {
self.bits & CollationMetadata::ALTERNATE_SHIFTED_MASK != 0
}
#[inline(always)]
pub(crate) fn case_first(self) -> CollationCaseFirst {
if self.bits & CollationMetadata::CASE_FIRST_MASK != 0 {
if self.bits & CollationMetadata::UPPER_FIRST_MASK != 0 {
CollationCaseFirst::Upper
} else {
CollationCaseFirst::Lower
}
} else {
CollationCaseFirst::False
}
}
}
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_collator::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
pub struct CollationSpecialPrimaries<'data> {
#[cfg_attr(feature = "serde", serde(borrow))]
pub last_primaries: ZeroVec<'data, u16>,
pub numeric_primary: u8,
}
#[derive(Debug, PartialEq, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
pub(crate) struct CollationSpecialPrimariesValidated<'data> {
pub last_primaries: ZeroVec<'data, u16>,
pub numeric_primary: u8,
pub compressible_bytes: &'data [<u16 as AsULE>::ULE; 16],
}
impl CollationSpecialPrimariesValidated<'static> {
pub(crate) const HARDCODED_COMPRESSIBLE_BYTES_FALLBACK: &'static [<u16 as AsULE>::ULE; 16] = &[
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b1111_1111_1111_1110u16.to_le_bytes()),
RawBytesULE(0b1111_1111_1111_1111u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0001u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0000_0000_0000_0000u16.to_le_bytes()),
RawBytesULE(0b0100_0000_0000_0000u16.to_le_bytes()),
];
}
icu_provider::data_struct!(
CollationSpecialPrimaries<'_>,
#[cfg(feature = "datagen")]
);
impl CollationSpecialPrimariesValidated<'_> {
#[expect(clippy::unwrap_used)]
pub(crate) fn last_primary_for_group(&self, max_variable: MaxVariable) -> u32 {
(u32::from(self.last_primaries.get(max_variable as usize).unwrap()) << 16) - 1
}
#[allow(dead_code)]
pub(crate) fn is_compressible(&self, b: u8) -> bool {
#[expect(clippy::indexing_slicing)]
let field = u16::from_unaligned(self.compressible_bytes[usize::from(b >> 4)]);
let mask = 1 << (b & 0b1111);
(field & mask) != 0
}
}