Source code for spinnman.processes.get_routes_process

from spinnman.messages.scp.impl import ReadMemory
from spinn_machine import MulticastRoutingEntry
from .abstract_multi_connection_process import AbstractMultiConnectionProcess
from spinnman.constants import UDP_MESSAGE_MAX_SIZE

import functools
import struct

# There are 1024 entries in a routing table
_N_ENTRIES = 1024

# 16 entries fit in a 256-byte read
_ENTRIES_PER_READ = 16

# 64 reads of 16 entries are required for 1024 entries
_N_READS = 64


[docs]class GetMultiCastRoutesProcess(AbstractMultiConnectionProcess): def __init__(self, connection_selector, app_id=None): AbstractMultiConnectionProcess.__init__(self, connection_selector) self._entries = [None] * _N_ENTRIES self._app_id = app_id def _add_routing_entry(self, route_no, offset, app_id, route, key, mask): if route >= 0xFF000000: return if self._app_id is not None and self._app_id != app_id: return processor_ids = list() for processor_id in range(0, 26): if (route & (1 << (6 + processor_id))) != 0: processor_ids.append(processor_id) link_ids = list() for link_id in range(0, 6): if (route & (1 << link_id)) != 0: link_ids.append(link_id) self._entries[route_no + offset] = MulticastRoutingEntry( key, mask, processor_ids, link_ids, False)
[docs] def handle_read_response(self, offset, response): for route_no in range(_ENTRIES_PER_READ): entry = struct.unpack_from( "<2xBxIII", response.data, response.offset + (route_no * 16)) self._add_routing_entry(route_no, offset, *entry)
[docs] def get_routes(self, x, y, base_address): # Create the read requests offset = 0 for _ in range(_N_READS): self._send_request( ReadMemory( x, y, base_address + (offset * 16), UDP_MESSAGE_MAX_SIZE), functools.partial(self.handle_read_response, offset)) offset += _ENTRIES_PER_READ self._finish() self.check_for_error() return [entry for entry in self._entries if entry is not None]