1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

/*!
 * This module defines routines to compute normals on various meshes.
 */

use math::Vector3;
use crate::prim::Triangle;
use math::{BaseNum};
use std::ops::Neg;

pub fn compute_vertex_area_weighted_normals<T, V3>(
    vertex_positions: &[V3],
    topo: &[[usize; 3]],
    normals: &mut [[T;3]],
) where
    T: BaseNum + Neg<Output = T>,
    V3: Into<[T; 3]> + Clone,
{
    // Clear the normals.
    for nml in normals.iter_mut() {
        *nml = [T::zero(); 3];
    }

    for tri_indices in topo.iter() {
        let tri = Triangle::from_indexed_slice(tri_indices, vertex_positions);
        let area_nml = Vector3::from(tri.area_normal());
        normals[tri_indices[0]] = (Vector3::from(normals[tri_indices[0]]) + area_nml).into();
        normals[tri_indices[1]] = (Vector3::from(normals[tri_indices[1]]) + area_nml).into();
        normals[tri_indices[2]] = (Vector3::from(normals[tri_indices[2]]) + area_nml).into();
    }
}