joeriPDEVS/models/dist_activity_synthetic/allocator.py

class MyAllocator(object):
    """
    Allocate all models at the start of the simulation. After this, model relocation is handed over to a relocator.
    """
    def allocate(self, models, edges, nrnodes, totalActivities):
        """
        Calculate allocations for the nodes, using the information provided.

        :param models: the models to allocte
        :param edges: the edges between the models
        :param nrnodes: the number of nodes to allocate over. Simply an upper bound!
        :param totalActivities: activity tracking information from each model
        :returns: allocation that was found
        """
        # Return something of the form: {0: 0, 1: 0, 2: 0, 3: 1}
        # To allocate model_ids 0, 1 and 2 to node 0 and model_id 3 to node 1
        avgload = sum(totalActivities.values()) / nrnodes
        alloc = {}
        runningload = 0.0
        currentnode = 0
        for node, activity in totalActivities.items():
            if runningload + (activity / 2) > avgload:
                currentnode = (currentnode + 1) % nrnodes
                runningload = 0.0
            runningload += activity
            alloc[node] = currentnode
        return alloc

    def getTerminationTime(self):
        """
        Returns the time it takes for the allocator to make an 'educated guess' of the advised allocation.
        This time will not be used exactly, but as soon as the GVT passes over it. While this is not exactly 
        necessary, it avoids the overhead of putting such a test in frequently used code.

        :returns: float -- the time at which to perform the allocations (and save them)
        """
        # No need for any run time information means 0.0
        return 2.0