Implement RawDataSubst for CowRepr

rust-ndarray · Jun 1, 2021 · 01d6713 · 01d6713
1 parent e7600e8
commit 01d6713
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/data_traits.rs b/src/data_traits.rs
@@ -675,3 +675,13 @@ impl<'a, A: 'a, B: 'a> RawDataSubst<B> for ViewRepr<&'a mut A> {
     }
 }
 
+impl<'a, A: 'a, B: 'a> RawDataSubst<B> for CowRepr<'a, A> {
+    type Output = CowRepr<'a, B>;
+
+    unsafe fn data_subst(self) -> Self::Output {
+        match self {
+            CowRepr::View(view) => CowRepr::View(view.data_subst()),
+            CowRepr::Owned(owned) => CowRepr::Owned(owned.data_subst()),
+        }
+    }
+}