mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Filippo Cremonese
56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
from revng.tupletree import Reference, init_reference_yaml_classes
|
|
from ._generated import *
|
|
from . import _generated
|
|
from ..metaaddress import MetaAddressType, MetaAddress, init_metaaddress_yaml_classes
|
|
from .._util import force_constructor_kwarg, enum_value_to_index
|
|
|
|
_types_to_kinds = [
|
|
(PrimitiveType, TypeKind.Primitive),
|
|
(EnumType, TypeKind.Enum),
|
|
(TypedefType, TypeKind.Typedef),
|
|
(StructType, TypeKind.Struct),
|
|
(UnionType, TypeKind.Union),
|
|
(CABIFunctionType, TypeKind.CABIFunctionType),
|
|
(RawFunctionType, TypeKind.RawFunctionType),
|
|
]
|
|
for t, k in _types_to_kinds:
|
|
force_constructor_kwarg(t, "Kind", k)
|
|
|
|
init_metaaddress_yaml_classes(YamlLoader, YamlDumper)
|
|
init_reference_yaml_classes(YamlLoader, YamlDumper)
|
|
|
|
|
|
# Create a derived PrimitiveType class that automatically computes the correct ID
|
|
class PrimitiveType(_generated.PrimitiveType):
|
|
def __init__(self, *args, **kwargs):
|
|
if "ID" not in kwargs:
|
|
primitive_kind = kwargs.get("PrimitiveKind")
|
|
if primitive_kind is None:
|
|
raise ValueError("Must provide PrimitiveKind as kwarg to PrimitiveType constructor")
|
|
|
|
size = kwargs.get("Size")
|
|
if size is None:
|
|
raise ValueError("Must provide Size as kwarg to PrimitiveType constructor")
|
|
|
|
primitive_kind_value = enum_value_to_index(primitive_kind)
|
|
kwargs["ID"] = primitive_kind_value << 8 | size
|
|
|
|
super(self.__class__, self).__init__(*args, **kwargs)
|
|
|
|
|
|
class Binary(_generated.Binary):
|
|
@classmethod
|
|
def get_reference_str(cls, t):
|
|
if hasattr(t, "Kind"):
|
|
typename = t.Kind.value
|
|
else:
|
|
typename = type(t).__name__
|
|
return f"/Types/{typename}-{t.ID}"
|
|
|
|
|
|
# Since we subclassed them we need to re-register their constructors and representers
|
|
YamlLoader.add_constructor("!PrimitiveType", PrimitiveType.yaml_constructor)
|
|
YamlDumper.add_representer(PrimitiveType, PrimitiveType.yaml_representer)
|
|
YamlLoader.add_constructor("!Binary", Binary.yaml_constructor)
|
|
YamlDumper.add_representer(Binary, Binary.yaml_representer)
|