Getter and setter for DenseMatrix

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -186,3 +186,46 @@ VALUE cmatrix_sparse_init(VALUE self, VALUE args)
 
     return self;
 }
+
+
+VALUE cmatrix_dense_get(VALUE self, VALUE r, VALUE c)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CDenseMatrix, this);
+    
+    basic_struct *cresult;
+    VALUE result;
+
+    unsigned long cbasic_r;
+    cbasic_r = NUM2ULONG(r);
+    unsigned long cbasic_c;
+    cbasic_c = NUM2ULONG(c);
+
+    cresult = basic_new_heap();
+    
+    dense_matrix_get_basic(cresult, this, cbasic_r, cbasic_c);
+    result = Data_Wrap_Struct(Klass_of_Basic(cresult), NULL, basic_free_heap,
+                              cresult);
+    return result;
+}
+
+VALUE cmatrix_dense_set(VALUE self, VALUE r, VALUE c, VALUE operand)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CDenseMatrix, this);
+    
+    basic cbasic_operand;
+    basic_new_stack(cbasic_operand);
+    sympify(operand, cbasic_operand);
+    
+    unsigned long cbasic_r;
+    cbasic_r = NUM2ULONG(r);
+    unsigned long cbasic_c;
+    cbasic_c = NUM2ULONG(c);
+    
+    dense_matrix_set_basic(this, cbasic_r, cbasic_c, cbasic_operand);
+    
+    basic_free_stack(cbasic_operand);
+    
+    return self;
+}

ext/symengine/ruby_matrix.h

@@ -4,10 +4,16 @@
 #include <ruby.h>
 #include <symengine/cwrapper.h>
 
+#include "symengine.h"
+#include "ruby_basic.h"
+#include "symengine_utils.h"
+
 void cmatrix_dense_free(void *ptr);
 VALUE cmatrix_dense_alloc(VALUE klass);
 VALUE cmatrix_dense_init(VALUE self, VALUE args);
 VALUE cmatrix_dense_to_str(VALUE self);
+VALUE cmatrix_dense_get(VALUE self, VALUE r, VALUE c);
+VALUE cmatrix_dense_set(VALUE self, VALUE r, VALUE c, VALUE operand);
 
 void cmatrix_sparse_free(void *ptr);
 VALUE cmatrix_sparse_alloc(VALUE klass);

ext/symengine/symengine.c

@@ -237,6 +237,8 @@ void Init_symengine()
     rb_define_alloc_func(c_dense_matrix, cmatrix_dense_alloc);
     rb_define_method(c_dense_matrix, "initialize", cmatrix_dense_init, -2);
     rb_define_method(c_dense_matrix, "to_s", cmatrix_dense_to_str, 0);
+    rb_define_method(c_dense_matrix, "get", cmatrix_dense_get, 2);
+    rb_define_method(c_dense_matrix, "set", cmatrix_dense_set, 3);
 
     // SparseMatrix Methods
     rb_define_alloc_func(c_sparse_matrix, cmatrix_sparse_alloc);