From 0a915305a6481f3b9553511376358ab81a51724c Mon Sep 17 00:00:00 2001 From: David Date: Thu, 5 Jun 2014 17:49:36 +1200 Subject: Clean-ups --- tray.c | 407 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 407 insertions(+) create mode 100644 tray.c (limited to 'tray.c') 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 * + * . * + ************************************************************************/ + +#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 +#include +#include +#include +#include +#include + + +#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); +} -- cgit v1.1