Changeset 587 for inundation/ga/storm_surge/pyvolution/sparse.py

Timestamp:

Nov 17, 2004, 11:40:28 PM (20 years ago)

Author:

steve

Message:

testing sparse

File:

• inundation/ga/storm_surge/pyvolution/sparse.py (modified) (2 diffs)

inundation/ga/storm_surge/pyvolution/sparse.py

@@ -302 +302 @@
             print 'FIXME: Only Numeric types implemented so far'
 
+
+##        R = csr_mv(self,B)
+        
         #Assume numeric types from now on
         
@@ -337 +340 @@
 
         return R
+
+def csr_mv_python(self, B):
+
+    from Numeric import zeros, Float
     
+    if len(B.shape) == 1:
+        #Vector
+        assert B.shape[0] == self.N, 'Mismatching dimensions'
+
+        R = zeros(self.M, Float) #Result
+            
+        #Multiply nonzero elements
+        for i in range(self.M):
+            for ckey in range(self.row_ptr[i],self.row_ptr[i+1]):
+                j = self.colind[ckey]
+                R[i] += self.data[ckey]*B[j]            
+
+        return R
+    else:
+        raise ValueError, 'Dimension too high: d=%d' %len(B.shape)
+
+    return R
+
+
+import compile
+if compile.can_use_C_extension('sparse_ext.c'):
+    #Replace python version with c implementations      
+    from sparse_ext import csr_mv
+
 
 if __name__ == '__main__':