42 lines
959 B
Python
42 lines
959 B
Python
|
"""
|
||
|
|
Serves as a global point for storing and retrieving node types needed during simulation.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pprint
|
||
|
|
|
||
|
|
from core.misc import log
|
||
|
|
|
||
|
|
logger = log.get_logger(__name__)
|
||
|
|
|
||
|
|
_NODE_MAP = None
|
||
|
|
|
||
|
|
|
||
|
|
def _convert_map(x, y):
|
||
|
|
x[y[0].name] = y[1]
|
||
|
|
return x
|
||
|
|
|
||
|
|
|
||
|
|
def set_node_map(node_map):
|
||
|
|
global _NODE_MAP
|
||
|
|
print_map = reduce(lambda x, y: _convert_map(x, y), node_map.items(), {})
|
||
|
|
logger.info("setting node class map: \n%s", pprint.pformat(print_map, indent=4))
|
||
|
|
_NODE_MAP = node_map
|
||
|
|
|
||
|
|
|
||
|
|
def get_node_class(node_type):
|
||
|
|
global _NODE_MAP
|
||
|
|
return _NODE_MAP[node_type]
|
||
|
|
|
||
|
|
|
||
|
|
def is_node(obj, node_types):
|
||
|
|
type_classes = []
|
||
|
|
if isinstance(node_types, (tuple, list)):
|
||
|
|
for node_type in node_types:
|
||
|
|
type_class = get_node_class(node_type)
|
||
|
|
type_classes.append(type_class)
|
||
|
|
else:
|
||
|
|
type_class = get_node_class(node_types)
|
||
|
|
type_classes.append(type_class)
|
||
|
|
|
||
|
|
return isinstance(obj, tuple(type_classes))
|