[gtk+] Make window icons work in GtkHeaderBar

[Matthias Clasen <matthiasc src gnome org>]

commit dc073b8020c7b9be5dc6d3935f06c9f633d27396
Author: Matthias Clasen <mclasen redhat com>
Date:   Fri Dec 13 00:39:59 2013 -0500

    Make window icons work in GtkHeaderBar

 gtk/gtkheaderbar.c        |   68 +++++++++++++++++++++++++++------------------
 gtk/gtkheaderbarprivate.h |    4 +-
 gtk/gtkwindow.c           |    2 +-
 3 files changed, 44 insertions(+), 30 deletions(-)
---
diff --git a/gtk/gtkheaderbar.c b/gtk/gtkheaderbar.c
index 64384b5..99902db 100644
--- a/gtk/gtkheaderbar.c
+++ b/gtk/gtkheaderbar.c
@@ -200,22 +200,30 @@ create_title_box (const char *title,
   return label_box;
 }
 
-void
+gboolean
 _gtk_header_bar_update_window_icon (GtkHeaderBar *bar,
-                                    GList        *list)
+                                    GtkWindow    *window)
 {
   GtkHeaderBarPrivate *priv = gtk_header_bar_get_instance_private (bar);
+  gint size;
+  GList *list;
+  const gchar *name;
+  GdkPixbuf *best = NULL;
+
+  if (priv->titlebar_icon == NULL)
+    return FALSE;
+
+  if (GTK_IS_BUTTON (gtk_widget_get_parent (priv->titlebar_icon)))
+    size = 16;
+  else
+    size = 20;
 
-  if (priv->titlebar_icon && list != NULL)
+  list = gtk_window_get_icon_list (window);
+
+  if (list != NULL)
     {
-      GdkPixbuf *pixbuf, *best;
+      GdkPixbuf *pixbuf;
       GList *l;
-      gint size;
-
-      if (GTK_IS_BUTTON (gtk_widget_get_parent (priv->titlebar_icon)))
-        size = 16;
-      else
-        size = 20;
 
       best = NULL;
       for (l = list; l; l = l->next)
@@ -231,11 +239,26 @@ _gtk_header_bar_update_window_icon (GtkHeaderBar *bar,
       if (best == NULL)
         best = gdk_pixbuf_scale_simple (GDK_PIXBUF (list->data), size, size, GDK_INTERP_BILINEAR);
 
+      g_list_free (list);
+    }
+  else
+    {
+      name = gtk_window_get_icon_name (window);
+      if (name != NULL)
+        best = gtk_icon_theme_load_icon (gtk_icon_theme_get_default (),
+                                         name, size, GTK_ICON_LOOKUP_FORCE_SIZE, NULL);
+    }
+
+  if (best)
+    {
       gtk_image_set_from_pixbuf (GTK_IMAGE (priv->titlebar_icon), best);
       g_object_unref (best);
-
       gtk_widget_show (priv->titlebar_icon);
+
+      return TRUE;
     }
+
+  return FALSE;
 }
 
 void
@@ -247,7 +270,6 @@ _gtk_header_bar_update_window_buttons (GtkHeaderBar *bar)
   gchar *layout_desc;
   gchar **tokens, **t;
   gint i, j;
-  GdkPixbuf *icon = NULL;
   GMenuModel *menu;
   gboolean shown_by_shell;
 
@@ -260,9 +282,6 @@ _gtk_header_bar_update_window_buttons (GtkHeaderBar *bar)
 
   if (priv->titlebar_icon)
     {
-      icon = gtk_image_get_pixbuf (GTK_IMAGE (priv->titlebar_icon));
-      if (icon)
-        g_object_ref (icon);
       gtk_widget_destroy (priv->titlebar_icon);
       priv->titlebar_icon = NULL;
     }
@@ -343,30 +362,27 @@ _gtk_header_bar_update_window_buttons (GtkHeaderBar *bar)
                   gtk_style_context_add_class (gtk_widget_get_style_context (button), "titlebutton");
                   gtk_widget_set_size_request (button, 20, 20);
                   gtk_widget_show (button);
-                  if (icon != NULL)
-                    gtk_image_set_from_pixbuf (GTK_IMAGE (button), icon);
-                  else
-                    gtk_widget_hide (button);
-
                   priv->titlebar_icon = button;
+                  if (!_gtk_header_bar_update_window_icon (bar, window))
+                    gtk_widget_hide (button);
                 }
               else if (strcmp (t[j], "menu") == 0 && menu != NULL)
                 {
                   button = gtk_menu_button_new ();
                   gtk_menu_button_set_menu_model (GTK_MENU_BUTTON (button), menu);
                   gtk_style_context_add_class (gtk_widget_get_style_context (button), "titlebutton");
-                  if (icon != NULL)
-                    image = gtk_image_new_from_pixbuf (icon);
-                  else
-                    image = gtk_image_new_from_icon_name ("process-stop-symbolic", GTK_ICON_SIZE_MENU);
+                  image = gtk_image_new ();
                   gtk_container_add (GTK_CONTAINER (button), image);
                   gtk_widget_set_can_focus (button, FALSE);
                   gtk_widget_show_all (button);
                   accessible = gtk_widget_get_accessible (button);
                   if (GTK_IS_ACCESSIBLE (accessible))
                     atk_object_set_name (accessible, _("Application menu"));
-                 priv->titlebar_icon = image;
+                  priv->titlebar_icon = image;
                   priv->titlebar_menu_button = button;
+
+                  if (!_gtk_header_bar_update_window_icon (bar, window))
+                    gtk_image_set_from_icon_name (GTK_IMAGE (priv->titlebar_icon), "process-stop-symbolic", 
GTK_ICON_SIZE_MENU);
                 }
               else if (strcmp (t[j], "minimize") == 0 &&
                        gtk_window_get_type_hint (window) == GDK_WINDOW_TYPE_HINT_NORMAL)
@@ -464,8 +480,6 @@ _gtk_header_bar_update_window_buttons (GtkHeaderBar *bar)
       g_strfreev (tokens);
     }
   g_free (layout_desc);
-  if (icon)
-    g_object_unref (icon);
 }
 
 gboolean
diff --git a/gtk/gtkheaderbarprivate.h b/gtk/gtkheaderbarprivate.h
index c4c04ba..58cd3bb 100644
--- a/gtk/gtkheaderbarprivate.h
+++ b/gtk/gtkheaderbarprivate.h
@@ -22,7 +22,7 @@
 
 gboolean     _gtk_header_bar_get_shows_app_menu    (GtkHeaderBar *bar);
 void         _gtk_header_bar_update_window_buttons (GtkHeaderBar *bar);
-void         _gtk_header_bar_update_window_icon    (GtkHeaderBar *bar,
-                                                    GList        *list);
+gboolean     _gtk_header_bar_update_window_icon    (GtkHeaderBar *bar,
+                                                    GtkWindow    *window);
 
 #endif /* __GTK_HEADER_BAR_PRIVATE_H__ */
diff --git a/gtk/gtkwindow.c b/gtk/gtkwindow.c
index 88b884c..1a8e691 100644
--- a/gtk/gtkwindow.c
+++ b/gtk/gtkwindow.c
@@ -3908,7 +3908,7 @@ gtk_window_realize_icon (GtkWindow *window)
 
   gdk_window_set_icon_list (gtk_widget_get_window (widget), icon_list);
   if (GTK_IS_HEADER_BAR (priv->title_box))
-    _gtk_header_bar_update_window_icon (GTK_HEADER_BAR (priv->title_box), icon_list);
+    _gtk_header_bar_update_window_icon (GTK_HEADER_BAR (priv->title_box), window);
 
   if (info->using_themed_icon) 
     {