use crate::refiner::LevelDataCommon;
use crate::{Adjacency, KernelError, LineageMaps, Mesh};
use super::super::topology::{Topology, build_topology};
pub(crate) struct DooSabinLevelData {
pub(crate) mesh: Mesh,
pub(crate) topo: Topology,
pub(crate) face_selected: Vec<bool>,
pub(crate) effective_edge_creases: Vec<f32>,
pub(crate) lineage: LineageMaps,
pub(crate) adjacency: Adjacency,
}
impl LevelDataCommon for DooSabinLevelData {
fn mesh(&self) -> &Mesh {
&self.mesh
}
fn lineage(&self) -> &LineageMaps {
&self.lineage
}
fn face_selected(&self) -> &[bool] {
&self.face_selected
}
fn adjacency(&self) -> &Adjacency {
&self.adjacency
}
}
pub(super) fn compute_effective_edge_creases(
topology: &Topology,
face_selected: &[bool],
selection_boundary_crease: f32,
) -> Vec<f32> {
let edge_count = topology.edge_vertices.len();
let all_selected = face_selected.iter().all(|&s| s);
let edge_is_sel_boundary: Vec<bool> = if all_selected {
vec![false; edge_count]
} else {
(0..edge_count)
.map(|ei| {
let ef = topology.edge_faces.row(ei);
let s0 = ef
.first()
.map(|&fi| face_selected[fi as usize])
.unwrap_or(false);
let s1 = ef
.get(1)
.map(|&fi| face_selected[fi as usize])
.unwrap_or(false);
s0 != s1
})
.collect()
};
topology
.edge_creases
.iter()
.copied()
.enumerate()
.map(|(ei, base)| {
if edge_is_sel_boundary[ei] && selection_boundary_crease > 0.0 {
base.max(selection_boundary_crease)
} else {
base
}
})
.collect()
}
pub(crate) fn base_level_data(
topo: &Mesh,
face_selected: Vec<bool>,
selection_boundary_crease: f32,
) -> Result<DooSabinLevelData, KernelError> {
let topology = build_topology(topo)?;
let face_count = topology.faces.len();
if face_selected.len() != face_count {
return Err(KernelError::InvalidTopology(
"selected-face mask length does not match face count",
));
}
let effective_edge_creases =
compute_effective_edge_creases(&topology, &face_selected, selection_boundary_crease);
Ok(DooSabinLevelData {
mesh: topo.clone(),
topo: topology,
face_selected,
effective_edge_creases,
lineage: LineageMaps::default(),
adjacency: Adjacency {
face_edges: Vec::new(),
edge_faces: Vec::new(),
vertex_edge_offsets: Vec::new(),
vertex_edges: Vec::new(),
vertex_face_offsets: Vec::new(),
vertex_faces: Vec::new(),
edge_is_boundary: Vec::new(),
vertex_is_boundary: Vec::new(),
},
})
}