mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Filippo Cremonese
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
from typing import IO, Text, Union
|
|
|
|
import yaml
|
|
|
|
from ._generated import *
|
|
from .base import YamlLoader, YamlDumper
|
|
from .metaaddress import MetaAddress, MetaAddressType
|
|
from .reference import Reference
|
|
from .._common.monkeypatches import (
|
|
autoassign_primitive_id,
|
|
autoassign_random_id,
|
|
autoassign_constructor_argument,
|
|
make_hashable_using_attribute,
|
|
)
|
|
|
|
# Automatically compute primitive type ID if we weren't provided one
|
|
autoassign_primitive_id(Primitive)
|
|
|
|
# Automatically assign a random ID if we weren't provided one
|
|
types_with_random_id = [
|
|
CABIFunctionType,
|
|
RawFunctionType,
|
|
UnionType,
|
|
Struct,
|
|
EnumType,
|
|
Typedef,
|
|
]
|
|
|
|
for t in types_with_random_id:
|
|
autoassign_random_id(t)
|
|
|
|
# Autoassign the Kind constructor argument
|
|
types_to_kind = [
|
|
(Primitive, TypeKind.Primitive),
|
|
(EnumType, TypeKind.Enum),
|
|
(Typedef, TypeKind.Typedef),
|
|
(Struct, TypeKind.Struct),
|
|
(UnionType, TypeKind.Union),
|
|
(CABIFunctionType, TypeKind.CABIFunctionType),
|
|
(RawFunctionType, TypeKind.RawFunctionType),
|
|
]
|
|
for t, kind_val in types_to_kind:
|
|
autoassign_constructor_argument(t, "Kind", kind_val)
|
|
|
|
# Implement __hash__ based on the types ID
|
|
hashable_types = [
|
|
(CABIFunctionType, "ID"),
|
|
(RawFunctionType, "ID"),
|
|
(UnionType, "ID"),
|
|
(Struct, "ID"),
|
|
(EnumType, "ID"),
|
|
(Typedef, "ID"),
|
|
(Primitive, "ID"),
|
|
(Function, None),
|
|
]
|
|
|
|
for t, attr_name in hashable_types:
|
|
make_hashable_using_attribute(t, attr_name)
|
|
|
|
|
|
def load_model(stream: Union[bytes, IO[bytes], Text, IO[Text]]):
|
|
return yaml.load(stream, Loader=YamlLoader)
|