All numpy-like functions wrapped for DenseMatrix

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -614,8 +614,34 @@ VALUE cmatrix_dense_diag(VALUE self, VALUE args)
 
     return result;
 }
-/*
+
 VALUE cmatrix_dense_eye(VALUE self, VALUE args)
 {
+    int argc = RARRAY_LEN(args);
+
+    if (argc != 1 && argc != 3){
+        rb_raise(rb_eTypeError, "Wrong number of arguments");
+    }
+
+    CDenseMatrix *cresult;
+    cresult = dense_matrix_new();
+    VALUE result;
 
-}*/
+    VALUE operand = rb_ary_shift(args);
+    unsigned long int N = NUM2ULONG(operand);
+    unsigned long int M = N;
+    int k = 0;
+
+    if(argc == 3) {
+        operand = rb_ary_shift(args);
+        M = NUM2ULONG(operand);
+        operand = rb_ary_shift(args);
+        k = NUM2INT(operand);
+    }
+
+    dense_matrix_eye(cresult, N, M, k);
+
+    result = Data_Wrap_Struct(c_dense_matrix, NULL, dense_matrix_free,
+                              cresult);
+    return result;
+}

ext/symengine/ruby_matrix.h

@@ -29,8 +29,8 @@ 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);*/
+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

@@ -257,7 +257,7 @@ void Init_symengine()
     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);*/
+    rb_define_module_function(m_symengine, "eye", cmatrix_dense_eye, -2);
 
 
     // SparseMatrix Methods