333 lines
11 KiB
Python
333 lines
11 KiB
Python
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:
|
|
node_typename: str
|
|
list_typename: str
|
|
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,
|
|
type_string_upper: str,
|
|
type_string_lower: str,
|
|
node_typename: str,
|
|
list_typename: str
|
|
):
|
|
return load_func_body_from_file(filename).format(
|
|
T=type_string,
|
|
NodeType=node_typename,
|
|
ListType=list_typename,
|
|
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] = [
|
|
CInclude(header="stdbool.h")
|
|
]
|
|
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(
|
|
node_typename="Str8Node",
|
|
list_typename="Str8List",
|
|
hdr_decl_types=[
|
|
CStruct(name="str8", cargs=[], typedef_name="Str8"),
|
|
],
|
|
),
|
|
}
|
|
|
|
for _type in CType:
|
|
if _type == CType.VOID:
|
|
datatypes["void *"] = DblListData(
|
|
node_typename="VoidPNode",
|
|
list_typename="VoidPList",
|
|
)
|
|
continue
|
|
elif _type == CType.BOOL:
|
|
datatypes[_type.value] = DblListData(
|
|
node_typename="BoolNode",
|
|
list_typename="BoolList",
|
|
)
|
|
continue
|
|
|
|
type_title = _type.value.title()
|
|
datatypes[_type.value] = DblListData(
|
|
node_typename=f"{type_title}Node",
|
|
list_typename=f"{type_title}List",
|
|
)
|
|
|
|
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)
|
|
clean_type_string = type_string.replace(" ", "").replace("*", "_ptr")
|
|
type_string_upper = clean_type_string.upper()
|
|
type_string_lower = clean_type_string.lower()
|
|
|
|
node = CStruct(
|
|
name=dbl_list_data.node_typename,
|
|
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=dbl_list_data.list_typename,
|
|
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(
|
|
filename=snippets_dir / "list_node",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
)
|
|
|
|
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(
|
|
filename=snippets_dir / "list_get",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
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(
|
|
filename=snippets_dir / "list_push_front",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
)
|
|
|
|
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(
|
|
filename=snippets_dir / "list_push_back",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
)
|
|
|
|
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(
|
|
filename=snippets_dir / "list_insert",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
)
|
|
|
|
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(
|
|
filename=snippets_dir / "list_pop_front",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
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(
|
|
filename=snippets_dir / "list_pop_back",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
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(
|
|
filename=snippets_dir / "list_remove",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
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(
|
|
filename=snippets_dir / "list_empty",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
)
|
|
|
|
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(
|
|
filename=snippets_dir / "node_to_list",
|
|
type_string=type_string,
|
|
type_string_upper=type_string_upper,
|
|
type_string_lower=type_string_lower,
|
|
node_typename=dbl_list_data.node_typename,
|
|
list_typename=dbl_list_data.list_typename
|
|
),
|
|
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)
|