Set the window icon from the module resource pool

Makes use of the stb_image library
Conversion to HICON seems to have broken somewhere

src/xenia/ui/window.h

@@ -53,6 +53,8 @@ class Window {
     return true;
   }
 
+  virtual bool set_icon_from_buffer(void *buffer, size_t size) = 0;
+
   virtual bool is_fullscreen() const { return false; }
   virtual void ToggleFullscreen(bool fullscreen) {}
 

src/xenia/ui/window_win.cc

@@ -14,6 +14,11 @@
 #include "xenia/base/platform_win.h"
 #include "xenia/ui/window_win.h"
 
+#include "third_party/stb/stb_image.h"
+
+#define STB_IMAGE_IMPLEMENTATION
+#include "third_party/stb/stb_image.h"
+
 namespace xe {
 namespace ui {
 
@@ -163,6 +168,46 @@ bool Win32Window::set_title(const std::wstring& title) {
   return true;
 }
 
+bool Win32Window::set_icon_from_buffer(void *buffer, size_t size) {
+
+	if (icon_ != nullptr) {
+		DestroyIcon(icon_);
+	}
+	
+	// Convert the buffer for Windows
+	int width, height, bpp;
+	unsigned char *data = stbi_load_from_memory(reinterpret_cast<const stbi_uc*>(buffer), static_cast<int>(size), &width, &height, &bpp, 0);
+
+	int iSize = sizeof(BITMAPINFOHEADER);
+	int iSizeImage = width * height * bpp;
+
+	std::vector<uint8_t> bitmap(iSize + iSizeImage);
+	BITMAPINFOHEADER *pBitmap(reinterpret_cast<BITMAPINFOHEADER *>(bitmap.data()));
+
+	pBitmap->biSize = sizeof(BITMAPINFOHEADER);
+	pBitmap->biWidth = width;
+	pBitmap->biHeight = height;
+	pBitmap->biPlanes = 1;
+	pBitmap->biBitCount = bpp * 8;
+	pBitmap->biCompression = BI_RGB;
+	pBitmap->biSizeImage = iSizeImage;
+
+	unsigned char *pImage = bitmap.data() + iSize;
+
+	std::copy(data, data + iSizeImage, pImage);
+
+	HICON icon = CreateIconFromResourceEx((BYTE *)pBitmap, iSize + iSizeImage, TRUE, 0x00030000, width, height, LR_DEFAULTCOLOR);
+
+	if (icon != nullptr) {
+		SendMessage(hwnd_, WM_SETICON, ICON_BIG, reinterpret_cast<LPARAM>(icon));
+		SendMessage(hwnd_, WM_SETICON, ICON_SMALL, reinterpret_cast<LPARAM>(icon));
+
+		icon_ = icon;
+	}
+	
+	return false;
+}
+
 bool Win32Window::is_fullscreen() const { return fullscreen_; }
 
 void Win32Window::ToggleFullscreen(bool fullscreen) {
@@ -294,6 +339,9 @@ void Win32Window::Close() {
   Close();
   OnClose();
   DestroyWindow(hwnd_);
+  if (icon_ != nullptr) {
+	  DestroyIcon(icon_);
+  }
 }
 
 void Win32Window::OnMainMenuChange() {

src/xenia/ui/window_win.h

@@ -31,6 +31,7 @@ class Win32Window : public Window {
   HWND hwnd() const { return hwnd_; }
 
   bool set_title(const std::wstring& title) override;
+  bool set_icon_from_buffer(void *buffer, size_t size) override;
 
   bool is_fullscreen() const override;
   void ToggleFullscreen(bool fullscreen) override;
@@ -68,6 +69,7 @@ class Win32Window : public Window {
   bool HandleKeyboard(UINT message, WPARAM wParam, LPARAM lParam);
 
   HWND hwnd_ = nullptr;
+  HICON icon_ = nullptr;
   bool closing_ = false;
   bool fullscreen_ = false;