1 files changed, 24 insertions, 9 deletions
diff --git a/src/python/m5/event.py b/src/python/m5/event.py
index 2d6497464..5d50448e7 100644
--- a/src/python/m5/event.py
+++ b/src/python/m5/event.py
@@ -26,17 +26,32 @@
 #
 # Authors: Nathan Binkert
 
-from internal.event import create
-from internal.event import SimLoopExitEvent as SimExit
+import internal.event
 
-class ProgressEvent(object):
-    def __init__(self, period):
-        self.period = int(period)
-        self.schedule()
+from internal.event import PythonEvent, SimLoopExitEvent as SimExit
+
+mainq = internal.event.cvar.mainEventQueue
+
+def create(obj, priority=None):
+    if priority is None:
+        priority = internal.event.Event.Default_Pri
+    return internal.event.PythonEvent(obj, priority)
 
-    def schedule(self):
-        create(self, m5.curTick() + self.period)
+class Event(PythonEvent):
+    def __init__(self, priority=None):
+        if priority is None:
+            priority = internal.event.Event.Default_Pri
+        super(PythonEvent, self).__init__(self, priority)
+
+class ProgressEvent(Event):
+    def __init__(self, eventq, period):
+        super(ProgressEvent, self).__init__()
+        self.period = int(period)
+        self.eventq = eventq
+        self.eventq.schedule(self, m5.curTick() + self.period)
 
     def __call__(self):
         print "Progress! Time now %fs" % (m5.curTick()/1e12)
-        self.schedule()
+        self.eventq.schedule(self, m5.curTick() + self.period)
+
+__all__ = [ 'create', 'Event', 'ProgressEvent', 'SimExit', 'mainq' ]