Merge branch 'codegen'

codegen/dbl_list/make_dbl_list.py

@@ -0,0 +1,224 @@
+from pathlib import Path
+from dataclasses import dataclass, field
+from codegen.constants import WAPP_SRC_ROOT
+from codegen.utils import load_func_body_from_file
+from codegen.datatypes import (
+    CDataType,
+    CMacro,
+    CStruct,
+    CFunc,
+    CHeader,
+    CSource,
+    CArg,
+    CType,
+    CPointer,
+    CPointerType,
+    CQualifier,
+    CInclude,
+    get_datatype_string,
+)
+
+
+@dataclass
+class DblListData:
+    hdr_decl_types: list[CStruct] = field(default_factory=list)
+    src_decl_types: list[CStruct] = field(default_factory=list)
+
+
+def make_dbl_list(user_datatypes: dict[CDataType, DblListData] = {}):
+    def __format_func_body(filename: Path, type_string: str):
+        return load_func_body_from_file(filename).format(
+            T=type_string,
+            Ttitle=type_string.title(),
+            Tupper=type_string.upper(),
+            Tlower=type_string.lower(),
+        )
+
+    out_dir = WAPP_SRC_ROOT / "containers" / "dbl_list"
+    out_dir.mkdir(parents=True, exist_ok=True)
+
+    common_local_include_files = [
+        (WAPP_SRC_ROOT / "common" / "aliases" / "aliases.h")
+    ]
+    common_includes: list[CInclude] = []
+    for local_file in common_local_include_files:
+        common_includes.append(
+            CInclude(
+                header=str(local_file.relative_to(out_dir, walk_up=True)),
+                local=True,
+            )
+        )
+
+    common_decl_types: list[CStruct] = []
+
+    datatypes: dict[CDataType, DblListData] = {
+        "Str8": DblListData(
+            hdr_decl_types=[
+                CStruct(name="str8", cargs=[], typedef_name="Str8"),
+            ],
+        ),
+    }
+    datatypes.update(user_datatypes)
+
+    snippets_dir = Path(__file__).parent / "snippets"
+
+    header = CHeader(
+        name="dbl_list",
+        decl_types=common_decl_types,
+        includes=[],
+        types=[],
+        funcs=[]
+    )
+
+    source = CSource(
+        name=header.name,
+        decl_types=common_decl_types,
+        includes=[CInclude(header, local=True, same_dir=True), CInclude(header="stddef.h")],
+        internal_funcs=[],
+        funcs=header.funcs
+    )
+
+    if len(common_includes) > 0:
+        header.includes.extend(common_includes)
+        source.includes.extend(common_includes)
+
+    for _type, dbl_list_data in datatypes.items():
+        type_string = get_datatype_string(_type)
+
+        node = CStruct(
+            name=f"{type_string.title()}Node",
+            cargs=[
+                CArg(name="item", _type=type_string, pointer=CPointer(_type=CPointerType.SINGLE)),
+            ],
+        )
+        node.cargs.extend([
+            CArg(name="prev", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
+            CArg(name="next", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
+        ])
+        
+        dl_list = CStruct(
+            name=f"{type_string.title()}List",
+            cargs=[
+                CArg(name="first", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
+                CArg(name="last", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
+                CArg(name="node_count", _type=CType.U64),
+            ],
+        )
+
+        node_macro = CMacro(
+            name=f"wapp_{type_string.lower()}_list_node(ITEM_PTR)",
+            value=__format_func_body(snippets_dir / "list_node", type_string),
+        )
+
+        get_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_get",
+            ret_type=node,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE), qualifier=CQualifier.CONST),
+                CArg(name="index", _type=CType.U64),
+            ],
+            body=__format_func_body(snippets_dir / "list_get", type_string),
+            pointer=CPointer(CPointerType.SINGLE),
+        )
+
+        push_front_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_push_front",
+            ret_type=CType.VOID,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "list_push_front", type_string),
+        )
+
+        push_back_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_push_back",
+            ret_type=CType.VOID,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "list_push_back", type_string),
+        )
+
+        insert_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_insert",
+            ret_type=CType.VOID,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
+                CArg(name="index", _type=CType.U64),
+            ],
+            body=__format_func_body(snippets_dir / "list_insert", type_string),
+        )
+
+        pop_front_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_pop_front",
+            ret_type=node,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "list_pop_front", type_string),
+            pointer=CPointer(CPointerType.SINGLE),
+        )
+
+        pop_back_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_pop_back",
+            ret_type=node,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "list_pop_back", type_string),
+            pointer=CPointer(CPointerType.SINGLE),
+        )
+
+        remove_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_remove",
+            ret_type=node,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+                CArg(name="index", _type=CType.U64),
+            ],
+            body=__format_func_body(snippets_dir / "list_remove", type_string),
+            pointer=CPointer(CPointerType.SINGLE),
+        )
+
+        empty_func = CFunc(
+            name=f"wapp_{type_string.lower()}_list_empty",
+            ret_type=CType.VOID,
+            args=[
+                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "list_empty", type_string),
+        )
+
+        node_to_list_func = CFunc(
+            name=f"{type_string.lower()}_node_to_list",
+            ret_type=dl_list,
+            args=[
+                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
+            ],
+            body=__format_func_body(snippets_dir / "node_to_list", type_string),
+            qualifiers=[CQualifier.INTERNAL],
+        )
+
+        header.decl_types.extend(dbl_list_data.hdr_decl_types)
+        header.macros.append(node_macro)
+        header.types.extend([node, dl_list])
+        header.funcs.extend([
+            get_func,
+            push_front_func,
+            push_back_func,
+            insert_func,
+            pop_front_func,
+            pop_back_func,
+            remove_func,
+            empty_func,
+        ])
+
+        source.decl_types.extend(dbl_list_data.src_decl_types)
+        source.internal_funcs.append(node_to_list_func)
+        source.funcs = header.funcs
+
+    header.save(out_dir)
+    source.save(out_dir)

codegen/dbl_list/snippets/list_empty

@@ -0,0 +1,8 @@
+  if (!list) {{
+    return;
+  }}
+
+  u64 count = list->node_count;
+  for (u64 i = 0; i < count; ++i) {{
+    wapp_{Tlower}_list_pop_back(list);
+  }}

codegen/dbl_list/snippets/list_get

@@ -0,0 +1,13 @@
+  if (index >= list->node_count) {{
+    return NULL;
+  }}
+
+  {Ttitle}Node *output  = NULL;
+  {Ttitle}Node *current = list->first;
+  for (u64 i = 1; i <= index; ++i) {{
+    current = current->next;
+  }}
+
+  output = current;
+
+  return output;

codegen/dbl_list/snippets/list_insert

@@ -0,0 +1,28 @@
+  if (!list || !node || !(node->item)) {{
+    return;
+  }}
+
+  if (index == 0) {{
+    wapp_{Tlower}_list_push_front(list, node);
+    return;
+  }} else if (index == list->node_count) {{
+    wapp_{Tlower}_list_push_back(list, node);
+    return;
+  }}
+
+  {Ttitle}Node *dst_node = wapp_{Tlower}_list_get(list, index);
+  if (!dst_node) {{
+    return;
+  }}
+
+  {Ttitle}List node_list = {Tlower}_node_to_list(node);
+
+  list->node_count += node_list.node_count;
+
+  {Ttitle}Node *prev = dst_node->prev;
+
+  dst_node->prev = node_list.last;
+  prev->next     = node_list.first;
+
+  node_list.first->prev = prev;
+  node_list.last->next  = dst_node;

codegen/dbl_list/snippets/list_node

@@ -0,0 +1 @@
+(({Ttitle}Node){{.item = ITEM_PTR}})

codegen/dbl_list/snippets/list_pop_back

@@ -0,0 +1,20 @@
+  {Ttitle}Node *output = NULL;
+
+  if (!list || list->node_count == 0) {{
+    goto RETURN_{Tupper}_LIST_POP_BACK;
+  }}
+
+  output = list->last;
+
+  if (list->node_count == 1) {{
+    *list = ({Ttitle}List){{0}};
+    goto RETURN_{Tupper}_LIST_POP_BACK;
+  }}
+
+  --(list->node_count);
+  list->last        = output->prev;
+
+  output->prev = output->next = NULL;
+
+RETURN_{Tupper}_LIST_POP_BACK:
+  return output;

codegen/dbl_list/snippets/list_pop_front

@@ -0,0 +1,20 @@
+  {Ttitle}Node *output = NULL;
+
+  if (!list || list->node_count == 0) {{
+    goto RETURN_{Tupper}_LIST_POP_FRONT;
+  }}
+
+  output = list->first;
+
+  if (list->node_count == 1) {{
+    *list = ({Ttitle}List){{0}};
+    goto RETURN_{Tupper}_LIST_POP_FRONT;
+  }}
+
+  --(list->node_count);
+  list->first       = output->next;
+
+  output->prev = output->next = NULL;
+
+RETURN_{Tupper}_LIST_POP_FRONT:
+  return output;

codegen/dbl_list/snippets/list_push_back

@@ -0,0 +1,20 @@
+  if (!list || !node || !(node->item)) {{
+    return;
+  }}
+
+  {Ttitle}List node_list = {Tlower}_node_to_list(node);
+
+  if (list->node_count == 0) {{
+    *list = node_list;
+    return;
+  }}
+
+  list->node_count += node_list.node_count;
+
+  {Ttitle}Node *last = list->last;
+  if (last) {{
+    last->next = node_list.first;
+  }}
+
+  list->last            = node_list.last;
+  node_list.first->prev = last;

codegen/dbl_list/snippets/list_push_front

@@ -0,0 +1,20 @@
+  if (!list || !node || !(node->item)) {{
+    return;
+  }}
+
+  {Ttitle}List node_list = {Tlower}_node_to_list(node);
+
+  if (list->node_count == 0) {{
+    *list = node_list;
+    return;
+  }}
+
+  list->node_count += node_list.node_count;
+
+  {Ttitle}Node *first = list->first;
+  if (first) {{
+    first->prev = node_list.last;
+  }}
+
+  list->first          = node_list.first;
+  node_list.last->next = first;

codegen/dbl_list/snippets/list_remove

@@ -0,0 +1,27 @@
+  {Ttitle}Node *output = NULL;
+  if (!list) {{
+    goto RETURN_{Tupper}_LIST_REMOVE;
+  }}
+
+  if (index == 0) {{
+    output = wapp_{Tlower}_list_pop_front(list);
+    goto RETURN_{Tupper}_LIST_REMOVE;
+  }} else if (index == list->node_count) {{
+    output = wapp_{Tlower}_list_pop_back(list);
+    goto RETURN_{Tupper}_LIST_REMOVE;
+  }}
+
+  output = wapp_{Tlower}_list_get(list, index);
+  if (!output) {{
+    goto RETURN_{Tupper}_LIST_REMOVE;
+  }}
+
+  output->prev->next = output->next;
+  output->next->prev = output->prev;
+
+  --(list->node_count);
+
+  output->prev = output->next = NULL;
+
+RETURN_{Tupper}_LIST_REMOVE:
+  return output;

codegen/dbl_list/snippets/node_to_list

@@ -0,0 +1,13 @@
+  {Ttitle}List output = {{.first = node, .last = node, .node_count = 1}};
+
+  while (output.first->prev != NULL) {{
+    output.first = output.first->prev;
+    ++(output.node_count);
+  }}
+
+  while (output.last->next != NULL) {{
+    output.last = output.last->next;
+    ++(output.node_count);
+  }}
+
+  return output;