Source code for spinn_front_end_common.interface.interface_functions.graph_provenance_gatherer
# Copyright (c) 2017-2019 The University of Manchester
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from spinn_utilities.progress_bar import ProgressBar
from spinn_front_end_common.interface.provenance import (
AbstractProvidesLocalProvenanceData)
[docs]class GraphProvenanceGatherer(object):
__slots__ = []
def __call__(
self, machine_graph, application_graph=None):
"""
:param machine_graph: The machine graph to inspect
:param application_graph: The optional application graph
:param provenance_data_objects: Any existing objects to append to
"""
prov_items = list()
progress = ProgressBar(
machine_graph.n_vertices +
machine_graph.n_outgoing_edge_partitions,
"Getting provenance data from machine graph")
for vertex in progress.over(machine_graph.vertices, False):
if isinstance(vertex, AbstractProvidesLocalProvenanceData):
prov_items.extend(vertex.get_local_provenance_data())
for partition in progress.over(machine_graph.outgoing_edge_partitions):
for edge in partition.edges:
if isinstance(edge, AbstractProvidesLocalProvenanceData):
prov_items.extend(edge.get_local_provenance_data())
if application_graph is not None:
progress = ProgressBar(
application_graph.n_vertices +
application_graph.n_outgoing_edge_partitions,
"Getting provenance data from application graph")
for vertex in progress.over(application_graph.vertices, False):
if isinstance(vertex, AbstractProvidesLocalProvenanceData):
prov_items.extend(vertex.get_local_provenance_data())
for partition in progress.over(
application_graph.outgoing_edge_partitions):
for edge in partition.edges:
if isinstance(edge, AbstractProvidesLocalProvenanceData):
prov_items.extend(edge.get_local_provenance_data())
return prov_items