59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
|
"""Utility functions not closely tied to other spec_tools types."""
|
||
|
|
# Copyright (c) 2018-2019 Collabora, Ltd.
|
||
|
|
# Copyright (c) 2013-2020 The Khronos Group Inc.
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
|
||
|
|
|
||
|
|
def getElemName(elem, default=None):
|
||
|
|
"""Get the name associated with an element, either a name child or name attribute."""
|
||
|
|
name_elem = elem.find('name')
|
||
|
|
if name_elem is not None:
|
||
|
|
return name_elem.text
|
||
|
|
# Fallback if there is no child.
|
||
|
|
return elem.get('name', default)
|
||
|
|
|
||
|
|
|
||
|
|
def getElemType(elem, default=None):
|
||
|
|
"""Get the type associated with an element, either a type child or type attribute."""
|
||
|
|
type_elem = elem.find('type')
|
||
|
|
if type_elem is not None:
|
||
|
|
return type_elem.text
|
||
|
|
# Fallback if there is no child.
|
||
|
|
return elem.get('type', default)
|
||
|
|
|
||
|
|
|
||
|
|
def findFirstWithPredicate(collection, pred):
|
||
|
|
"""Return the first element that satisfies the predicate, or None if none exist.
|
||
|
|
|
||
|
|
NOTE: Some places where this is used might be better served by changing to a dictionary.
|
||
|
|
"""
|
||
|
|
for elt in collection:
|
||
|
|
if pred(elt):
|
||
|
|
return elt
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def findNamedElem(elems, name):
|
||
|
|
"""Traverse a collection of elements with 'name' nodes or attributes, looking for and returning one with the right name.
|
||
|
|
|
||
|
|
NOTE: Many places where this is used might be better served by changing to a dictionary.
|
||
|
|
"""
|
||
|
|
return findFirstWithPredicate(elems, lambda elem: getElemName(elem) == name)
|
||
|
|
|
||
|
|
|
||
|
|
def findTypedElem(elems, typename):
|
||
|
|
"""Traverse a collection of elements with 'type' nodes or attributes, looking for and returning one with the right typename.
|
||
|
|
|
||
|
|
NOTE: Many places where this is used might be better served by changing to a dictionary.
|
||
|
|
"""
|
||
|
|
return findFirstWithPredicate(elems, lambda elem: getElemType(elem) == typename)
|
||
|
|
|
||
|
|
|
||
|
|
def findNamedObject(collection, name):
|
||
|
|
"""Traverse a collection of elements with 'name' attributes, looking for and returning one with the right name.
|
||
|
|
|
||
|
|
NOTE: Many places where this is used might be better served by changing to a dictionary.
|
||
|
|
"""
|
||
|
|
return findFirstWithPredicate(collection, lambda elt: elt.name == name)
|