Source code for spynnaker.pyNN.external_devices_models.push_bot.spinnaker_link.push_bot_retina_device

# 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.overrides import overrides
from pacman.executor.injection_decorator import inject, supports_injection
from pacman.model.graphs.application import ApplicationSpiNNakerLinkVertex
from spynnaker.pyNN.utilities.constants import SPIKE_PARTITION_ID
from spynnaker.pyNN.external_devices_models.push_bot import (
    AbstractPushBotRetinaDevice)


[docs]@supports_injection class PushBotSpiNNakerLinkRetinaDevice( AbstractPushBotRetinaDevice, ApplicationSpiNNakerLinkVertex): __slots__ = ["__new_key_command"] default_parameters = {'label': None, 'board_address': None} def __init__( self, spinnaker_link_id, protocol, resolution, board_address=default_parameters['board_address'], label=default_parameters['label']): """ :param spinnaker_link_id: :param protocol: :type protocol: ~spynnaker.pyNN.protocols.MunichIoSpiNNakerLinkProtocol :param resolution: :type resolution: ~spynnaker.pyNN.external_devices_models.push_bot.parameters.PushBotRetinaResolution :param board_address: :param label: """ super().__init__(protocol, resolution) ApplicationSpiNNakerLinkVertex.__init__( self, spinnaker_link_id=spinnaker_link_id, n_atoms=resolution.value.n_neurons, board_address=board_address, label=label) # stores for the injection aspects self.__new_key_command = None
[docs] @inject("MemoryRoutingInfos") def routing_info(self, routing_info): key = routing_info.get_first_key_from_pre_vertex( list(self.machine_vertices)[0], SPIKE_PARTITION_ID) self.__new_key_command.payload = key
@property @overrides(AbstractPushBotRetinaDevice.start_resume_commands) def start_resume_commands(self): # Update the commands with the additional one to set the key new_commands = list() for command in super().start_resume_commands: if command.key == self._protocol.disable_retina_key: # This has to be stored so that the payload can be updated self.__new_key_command = self._protocol.set_retina_key(0) new_commands.append(self.__new_key_command) new_commands.append(command) return new_commands