GA144.py

import itertools
import GA144_rom

from simSched import OperationSched


'''The goal of this module is to emulate a GA144 processor,
   comprising of 144 FA18A nodes.  To do this we:
   0) Write some infrastructure code to feed the others
   1) Make a schedular to deal with all the async handling
   2) Invoke and populate the 144 nodes with their respective ROM's
   3) Register all their IO read deps
   4) Run (fake) one or more boot streams
   5) Let the schedular take over'''


def maybeScheduleStep(pInstance, schedular):
    nextTime = pInstance.getNextStepTime()
    # Waiting/Blocked on IO
    if nextTime is None:
        return
    # Resechedule next operation to be run
    schedular.enter(nextTime, runStepAndMaybeSchedule, (pInstance, schedular))


def runStepAndMaybeSchedule(pInstance, schedular):
    '''This function is called when a operation is schedulaed to finish.
    It runs the state update step, then possibly schedules another operation
    to take place based on whether there is a known run-time.'''
    print(pInstance.nextOp)
    pInstance.runStep()

    maybeScheduleStep(pInstance, schedular)


class GA144(object):
    def __init__(self, romFile="GA144.rom"):
        self._setupNodes()
        self._setupSched()
        self._setupInterconnects()
        self._setupExtraconnects()
        self.chipReset()

    def _setupNodes(self):
        # Make an iterator that has the array geometry of 8 x 18
        cooridinateIterator = itertools.product(range(8), range(18))

        # Map the iterator over the coordinates, making new nodes
        self.cores = {coord: GA144_rom.makeNodeByCoord(*coord) for coord in cooridinateIterator}

        # Put them in their starting locations
        [node.jumpToWarm() for node in self.cores.values()]

    def _setupInterconnects(self):
        pass

    def _setupExtraconnects(self):
        pass

    def _setupSched(self):
        # Now create a class to track what runs when
        self.scheduler = OperationSched()

        # Register all the nodes in the schedular
        [maybeScheduleStep(node, self.scheduler) for node in self.cores.values()]

    def chipReset(self):
        # Reset all nodes on the chip
        [node.reset() for node in self.cores.values()]

    def fakeBootStream():
        # Fake the boot stream, for now just based on the 2 wire async boot
        # asyncNode = self.cores[(7, 8)]

        # asyncNode.fakeBootStream()
        # MRG TODO: during boot stream, all nodes are actully already
        # running the jump to get get to the right port read pattern
        # More accurate to start the schedular earlier, the fake the
        # boot stream as an event in it.
        pass

    def run(self):
        self.scheduler.run()


if __name__ == "__main__":
    p = GA144()
    p.run()