SparseMatrix get and set methods

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -645,3 +645,45 @@ VALUE cmatrix_dense_eye(VALUE self, VALUE args)
                               cresult);
     return result;
 }
+
+VALUE cmatrix_sparse_get(VALUE self, VALUE r, VALUE c)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CSparseMatrix, 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();
+    
+    sparse_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_sparse_set(VALUE self, VALUE r, VALUE c, VALUE operand)
+{
+    CSparseMatrix *this;
+    Data_Get_Struct(self, CSparseMatrix, 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);
+    
+    sparse_matrix_set_basic(this, cbasic_r, cbasic_c, cbasic_operand);
+    
+    basic_free_stack(cbasic_operand);
+    
+    return self;
+}

ext/symengine/ruby_matrix.h

@@ -36,5 +36,7 @@ void cmatrix_sparse_free(void *ptr);
 VALUE cmatrix_sparse_alloc(VALUE klass);
 VALUE cmatrix_sparse_init(VALUE self, VALUE args);
 VALUE cmatrix_sparse_to_str(VALUE self);
+VALUE cmatrix_sparse_get(VALUE self, VALUE r, VALUE c);
+VALUE cmatrix_sparse_set(VALUE self, VALUE r, VALUE c, VALUE operand);
 
 #endif // RUBY_MATRIX_H_

ext/symengine/symengine.c

@@ -264,6 +264,8 @@ void Init_symengine()
     rb_define_alloc_func(c_sparse_matrix, cmatrix_sparse_alloc);
     rb_define_method(c_sparse_matrix, "initialize", cmatrix_sparse_init, -2);
     rb_define_method(c_sparse_matrix, "to_s", cmatrix_sparse_to_str, 0);
+    rb_define_method(c_sparse_matrix, "get", cmatrix_sparse_get, 2);
+    rb_define_method(c_sparse_matrix, "set", cmatrix_sparse_set, 3);
 
 
     symengine_print_stack_on_segfault();