NTheory Wrappers

Author: rajithv

ext/symengine/ruby_ntheory.c

@@ -0,0 +1,58 @@
+#include "ruby_ntheory.h"
+
+VALUE cfunction_onearg(void (*cwfunc_ptr)(basic_struct*, const basic_struct*), VALUE operand1) {
+    basic_struct *cresult;
+    VALUE result;
+
+    basic cbasic_operand1;
+    basic_new_stack(cbasic_operand1);
+    sympify(operand1, cbasic_operand1);
+
+    cresult = basic_new_heap();
+    cwfunc_ptr(cresult, cbasic_operand1);
+    result = Data_Wrap_Struct(Klass_of_Basic(cresult), NULL , cbasic_free_heap, cresult);
+    basic_free_stack(cbasic_operand1);
+
+    return result;
+}
+
+VALUE cfunction_twoarg(void (*cwfunc_ptr)(basic_struct*, const basic_struct*, const basic_struct*), VALUE operand1, VALUE operand2) {
+    basic_struct *cresult;
+    VALUE result;
+
+    basic cbasic_operand1;
+    basic_new_stack(cbasic_operand1);
+    sympify(operand1, cbasic_operand1);
+
+    basic cbasic_operand2;
+    basic_new_stack(cbasic_operand2);
+    sympify(operand2, cbasic_operand2);
+
+    cresult = basic_new_heap();
+    cwfunc_ptr(cresult, cbasic_operand1, cbasic_operand2);
+    result = Data_Wrap_Struct(Klass_of_Basic(cresult), NULL , cbasic_free_heap, cresult);
+    basic_free_stack(cbasic_operand1);
+    basic_free_stack(cbasic_operand2);
+
+    return result;
+}
+
+VALUE cfunction_nextprime(VALUE self, VALUE operand1) {
+    return cfunction_onearg(nthoery_gcd, operand1);
+}
+
+VALUE cfunction_gcd(VALUE self, VALUE operand1, VALUE operand2) {
+    return cfunction_twoarg(ntheory_gcd, operand1, operand2);
+}
+
+VALUE cfunction_lcm(VALUE self, VALUE operand1, VALUE operand2) {
+    return cfunction_twoarg(ntheory_lcm, operand1, operand2);
+}
+
+VALUE cfunction_mod(VALUE self, VALUE operand1, VALUE operand2) {
+    return cfunction_twoarg(ntheory_mod, operand1, operand2);
+}
+
+VALUE cfunction_quotient(VALUE self, VALUE operand1, VALUE operand2) {
+    return cfunction_twoarg(ntheory_quotient, operand1, operand2);
+}

ext/symengine/ruby_ntheory.h

@@ -0,0 +1,23 @@
+#ifndef RUBY_NTHEORY_H_
+#define RUBY_NTHEORY_H_
+
+#include <ruby.h>
+#include <symengine/cwrapper.h>
+
+#include "ruby_basic.h"
+#include "symengine.h"
+#include "symengine_macros.h"
+
+VALUE cfunction_onearg(void (*cwfunc_ptr)(basic_struct*, const basic_struct*), VALUE operand1);
+VALUE cfunction_twoarg(void (*cwfunc_ptr)(basic_struct*, const basic_struct*, const basic_struct*), VALUE operand1, VALUE operand2);
+
+VALUE cfunction_gcd(VALUE self, VALUE operand1, VALUE operand2);
+VALUE cfunction_lcm(VALUE self, VALUE operand1, VALUE operand2);
+VALUE cfunction_nextprime(VALUE self, VALUE operand1);
+VALUE cfunction_mod(VALUE self, VALUE operand1, VALUE operand2);
+VALUE cfunction_quotient(VALUE self, VALUE operand1, VALUE operand2);
+//VALUE cfunction_fibonacci(VALUE self, VALUE operand1);
+//VALUE cfunction_lucas(VALUE self, VALUE operand1);
+//VALUE cfunction_binomial(VALUE self, VALUE operand1, VALUE operand2);
+
+#endif //RUBY_NTHEORY_H_