Initial commit

models/dist_activity_synthetic/allocator.py

@@ -0,0 +1,38 @@
+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