lib/tty/color.c

   1 /* Color setup.
   2    Interface functions.
   3
   4    Copyright (C) 1994, 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005,
   5    2007, 2008, 2009 Free Software Foundation, Inc.
   6
   7    Written by:
   8    Andrew Borodin <aborodin@vmail.ru>, 2009.
   9    Slava Zanko <slavazanko@gmail.com>, 2009.
  10
  11    This program is free software; you can redistribute it and/or modify
  12    it under the terms of the GNU General Public License as published by
  13    the Free Software Foundation; either version 2 of the License, or
  14    (at your option) any later version.
  15
  16    This program is distributed in the hope that it will be useful,
  17    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19    GNU General Public License for more details.
  20
  21    You should have received a copy of the GNU General Public License
  22    along with this program; if not, write to the Free Software
  23    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  24
  25 /** \file color.c
  26  *  \brief Source: color setup
  27  */
  28
  29 #include <config.h>
  30
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <sys/types.h>          /* size_t */
  35
  36 #include "lib/global.h"
  37
  38 #include "tty.h"
  39 #include "color.h"
  40
  41 #include "color-internal.h"
  42
  43 /*** global variables ****************************************************************************/
  44
  45 char *command_line_colors = NULL;
  46
  47 static char *tty_color_defaults__fg = NULL;
  48 static char *tty_color_defaults__bg = NULL;
  49
  50 /* Set if we are actually using colors */
  51 gboolean use_colors = FALSE;
  52
  53 /*** file scope macro definitions ****************************************************************/
  54
  55 /*** file scope type declarations ****************************************************************/
  56
  57 /*** file scope variables ************************************************************************/
  58
  59 static GHashTable *mc_tty_color__hashtable = NULL;
  60
  61 /*** file scope functions ************************************************************************/
  62 /* --------------------------------------------------------------------------------------------- */
  63
  64 static inline void
  65 color_hash_destroy_key (gpointer data)
  66 {
  67     g_free (data);
  68 }
  69
  70 /* --------------------------------------------------------------------------------------------- */
  71
  72 static void
  73 color_hash_destroy_value (gpointer data)
  74 {
  75     tty_color_pair_t *mc_color_pair = (tty_color_pair_t *) data;
  76     g_free (mc_color_pair);
  77 }
  78
  79 /* --------------------------------------------------------------------------------------------- */
  80
  81 static gboolean
  82 tty_color_free_condition_cb (gpointer key, gpointer value, gpointer user_data)
  83 {
  84     gboolean is_temp_color;
  85     tty_color_pair_t *mc_color_pair;
  86     (void) key;
  87
  88     is_temp_color = user_data != NULL;
  89     mc_color_pair = (tty_color_pair_t *) value;
  90     return (mc_color_pair->is_temp == is_temp_color);
  91 }
  92
  93 /* --------------------------------------------------------------------------------------------- */
  94
  95 static void
  96 tty_color_free_all (gboolean is_temp_color)
  97 {
  98     g_hash_table_foreach_remove (mc_tty_color__hashtable, tty_color_free_condition_cb,
  99                                  (is_temp_color) ? (gpointer) 1 : NULL);
 100 }
 101
 102 /* --------------------------------------------------------------------------------------------- */
 103
 104 static gboolean
 105 tty_color_get_next_cpn_cb (gpointer key, gpointer value, gpointer user_data)
 106 {
 107     size_t cp;
 108     tty_color_pair_t *mc_color_pair;
 109     (void) key;
 110
 111     cp = (size_t) user_data;
 112     mc_color_pair = (tty_color_pair_t *) value;
 113
 114     if (cp == mc_color_pair->pair_index)
 115         return TRUE;
 116
 117     return FALSE;
 118 }
 119
 120 /* --------------------------------------------------------------------------------------------- */
 121
 122 static int
 123 tty_color_get_next__color_pair_number ()
 124 {
 125     size_t cp_count = g_hash_table_size (mc_tty_color__hashtable);
 126     size_t cp = 0;
 127
 128     for (cp = 0; cp < cp_count; cp++)
 129     {
 130         if (g_hash_table_find (mc_tty_color__hashtable, tty_color_get_next_cpn_cb, (gpointer) cp) ==
 131             NULL)
 132             return cp;
 133     }
 134     return cp;
 135 }
 136
 137 /* --------------------------------------------------------------------------------------------- */
 138 /*** public functions ****************************************************************************/
 139 /* --------------------------------------------------------------------------------------------- */
 140
 141 void
 142 tty_init_colors (gboolean disable, gboolean force)
 143 {
 144     tty_color_init_lib (disable, force);
 145     mc_tty_color__hashtable = g_hash_table_new_full (g_str_hash, g_str_equal,
 146                                                      color_hash_destroy_key,
 147                                                      color_hash_destroy_value);
 148 }
 149
 150 /* --------------------------------------------------------------------------------------------- */
 151
 152 void
 153 tty_colors_done (void)
 154 {
 155     tty_color_deinit_lib ();
 156     g_free (tty_color_defaults__fg);
 157     g_free (tty_color_defaults__bg);
 158
 159     g_hash_table_destroy (mc_tty_color__hashtable);
 160 }
 161
 162 /* --------------------------------------------------------------------------------------------- */
 163
 164 gboolean
 165 tty_use_colors (void)
 166 {
 167     return use_colors;
 168 }
 169
 170 /* --------------------------------------------------------------------------------------------- */
 171
 172 int
 173 tty_try_alloc_color_pair2 (const char *fg, const char *bg, gboolean is_temp_color)
 174 {
 175     gchar *color_pair;
 176     tty_color_pair_t *mc_color_pair;
 177     const char *c_fg, *c_bg;
 178
 179     if (fg == NULL)
 180         fg = tty_color_defaults__fg;
 181
 182     if (bg == NULL)
 183     {
 184         bg = tty_color_defaults__bg;
 185     }
 186     c_fg = tty_color_get_valid_name (fg);
 187     c_bg = tty_color_get_valid_name (bg);
 188
 189     color_pair = g_strdup_printf ("%s.%s", c_fg, c_bg);
 190     if (color_pair == NULL)
 191         return 0;
 192
 193     mc_color_pair =
 194         (tty_color_pair_t *) g_hash_table_lookup (mc_tty_color__hashtable, (gpointer) color_pair);
 195
 196     if (mc_color_pair != NULL)
 197     {
 198         g_free (color_pair);
 199         return mc_color_pair->pair_index;
 200     }
 201
 202     mc_color_pair = g_try_new0 (tty_color_pair_t, 1);
 203     if (mc_color_pair == NULL)
 204     {
 205         g_free (color_pair);
 206         return 0;
 207     }
 208
 209     mc_color_pair->is_temp = is_temp_color;
 210     mc_color_pair->cfg = c_fg;
 211     mc_color_pair->cbg = c_bg;
 212     mc_color_pair->ifg = tty_color_get_index_by_name (c_fg);
 213     mc_color_pair->ibg = tty_color_get_index_by_name (c_bg);
 214     mc_color_pair->pair_index = tty_color_get_next__color_pair_number ();
 215
 216     tty_color_try_alloc_pair_lib (mc_color_pair);
 217
 218     g_hash_table_insert (mc_tty_color__hashtable, (gpointer) color_pair, (gpointer) mc_color_pair);
 219
 220     return mc_color_pair->pair_index;
 221 }
 222
 223 /* --------------------------------------------------------------------------------------------- */
 224
 225 int
 226 tty_try_alloc_color_pair (const char *fg, const char *bg)
 227 {
 228     return tty_try_alloc_color_pair2 (fg, bg, TRUE);
 229 }
 230
 231 /* --------------------------------------------------------------------------------------------- */
 232
 233 void
 234 tty_color_free_all_tmp (void)
 235 {
 236     tty_color_free_all (TRUE);
 237 }
 238
 239 /* --------------------------------------------------------------------------------------------- */
 240
 241 void
 242 tty_color_free_all_non_tmp (void)
 243 {
 244     tty_color_free_all (FALSE);
 245 }
 246
 247 /* --------------------------------------------------------------------------------------------- */
 248
 249 void
 250 tty_color_set_defaults (const char *fgcolor, const char *bgcolor)
 251 {
 252     g_free (tty_color_defaults__fg);
 253     g_free (tty_color_defaults__fg);
 254
 255     tty_color_defaults__fg = (fgcolor != NULL) ? g_strdup (fgcolor) : NULL;
 256     tty_color_defaults__bg = (bgcolor != NULL) ? g_strdup (bgcolor) : NULL;
 257 }
 258
 259 /* --------------------------------------------------------------------------------------------- */