russell_tensor 3.0.1

use crate::StrError;
use crate::{Tensor2, Tensor3, Tensor4};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fs;
use std::path::Path;

/// Holds the Vacuum Electric Permittivity constant
///
/// ε₀ Units: F/m
///
/// See: <https://physics.nist.gov/cgi-bin/cuu/Value?ep0>
const VACUUM_ELECTRIC_PERMITTIVITY: f64 = 8.854_187_818_8e-12;

/// Represents the entire database deserialized from the JSON file.
///
/// The JSON structure contains a specific `_metadata` key with global metadata,
/// and dynamically namespaced keys containing tensor data mapped by their
/// Materials Project ID (e.g., `"mp-3731"`). The `#[serde(flatten)]` macro
/// cleanly separates the metadata from the dynamic material dictionary.
#[derive(Debug, Serialize, Deserialize, PartialEq)]
pub struct PiezoDatabase {
    /// Global metadata describing tensor mappings and formats used in the DB.
    #[serde(rename = "_metadata")]
    pub metadata: Metadata,

    /// A map holding all material objects, keyed by their Materials Project ID.
    #[serde(flatten)]
    pub materials: HashMap<String, Material>,
}

impl PiezoDatabase {
    /// Reads and parses the PiezoDatabase directly from a JSON file.
    ///
    /// # Arguments
    /// * `path` - A path reference to the JSON file to be loaded.
    ///
    /// # Errors
    /// Returns an error if the file cannot be opened/read or if the JSON format
    /// does not perfectly match the struct schema.
    pub fn from_file<P: AsRef<Path>>(path: P) -> Result<Self, Box<dyn std::error::Error>> {
        let content = fs::read_to_string(path)?;
        let db: PiezoDatabase = serde_json::from_str(&content)?;
        Ok(db)
    }

    /// Gets a reference to a material in the database
    pub fn get(&self, material_id: &str) -> Result<&Material, StrError> {
        self.materials
            .get(material_id)
            .ok_or("material not found in the database")
    }
}

/// Physical units mapping for the exported tensors and moduli.
#[derive(Debug, Serialize, Deserialize, PartialEq)]
pub struct Units {
    pub e_tensor: String,
    pub e_voigt: String,
    pub cc_tensor: String,
    pub cc_voigt: String,
    pub relative_permittivity_tensor: String,
}

/// Stores standard definitions and rules applied across the database.
#[derive(Debug, Serialize, Deserialize, PartialEq)]
pub struct Metadata {
    /// The string mapping denoting the Voigt reduction rules.
    pub voigt_mapping: String,

    /// Detailed notes regarding the specific IEEE standard mapping used.
    pub voigt_mapping_notes: String,

    /// The physical units corresponding to the numerical tensor values.
    pub units: Units,
}

/// Represents the physical properties and tensors for a single crystalline material.
///
/// Note: Depending on the API availability from Materials Project, certain materials
/// (e.g. Ag3SI) may lack calculated elasticity data. As such, the strain tensor `d`
/// as well as the elastic tensors `C` and `S` are strictly optional.
#[derive(Debug, Serialize, Deserialize, PartialEq)]
pub struct Material {
    /// The Materials Project unique identifier.
    pub material_id: String,

    /// The chemical formula of the material (e.g. "LiNbO3").
    pub formula: String,

    /// The crystal system classifying the lattice (e.g. "Trigonal", "Triclinic").
    pub crystal_system: String,

    /// The crystallographic point group notation.
    pub point_group: String,

    /// The space group symbol (e.g. "R3c").
    pub space_group_symbol: String,

    /// The standard space group number (1 to 230).
    pub space_group_number: i32,

    /// The fully expanded Cartesian Piezoelectric Stress Tensor ($e_{ijk}$) (3x3x3 array).
    ///
    /// Represents polarization generated per unit strain.
    pub e_tensor: Vec<Vec<Vec<f64>>>,

    /// The Voigt-reduced Piezoelectric Stress Tensor ($e_{i\alpha}$) (3x6 matrix).
    pub e_voigt: Vec<Vec<f64>>,

    /// The fully expanded Cartesian Elastic Stiffness Tensor ($C_{ijkl}$) (3x3x3x3 array).
    ///
    /// Derived from the `elasticity` dataset in GPa.
    pub cc_tensor: Vec<Vec<Vec<Vec<f64>>>>,

    /// The Voigt-reduced Elastic Stiffness Tensor ($C_{\alpha\beta}$) (6x6 matrix).
    pub cc_voigt: Vec<Vec<f64>>,

    //
    // Dielectric permittivity tensors ($\epsilon_{ij}$) (3x3 Cartesian array).
    //
    /// Extracted directly from the `dielectric` dataset. Represents relative permittivity (dimensionless).
    pub relative_permittivity_tensor: Vec<Vec<f64>>,
}

impl Material {
    /// Returns information about this material
    pub fn info(&self) -> String {
        format!(
            "{} ({}) : Crystal System {} : Space Group {}",
            self.formula, self.material_id, self.crystal_system, self.space_group_symbol
        )
    }

    /// Returns the p, e, and C moduli
    ///
    /// Returns `(p, e, cc)` where:
    ///
    /// * `p` -- Dielectric permittivity tensor in F/m
    /// * `e` -- Piezoelectric stress tensor in C/m^2
    /// * `cc` -- elastic stiffness tensor in Pa
    pub fn moduli(&self) -> Result<(Tensor2<6>, Tensor3<3, 6>, Tensor4<6>), StrError> {
        let mut p = Tensor2::<6>::from_std_matrix(&symmetrize3(&self.relative_permittivity_tensor))?;
        let e = Tensor3::<3, 6>::from_std_array(&vec_to_std_array_3(&self.e_tensor))?;
        let mut cc = Tensor4::<6>::from_std_array(&vec_to_std_array_4(&self.cc_tensor))?;
        p.scale(VACUUM_ELECTRIC_PERMITTIVITY);
        cc.scale(1e9); // GPa to Pa
        Ok((p, e, cc))
    }
}

/// Converts a Vec-based 3×3×3 array into a fixed-size `[[[f64; 3]; 3]; 3]`
fn vec_to_std_array_3(v: &[Vec<Vec<f64>>]) -> [[[f64; 3]; 3]; 3] {
    let mut a = [[[0.0; 3]; 3]; 3];
    for i in 0..3 {
        for j in 0..3 {
            for k in 0..3 {
                a[i][j][k] = v[i][j][k];
            }
        }
    }
    a
}

/// Converts a Vec-based 3×3×3×3 array into a fixed-size `[[[[f64; 3]; 3]; 3]; 3]`
fn vec_to_std_array_4(v: &[Vec<Vec<Vec<f64>>>]) -> [[[[f64; 3]; 3]; 3]; 3] {
    let mut a = [[[[0.0; 3]; 3]; 3]; 3];
    for i in 0..3 {
        for j in 0..3 {
            for k in 0..3 {
                for l in 0..3 {
                    a[i][j][k][l] = v[i][j][k][l];
                }
            }
        }
    }
    a
}

/// Symmetrizes a Vec-based 3×3 matrix by averaging the off-diagonal components
fn symmetrize3(v: &[Vec<f64>]) -> [[f64; 3]; 3] {
    let mut a = [[0.0; 3]; 3];
    for i in 0..3 {
        for j in 0..3 {
            a[i][j] = 0.5 * (v[i][j] + v[j][i]);
        }
    }
    a
}

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

#[cfg(test)]
mod tests {
    use super::*;
    use russell_lab::approx_eq;
    use std::env;
    use std::path::PathBuf;

    #[test]
    fn test_parse_piezo_data_json() {
        // get the asset's full path (the JSON file is in the crate's data/ directory)
        let root = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
        let path = root.join("data/piezo_data.json");

        let db = PiezoDatabase::from_file(path).expect("Failed to parse JSON database");

        // Test metadata
        assert_eq!(db.metadata.voigt_mapping, "11, 22, 33, 23, 13, 12");

        // Test parsing a specific material (e.g. LiNbO3 -> mp-3731)
        assert!(db.materials.contains_key("mp-3731"), "Missing mp-3731");
        let linbo3 = &db.materials["mp-3731"];
        assert_eq!(linbo3.material_id, "mp-3731");
        assert_eq!(linbo3.formula, "LiNbO3");
        assert_eq!(linbo3.crystal_system, "Trigonal");
        assert_eq!(linbo3.space_group_number, 161);

        // Check tensor structures
        let e_voigt = &linbo3.e_voigt;
        assert_eq!(e_voigt.len(), 3);
        assert_eq!(e_voigt[0].len(), 6);

        let e_tensor = &linbo3.e_tensor;
        assert_eq!(e_tensor.len(), 3);
        assert_eq!(e_tensor[0].len(), 3);
        assert_eq!(e_tensor[0][0].len(), 3);

        let cc_voigt = &linbo3.cc_voigt;
        assert_eq!(cc_voigt.len(), 6);
        assert_eq!(cc_voigt[0].len(), 6);
    }

    #[test]
    fn material_moduli_works() {
        let root = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
        let path = root.join("data/piezo_data.json");
        let db = PiezoDatabase::from_file(path).expect("Failed to parse JSON database");

        let mat = db.get("mp-3731").unwrap();
        let (p, e, cc) = mat.moduli().unwrap();
        let linbo3 = &db.materials["mp-3731"];

        // dielectric permittivity
        for i in 0..3 {
            for j in 0..3 {
                approx_eq(
                    p.get_std(i, j),
                    VACUUM_ELECTRIC_PERMITTIVITY * linbo3.relative_permittivity_tensor[i][j],
                    1e-13,
                );
            }
        }

        // piezoelectric stress tensor
        for i in 0..3 {
            for j in 0..3 {
                for k in 0..3 {
                    approx_eq(e.get_std(i, j, k), linbo3.e_tensor[i][j][k], 1e-13);
                }
            }
        }

        // elastic stiffness tensor (Pa)
        for i in 0..3 {
            for j in 0..3 {
                for k in 0..3 {
                    for l in 0..3 {
                        approx_eq(cc.get_std(i, j, k, l), 1e9 * linbo3.cc_tensor[i][j][k][l], 1e-4);
                    }
                }
            }
        }
    }
}