1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
import logging
import pyGHDL.libghdl.vhdl.nodes as nodes
import pyGHDL.libghdl.vhdl.nodes_meta as nodes_meta
import pyGHDL.libghdl.name_table as name_table
import pyGHDL.libghdl.utils as pyutils
log = logging.getLogger(__name__)
def find_def_chain(first, loc):
n1 = first
while n1 != nodes.Null_Iir:
res = find_def(n1, loc)
if res is not None:
return res
n1 = nodes.Get_Chain(n1)
return None
def find_def(n, loc):
"""Return the node at location :param loc:, or None if not under :param n:."""
if n == nodes.Null_Iir:
return None
k = nodes.Get_Kind(n)
if k in (
nodes.Iir_Kind.Simple_Name,
nodes.Iir_Kind.Character_Literal,
nodes.Iir_Kind.Operator_Symbol,
nodes.Iir_Kind.Selected_Name,
nodes.Iir_Kind.Attribute_Name,
nodes.Iir_Kind.Selected_Element,
):
n_loc = nodes.Get_Location(n)
if loc >= n_loc:
ident = nodes.Get_Identifier(n)
id_len = name_table.Get_Name_Length(ident)
if loc < n_loc + id_len:
return n
if k == nodes.Iir_Kind.Simple_Name:
return None
elif k == nodes.Iir_Kind.Design_File:
return find_def_chain(nodes.Get_First_Design_Unit(n), loc)
elif k == nodes.Iir_Kind.Design_Unit:
# if loc > elocations.Get_End_Location(unit):
# return None
res = find_def_chain(nodes.Get_Context_Items(n), loc)
if res is not None:
return res
unit = nodes.Get_Library_Unit(n)
return find_def(unit, loc)
elif k in (
nodes.Iir_Kind.Identity_Operator,
nodes.Iir_Kind.Negation_Operator,
nodes.Iir_Kind.Addition_Operator,
nodes.Iir_Kind.Substraction_Operator,
nodes.Iir_Kind.Multiplication_Operator,
nodes.Iir_Kind.Division_Operator,
nodes.Iir_Kind.Concatenation_Operator,
nodes.Iir_Kind.Equality_Operator,
nodes.Iir_Kind.Less_Than_Operator,
nodes.Iir_Kind.Greater_Than_Operator,
):
n_loc = nodes.Get_Location(n)
if loc == n_loc:
return n
elif k in (
nodes.Iir_Kind.Or_Operator,
nodes.Iir_Kind.Inequality_Operator,
nodes.Iir_Kind.Less_Than_Or_Equal_Operator,
nodes.Iir_Kind.Greater_Than_Or_Equal_Operator,
nodes.Iir_Kind.Condition_Operator,
nodes.Iir_Kind.Exponentiation_Operator,
):
n_loc = nodes.Get_Location(n)
if n_loc <= loc <= n_loc + 1:
return n
elif k in (
nodes.Iir_Kind.Absolute_Operator,
nodes.Iir_Kind.Not_Operator,
nodes.Iir_Kind.And_Operator,
nodes.Iir_Kind.Nor_Operator,
nodes.Iir_Kind.Xor_Operator,
nodes.Iir_Kind.Sll_Operator,
nodes.Iir_Kind.Sla_Operator,
nodes.Iir_Kind.Srl_Operator,
nodes.Iir_Kind.Sra_Operator,
nodes.Iir_Kind.Rol_Operator,
nodes.Iir_Kind.Ror_Operator,
nodes.Iir_Kind.Modulus_Operator,
nodes.Iir_Kind.Remainder_Operator,
):
n_loc = nodes.Get_Location(n)
if n_loc <= loc <= n_loc + 2:
return n
elif k in (
nodes.Iir_Kind.Nand_Operator,
nodes.Iir_Kind.Xnor_Operator,
):
n_loc = nodes.Get_Location(n)
if n_loc <= loc <= n_loc + 3:
return n
# This is *much* faster than using node_iter!
for f in pyutils.fields_iter(n):
typ = nodes_meta.get_field_type(f)
if typ == nodes_meta.types.Iir:
attr = nodes_meta.get_field_attribute(f)
if attr == nodes_meta.Attr.ANone:
res = find_def(nodes_meta.Get_Iir(n, f), loc)
if res is not None:
return res
elif attr == nodes_meta.Attr.Chain:
res = find_def_chain(nodes_meta.Get_Iir(n, f), loc)
if res is not None:
return res
elif attr == nodes_meta.Attr.Maybe_Ref:
if not nodes.Get_Is_Ref(n):
res = find_def(nodes_meta.Get_Iir(n, f), loc)
if res is not None:
return res
elif typ == nodes_meta.types.Iir_List:
attr = nodes_meta.get_field_attribute(f)
if attr == nodes_meta.Attr.ANone:
for n1 in pyutils.list_iter(nodes_meta.Get_Iir_List(n, f)):
res = find_def(n1, loc)
if res is not None:
return res
elif typ == nodes_meta.types.Iir_Flist:
attr = nodes_meta.get_field_attribute(f)
if attr == nodes_meta.Attr.ANone or (attr == nodes_meta.Attr.Of_Maybe_Ref and not nodes.Get_Is_Ref(n)):
for n1 in pyutils.flist_iter(nodes_meta.Get_Iir_Flist(n, f)):
res = find_def(n1, loc)
if res is not None:
return res
return None
def find_node_by_loc(n, loc):
"""Return the denoting node for :param loc: or None."""
ref = find_def(n, loc)
log.debug("for loc %u found node %s", loc, ref)
return ref
def find_definition_by_loc(n, loc):
"""Return the declaration (as a node) under :param loc: or None."""
ref = find_node_by_loc(n, loc)
if ref is None:
return None
k = nodes.Get_Kind(ref)
if k in nodes.Iir_Kinds.Denoting_Name or k == nodes.Iir_Kind.Selected_Element:
ent = nodes.Get_Named_Entity(ref)
else:
ent = nodes.Get_Implementation(ref)
return None if ent == nodes.Null_Iir else ent
|
