Real part and Imaginary part wrappers

Author: rajithv

ext/symengine/CMakeLists.txt

@@ -2,6 +2,7 @@ set(RUBY_WRAPPER_SRC
     ruby_basic.c
     ruby_symbol.c
     ruby_integer.c
+    ruby_complex.c
     ruby_constant.c
     ruby_function.c
     ruby_ntheory.c

ext/symengine/ruby_complex.c

@@ -0,0 +1,37 @@
+#include "ruby_complex.h"
+
+VALUE ccomplex_real_part(VALUE self) {
+    basic_struct *c_result = basic_new_heap();
+
+    basic cbasic_operand;
+    basic_new_stack(cbasic_operand);
+
+    VALUE result;
+
+    sympify(self, cbasic_operand);
+    complex_real_part(c_result, cbasic_operand);
+
+    result = Data_Wrap_Struct(Klass_of_Basic(c_result), NULL, cbasic_free_heap, c_result);
+    
+    basic_free_stack(cbasic_operand);
+
+    return result;
+}
+
+VALUE ccomplex_imaginary_part(VALUE self) {
+    basic_struct *c_result = basic_new_heap();
+
+    basic cbasic_operand;
+    basic_new_stack(cbasic_operand);
+
+    VALUE result;
+
+    sympify(self, cbasic_operand);
+    complex_imaginary_part(c_result, cbasic_operand);
+
+    result = Data_Wrap_Struct(Klass_of_Basic(c_result), NULL, cbasic_free_heap, c_result);
+    
+    basic_free_stack(cbasic_operand);
+
+    return result;
+}

ext/symengine/ruby_complex.h

@@ -0,0 +1,13 @@
+#ifndef RUBY_COMPLEX_H_
+#define RUBY_COMPLEX_H_
+
+#include <ruby.h>
+#include <symengine/cwrapper.h>
+
+#include "symengine.h"
+#include "symengine_utils.h"
+
+VALUE ccomplex_real_part(VALUE self);
+VALUE ccomplex_imaginary_part(VALUE self);
+
+#endif //RUBY_COMPLEX_H_

ext/symengine/symengine.c

@@ -3,6 +3,7 @@
 #include "ruby_symbol.h"
 #include "ruby_integer.h"
 #include "ruby_constant.h"
+#include "ruby_complex.h"
 #include "ruby_function.h"
 #include "ruby_ntheory.h"
 #include "ruby_utils.h"
@@ -67,6 +68,8 @@ void Init_symengine() {
     //Complex class
     c_complex = rb_define_class_under(m_symengine, "Complex", c_basic);
     rb_define_alloc_func(c_complex, cbasic_alloc);
+    rb_define_method(c_complex, "real", ccomplex_real_part, 0);
+    rb_define_method(c_complex, "imaginary", ccomplex_imaginary_part, 0);
 
     //ComplexDouble class
     c_complex_double = rb_define_class_under(m_symengine, "ComplexDouble", c_basic);

lib/symengine/complex.rb

@@ -4,4 +4,4 @@ def to_c
         self.to_s.sub('I', 'i').sub('*','').gsub(' ','').to_c
     end
   end
-end
+end 

spec/complex_spec.rb

@@ -13,12 +13,50 @@
       end
       context 'with a Ruby Integer as input' do
         it 'returns an instance of SymEngine::Integer class' do
-          a = Complex(2, 0);
+          a = Complex(2, 0)
           a = SymEngine::sympify(a)
           expect(a).to be_an_instance_of SymEngine::Integer
           expect(a.to_s).to eq('2')
         end
       end
+      context 'Obtaining the integer real part' do
+        it 'returns an instance of SymEngine::Integer class' do
+          a = Complex(2, 7)
+          a = SymEngine::S(a)
+          a = a.real
+          expect(a).to be_an_instance_of SymEngine::Integer
+          expect(a.to_s).to eq('2');
+        end
+      end
+      context 'Obtaining the integer imaginary part' do
+        it 'returns an instance of SymEngine::Integer class' do
+          a = Complex(2, 7)
+          a = SymEngine::S(a)
+          a = a.imaginary
+          expect(a).to be_an_instance_of SymEngine::Integer
+          expect(a.to_s).to eq('7');
+        end
+      end
+      context 'Obtaining the rational real part ' do
+        it 'returns an instance of SymEngine::Rational class' do
+          a = Rational('5/4')
+          a = Complex(a, 7)
+          a = SymEngine::S(a)
+          a = a.real
+          expect(a).to be_an_instance_of SymEngine::Rational
+          expect(a.to_s).to eq('5/4');
+        end
+      end
+      context 'Obtaining the rational imaginary part' do
+        it 'returns an instance of SymEngine::Integer class' do
+          a = Rational('5/4')
+          a = Complex(2, a)
+          a = SymEngine::S(a)
+          a = a.imaginary
+          expect(a).to be_an_instance_of SymEngine::Rational
+          expect(a.to_s).to eq('5/4');
+        end
+      end
     end
   end
 end

symengine_version.txt

@@ -1,2 +1 @@
-5e88e62b42c38200573eecf688d90b46beb38a77
-
+382a917700e6c6116e025e64baff66d0668c64f2