numpy-like functions ones and zeros

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -532,3 +532,48 @@ VALUE cmatrix_dense_FFLDU(VALUE self)
 
     return result;
 }
+
+
+VALUE cmatrix_dense_ones(VALUE self, VALUE r, VALUE c)
+{
+    unsigned long int r_ = NUM2ULONG(r);
+    unsigned long int c_ = NUM2ULONG(c);
+
+    CDenseMatrix *cresult;
+    cresult = dense_matrix_new();
+    VALUE result;
+
+    dense_matrix_ones(cresult, r_, c_);
+
+    result = Data_Wrap_Struct(c_dense_matrix, NULL, dense_matrix_free,
+                              cresult);
+
+    return result;
+}
+
+VALUE cmatrix_dense_zeros(VALUE self, VALUE r, VALUE c)
+{
+    unsigned long int r_ = NUM2ULONG(r);
+    unsigned long int c_ = NUM2ULONG(c);
+
+    CDenseMatrix *cresult;
+    cresult = dense_matrix_new();
+    VALUE result;
+
+    dense_matrix_zeros(cresult, r_, c_);
+
+    result = Data_Wrap_Struct(c_dense_matrix, NULL, dense_matrix_free,
+                              cresult);
+
+    return result;
+}
+/*
+VALUE cmatrix_dense_diag(VALUE self, VALUE args)
+{
+
+}
+
+VALUE cmatrix_dense_eye(VALUE self, VALUE args)
+{
+
+}*/

ext/symengine/ruby_matrix.h

@@ -27,6 +27,10 @@ VALUE cmatrix_dense_LDL(VALUE self);
 VALUE cmatrix_dense_LU_solve(VALUE self, VALUE b);
 VALUE cmatrix_dense_FFLU(VALUE self);
 VALUE cmatrix_dense_FFLDU(VALUE self);
+VALUE cmatrix_dense_ones(VALUE self, VALUE r, VALUE c);
+VALUE cmatrix_dense_zeros(VALUE self, VALUE r, VALUE c);
+/*VALUE cmatrix_dense_diag(VALUE self, VALUE args);
+VALUE cmatrix_dense_eye(VALUE self, VALUE args);*/
 
 void cmatrix_sparse_free(void *ptr);
 VALUE cmatrix_sparse_alloc(VALUE klass);

ext/symengine/symengine.c

@@ -253,6 +253,12 @@ void Init_symengine()
     rb_define_method(c_dense_matrix, "FFLU", cmatrix_dense_FFLU, 0);
     rb_define_method(c_dense_matrix, "FFLDU", cmatrix_dense_FFLDU, 0);
 
+    // Numpy-like methods for DenseMatrix as module level methods
+    rb_define_module_function(m_symengine, "ones", cmatrix_dense_ones, 2);
+    rb_define_module_function(m_symengine, "zeros", cmatrix_dense_zeros, 2);
+    /*rb_define_module_function(m_symengine, "diag", cmatrix_dense_diag, -2);
+    rb_define_module_function(m_symengine, "eye", cmatrix_dense_eye, -2);*/
+    
 
     // SparseMatrix Methods
     rb_define_alloc_func(c_sparse_matrix, cmatrix_sparse_alloc);