use castep_cell_fmt::{Cell, CellValue, ToCell, ToCellValue};
use castep_cell_fmt::parse::{FromCellValue, FromKeyValue};
use castep_cell_fmt::{CResult, Error};
use castep_cell_fmt::query::value_as_str;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename = "BASIS_PRECISION")]
#[derive(Default)]
pub enum BasisPrecision {
#[serde(alias = "coarse", alias = "COARSE")]
Coarse,
#[serde(alias = "medium", alias = "MEDIUM")]
Medium,
#[serde(alias = "fine", alias = "FINE")]
#[default]
Fine,
#[serde(alias = "precise", alias = "PRECISE")]
Precise,
#[serde(alias = "extreme", alias = "EXTREME")]
Extreme,
}
impl FromCellValue for BasisPrecision {
fn from_cell_value(value: &CellValue<'_>) -> CResult<Self> {
match value_as_str(value)?.to_ascii_lowercase().as_str() {
"coarse" => Ok(Self::Coarse),
"medium" => Ok(Self::Medium),
"fine" => Ok(Self::Fine),
"precise" => Ok(Self::Precise),
"extreme" => Ok(Self::Extreme),
other => Err(Error::Message(format!("unknown BasisPrecision: {other}"))),
}
}
}
impl FromKeyValue for BasisPrecision {
const KEY_NAME: &'static str = "BASIS_PRECISION";
fn from_cell_value_kv(value: &CellValue<'_>) -> CResult<Self> {
Self::from_cell_value(value)
}
}
impl ToCell for BasisPrecision {
fn to_cell(&self) -> Cell<'_> {
Cell::KeyValue("BASIS_PRECISION", self.to_cell_value())
}
}
impl ToCellValue for BasisPrecision {
fn to_cell_value(&self) -> CellValue<'_> {
CellValue::String(
match self {
BasisPrecision::Coarse => "COARSE",
BasisPrecision::Medium => "MEDIUM",
BasisPrecision::Fine => "FINE",
BasisPrecision::Precise => "PRECISE",
BasisPrecision::Extreme => "EXTREME",
}
.to_string(),
)
}
}
#[cfg(test)]
mod tests {
use super::*;
use castep_cell_fmt::CellValue;
#[test]
fn test_case_insensitive() {
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("coarse")).unwrap(), BasisPrecision::Coarse);
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("COARSE")).unwrap(), BasisPrecision::Coarse);
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("fine")).unwrap(), BasisPrecision::Fine);
}
#[test]
fn test_all_variants() {
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("medium")).unwrap(), BasisPrecision::Medium);
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("precise")).unwrap(), BasisPrecision::Precise);
assert_eq!(BasisPrecision::from_cell_value(&CellValue::Str("extreme")).unwrap(), BasisPrecision::Extreme);
}
#[test]
fn test_invalid() {
assert!(BasisPrecision::from_cell_value(&CellValue::Str("invalid")).is_err());
}
#[test]
fn test_key_name() {
assert_eq!(BasisPrecision::KEY_NAME, "BASIS_PRECISION");
}
}