# HG changeset patch
# User Simon Cross <hodgestar@gmail.com>
# Date 1315740858 -7200
# Node ID ad2bcbf492bf199e85040b40d7bd7bcd713134e0
# Parent  4e0d6ebf9127c42bb564f7143837cff74470cba3
Hook up top-level habitat support.

diff -r 4e0d6ebf9127 -r ad2bcbf492bf mamba/__main__.py
--- a/mamba/__main__.py	Sun Sep 11 13:30:55 2011 +0200
+++ b/mamba/__main__.py	Sun Sep 11 13:34:18 2011 +0200
@@ -7,6 +7,7 @@
 
 from mamba.constants import SCREEN
 from mamba.engine import Engine
+from mamba.habitats.mainmenu import MainMenu
 
 # For future use
 DEBUG = False
@@ -24,4 +25,5 @@
     pygame.display.set_caption('Mamba')
 
     engine = Engine()
+    engine.set_habitat(MainMenu())
     engine.run()
diff -r 4e0d6ebf9127 -r ad2bcbf492bf mamba/engine.py
--- a/mamba/engine.py	Sun Sep 11 13:30:55 2011 +0200
+++ b/mamba/engine.py	Sun Sep 11 13:34:18 2011 +0200
@@ -7,7 +7,10 @@
 
 class Engine(object):
     def __init__(self):
-        pass
+        self._habitat = None
+
+    def set_habitat(self, habitat):
+        self._habitat = habitat
 
     def run(self):
         """Game loop."""
@@ -18,9 +21,19 @@
             for ev in events:
                 if ev.type is QUIT:
                     return
+                self._habitat.dispatch(ev)
+            self._habitat.draw()
             flip()
 
 
+class Habitat(object):
+    def dispatch(self, ev):
+        pass
+
+    def draw(self):
+        pass
+
+
 class UserEvent(object):
 
     utype = "UNKNOWN"
diff -r 4e0d6ebf9127 -r ad2bcbf492bf mamba/habitats/mainmenu.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mamba/habitats/mainmenu.py	Sun Sep 11 13:34:18 2011 +0200
@@ -0,0 +1,8 @@
+"""Main menu."""
+
+from mamba.engine import Habitat
+
+
+class MainMenu(Habitat):
+    # TODO: make something interesting happen here.
+    pass