| """Utility functions not closely tied to other spec_tools types.""" |
| # Copyright (c) 2018-2019 Collabora, Ltd. |
| # Copyright 2013-2024 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) |