Clean-ups

1 files changed, 407 insertions, 0 deletions

diff --git a/tray.c b/tray.c
new file mode 100644
index 0000000..e1c2940
--- /dev/null
+++ b/tray.c
@@ -0,0 +1,407 @@
+/************************************************************************
+ * This file is part of trayfreq-archlinux.                             *
+ *                                                                      *
+ * trayfreq-archlinux is free software; you can redistribute it and/or  *
+ * modify it under the terms of the GNU General Public License as       *
+ * published by the Free Software Foundation; either version 3 of the   *
+ * License, or (at your option) any later version.                      *
+ *                                                                      *
+ * trayfreq-archlinux is distributed in the hope that it will be useful,*
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
+ * GNU General Public License for more details.                         *
+ *                                                                      *
+ * You should have received a copy of the GNU General Public License    *
+ * along with trayfreq-archlinux. If not, see                           *
+ * <http://www.gnu.org/licenses/>.                                      *
+ ************************************************************************/
+
+#include "tray.h"
+
+#include "getfreq.h"
+#include "getcore.h"
+#include "getgov.h"
+#include "trayfreq_set_interface.h"
+#include "bat_tray.h"
+#include "defaults.h"
+#include "common.h"
+
+#include <glib.h>
+#include <gtk/gtk.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <libintl.h>
+
+
+#define TOOLTIP_TEXT_SIZE 500
+
+GtkStatusIcon* tray;
+char tooltip_text[TOOLTIP_TEXT_SIZE];
+
+GtkWidget* menu;
+GSList* menu_items;
+GtkWidget* checked_menu_item;
+
+
+/***********************************************************************
+ * Handler for when freq is chosen
+ **********************************************************************/
+static void freq_menu_item_toggled(GtkCheckMenuItem* item, gpointer data)
+{
+	debug("[checking in]\n");
+	if(gtk_check_menu_item_get_active(item))
+	{
+		checked_menu_item = GTK_WIDGET(item);
+		gint freq = GPOINTER_TO_INT(data);
+		int i = 0;
+		for(i = 0; i < gc_number(); ++i)
+			si_freq(freq, i);
+	}
+}
+
+
+/***********************************************************************
+ * Handler for when governor is chosen
+ **********************************************************************/
+static void gov_menu_item_toggled(GtkCheckMenuItem* item, gpointer data)
+{
+	debug("[checking in]\n");
+	if(gtk_check_menu_item_get_active(item))
+	{
+		checked_menu_item = GTK_WIDGET(item);
+		char* gov = (char*)data;
+		int i = 0;
+		for(i = 0; i < gc_number(); ++i)
+			si_gov(gov, i);
+	}
+}
+/*
+static gboolean governor_exists(gchar* governor)
+{
+	int i = 0;
+	for(i = 0; i < gf_number(); ++i)
+	{
+		if(g_strcmp0(governor, gg_gov(0, i)) == 0)
+			return TRUE;
+	}
+	return FALSE;
+}*/
+
+
+/***********************************************************************
+ * Destroy a menu item
+ **********************************************************************/
+static void remove_menu_item(GtkWidget* menu_item, gpointer data)
+{
+	debug("Destroying menu_item\n");
+	gtk_widget_destroy(menu_item);
+}
+
+
+/***********************************************************************
+ * Remove all items in menu
+ **********************************************************************/
+static void tray_clear_menu()
+{
+	debug("Clearing the menu\n");
+	GtkContainer* cont = GTK_CONTAINER(menu);
+	gtk_container_foreach(cont, remove_menu_item, NULL);
+	menu_items = NULL;
+}
+
+
+/***********************************************************************
+ * Create new instance of gtk_tray_new for gov/freq tray
+ **********************************************************************/
+static void tray_init_menu()
+{
+	debug("Spawning new menu");
+	menu = gtk_menu_new();
+}
+
+/***********************************************************************
+ * Populate gtk menu object with all the entries
+ **********************************************************************/
+static void tray_generate_menu()
+{
+	tray_clear_menu();
+	gg_init();
+
+	char label[20];
+	int i = 0;
+
+	char current_governor[20];
+	memset(current_governor, '\0', sizeof(current_governor) );
+	gg_current(0, current_governor, sizeof(current_governor) );
+
+	gint current_frequency = gf_current(0);
+
+	// Add available frequencies
+	for(i = 0; i < gf_number(); ++i)
+	{
+		memset(label, '\0', 20);
+		gf_get_frequency_label(gf_freqi(0, i), label);
+		debug("Got freq label '%s', i=%d\n",label,i);
+
+		GtkWidget* item = gtk_radio_menu_item_new_with_label(menu_items, label);
+		menu_items = gtk_radio_menu_item_get_group(GTK_RADIO_MENU_ITEM (item));
+
+		if(g_strcmp0(current_governor, "userspace") == 0 && gf_freqi(0, i) == current_frequency)
+		{
+			debug("This freq is current freq, ticking radio button\n");
+			gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(item), TRUE);
+		}
+
+		debug("Setting connection/callback\n");
+		g_signal_connect(G_OBJECT(item), "toggled", GTK_SIGNAL_FUNC(freq_menu_item_toggled), GINT_TO_POINTER(gf_freqi(0, i)));
+		
+		debug("Adding item to menu\n");
+		gtk_menu_shell_append(GTK_MENU_SHELL(menu), item);
+	}
+
+	// Add a seperator
+	debug("Adding separator\n");
+	GtkWidget* seperator = gtk_separator_menu_item_new();
+	gtk_menu_append(menu, seperator);
+
+	// Add available governors
+	for(i = 0; i < gg_number(); i++)
+	{
+		if(g_strcmp0(gg_gov(0, i), "userspace") == 0)
+		{
+			debug("Gov is userspace, not adding\n");
+			continue;
+		}
+
+		GtkWidget* item = gtk_radio_menu_item_new_with_label(menu_items, gg_gov(0, i));
+		menu_items = gtk_radio_menu_item_get_group(GTK_RADIO_MENU_ITEM (item));
+
+		if(g_strcmp0(gg_gov(0, i), current_governor) == 0)
+		{
+			debug("Governor is current one, ticking radio button\n");
+			gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(item), TRUE);
+		}
+		
+		debug("Adding callback");
+		g_signal_connect(G_OBJECT(item), "toggled", GTK_SIGNAL_FUNC(gov_menu_item_toggled), gg_gov(0, i));
+		
+		debug("Adding item to menu\n");
+		gtk_menu_shell_append(GTK_MENU_SHELL(menu), item);
+	}
+	debug("Showing menu\n");
+	gtk_widget_show_all(menu);
+}
+
+/***********************************************************************
+ * Refresh the tooltip message
+ **********************************************************************/
+static gboolean update_tooltip(GtkStatusIcon* status_icon,gint x,gint y,gboolean keyboard_mode,GtkTooltip* tooltip,gpointer data)
+{
+	char msg[TOOLTIP_TEXT_SIZE];
+	char current_governor[20];
+	char label[20];
+	int i = 0;
+
+	memset(msg, '\0', sizeof(msg));
+	memset(current_governor, '\0', sizeof(current_governor) );
+
+	gg_current(0, current_governor, sizeof(current_governor) );
+	sprintf(msg+strlen(msg), _("Governor: %s\n"), current_governor);
+
+	for(i = 0; i < gc_number(); ++i)
+	{
+		debug("Adding CPU%i's frequency\n",i);
+		memset(label, '\0', sizeof(label));
+		gf_get_frequency_label(gf_current(i), label);
+		sprintf(msg+strlen(msg), _("CPU%i: %s%s"), i, label, i == gc_number()-1 ? "" : "\n");
+	}
+
+	debug("Setting tooltip text\n");
+	tray_set_tooltip(msg);
+	gtk_tooltip_set_text(tooltip, tooltip_text);
+
+	return TRUE;
+}
+
+
+/***********************************************************************
+ * Handler for spawning the popup/context menu
+ **********************************************************************/
+static void popup_menu(GtkStatusIcon* statuc_icon,guint button,guint activate_time,gpointer data)
+{
+	debug("Spawning popup menu\n");
+	tray_generate_menu();
+	gtk_menu_popup(GTK_MENU(menu),NULL,NULL,gtk_status_icon_position_menu,tray,button,activate_time);
+}
+
+
+/***********************************************************************
+ * Update the freq/gov tray icon
+ **********************************************************************/
+static gboolean update_icon(gpointer user_data)
+{
+	int i;
+	switch ( get_battery_state() )
+	{
+		case STATE_DISCHARGING:
+			debug("Discharging\n");
+			if(_DEFAULT_BAT_GOV)
+			{
+				for(i = 0; i < gc_number(); ++i)
+					si_gov(_DEFAULT_BAT_GOV, i);
+			}
+			break;
+
+		case STATE_CHARGING:
+		case STATE_FULL:
+			debug("Charging/Full\n");
+			if(_DEFAULT_AC_GOV)
+			{
+				for(i = 0; i < gc_number(); ++i)
+					si_gov(_DEFAULT_AC_GOV, i);
+			}
+
+			break;
+	}
+
+	debug("Updating icon\n");
+	tray_update_icon_percent();
+	return TRUE;
+}
+
+
+/***********************************************************************
+ * Apply default gov/freq for current situation
+ **********************************************************************/
+void tray_set_defaults()
+{
+	// Set defaults
+	int i = 0;
+	if(_DEFAULT_GOV)
+	{
+		for(i = 0; i < gc_number(); ++i)
+			si_gov(_DEFAULT_GOV, i);
+
+	} else {
+		for(i = 0; i < gc_number(); ++i)
+			si_gov("ondemand", i);
+	}
+
+	if(_DEFAULT_FREQ)
+	{
+		for(i = 0; i < gc_number(); ++i)
+			si_freq(atoi(_DEFAULT_FREQ), i);
+	}
+
+}
+
+
+/***********************************************************************
+ * Start the tray icon thingy
+ **********************************************************************/
+void tray_init()
+{
+	tray_set_defaults();
+	tray = gtk_status_icon_new();
+	char* icon_file = g_strconcat(_DEFAULT_THEME, "/cpufreq-0.png", NULL);
+
+	debug("Setting icon to '%s'\n",icon_file);
+	gtk_status_icon_set_from_file(tray, icon_file);
+	gtk_status_icon_set_has_tooltip(tray, TRUE);
+
+	debug("Setting up callbacks\n");
+	g_signal_connect(G_OBJECT(tray), "query-tooltip", GTK_SIGNAL_FUNC(update_tooltip), NULL);
+	g_signal_connect(G_OBJECT(tray), "popup-menu", GTK_SIGNAL_FUNC(popup_menu), NULL);
+
+	debug("Adding timeout\n");
+	g_timeout_add(1000, update_icon, NULL);
+	tray_init_menu();
+}
+
+
+/***********************************************************************
+ * Set the tooltip message
+ **********************************************************************/
+void tray_set_tooltip(const char* msg)
+{
+	debug("Setting up toolip var with text '%s'\n",msg);
+	memset(tooltip_text, '\0', TOOLTIP_TEXT_SIZE);
+	memmove(tooltip_text, msg, strlen(msg));
+}
+
+/**********************************************************************
+ * Set icon based on current freq/governor
+ **********************************************************************/
+void tray_update_icon_percent()
+{
+	gulong max_frequency = gf_freqi(0, 0);
+	gint adjusted_percent = 0;
+	// If no governor, set percentage to 0. This if statement fixes an FPE a few lines down
+	if (gg_number() == 0)
+	{
+		adjusted_percent = 0;
+	} else {
+		// Percentages need to be {25,50,75,100}. Round to one of these numbers.
+		// TO DO: round instead of lots of ifs
+		gint percent = (gf_current(0) * 100)/max_frequency;
+		if(percent == 100) {
+			adjusted_percent = 100;
+		} else if(percent >= 65.5) {
+			adjusted_percent = 75;
+		} else if(percent >= 37.5) {
+			adjusted_percent = 50;
+		} else if(percent >= 12.5) {
+			adjusted_percent = 25;
+		} else {
+			adjusted_percent = 0;
+		}
+	}
+
+	debug("Rounded/adjusted bat percentage: %d\n",adjusted_percent);
+	/* convert the int to a string */
+	char adjusted_percent_string[] = {'\0', '\0', '\0', '\0'};
+	sprintf(adjusted_percent_string, "%i", adjusted_percent);
+
+	char* file = g_strconcat(_DEFAULT_THEME, "/cpufreq-", adjusted_percent_string, ".png", NULL);
+	debug("Setting tray icon to '%s'\n",file);
+	gtk_status_icon_set_from_file(tray, file);
+
+	g_free(file);
+}
+
+
+/***********************************************************************
+ * Show the tray
+ **********************************************************************/
+void tray_show()
+{
+	debug("Showing tray\n");
+	gtk_status_icon_set_visible(tray, TRUE);
+}
+
+/***********************************************************************
+ * Hide the tray
+ **********************************************************************/
+void tray_hide()
+{
+	debug("Hiding tray");
+	gtk_status_icon_set_visible(tray, FALSE);
+}
+
+
+/***********************************************************************
+ * Return bool for visiblity of tray
+ **********************************************************************/
+bool tray_visible()
+{
+	return gtk_status_icon_get_visible(tray);
+}
+
+/***********************************************************************
+ * Return bool for embeddedness of tray
+ **********************************************************************/
+bool tray_embedded()
+{
+	return gtk_status_icon_is_embedded(tray);
+}