Changeset 7886 for trunk/anuga_core

Timestamp:

Jul 1, 2010, 5:09:54 PM (15 years ago)

Author:

steve

Message:

Copied sudi's directory

Location:

trunk/anuga_core/source/anuga/utilities

Files:

• sparse.py (modified) (2 diffs)
• util_ext.h (modified) (1 diff)

trunk/anuga_core/source/anuga/utilities/sparse.py

@@ -297 +297 @@
     def __mul__(self, other):
         """Multiply this matrix onto 'other' which can either be
-        a numeric vector, a numeric matrix or another sparse matrix.
+        a numeric vector.
         """
 
@@ -303 +303 @@
             B = num.array(other)
         except:
-            print 'FIXME: Only numeric types implemented so far'
-
-        return csr_mv(self,B) 
+            raise ValueError, 'FIXME: Only numeric types implemented so far'
+
+
+        # Assume numeric types from now on
+
+        if len(B.shape) == 0:
+            # Scalar - use __rmul__ method
+            #R = B*self
+            raise TypeError, 'Sparse matrix X scalar not implemented'
+
+        elif len(B.shape) == 1:
+            # Vector
+            msg = 'Mismatching dimensions: You cannot multiply (%d x %d) matrix onto %d-vector'\
+                  %(self.M, self.N, B.shape[0])
+            assert B.shape[0] == self.N, msg
+
+            R = csr_mv(self,B)
+
+
+        elif len(B.shape) == 2:
+            raise TypeError, 'Sparse matrix X matrix multiplication not implemented'
+
+        else:
+            raise ValueError, 'Dimension too high: d=%d' %len(B.shape)
+
+        return R
+
 
 

trunk/anuga_core/source/anuga/utilities/util_ext.h

@@ -31 +31 @@
 
 
+double sign(double x) {
+  //Return sign of a double
+
+  if (x>0.0) return 1.0;
+  else if (x<0.0) return -1.0;
+  else return 0.0;
+}
+
 int _gradient(double x0, double y0, 
               double x1, double y1,