"""Filter factories to hide or show sets of nodes and edges.
These filters return the function used when creating `SubGraph`.
"""
__all__ = [
"no_filter",
"hide_nodes",
"hide_edges",
"hide_multiedges",
"hide_diedges",
"hide_multidiedges",
"show_nodes",
"show_edges",
"show_multiedges",
"show_diedges",
"show_multidiedges",
]
[docs]def no_filter(*items):
return True
[docs]def hide_nodes(nodes):
nodes = set(nodes)
return lambda node: node not in nodes
[docs]def hide_diedges(edges):
edges = {(u, v) for u, v in edges}
return lambda u, v: (u, v) not in edges
[docs]def hide_edges(edges):
alledges = set(edges) | {(v, u) for (u, v) in edges}
return lambda u, v: (u, v) not in alledges
[docs]def hide_multidiedges(edges):
edges = {(u, v, k) for u, v, k in edges}
return lambda u, v, k: (u, v, k) not in edges
[docs]def hide_multiedges(edges):
alledges = set(edges) | {(v, u, k) for (u, v, k) in edges}
return lambda u, v, k: (u, v, k) not in alledges
# write show_nodes as a class to make SubGraph pickleable
[docs]class show_nodes:
[docs] def __init__(self, nodes):
self.nodes = set(nodes)
def __call__(self, node):
return node in self.nodes
[docs]def show_diedges(edges):
edges = {(u, v) for u, v in edges}
return lambda u, v: (u, v) in edges
[docs]def show_edges(edges):
alledges = set(edges) | {(v, u) for (u, v) in edges}
return lambda u, v: (u, v) in alledges
[docs]def show_multidiedges(edges):
edges = {(u, v, k) for u, v, k in edges}
return lambda u, v, k: (u, v, k) in edges
[docs]def show_multiedges(edges):
alledges = set(edges) | {(v, u, k) for (u, v, k) in edges}
return lambda u, v, k: (u, v, k) in alledges