| 1 | from asdl import pybase
|
| 2 | from mycpp import mops
|
| 3 | from typing import Optional, List, Tuple, Dict, Any, cast, TYPE_CHECKING
|
| 4 |
|
| 5 | class color_t(pybase.SimpleObj):
|
| 6 | pass
|
| 7 |
|
| 8 | class color_e(object):
|
| 9 | TypeName = color_t(1)
|
| 10 | StringConst = color_t(2)
|
| 11 | OtherConst = color_t(3)
|
| 12 | UserType = color_t(4)
|
| 13 | External = color_t(5)
|
| 14 |
|
| 15 | _color_str = {
|
| 16 | 1: 'color.TypeName',
|
| 17 | 2: 'color.StringConst',
|
| 18 | 3: 'color.OtherConst',
|
| 19 | 4: 'color.UserType',
|
| 20 | 5: 'color.External',
|
| 21 | }
|
| 22 |
|
| 23 | def color_str(val):
|
| 24 | # type: (color_t) -> str
|
| 25 | return _color_str[val]
|
| 26 |
|
| 27 | class hnode_e(object):
|
| 28 | AlreadySeen = 1
|
| 29 | Record = 2
|
| 30 | Array = 3
|
| 31 | Leaf = 4
|
| 32 | External = 5
|
| 33 |
|
| 34 | _hnode_str = {
|
| 35 | 1: 'AlreadySeen',
|
| 36 | 2: 'Record',
|
| 37 | 3: 'Array',
|
| 38 | 4: 'Leaf',
|
| 39 | 5: 'External',
|
| 40 | }
|
| 41 |
|
| 42 | def hnode_str(tag, dot=True):
|
| 43 | # type: (int, bool) -> str
|
| 44 | v = _hnode_str[tag]
|
| 45 | if dot:
|
| 46 | return "hnode.%s" % v
|
| 47 | else:
|
| 48 | return v
|
| 49 |
|
| 50 | class hnode_t(pybase.CompoundObj):
|
| 51 | def tag(self):
|
| 52 | # type: () -> int
|
| 53 | return self._type_tag
|
| 54 | pass
|
| 55 |
|
| 56 | class hnode(object):
|
| 57 | class AlreadySeen(hnode_t):
|
| 58 | _type_tag = 1
|
| 59 | __slots__ = ('heap_id',)
|
| 60 |
|
| 61 | def __init__(self, heap_id):
|
| 62 | # type: (int) -> None
|
| 63 | self.heap_id = heap_id
|
| 64 |
|
| 65 | class Record(hnode_t):
|
| 66 | _type_tag = 2
|
| 67 | __slots__ = ('node_type', 'fields', 'abbrev', 'left', 'right',
|
| 68 | 'unnamed_fields')
|
| 69 |
|
| 70 | def __init__(self, node_type, fields, abbrev, left, right, unnamed_fields):
|
| 71 | # type: (str, List[Field], bool, str, str, List[hnode_t]) -> None
|
| 72 | self.node_type = node_type
|
| 73 | self.fields = fields
|
| 74 | self.abbrev = abbrev
|
| 75 | self.left = left
|
| 76 | self.right = right
|
| 77 | self.unnamed_fields = unnamed_fields
|
| 78 |
|
| 79 | class Array(hnode_t):
|
| 80 | _type_tag = 3
|
| 81 | __slots__ = ('children',)
|
| 82 |
|
| 83 | def __init__(self, children):
|
| 84 | # type: (List[hnode_t]) -> None
|
| 85 | self.children = children
|
| 86 |
|
| 87 | class Leaf(hnode_t):
|
| 88 | _type_tag = 4
|
| 89 | __slots__ = ('s', 'color')
|
| 90 |
|
| 91 | def __init__(self, s, color):
|
| 92 | # type: (str, color_t) -> None
|
| 93 | self.s = s
|
| 94 | self.color = color
|
| 95 |
|
| 96 | class External(hnode_t):
|
| 97 | _type_tag = 5
|
| 98 | __slots__ = ('obj',)
|
| 99 |
|
| 100 | def __init__(self, obj):
|
| 101 | # type: (Any) -> None
|
| 102 | self.obj = obj
|
| 103 |
|
| 104 | pass
|
| 105 |
|
| 106 | class alloc_members_t(pybase.SimpleObj):
|
| 107 | pass
|
| 108 |
|
| 109 | class alloc_members_e(object):
|
| 110 | List = alloc_members_t(1)
|
| 111 | Dict = alloc_members_t(2)
|
| 112 | Struct = alloc_members_t(3)
|
| 113 |
|
| 114 | _alloc_members_str = {
|
| 115 | 1: 'alloc_members.List',
|
| 116 | 2: 'alloc_members.Dict',
|
| 117 | 3: 'alloc_members.Struct',
|
| 118 | }
|
| 119 |
|
| 120 | def alloc_members_str(val):
|
| 121 | # type: (alloc_members_t) -> str
|
| 122 | return _alloc_members_str[val]
|
| 123 |
|
| 124 | class Field(pybase.CompoundObj):
|
| 125 | _type_tag = 64
|
| 126 | __slots__ = ('name', 'val')
|
| 127 |
|
| 128 | def __init__(self, name, val):
|
| 129 | # type: (str, hnode_t) -> None
|
| 130 | self.name = name
|
| 131 | self.val = val
|
| 132 |
|