XEEN: Fixes for the buttons in the Items dialog

1 files changed, 25 insertions, 3 deletions

diff --git a/engines/xeen/dialogs/dialogs.h b/engines/xeen/dialogs/dialogs.h
index 08c17e8881..9e79b1ea5b 100644
--- a/engines/xeen/dialogs/dialogs.h
+++ b/engines/xeen/dialogs/dialogs.h
@@ -39,14 +39,36 @@ public:
 	Common::Rect _bounds;
 	SpriteResource *_sprites;
 	int _value;
-	uint _frameNum;
+	uint _frameNum, _selectedFrame;
 	bool _draw;
 
+	/**
+	 * Constructor
+	 */
 	UIButton(const Common::Rect &bounds, int value, uint frameNum, SpriteResource *sprites, bool draw) :
-		_bounds(bounds), _value(value), _frameNum(frameNum),
+		_bounds(bounds), _value(value), _frameNum(frameNum), _selectedFrame(frameNum | 1),
 		_sprites(sprites), _draw(draw) {}
 
-	UIButton() : _value(0), _frameNum(0), _sprites(nullptr), _draw(false) {}
+	/**
+	 * Constructor
+	 */
+	UIButton() : _value(0), _frameNum(0), _selectedFrame(0), _sprites(nullptr), _draw(false) {}
+
+	/**
+	 * Set the frame
+	 */
+	void setFrame(uint frameNum) {
+		_frameNum = frameNum;
+		_selectedFrame = frameNum | 1;
+	}
+
+	/**
+	 * Set the frame
+	 */
+	void setFrame(uint frameNum, uint selectedFrame) {
+		_frameNum = frameNum;
+		_selectedFrame = selectedFrame;
+	}
 };
 
 class ButtonContainer : public Cutscenes {