Added Classes for Function types

Author: rajithv

ext/symengine/symengine.c

@@ -69,6 +69,42 @@ void Init_symengine() {
     //Functions Class
     c_function = rb_define_class_under(m_symengine, "Function", c_basic);
 
+    //Function SubClasses
+    c_trig_function = rb_define_class_under(m_symengine, "TrigFunction", c_function);
+    c_hyperbolic_function = rb_define_class_under(m_symengine, "HyperbolicFunction", c_function);
+    c_lambertw = rb_define_class_under(m_symengine, "LambertW", c_function);
+    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);
+
+    //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);
+    c_tan = rb_define_class_under(m_symengine, "Tan", c_trig_function);
+    c_csc = rb_define_class_under(m_symengine, "Csc", c_trig_function);
+    c_sec = rb_define_class_under(m_symengine, "Sec", c_trig_function);
+    c_cot = rb_define_class_under(m_symengine, "Cot", c_trig_function);
+    c_asin = rb_define_class_under(m_symengine, "ASin", c_trig_function);
+    c_acos = rb_define_class_under(m_symengine, "ACos", c_trig_function);
+    c_atan = rb_define_class_under(m_symengine, "ATan", c_trig_function);
+    c_acsc = rb_define_class_under(m_symengine, "ACsc", c_trig_function);
+    c_asec = rb_define_class_under(m_symengine, "ASec", c_trig_function);
+    c_acot = rb_define_class_under(m_symengine, "ACot", c_trig_function);
+
+    //HyperbolicFunction SubClasses
+    c_sinh = rb_define_class_under(m_symengine, "Sinh", c_hyperbolic_function);
+    c_cosh = rb_define_class_under(m_symengine, "Cosh", c_hyperbolic_function);
+    c_tanh = rb_define_class_under(m_symengine, "Tanh", c_hyperbolic_function);
+    c_csch = rb_define_class_under(m_symengine, "Csch", c_hyperbolic_function);
+    c_sech = rb_define_class_under(m_symengine, "Sech", c_hyperbolic_function);
+    c_coth = rb_define_class_under(m_symengine, "Coth", c_hyperbolic_function);
+    c_asinh = rb_define_class_under(m_symengine, "ASinh", c_hyperbolic_function);
+    c_acosh = rb_define_class_under(m_symengine, "ACosh", c_hyperbolic_function);
+    c_atanh = rb_define_class_under(m_symengine, "ATanh", c_hyperbolic_function);
+    c_acsch = rb_define_class_under(m_symengine, "ACsch", c_hyperbolic_function);
+    c_asech = rb_define_class_under(m_symengine, "ASech", c_hyperbolic_function);
+    c_acoth = rb_define_class_under(m_symengine, "ACoth", c_hyperbolic_function);
+
     //SymEngine Functions as Module Level Funcions
     rb_define_module_function(m_symengine, "abs", cfunction_abs, 1);
     rb_define_module_function(m_symengine, "sin", cfunction_sin, 1);

ext/symengine/symengine.h

@@ -16,5 +16,36 @@ VALUE c_add;
 VALUE c_mul;
 VALUE c_pow;
 VALUE c_function;
+VALUE c_trig_function;
+VALUE c_hyperbolic_function;
+VALUE c_lambertw;
+VALUE c_dirichlet_eta;
+VALUE c_zeta;
+VALUE c_gamma;
+VALUE c_sin;
+VALUE c_cos;
+VALUE c_tan;
+VALUE c_csc;
+VALUE c_sec;
+VALUE c_cot;
+VALUE c_asin;
+VALUE c_acos;
+VALUE c_atan;
+VALUE c_acsc;
+VALUE c_asec;
+VALUE c_acot;
+VALUE c_sinh;
+VALUE c_cosh;
+VALUE c_tanh;
+VALUE c_csch;
+VALUE c_sech;
+VALUE c_coth;
+VALUE c_asinh;
+VALUE c_acosh;
+VALUE c_atanh;
+VALUE c_acsch;
+VALUE c_asech;
+VALUE c_acoth;
+
 
 #endif //SYMENGINE_H_

ext/symengine/symengine_macros.c

@@ -51,6 +51,62 @@ VALUE Klass_of_Basic(const basic_struct *basic_ptr) {
             return c_mul;
         case SYMENGINE_POW:
             return c_pow;
+        case SYMENGINE_SIN:
+            return c_sin;
+        case SYMENGINE_COS:
+            return c_cos;
+        case SYMENGINE_TAN:
+            return c_tan;
+        case SYMENGINE_CSC:
+            return c_csc;
+        case SYMENGINE_SEC:
+            return c_sec;
+        case SYMENGINE_COT:
+            return c_cot;
+        case SYMENGINE_ASIN:
+            return c_asin;
+        case SYMENGINE_ACOS:
+            return c_acos;
+        case SYMENGINE_ATAN:
+            return c_atan;
+        case SYMENGINE_ACSC:
+            return c_acsc;
+        case SYMENGINE_ASEC:
+            return c_asec;
+        case SYMENGINE_ACOT:
+            return c_acot;
+        case SYMENGINE_SINH:
+            return c_sinh;
+        case SYMENGINE_COSH:
+            return c_cosh;
+        case SYMENGINE_TANH:
+            return c_tanh;
+        case SYMENGINE_CSCH:
+            return c_csch;
+        case SYMENGINE_SECH:
+            return c_sech;
+        case SYMENGINE_COTH:
+            return c_coth;
+        case SYMENGINE_ASINH:
+            return c_asinh;
+        case SYMENGINE_ACOSH:
+            return c_acosh;
+        case SYMENGINE_ATANH:
+            return c_atanh;
+        case SYMENGINE_ACSCH:
+            return c_acsch;
+        case SYMENGINE_ASECH:
+            return c_asech;
+        case SYMENGINE_ACOTH:
+            return c_acoth;
+        case SYMENGINE_LAMBERTW:
+            return c_lambertw;
+        case SYMENGINE_DIRICHLET_ETA:
+            return c_dirichlet_eta;
+        case SYMENGINE_ZETA:
+            return c_zeta;
+        case SYMENGINE_GAMMA:
+            return c_gamma;
         default:
             return c_basic;
     }