From 17cb701324fd442c5c9cc9ea6b662bf829d92ff2 Mon Sep 17 00:00:00 2001
From: Ash Tyndall <git@atyndall.net>
Date: Sat, 8 Oct 2011 13:44:22 +0800
Subject: [PATCH]

---
 scene.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/scene.c b/scene.c
index 8fddc81..2733cf8 100644
--- a/scene.c
+++ b/scene.c
@@ -269,10 +269,13 @@ void makeMenu() {
 
   // Construct add object submenus
   int addObjectSubmenu[6];
+  int objectMenuEntriesSize = sizeof(objectMenuEntries) / sizeof(objectMenuEntries[0]);
+  int menuNumber = objectMenuEntriesSize / 10 + 1;
 
-  for( int i = 0; i < 6; i++ ) {
+  for( int i = 0; i < menuNumber; i++ ) {
     addObjectSubmenu[i] = glutCreateMenu(processObjectEvents);
     for ( int j = i; j < (i+9); j++ ) {
+      if ( !objectMenuEntries[j] ) break; // Detect if we've reached the end of the array
       glutAddMenuEntry( objectMenuEntries[j], j );
     }
   }
-- 
2.20.1