conspire 0.7.7

The Rust interface to conspire.
Documentation
use crate::{
    constitutive::{ConstitutiveError, solid::hyperelastic_viscoplastic::HyperelasticViscoplastic},
    fem::block::element::{
        Element, ElementNodalCoordinates, FiniteElement, FiniteElementError,
        solid::{
            SolidFiniteElement, elastic_viscoplastic::ElasticViscoplasticFiniteElement,
            viscoplastic::ViscoplasticStateVariables,
        },
    },
    math::{Differentiate, Quantity, Tensor},
    units::Energy,
};

pub trait HyperelasticViscoplasticFiniteElement<
    C,
    const G: usize,
    const M: usize,
    const N: usize,
    const P: usize,
    Y,
> where
    C: HyperelasticViscoplastic<Y>,
    Self: ElasticViscoplasticFiniteElement<C, G, M, N, P, Y>,
    Y: Differentiate + Tensor,
{
    fn helmholtz_free_energy(
        &self,
        constitutive_model: &C,
        nodal_coordinates: &ElementNodalCoordinates<N>,
        state_variables: &ViscoplasticStateVariables<G, Y>,
    ) -> Result<Quantity<Energy>, FiniteElementError>;
}

impl<C, const G: usize, const N: usize, const O: usize, const P: usize, Y>
    HyperelasticViscoplasticFiniteElement<C, G, 3, N, P, Y> for Element<3, G, N, O>
where
    C: HyperelasticViscoplastic<Y>,
    Self: ElasticViscoplasticFiniteElement<C, G, 3, N, P, Y>,
    Y: Differentiate + Tensor,
{
    fn helmholtz_free_energy(
        &self,
        constitutive_model: &C,
        nodal_coordinates: &ElementNodalCoordinates<N>,
        state_variables: &ViscoplasticStateVariables<G, Y>,
    ) -> Result<Quantity<Energy>, FiniteElementError> {
        self.deformation_gradients(nodal_coordinates)
            .iter()
            .zip(state_variables)
            .zip(self.integration_weights())
            .map(
                |((deformation_gradient, state_variable), integration_weight)| {
                    let (deformation_gradient_p, _) = state_variable.into();
                    Ok::<_, ConstitutiveError>(
                        constitutive_model.helmholtz_free_energy_density(
                            deformation_gradient,
                            deformation_gradient_p,
                        )? * integration_weight,
                    )
                },
            )
            .sum::<Result<_, ConstitutiveError>>()
            .map_err(|error| FiniteElementError::upstream(error, self))
    }
}