use crate::{
objects::{Cycle, HalfEdge},
operations::{derive::DeriveFrom, insert::Insert},
storage::Handle,
Core,
};
pub trait UpdateCycle {
#[must_use]
fn add_half_edges<T>(
&self,
half_edges: impl IntoIterator<Item = T>,
core: &mut Core,
) -> Self
where
T: Insert<Inserted = Handle<HalfEdge>>;
#[must_use]
fn update_half_edge<T, R>(
&self,
handle: &Handle<HalfEdge>,
update: impl FnOnce(&Handle<HalfEdge>, &mut Core) -> R,
core: &mut Core,
) -> Self
where
T: Insert<Inserted = Handle<HalfEdge>>,
R: IntoIterator<Item = T>;
}
impl UpdateCycle for Cycle {
fn add_half_edges<T>(
&self,
half_edges: impl IntoIterator<Item = T>,
core: &mut Core,
) -> Self
where
T: Insert<Inserted = Handle<HalfEdge>>,
{
let half_edges = half_edges
.into_iter()
.map(|half_edge| half_edge.insert(core));
let half_edges = self.half_edges().iter().cloned().chain(half_edges);
Cycle::new(half_edges)
}
fn update_half_edge<T, R>(
&self,
handle: &Handle<HalfEdge>,
update: impl FnOnce(&Handle<HalfEdge>, &mut Core) -> R,
core: &mut Core,
) -> Self
where
T: Insert<Inserted = Handle<HalfEdge>>,
R: IntoIterator<Item = T>,
{
let edges = self
.half_edges()
.replace(
handle,
update(handle, core).into_iter().map(|object| {
object.insert(core).derive_from(handle, core)
}),
)
.expect("Half-edge not found");
Cycle::new(edges)
}
}