Added nextEventTime(); added NewEventListeners

[Mograsim.git] / era.mi / src / era / mi / logic / timeline / Timeline.java

diff --git a/era.mi/src/era/mi/logic/timeline/Timeline.java b/era.mi/src/era/mi/logic/timeline/Timeline.java
index 186e946..616db7c 100644 (file)
--- a/era.mi/src/era/mi/logic/timeline/Timeline.java
+++ b/era.mi/src/era/mi/logic/timeline/Timeline.java
@@ -1,6 +1,9 @@
 package era.mi.logic.timeline;\r
 \r
+import java.util.ArrayList;\r
+import java.util.List;\r
 import java.util.PriorityQueue;\r
+import java.util.function.Consumer;\r
 \r
 /**\r
  * Orders Events by the time they are due to be executed. Can execute Events individually.\r
@@ -12,6 +15,8 @@ public class Timeline
        private PriorityQueue<InnerEvent> events;\r
        private long currentTime = 0;\r
        \r
+       private final List<Consumer<TimelineEvent>> eventAddedListener;\r
+       \r
        public Timeline(int initCapacity)\r
        {\r
                events = new PriorityQueue<InnerEvent>(initCapacity, (a, b) -> {\r
@@ -20,6 +25,8 @@ public class Timeline
                                return 0;\r
                        return difference < 0 ? -1 : 1;\r
                });\r
+               \r
+               eventAddedListener = new ArrayList<>();\r
        }\r
        \r
        public boolean hasNext()\r
@@ -39,17 +46,69 @@ public class Timeline
                while (hasNext())\r
                        executeNext();\r
        }\r
+       \r
+       /**\r
+        * Executes all events up to a given simulation timestamp. The simulation\r
+        * process can be constrained by a real world timestamp.\r
+        * \r
+        * @param timestamp  the simulation timestamp up to which the events will be\r
+        *                   processed\r
+        * @param stopMillis the System.currentTimeMillis() when simulation definitely\r
+        *                   needs to stop.\r
+        * @return if it was possible to fulfil the goal in the given real world time.\r
+        * @author Christian Femers\r
+        */\r
+       public ExecutionResult executeUpTo(long timestamp, long stopMillis)\r
+       {\r
+               if (events.isEmpty())\r
+               {\r
+                       currentTime = timestamp;\r
+                       return ExecutionResult.NOTHING_DONE;\r
+               }\r
+               int checkStop = 0;\r
+               InnerEvent first = events.peek();\r
+               while (first != null && first.getTiming() <= timestamp)\r
+               {\r
+                       events.remove();\r
+                       currentTime = first.getTiming();\r
+                       first.run();\r
+                       // Don't check after every run\r
+                       checkStop = (checkStop + 1) % 10;\r
+                       if (checkStop == 0 && System.currentTimeMillis() >= stopMillis)\r
+                               return ExecutionResult.RAN_OUT_OF_TIME;\r
+                       first = events.peek();\r
+               }\r
+               currentTime = timestamp;\r
+               return ExecutionResult.DONE_IN_TIME;\r
+       }\r
 \r
        public long getSimulationTime()\r
        {\r
                return currentTime;\r
        }\r
+\r
+       public long nextEventTime()\r
+       {\r
+               if(!hasNext())\r
+                       return -1;\r
+               else\r
+                       return events.peek().timing;\r
+       }\r
        \r
        public void reset()\r
        {\r
                events.clear();\r
                currentTime = 0;\r
        }\r
+\r
+       public void addEventAddedListener(Consumer<TimelineEvent> listener)\r
+       {\r
+               eventAddedListener.add(listener);\r
+       }\r
+       public void removeEventAddedListener(Consumer<TimelineEvent> listener)\r
+       {\r
+               eventAddedListener.remove(listener);\r
+       }\r
        \r
        /**\r
         * Adds an Event to the {@link Timeline}\r
@@ -59,7 +118,9 @@ public class Timeline
        public void addEvent(TimelineEventHandler function, int relativeTiming)\r
        {\r
                long timing = currentTime + relativeTiming;\r
-               events.add(new InnerEvent(function, new TimelineEvent(timing), timing));\r
+               TimelineEvent event = new TimelineEvent(timing);\r
+               events.add(new InnerEvent(function, event, timing));\r
+               eventAddedListener.forEach(l -> l.accept(event));\r
        }\r
        \r
        private class InnerEvent\r
@@ -108,4 +169,9 @@ public class Timeline
        {\r
                return ticks; //TODO: Alter this when it has been determined how ticks should relate to real time.\r
        }\r
+       \r
+       public enum ExecutionResult\r
+       {\r
+               NOTHING_DONE, DONE_IN_TIME, RAN_OUT_OF_TIME \r
+       }\r
 }
\ No newline at end of file