diff --git a/src/rstd/singular.rs b/src/rstd/singular.rs index cdb7e0c..262dd05 100644 --- a/src/rstd/singular.rs +++ b/src/rstd/singular.rs @@ -1,3 +1,4 @@ +use crate::func::context::*; use crate::rcore::*; use super::*; @@ -39,6 +40,10 @@ pub struct SingularResolver<'a, Ctx: Context<'a>> { points: Vec>>, } +pub struct SingularRobust<'a, Ctx: Context<'a>> { + resolver: Arc>, +} + #[derive(Debug)] pub enum SingularityError { OutOfBounds { @@ -108,6 +113,21 @@ impl<'a, Ctx: Context<'a>> Resolver<'a, Ctx> for SingularResolver<'a, Ctx> { } } +trait SingularCtx<'a>: Context<'a> { + fn from_singular(error: SingularityError) -> Self::LookupError; +} + +impl<'a, Ctx: SingularCtx<'a>> Resolver<'a, Ctx> for SingularRobust<'a, Ctx> { + fn resolve(self: Arc, address: Address) -> HashResolution<'a, Ctx> { + self.resolver + .clone() + .resolve_robust(address) + .map_err(Ctx::from_singular) + .map_err(Err) + .unwrap_or_else(Ctx::pure) + } +} + impl<'a, Ctx: Context<'a>> PointsVisitor<'a, Ctx> for Vec>> { fn visit>(&mut self, point: &Point<'a, Ctx, A>) { self.push(Arc::new(point.clone()) as _);