FunctionSymbol class and its methods

Author: rajithv

ext/symengine/ruby_function.c

@@ -37,3 +37,31 @@ IMPLEMENT_ONE_ARG_FUNC(gamma);
 
 #undef IMPLEMENT_ONE_ARG_FUNC
 
+VALUE cfunction_functionsymbol_init(VALUE self, VALUE args)
+{
+    int argc = NUM2INT(rb_funcall(args, rb_intern("length"), 0, NULL));
+    printf("%d\n", argc);
+    VALUE first = rb_ary_shift(args);
+    if( TYPE(first) != T_STRING ){
+        rb_raise(rb_eTypeError, "String expected as first argument");
+    }
+    char *name = StringValueCStr( first );
+
+    basic_struct *cargs[argc];
+    int i;
+    for(i = 0; i < argc; i++){
+        cargs[i] = basic_new_heap();
+        sympify(rb_ary_shift(args), cargs[i]);
+    }
+    
+    basic_struct *this;
+    Data_Get_Struct(self, basic_struct, this);
+
+    function_symbol_set(this, name, cargs);
+    
+    for(i = 0; i < argc; i++){
+        basic_free_heap(cargs[i]);
+    }
+    return self;
+}
+

ext/symengine/ruby_function.h

@@ -37,4 +37,6 @@ VALUE cfunction_dirichlet_eta(VALUE self, VALUE operand1);
 VALUE cfunction_zeta(VALUE self, VALUE operand1);
 VALUE cfunction_gamma(VALUE self, VALUE operand1);
 
+VALUE cfunction_functionsymbol_init(VALUE self, VALUE args);
+
 #endif //RUBY_FUNCTION_H_

ext/symengine/symengine.c

@@ -113,10 +113,12 @@ void Init_symengine() {
     c_dirichlet_eta = rb_define_class_under(m_symengine, "Dirichlet_eta", c_function);
     c_zeta = rb_define_class_under(m_symengine, "Zeta", c_function);
     c_gamma = rb_define_class_under(m_symengine, "Gamma", c_function);
-
-    //Abs Class
     c_abs = rb_define_class_under(m_symengine, "Abs", c_function);
 
+    //FunctionSymbol Class
+    c_function_symbol = rb_define_class_under(m_symengine, "FunctionSymbol", c_function);
+    rb_define_method(c_function_symbol, "initialize", cfunction_functionsymbol_init, -2);
+
     //TrigFunction SubClasses
     c_sin = rb_define_class_under(m_symengine, "Sin", c_trig_function);
     c_cos = rb_define_class_under(m_symengine, "Cos", c_trig_function);

ext/symengine/symengine.h

@@ -19,6 +19,7 @@ VALUE c_add;
 VALUE c_mul;
 VALUE c_pow;
 VALUE c_function;
+VALUE c_function_symbol;
 VALUE c_trig_function;
 VALUE c_hyperbolic_function;
 VALUE c_lambertw;

ext/symengine/symengine_utils.c

@@ -101,6 +101,8 @@ VALUE Klass_of_Basic(const basic_struct *basic_ptr) {
             return c_mul;
         case SYMENGINE_POW:
             return c_pow;
+        case SYMENGINE_FUNCTIONSYMBOL:
+            return c_function_symbol;
         case SYMENGINE_ABS:
             return c_abs;
         case SYMENGINE_SIN: