WIP WIP WIP WIP Scaled

Author: bluss

src/impl_ops.rs

@@ -355,4 +355,28 @@ mod assign_ops {
                     "Perform `self <<= rhs` as elementwise left shift (in place).\n");
     impl_assign_op!(ShrAssign, shr_assign,
                     "Perform `self >>= rhs` as elementwise right shift (in place).\n");
+
+    use LinalgScalar;
+    use Scaled;
+    /// If their shapes disagree, `rhs` is broadcast to the shape of `self`.
+    ///
+    /// **Panics** if broadcasting isn’t possible.
+    ///
+    /// **Requires crate feature `"assign_ops"`**
+    impl<'a, A, S, S2, D, E> AddAssign<Scaled<'a, S2, E>> for ArrayBase<S, D>
+        where A: LinalgScalar,
+              S: DataMut<Elem=A>,
+              S2: Data<Elem=A>,
+              D: Dimension,
+              E: Dimension,
+    {
+        fn add_assign(&mut self, rhs: Scaled<'a, S2, E>) {
+            self.zip_mut_with(rhs.1, |x, y| {
+                *x = x.clone() + rhs.0 * y.clone();
+            });
+        }
+    }
+
+
+
 }

src/lib.rs

@@ -722,3 +722,16 @@ enum ElementsRepr<S, C> {
     Slice(S),
     Counted(C),
 }
+
+pub struct Scaled<'a, S, D>(pub S::Elem, pub &'a ArrayBase<S, D>)
+    where S: 'a + Data,
+          D: 'a;
+
+impl<'a, S, D> Clone for Scaled<'a, S, D>  where S::Elem: Clone, S: Data {
+    #[inline]
+    fn clone(&self) -> Self {
+        Scaled(self.0.clone(), self.1)
+    }
+}
+impl<'a, S, D> Copy for Scaled<'a, S, D>  where S::Elem: Copy, S: Data { }
+