Merge branch 'users/chapuni/cov/single/base' into users/chapuni/cov/single/switchusers/chapuni/cov/single/switch

1 files changed, 30 insertions, 0 deletions

diff --git a/libc/utils/hdrgen/function.py b/libc/utils/hdrgen/function.py
new file mode 100644
index 0000000..d97df7f
--- /dev/null
+++ b/libc/utils/hdrgen/function.py
@@ -0,0 +1,30 @@
+# ====-- Function class for libc function headers -------------*- python -*--==#
+#
+# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+#
+# ==-------------------------------------------------------------------------==#
+
+
+class Function:
+    def __init__(
+        self, return_type, name, arguments, standards, guard=None, attributes=[]
+    ):
+        self.return_type = return_type
+        self.name = name
+        self.arguments = [
+            arg if isinstance(arg, str) else arg["type"] for arg in arguments
+        ]
+        self.standards = standards
+        self.guard = guard
+        self.attributes = attributes or ""
+
+    def __str__(self):
+        attributes_str = " ".join(self.attributes)
+        arguments_str = ", ".join(self.arguments)
+        if attributes_str == "":
+            result = f"{self.return_type} {self.name}({arguments_str})"
+        else:
+            result = f"{attributes_str} {self.return_type} {self.name}({arguments_str})"
+        return result