+ operator for DenceMatrix

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -308,3 +308,55 @@ VALUE cmatrix_dense_submatrix(VALUE self, VALUE r1, VALUE c1, VALUE r2, VALUE c2
                               cresult);
     return result;
 }
+
+VALUE cmatrix_dense_rows(VALUE self)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CDenseMatrix, this);
+    return ULONG2NUM( dense_matrix_rows(this));
+}
+
+
+VALUE cmatrix_dense_cols(VALUE self)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CDenseMatrix, this);
+    return ULONG2NUM( dense_matrix_cols(this));
+}
+
+VALUE cmatrix_dense_add(VALUE self, VALUE operand)
+{
+    CDenseMatrix *this;
+    Data_Get_Struct(self, CDenseMatrix, this);
+    
+    CDenseMatrix *cresult;
+    VALUE result;
+
+    cresult = dense_matrix_new();
+    
+    char *s = rb_obj_classname(operand);
+    
+    if(strcmp(s, "SymEngine::DenseMatrix") == 0) {
+    // Matrix Addition
+        CDenseMatrix *coperand;
+        Data_Get_Struct(operand, CDenseMatrix, coperand);
+        
+        dense_matrix_add_matrix(cresult, this, coperand);
+        
+        result = Data_Wrap_Struct(c_dense_matrix, NULL, dense_matrix_free,
+                              cresult);
+                              
+        dense_matrix_free(coperand);
+    } else {
+    // Scalar Addition
+        basic_struct *coperand = basic_new_heap();
+        sympify(operand, coperand);
+        
+        dense_matrix_add_scalar(cresult, this, coperand);
+        result = Data_Wrap_Struct(c_dense_matrix, NULL, dense_matrix_free,
+                              cresult);
+        basic_free_heap(coperand);
+    }
+    
+    return result;
+}

ext/symengine/ruby_matrix.h

@@ -18,6 +18,10 @@ VALUE cmatrix_dense_det(VALUE self);
 VALUE cmatrix_dense_inv(VALUE self);
 VALUE cmatrix_dense_transpose(VALUE self);
 VALUE cmatrix_dense_submatrix(VALUE self, VALUE r1, VALUE c1, VALUE r2, VALUE c2, VALUE r_, VALUE c_);
+VALUE cmatrix_dense_rows(VALUE self);
+VALUE cmatrix_dense_cols(VALUE self);
+VALUE cmatrix_dense_add(VALUE self, VALUE operand);
+VALUE cmatrix_dense_mul(VALUE self, VALUE operand);
 
 void cmatrix_sparse_free(void *ptr);
 VALUE cmatrix_sparse_alloc(VALUE klass);

ext/symengine/symengine.c

@@ -243,6 +243,11 @@ void Init_symengine()
     rb_define_method(c_dense_matrix, "inv", cmatrix_dense_inv, 0);
     rb_define_method(c_dense_matrix, "transpose", cmatrix_dense_transpose, 0);
     rb_define_method(c_dense_matrix, "submatrix", cmatrix_dense_submatrix, 6);
+    rb_define_method(c_dense_matrix, "rows", cmatrix_dense_rows, 0);
+    rb_define_method(c_dense_matrix, "cols", cmatrix_dense_cols, 0);
+    rb_define_method(c_dense_matrix, "+", cmatrix_dense_add, 1);
+    //rb_define_method(c_dense_matrix, "*", cmatrix_dense_mul, 1);
+    
 
     // SparseMatrix Methods
     rb_define_alloc_func(c_sparse_matrix, cmatrix_sparse_alloc);