Merge pull request #17 from lynnporu/dev

Compile function, process function pointers
lynnporu · Jan 14, 2022 · e46c6d3 · e46c6d3
2 parents ce9abb1 + d584780
commit e46c6d3
Showing 1 changed file with 19 additions and 4 deletions.
diff --git a/annotatec/declarations.py b/annotatec/declarations.py
@@ -68,6 +68,11 @@ def unwrap(self, wrapped):
         else:
             return wrapped
 
+    def is_function(self, name: str):
+        """Checks if passed name is a function.
+        """
+        return hasattr(self.lib, name)
+
     def compile(self, name: str, unwrap: bool = False):
         """
         Arguments:
@@ -96,6 +101,9 @@ def compile(self, name: str, unwrap: bool = False):
         if name in BASE_C_TYPES:
             return BASE_C_TYPES[name]
 
+        if self.is_function(name):
+            return name
+
         if name not in self:
             raise NamespaceError(
                 f"Trying to get `{name}` object, but there's no objects with "
@@ -193,10 +201,17 @@ def compile(self):
 
         if not self.compiled:
 
-            prototype = ctypes.CFUNCTYPE(*list(map(
-                self.namespace.compile_unwrap,
-                [self.return_type] + self.argument_types
-            )))
+            prototype = ctypes.CFUNCTYPE(
+                self.namespace.compile_unwrap(self.return_type),
+                *[
+                    # compile the name
+                    self.namespace.compile_unwrap(argument_type)
+                    if not self.namespace.is_function(argument_type)
+                    # or give (void*) if the name defines a function
+                    else ctypes.c_void_p
+                    for argument_type in self.argument_types
+                ]
+            )
 
             self.compilation_result = prototype(
                 (self.name, self.namespace.lib))