STARTREK: Quit game menu and mouse warp function

6 files changed, 192 insertions, 3 deletions

diff --git a/engines/startrek/graphics.h b/engines/startrek/graphics.h
index c63b2f6943..b3c84930a7 100644
--- a/engines/startrek/graphics.h
+++ b/engines/startrek/graphics.h
@@ -143,8 +143,15 @@ private:
 	void enableMenuButtons(uint32 bits);
 
 public:
+	void choseMousePositionFromSprites(Sprite *sprites, int numSprites, int spriteIndex, int mode);
 	void showOptionsMenu(int x, int y);
+	void showSaveMenu();
+	void showLoadMenu();
+	void setMusicEnabled(bool enabled);
+	void setSfxEnabled(bool enabled);
+	void showQuitGamePrompt(int x, int y);
 	void showTextConfigurationMenu(bool fromOptionMenu);
+
 	int loadTextDisplayMode();
 	void saveTextDisplayMode(int value);
 
diff --git a/engines/startrek/sprite.cpp b/engines/startrek/sprite.cpp
index d58cd3077e..ba5fa76072 100644
--- a/engines/startrek/sprite.cpp
+++ b/engines/startrek/sprite.cpp
@@ -32,4 +32,13 @@ void Sprite::setBitmap(SharedPtr<Bitmap> b) {
 	bitmapChanged = 1;
 }
 
+/**
+ * Returns a rect containing the sprite's bitmap on the screen.
+ */
+Common::Rect Sprite::getRect() {
+	Common::Rect rect(bitmap->width, bitmap->height);
+	rect.translate(pos.x - bitmap->xoffset, pos.y - bitmap->yoffset);
+	return rect;
+}
+
 }
diff --git a/engines/startrek/sprite.h b/engines/startrek/sprite.h
index c7a26dee4a..fa57332d58 100644
--- a/engines/startrek/sprite.h
+++ b/engines/startrek/sprite.h
@@ -62,6 +62,7 @@ struct Sprite {
 
 	Sprite() { memset(this, 0, sizeof(Sprite)); }
 	void setBitmap(SharedPtr<Bitmap> b);
+	Common::Rect getRect();
 };
 
 
diff --git a/engines/startrek/startrek.cpp b/engines/startrek/startrek.cpp
index de851b36a2..82df4beac1 100644
--- a/engines/startrek/startrek.cpp
+++ b/engines/startrek/startrek.cpp
@@ -58,6 +58,8 @@ StarTrekEngine::StarTrekEngine(OSystem *syst, const StarTrekGameDescription *gam
 
 	_mouseControllingShip = false;
 	_keyboardControlsMouse = true;
+
+	_inQuitGameMenu = false;
 }
 
 StarTrekEngine::~StarTrekEngine() {
diff --git a/engines/startrek/startrek.h b/engines/startrek/startrek.h
index 5f57f9bf33..0453c6af7c 100644
--- a/engines/startrek/startrek.h
+++ b/engines/startrek/startrek.h
@@ -161,6 +161,8 @@ public:
 	// between items in a menu).
 	bool _keyboardControlsMouse;
 
+	bool _inQuitGameMenu;
+
 	
 private:
 	Graphics *_gfx;
diff --git a/engines/startrek/text.cpp b/engines/startrek/text.cpp
index 8c4f05ca1e..de00cb1919 100644
--- a/engines/startrek/text.cpp
+++ b/engines/startrek/text.cpp
@@ -17,6 +17,7 @@
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
  */
 
 #include "common/events.h"
@@ -31,7 +32,7 @@ enum MenuEvent {
 	MENUEVENT_RCLICK_OFFBUTTON = -4,
 	MENUEVENT_ENABLEINPUT,          // Makes buttons selectable (occurs after a delay)
 	MENUEVENT_RCLICK_ONBUTTON,
-	MENUEVENT_LCLICK_OFFBUTTON,
+	MENUEVENT_LCLICK_OFFBUTTON
 };
 
 // Buttons for standard text display
@@ -896,6 +897,122 @@ void Graphics::enableMenuButtons(uint32 bits) {
 	_activeMenu->disabledButtons &= ~bits;
 }
 
+/**
+ * This chooses a sprite from the list to place the mouse cursor at. The sprite it chooses
+ * may be, for example, the top-leftmost one in the list. Exact behaviour is determined by
+ * the "mode" parameter.
+ *
+ * If "containMouseSprite" is a valid index, it's ensured that the mouse is contained
+ * within it. "mode" should be -1 in this case.
+ */
+void Graphics::choseMousePositionFromSprites(Sprite *sprites, int numSprites, int containMouseSprite, int mode) {
+	uint16 mouseX1 = 0x7fff; // Candidate positions to warp mouse to
+	uint16 mouseY1 = 0x7fff;
+	uint16 mouseX2 = 0x7fff;
+	uint16 mouseY2 = 0x7fff;
+
+	Common::Point mousePos = getMousePos();
+
+	// Ensure the cursor is contained within one of the sprites
+	if (containMouseSprite >= 0 && containMouseSprite < numSprites) {
+		Common::Rect rect = sprites[containMouseSprite].getRect();
+
+		if (mousePos.x < rect.left || mousePos.x >= rect.right
+				|| mousePos.y < rect.top || mousePos.y >= rect.bottom) {
+			mousePos.x = (rect.left + rect.right) / 2;
+			mousePos.y = (rect.top + rect.bottom) / 2;
+		}
+	}
+
+	// Choose a sprite to warp the cursor to
+	for (int i = 0; i < numSprites; i++) {
+		Sprite *sprite = &sprites[i];
+		if (sprite->drawMode != 2)
+			continue;
+
+		Common::Rect rect = sprite->getRect();
+
+		int hCenter = (rect.left + rect.right) / 2;
+		int vCenter = (rect.top + rect.bottom) / 2;
+
+		// Choose which sprite is closest based on certain criteria?
+		switch(mode) {
+		case 0: // Choose topmost, leftmost sprite that's below the cursor
+			if (((vCenter == mousePos.y && hCenter > mousePos.x) || vCenter > mousePos.y)
+					&& (vCenter < mouseY1 || (vCenter == mouseY1 && hCenter < mouseX1))) {
+				mouseX1 = hCenter;
+				mouseY1 = vCenter;
+			}
+			// fall through
+
+		case 4: // Choose topmost, leftmost sprite
+			if (vCenter < mouseY2 || (vCenter == mouseY2 && hCenter < mouseX2)) {
+				mouseX2 = hCenter;
+				mouseY2 = vCenter;
+			}
+			break;
+
+		case 1: // Choose bottommost, rightmost sprite that's above the cursor
+			if (((vCenter == mousePos.y && hCenter < mousePos.x) || vCenter < mousePos.y)
+					&& (mouseY1 == 0x7fff || vCenter > mouseY1
+						|| (vCenter == mouseY1 && hCenter > mouseX1))) {
+				mouseX1 = hCenter;
+				mouseY1 = vCenter;
+			}
+			// fall through
+
+		case 5: // Choose bottommost, rightmost sprite
+			if (mouseY2 == 0x7fff || vCenter > mouseY2
+					|| (vCenter == mouseY2 && hCenter > mouseX2)) {
+				mouseX2 = hCenter;
+				mouseY2 = vCenter;
+			}
+			break;
+
+		case 2:
+			// This seems broken... OR condition on first line has no affect on the logic...
+			if ((vCenter < mousePos.y || (vCenter == mouseY1 && hCenter == mousePos.x))
+					&& (mouseX1 == 0x7fff || vCenter >= mouseY1)) {
+				mouseX1 = hCenter;
+				mouseY1 = vCenter;
+				debug("Try %d %d", mouseX1, mouseY1);
+			}
+			if (mouseX2 == 0x7fff || vCenter > mouseY2
+					|| (hCenter == mouseX2 && vCenter == mouseY2)) {
+				mouseX2 = hCenter;
+				mouseY2 = vCenter;
+			}
+			break;
+
+		case 3:
+			// Similar to above...
+			if ((vCenter > mousePos.y || (vCenter == mouseY1 && hCenter == mousePos.x))
+					&& (mouseX1 == 0x7fff || vCenter <= mouseY1)) {
+				mouseX1 = hCenter;
+				mouseY1 = vCenter;
+			}
+			if (mouseX2 == 0x7fff || vCenter < mouseY2
+					|| (hCenter == mouseX2 && vCenter == mouseY2)) {
+				mouseX2 = hCenter;
+				mouseY2 = vCenter;
+			}
+			break;
+		}
+	}
+
+	// Warp mouse to one of the coordinates, if one is valid
+	if (mouseX1 != 0x7fff) {
+		mousePos.x = mouseX1;
+		mousePos.y = mouseY1;
+	}
+	else if (mouseX2 != 0x7fff) {
+		mousePos.x = mouseX2;
+		mousePos.y = mouseY2;
+	}
+
+	_vm->_system->warpMouse(mousePos.x, mousePos.y);
+
+}
 
 void Graphics::showOptionsMenu(int x, int y) {
 	bool tmpMouseControllingShip = _vm->_mouseControllingShip;
@@ -927,7 +1044,7 @@ void Graphics::showOptionsMenu(int x, int y) {
 		disabledButtons |= (1 << OPTIONBUTTON_ENABLESFX) | (1 << OPTIONBUTTON_DISABLESFX);
 
 	disableMenuButtons(disabledButtons);
-	// sub_28b5d();
+	choseMousePositionFromSprites(_activeMenu->sprites, _activeMenu->numButtons, -1, 4);
 	int event = handleMenuEvents(0, false);
 
 	unloadMenuButtons();
@@ -940,14 +1057,28 @@ void Graphics::showOptionsMenu(int x, int y) {
 
 	// Can't use OPTIONBUTTON constants since the button retvals differ from the button
 	// indices...
-	switch(event) { // TODO
+	switch(event) {
 	case 0: // Save
+		showSaveMenu();
+		break;
 	case 1: // Load
+		showLoadMenu();
+		break;
 	case 2: // Enable music
+		setMusicEnabled(true);
+		break;
 	case 3: // Disable music
+		setMusicEnabled(false);
+		break;
 	case 4: // Enable sfx
+		setSfxEnabled(true);
+		break;
 	case 5: // Disable sfx
+		setSfxEnabled(false);
+		break;
 	case 6: // Quit
+		showQuitGamePrompt(120, 20); // TODO: revert 120 to 20
+		break;
 	case 7: // Text
 		showTextConfigurationMenu(true);
 		break;
@@ -956,6 +1087,43 @@ void Graphics::showOptionsMenu(int x, int y) {
 	}
 }
 
+void Graphics::showSaveMenu() {
+	// TODO
+}
+
+void Graphics::showLoadMenu() {
+	// TODO
+}
+
+void Graphics::setMusicEnabled(bool enabled) {
+	// TODO
+}
+
+void Graphics::setSfxEnabled(bool enabled) {
+	// TODO
+}
+
+void Graphics::showQuitGamePrompt(int x, int y) {
+	const char *options[] = {
+		"Quit Game",
+		"#GENE\\GENER028#Yes, quit the game.",
+		"#GENE\\GENER008#No, do not quit the game.",
+		""
+	};
+
+	if (_vm->_inQuitGameMenu)
+		return;
+
+	_vm->_inQuitGameMenu = true;
+	int val = showText(&Graphics::readTextFromArray, (uintptr)options, x, y, 0xb0, true, 0, 1);
+	_vm->_inQuitGameMenu = false;
+
+	if (val == 0) {
+		// sub_1e70d();
+		_vm->_system->quit();
+	}
+}
+
 /**
  * This can be called from startup or from the options menu.
  * On startup, this tries to load the setting without user input.