printing for matrix like objects (except for LLSparseMatrixView)

Author: counterclocker

cysparse/sparse/csc_mat_matrices/csc_mat.cpx

@@ -12,6 +12,8 @@ from cysparse.sparse.s_mat cimport unexposed_value
 from cysparse.sparse.s_mat_matrices.s_mat_@index@_@type@ cimport ImmutableSparseMatrix_@index@_@type@
 from cysparse.sparse.ll_mat_matrices.ll_mat_@index@_@type@ cimport LLSparseMatrix_@index@_@type@
 
+from cysparse.sparse.sparse_utils.generic.print_@type@ cimport element_to_string_@type@, conjugated_element_to_string_@type@, empty_to_string_@type@
+
 ########################################################################################################################
 # Cython, NumPy import/cimport
 ########################################################################################################################
@@ -377,9 +379,41 @@ YOU HAVE TO EXPLICITY CAST YOUR TYPE
     ####################################################################################################################
     # String representations
     ####################################################################################################################
-    def __repr__(self):
-        s = "CSCSparseMatrix of size %d by %d with %d non zero values" % (self.nrow, self.ncol, self.nnz)
-        return s
+    def at_to_string(self, @index@ i, @index@ j, int cell_width=10):
+        """
+        Return a string with a given element if it exists or an "empty" string.
+
+
+        """
+        cdef:
+            @index@ k
+
+        for k from self.ind[j] <= k < self.ind[j+1]:
+            if i == self.row[k]:
+                return element_to_string_@type@(self.val[k], cell_width=cell_width)
+
+        # element not found -> return empty cell
+        return empty_to_string_@type@(cell_width=cell_width)
+
+    def at_conj_to_string(self, @index@ i, @index@ j, int cell_width=10):
+        """
+        Return a string with a given element if it exists or an "empty" string.
+
+
+        """
+        cdef:
+            @index@ k
+
+        for k from self.ind[j] <= k < self.ind[j+1]:
+            if i == self.row[k]:
+                return conjugated_element_to_string_@type@(self.val[k], cell_width=cell_width)
+
+        # element not found -> return empty cell
+        return empty_to_string_@type@(cell_width=cell_width)
+
+    #def __repr__(self):
+    #    s = "CSCSparseMatrix of size %d by %d with %d non zero values" % (self.nrow, self.ncol, self.nnz)
+    #    return s
 
     def print_to(self, OUT):
         """

cysparse/sparse/csc_mat_matrices/csc_mat_INT32_t_COMPLEX128_t.c

@@ -12,6 +12,8 @@ from cysparse.sparse.s_mat cimport unexposed_value
 from cysparse.sparse.s_mat_matrices.s_mat_INT32_t_COMPLEX128_t cimport ImmutableSparseMatrix_INT32_t_COMPLEX128_t
 from cysparse.sparse.ll_mat_matrices.ll_mat_INT32_t_COMPLEX128_t cimport LLSparseMatrix_INT32_t_COMPLEX128_t
 
+from cysparse.sparse.sparse_utils.generic.print_COMPLEX128_t cimport element_to_string_COMPLEX128_t, conjugated_element_to_string_COMPLEX128_t, empty_to_string_COMPLEX128_t
+
 ########################################################################################################################
 # Cython, NumPy import/cimport
 ########################################################################################################################
@@ -369,9 +371,41 @@ cdef class CSCSparseMatrix_INT32_t_COMPLEX128_t(ImmutableSparseMatrix_INT32_t_CO
     ####################################################################################################################
     # String representations
     ####################################################################################################################
-    def __repr__(self):
-        s = "CSCSparseMatrix of size %d by %d with %d non zero values" % (self.nrow, self.ncol, self.nnz)
-        return s
+    def at_to_string(self, INT32_t i, INT32_t j, int cell_width=10):
+        """
+        Return a string with a given element if it exists or an "empty" string.
+
+
+        """
+        cdef:
+            INT32_t k
+
+        for k from self.ind[j] <= k < self.ind[j+1]:
+            if i == self.row[k]:
+                return element_to_string_COMPLEX128_t(self.val[k], cell_width=cell_width)
+
+        # element not found -> return empty cell
+        return empty_to_string_COMPLEX128_t(cell_width=cell_width)
+
+    def at_conj_to_string(self, INT32_t i, INT32_t j, int cell_width=10):
+        """
+        Return a string with a given element if it exists or an "empty" string.
+
+
+        """
+        cdef:
+            INT32_t k
+
+        for k from self.ind[j] <= k < self.ind[j+1]:
+            if i == self.row[k]:
+                return conjugated_element_to_string_COMPLEX128_t(self.val[k], cell_width=cell_width)
+
+        # element not found -> return empty cell
+        return empty_to_string_COMPLEX128_t(cell_width=cell_width)
+
+    #def __repr__(self):
+    #    s = "CSCSparseMatrix of size %d by %d with %d non zero values" % (self.nrow, self.ncol, self.nnz)
+    #    return s
 
     def print_to(self, OUT):
         """