use na;
use na::Pnt3;
use super::{TriMesh, IndexBuffer};
use super::{sphere, utils};
use math::Scalar;
pub fn capsule<N>(caps_diameter: &N,
cylinder_height: &N,
ntheta_subdiv: u32,
nphi_subdiv: u32)
-> TriMesh<Pnt3<N>>
where N: Scalar {
let top = sphere::unit_hemisphere::<N>(ntheta_subdiv, nphi_subdiv);
let TriMesh { coords, normals, indices, .. } = top.clone();
let mut bottom_coords = coords;
let mut bottom_normals = normals.unwrap();
let mut bottom_indices = indices.unwrap_unified();
utils::reverse_clockwising(&mut bottom_indices[..]);
let TriMesh { coords, normals, indices, .. } = top;
let mut top_coords = coords;
let top_normals = normals.unwrap();
let mut top_indices = indices.unwrap_unified();
let half_height = *cylinder_height * na::cast(0.5);
for coord in top_coords.iter_mut() {
coord.x = coord.x * *caps_diameter;
coord.y = coord.y * *caps_diameter + half_height;
coord.z = coord.z * *caps_diameter;
}
for coord in bottom_coords.iter_mut() {
coord.x = coord.x * *caps_diameter;
coord.y = -(coord.y * *caps_diameter) - half_height;
coord.z = coord.z * *caps_diameter;
}
for normal in bottom_normals.iter_mut() {
normal.y = -normal.y;
}
let base_top_coords = bottom_coords.len() as u32;
for idx in top_indices.iter_mut() {
idx.x = idx.x + base_top_coords;
idx.y = idx.y + base_top_coords;
idx.z = idx.z + base_top_coords;
}
bottom_coords.extend(top_coords.into_iter());
bottom_normals.extend(top_normals.into_iter());
bottom_indices.extend(top_indices.into_iter());
utils::push_ring_indices(0, base_top_coords, ntheta_subdiv, &mut bottom_indices);
TriMesh::new(bottom_coords,
Some(bottom_normals),
None,
Some(IndexBuffer::Unified(bottom_indices)))
}