From f95d68f3d2ee565dd2a4aa0102f11f55e8d5071c Mon Sep 17 00:00:00 2001
From: Abdelrahman <said.abdelrahman89@gmail.com>
Date: Sat, 8 Feb 2025 16:09:25 +0000
Subject: [PATCH] Add a test for a generic typed array
---
codegen/__main__.py | 5 +--
codegen/codegen.py | 79 ++++++++++++++++++++++++++++++++++++++++++---
2 files changed, 77 insertions(+), 7 deletions(-)
diff --git a/codegen/__main__.py b/codegen/__main__.py
index a5c96ca..7d6135a 100644
--- a/codegen/__main__.py
+++ b/codegen/__main__.py
@@ -1,4 +1,5 @@
-from .codegen import test_codegen
+from .codegen import test_str8, test_typed_array
if __name__ == "__main__":
- test_codegen()
+ test_str8()
+ test_typed_array()
diff --git a/codegen/codegen.py b/codegen/codegen.py
index 96afb6e..5b0a23f 100644
--- a/codegen/codegen.py
+++ b/codegen/codegen.py
@@ -12,10 +12,11 @@ from .datatypes import (
CPointerType,
CQualifier,
CInclude,
+ CUserType,
)
-def test_codegen():
+def test_str8():
struct = CStruct(
name="Str8",
cargs=[
@@ -55,7 +56,7 @@ def test_codegen():
)
header = CHeader(
- name="str.h",
+ name="str",
includes=[
CInclude(header="aliases.h", local=True),
],
@@ -64,7 +65,7 @@ def test_codegen():
)
source = CSource(
- name="str.c",
+ name="str",
includes=[
CInclude(header=header, local=True),
CInclude(header="aliases.h", local=True),
@@ -78,5 +79,73 @@ def test_codegen():
source.save(Path("."))
-if __name__ == "__main__":
- test_codegen()
+def test_typed_array():
+ datatypes = [CType.U8, CType.F32, CType.CHAR]
+ types: list[CUserType] = []
+ funcs: list[CFunc] = []
+
+ getter_func_body = """\
+ if(idx >= arr->size) {
+ return 0;
+ }
+
+ return arr->data[idx];"""
+
+ setter_func_body = """\
+ if(idx >= arr->size) {
+ return;
+ }
+
+ arr->data[idx] = val;"""
+
+ for _type in datatypes:
+ struct = CStruct(
+ name=f"Array{str(_type).title()}".strip(),
+ cargs=[
+ CArg(name="size", _type=CType.U64),
+ CArg(name="data", _type=_type, pointer=CPointer(_type=CPointerType.SINGLE)),
+ ],
+ )
+ types.append(struct)
+
+ funcs.append(CFunc(
+ name=f"array_{str(_type)}_get",
+ ret_type=_type,
+ args=[
+ CArg(name="arr", _type=struct, pointer=CPointer(_type=CPointerType.SINGLE), qualifier=CQualifier.CONST),
+ CArg(name="idx", _type=CType.U64),
+ ],
+ body=getter_func_body,
+ ))
+
+ funcs.append(CFunc(
+ name=f"array_{str(_type)}_set",
+ ret_type=CType.VOID,
+ args=[
+ CArg(name="arr", _type=struct, pointer=CPointer(_type=CPointerType.SINGLE)),
+ CArg(name="idx", _type=CType.U64),
+ CArg(name="val", _type=_type)
+ ],
+ body=setter_func_body,
+ ))
+
+ header = CHeader(
+ name="typed_array",
+ includes=[
+ CInclude(header="aliases.h", local=True)
+ ],
+ types=types,
+ funcs=funcs
+ )
+
+ source = CSource(
+ name="typed_array",
+ includes=[
+ CInclude(header="aliases.h", local=True),
+ CInclude(header=header, local=True)
+ ],
+ funcs=funcs
+ )
+
+ header.save(Path("."))
+ source.save(Path("."))