use super::bits::{Bits, Bits11};
use alloc::{collections::BTreeMap, vec::Vec};
#[derive(Copy, Clone, Debug, Default)]
pub enum Language {
#[default]
English,
}
impl Language {
pub(crate) fn wordlist(&self) -> &'static WordList {
match *self {
Language::English => &lazy::WORDLIST_ENGLISH,
}
}
pub(crate) fn wordmap(&self) -> &'static WordMap {
match *self {
Language::English => &lazy::WORDMAP_ENGLISH,
}
}
}
pub(crate) struct WordMap {
inner: BTreeMap<&'static str, Bits11>,
}
pub(crate) struct WordList {
inner: Vec<&'static str>,
}
impl WordMap {
pub fn get_bits(&self, word: &str) -> Option<Bits11> {
self.inner.get(word).cloned()
}
}
impl WordList {
pub fn get_word(&self, bits: Bits11) -> &'static str {
self.inner[bits.bits() as usize]
}
}
mod lazy {
use super::{Bits11, WordList, WordMap};
use alloc::vec::Vec;
use once_cell::sync::Lazy;
fn gen_wordlist(lang_words: &'static str) -> WordList {
let inner: Vec<_> = lang_words.split_whitespace().collect();
debug_assert!(inner.len() == 2048, "Invalid wordlist length");
WordList { inner }
}
fn gen_wordmap(wordlist: &WordList) -> WordMap {
let inner = wordlist
.inner
.iter()
.enumerate()
.map(|(i, item)| (*item, Bits11::from(i as u16)))
.collect();
WordMap { inner }
}
pub(crate) static WORDLIST_ENGLISH: Lazy<WordList> =
Lazy::new(|| gen_wordlist(include_str!("langs/english.txt")));
pub(crate) static WORDMAP_ENGLISH: Lazy<WordMap> = Lazy::new(|| gen_wordmap(&WORDLIST_ENGLISH));
}