Merge pull request #1020 from jturner314/impl-RawDataSubst-for-CowRepr

Implement RawDataSubst for CowRepr
rust-ndarray · Jun 5, 2021 · 35e9ca2 · 35e9ca2
2 parents e7600e8 + 01d6713
commit 35e9ca2
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()),
+        }
+    }
+}