Source code for pacman.model.graphs.abstract_single_source_partition
# Copyright (c) 2019 The University of Manchester
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from spinn_utilities.abstract_base import AbstractBase
from spinn_utilities.overrides import overrides
from pacman.exceptions import PacmanConfigurationException
from pacman.model.graphs import AbstractEdgePartition
[docs]class AbstractSingleSourcePartition(
AbstractEdgePartition, metaclass=AbstractBase):
"""
An edge partition that has a single source vertex.
"""
__slots__ = [
# The vertex at the start of all the edges
"_pre_vertex"
]
def __init__(
self, pre_vertex, identifier, allowed_edge_types):
super().__init__(
identifier=identifier, allowed_edge_types=allowed_edge_types)
self._pre_vertex = pre_vertex
[docs] @overrides(AbstractEdgePartition.add_edge)
def add_edge(self, edge):
super().add_edge(edge)
if edge.pre_vertex != self._pre_vertex:
raise PacmanConfigurationException(
"A partition can only contain edges with the same pre_vertex")
@property
def pre_vertex(self):
"""
The vertex at which all edges in this outgoing edge partition start.
:rtype: ~pacman.model.graphs.AbstractVertex
"""
return self._pre_vertex
@property
@overrides(AbstractEdgePartition.pre_vertices)
def pre_vertices(self):
yield self.pre_vertex