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 155 156 157 158 159 160 161
|
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,
):
# One character operators
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,
):
# Two characters operators
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,
):
# Three characters operators
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,
):
# Four characters operators
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:
# Only sensitivity lists are interesting.
if f == nodes_meta.fields.Sensitivity_List:
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)
# TODO: character literal, attribute name...
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
|