equal and not equal operators for DenseMatrix

Author: rajithv

ext/symengine/ruby_matrix.c

@@ -404,6 +404,55 @@ VALUE cmatrix_dense_mul(VALUE self, VALUE operand)
     return result;
 }
 
+VALUE cmatrix_dense_eq(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) {
+        CDenseMatrix *coperand;
+        Data_Get_Struct(operand, CDenseMatrix, coperand);
+        
+        result = dense_matrix_eq(this, coperand) == 1 ? Qtrue : Qfalse;
+                              
+        dense_matrix_free(coperand);
+    } else {
+        result = Qfalse;
+    }
+    return result;
+}
+
+VALUE cmatrix_dense_neq(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) {
+        CDenseMatrix *coperand;
+        Data_Get_Struct(operand, CDenseMatrix, coperand);
+        
+        result = dense_matrix_eq(this, coperand) == 1 ? Qfalse : Qtrue;
+                              
+        dense_matrix_free(coperand);
+    } else {
+        result = Qtrue;
+    }
+    return result;
+}
 
 VALUE cmatrix_dense_LU(VALUE self)
 {

ext/symengine/ruby_matrix.h

@@ -22,6 +22,8 @@ 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);
+VALUE cmatrix_dense_eq(VALUE self, VALUE operand);
+VALUE cmatrix_dense_neq(VALUE self, VALUE operand);
 VALUE cmatrix_dense_LU(VALUE self);
 VALUE cmatrix_dense_LDL(VALUE self);
 VALUE cmatrix_dense_LU_solve(VALUE self, VALUE b);

ext/symengine/symengine.c

@@ -247,6 +247,9 @@ void Init_symengine()
     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);
+    rb_define_method(c_dense_matrix, "==", cmatrix_dense_eq, 1);
+    rb_define_method(c_dense_matrix, "eql?", cmatrix_dense_eq, 1);
+    rb_define_method(c_dense_matrix, "!=", cmatrix_dense_neq, 1);
     rb_define_method(c_dense_matrix, "LU", cmatrix_dense_LU, 0);
     rb_define_method(c_dense_matrix, "LDL", cmatrix_dense_LDL, 0);
     rb_define_method(c_dense_matrix, "LU_solve", cmatrix_dense_LU_solve, 1);