pub mod data;
use crate::atom::UffAtomType;
use data::UFF_DATA;
#[derive(Debug, Clone, Copy)]
pub struct UffParams {
pub r1: f64, pub theta0: f64, pub x1: f64, pub d1: f64, pub zeta: f64, pub z_star: f64, pub chi: f64, pub u_i: f64, }
pub fn get_uff_params(atom_type: &UffAtomType) -> Option<UffParams> {
let label = atom_type.as_str();
let label_stem = label.trim_end_matches('_');
UFF_DATA.iter()
.find(|&&(id, ..)| {
let id_stem = id.trim_end_matches('_');
id == label ||
id.starts_with(label) ||
label.starts_with(id) ||
id.starts_with(label_stem) ||
label_stem.starts_with(id_stem)
})
.map(|&(_, r1, theta0, x1, d1, zeta, z_star, chi, u_i)| UffParams {
r1, theta0, x1, d1, zeta, z_star, chi, u_i,
})
}
pub fn element_symbol(z: usize) -> &'static str {
let symbols = [
"H", "He", "Li", "Be", "B", "C", "N", "O", "F", "Ne",
"Na", "Mg", "Al", "Si", "P", "S", "Cl", "Ar", "K", "Ca",
"Sc", "Ti", "V", "Cr", "Mn", "Fe", "Co", "Ni", "Cu", "Zn",
"Ga", "Ge", "As", "Se", "Br", "Kr", "Rb", "Sr", "Y", "Zr",
"Nb", "Mo", "Tc", "Ru", "Rh", "Pd", "Ag", "Cd", "In", "Sn",
"Sb", "Te", "I", "Xe", "Cs", "Ba", "La", "Ce", "Pr", "Nd",
"Pm", "Sm", "Eu", "Gd", "Tb", "Dy", "Ho", "Er", "Tm", "Yb",
"Lu", "Hf", "Ta", "W", "Re", "Os", "Ir", "Pt", "Au", "Hg",
"Tl", "Pb", "Bi", "Po", "At", "Rn", "Fr", "Ra", "Ac", "Th",
"Pa", "U", "Np", "Pu", "Am", "Cm", "Bk", "Cf", "Es", "Fm",
"Md", "No", "Lr",
];
if z > 0 && z <= symbols.len() {
symbols[z - 1]
} else {
"X"
}
}