From 407877ff24d3c54747c8b15bba73b93b38c8b6e7 Mon Sep 17 00:00:00 2001
From: Fabian Mastenbroek <mail.fabianm@gmail.com>
Date: Mon, 12 Feb 2018 22:27:44 +0100
Subject: refactor(#18): Launch processes at initial run

This change will make the simulation kernel launch the processes at the
initial run instead of when the processes are registered.
---
 .../kotlin/com/atlarge/opendc/omega/OmegaKernel.kt | 37 ++++++++++++++++------
 1 file changed, 27 insertions(+), 10 deletions(-)

(limited to 'opendc-kernel-omega/src/main')

diff --git a/opendc-kernel-omega/src/main/kotlin/com/atlarge/opendc/omega/OmegaKernel.kt b/opendc-kernel-omega/src/main/kotlin/com/atlarge/opendc/omega/OmegaKernel.kt
index c729a63d..91150078 100644
--- a/opendc-kernel-omega/src/main/kotlin/com/atlarge/opendc/omega/OmegaKernel.kt
+++ b/opendc-kernel-omega/src/main/kotlin/com/atlarge/opendc/omega/OmegaKernel.kt
@@ -55,6 +55,11 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
      */
     private val queue: Queue<MessageContainer> = PriorityQueue(Comparator.comparingLong(MessageContainer::time))
 
+    /**
+     * The processes to be spawned.
+     */
+    private val spawnings: Queue<Process<*, M>> = ArrayDeque()
+
     /**
      * The simulation time.
      */
@@ -80,14 +85,8 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
             return false
         }
 
-        @Suppress("UNCHECKED_CAST")
-        val process = entity as Process<Any, M>
-        val context = OmegaContext(entity).also { registry[entity] = it }
-
-        // Bootstrap the process coroutine
-        val block: suspend () -> Unit = { process.run { context.run() } }
-        block.startCoroutine(context)
-
+        val process = entity as Process<*, M>
+        spawnings.add(process)
         return true
     }
 
@@ -102,6 +101,17 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
 
     override fun step() {
         while (true) {
+            // Initialise all spawned processes
+            while (spawnings.isNotEmpty()) {
+                val process = spawnings.poll()
+                val context = OmegaContext(process).also { registry[process] = it }
+
+                // Bootstrap the process coroutine
+                val block: suspend () -> Unit = { context.start() }
+                block.startCoroutine(context)
+
+            }
+
             val envelope = queue.peek() ?: return
             val delivery = envelope.time
 
@@ -136,7 +146,7 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
 
 
     override fun run() {
-        while (queue.isNotEmpty()) {
+        while (queue.isNotEmpty() || spawnings.isNotEmpty()) {
             step()
         }
     }
@@ -148,7 +158,7 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
             return
         }
 
-        while (time < until && queue.isNotEmpty()) {
+        while (time < until && (queue.isNotEmpty() || spawnings.isNotEmpty())) {
             step()
         }
 
@@ -219,6 +229,13 @@ internal class OmegaKernel<M>(bootstrap: Bootstrap<M>) : Kernel<M>, Bootstrap.Co
         override val <T : Entity<S, *>, S> T.state: S
             get() = context?.state ?: initialState
 
+        /**
+         * Start the process associated with this context.
+         */
+        internal suspend fun start() = process.run {
+            run()
+        }
+
         /**
          * Retrieve and remove and single message from the mailbox of the [Entity] and suspend the [Context] until the
          * message has been received.
-- 
cgit v1.2.3