add icon when opening a window

Author: yunline

src_c/window.c

@@ -6,6 +6,92 @@
 
 #include "doc/sdl2_video_doc.h"
 
+// Copied from display.c
+#if !defined(__APPLE__)
+static char *icon_defaultname = "pygame_icon.bmp";
+static int icon_colorkey = 0;
+#else
+static char *icon_defaultname = "pygame_icon_mac.bmp";
+static int icon_colorkey = -1;
+#endif
+
+static char *pkgdatamodule_name = "pygame.pkgdata";
+static char *imagemodule_name = "pygame.image";
+static char *resourcefunc_name = "getResource";
+static char *load_basicfunc_name = "load_basic";
+
+// Copied from display.c
+static void
+pg_close_file(PyObject *fileobj)
+{
+    PyObject *result = PyObject_CallMethod(fileobj, "close", NULL);
+    if (result) {
+        Py_DECREF(result);
+    }
+    else {
+        PyErr_Clear();
+    }
+}
+
+// Copied from display.c
+static PyObject *
+pg_display_resource(char *filename)
+{
+    PyObject *imagemodule = NULL;
+    PyObject *load_basicfunc = NULL;
+    PyObject *pkgdatamodule = NULL;
+    PyObject *resourcefunc = NULL;
+    PyObject *fresult = NULL;
+    PyObject *result = NULL;
+    PyObject *name = NULL;
+
+    pkgdatamodule = PyImport_ImportModule(pkgdatamodule_name);
+    if (!pkgdatamodule)
+        goto display_resource_end;
+
+    resourcefunc = PyObject_GetAttrString(pkgdatamodule, resourcefunc_name);
+    if (!resourcefunc)
+        goto display_resource_end;
+
+    imagemodule = PyImport_ImportModule(imagemodule_name);
+    if (!imagemodule)
+        goto display_resource_end;
+
+    load_basicfunc = PyObject_GetAttrString(imagemodule, load_basicfunc_name);
+    if (!load_basicfunc)
+        goto display_resource_end;
+
+    fresult = PyObject_CallFunction(resourcefunc, "s", filename);
+    if (!fresult)
+        goto display_resource_end;
+
+    name = PyObject_GetAttrString(fresult, "name");
+    if (name != NULL) {
+        if (PyUnicode_Check(name)) {
+            pg_close_file(fresult);
+            Py_DECREF(fresult);
+            fresult = name;
+            name = NULL;
+        }
+    }
+    else {
+        PyErr_Clear();
+    }
+
+    result = PyObject_CallFunction(load_basicfunc, "O", fresult);
+    if (!result)
+        goto display_resource_end;
+
+display_resource_end:
+    Py_XDECREF(pkgdatamodule);
+    Py_XDECREF(resourcefunc);
+    Py_XDECREF(imagemodule);
+    Py_XDECREF(load_basicfunc);
+    Py_XDECREF(fresult);
+    Py_XDECREF(name);
+    return result;
+}
+
 static PyTypeObject pgWindow_Type;
 
 #define pgWindow_Check(x) \
@@ -553,6 +639,19 @@ window_init(pgWindowObject *self, PyObject *args, PyObject *kwargs)
 
     SDL_SetWindowData(_win, "pg_window", self);
 
+    PyObject *icon = pg_display_resource(icon_defaultname);
+    if (!icon) {
+        return -1;
+    }
+    if (icon_colorkey != -1) {
+        if (SDL_SetColorKey(pgSurface_AsSurface(icon), SDL_TRUE,
+                            icon_colorkey) < 0) {
+            PyErr_SetString(pgExc_SDLError, SDL_GetError());
+            return -1;
+        }
+    }
+    SDL_SetWindowIcon(self->_win, pgSurface_AsSurface(icon));
+
     return 0;
 }