Source code for spinnman.processes.get_tags_process

# 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/>.

import functools
from spinn_machine.tags import ReverseIPTag, IPTag
from .abstract_multi_connection_process import AbstractMultiConnectionProcess
from spinnman.messages.scp.impl import IPTagGetInfo, IPTagGet


[docs]class GetTagsProcess(AbstractMultiConnectionProcess): __slots__ = [ "_tags", "_tag_info"] def __init__(self, connection_selector): """ :param connection_selector: :type connection_selector: AbstractMultiConnectionProcessConnectionSelector """ super().__init__(connection_selector) self._tag_info = None self._tags = None def __handle_tag_info_response(self, response): self._tag_info = response def __handle_get_tag_response(self, tag, board_address, response): if response.in_use: ip_address = response.ip_address host = "{}.{}.{}.{}".format(ip_address[0], ip_address[1], ip_address[2], ip_address[3]) if response.is_reverse: self._tags[tag] = ReverseIPTag( board_address, tag, response.rx_port, response.spin_chip_x, response.spin_chip_y, response.spin_cpu, response.spin_port) else: self._tags[tag] = IPTag( board_address, response.sdp_header.source_chip_x, response.sdp_header.source_chip_y, tag, host, response.port, response.strip_sdp)
[docs] def get_tags(self, connection): """ :param SCAMPConnection connection: :rtype: list(~spinn_machine.tags.IPTag or ~spinn_machine.tags.ReverseIPTag) """ # Get the tag information, without which we cannot continue self._send_request(IPTagGetInfo( connection.chip_x, connection.chip_y), self.__handle_tag_info_response) self._finish() self.check_for_error() # Get the tags themselves n_tags = self._tag_info.pool_size + self._tag_info.fixed_size self._tags = [None] * n_tags for tag in range(n_tags): self._send_request(IPTagGet( connection.chip_x, connection.chip_y, tag), functools.partial( self.__handle_get_tag_response, tag, connection.remote_ip_address)) self._finish() self.check_for_error() # Return the tags return [tag for tag in self._tags if tag is not None]