Check if matrix is column or row vector

1 files changed, 6 insertions, 0 deletions

diff --git a/matrix.h b/matrix.h
index dc09ef6..2546152 100644
--- a/matrix.h
+++ b/matrix.h
@@ -11,6 +11,12 @@
 
 #define matrix_foreach_idx(mat, it, i, j) double *it = (mat)->xs; for (size_t i = 0; i < (mat)->m; i++) for (size_t j = 0; j < (mat)->n; j++, ++it)
 
+#define matrix_is_colvec(mat) ((mat)->n == 1)
+
+#define matrix_is_rowvec(mat) ((mat)->m == 1)
+
+#define matrix_is_vec(mat) (matrix_is_colvec(mat) || matrix_is_rowvec(mat))
+
 #define TRUE 1
 #define FALSE 0