src/oping.h: Don't use the ncurses internal _nc_unicode_locale() function.

[liboping.git] / src / oping.c

diff --git a/src/oping.c b/src/oping.c
index 068c55d..eb16469 100644 (file)
--- a/src/oping.c
+++ b/src/oping.c
@@ -75,6 +75,7 @@
 #endif
 
 #include <locale.h>
+#include <langinfo.h>
 
 #if USE_NCURSES
 # define NCURSES_OPAQUE 1
@@ -89,15 +90,31 @@
 # define OPING_YELLOW_HIST 5
 # define OPING_RED_HIST 6
 
-static char const * const hist_symbols[] = {
+static char const * const hist_symbols_utf8[] = {
        "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█" };
-static size_t const hist_symbols_num = sizeof (hist_symbols)
-       / sizeof (hist_symbols[0]);
-
-static int const hist_colors[] = {
+static size_t const hist_symbols_utf8_num = sizeof (hist_symbols_utf8)
+       / sizeof (hist_symbols_utf8[0]);
+
+/* scancodes for 6 levels of horizontal bars, ncurses-specific */
+/* those are not the usual constants because those are not constant */
+static int const hist_symbols_acs[] = {
+       115, /* ACS_S9 "⎽" */
+       114, /* ACS_S7 "⎼" */
+       113, /* ACS_S5 "─" */
+       112, /* ACS_S3 "⎻" */
+       111  /* ACS_S1 "⎺" */
+};
+static size_t const hist_symbols_acs_num = sizeof (hist_symbols_acs)
+       / sizeof (hist_symbols_acs[0]);
+
+/* use different colors without a background for scancodes */
+static int const hist_colors_utf8[] = {
        OPING_GREEN_HIST, OPING_YELLOW_HIST, OPING_RED_HIST };
-static size_t const hist_colors_num = sizeof (hist_colors)
-       / sizeof (hist_colors[0]);
+static int const hist_colors_acs[] = {
+       OPING_GREEN, OPING_YELLOW, OPING_RED };
+/* assuming that both arrays are the same size */
+static size_t const hist_colors_num = sizeof (hist_colors_utf8)
+       / sizeof (hist_colors_utf8[0]);
 #endif
 
 #include "oping.h"
@@ -138,6 +155,7 @@ static char   *opt_filename   = NULL;
 static int     opt_count      = -1;
 static int     opt_send_ttl   = 64;
 static uint8_t opt_send_qos   = 0;
+static int     opt_utf8       = 0;
 
 static int host_num = 0;
 
@@ -484,7 +502,7 @@ static int read_options (int argc, char **argv) /* {{{ */
 
        while (1)
        {
-               optchar = getopt (argc, argv, "46c:hi:I:t:Q:f:D:");
+               optchar = getopt (argc, argv, "46uUc:hi:I:t:Q:f:D:");
 
                if (optchar == -1)
                        break;
@@ -496,6 +514,13 @@ static int read_options (int argc, char **argv) /* {{{ */
                                opt_addrfamily = (optchar == '4') ? AF_INET : AF_INET6;
                                break;
 
+                       case 'u':
+                               opt_utf8 = 2;
+                               break;
+                       case 'U':
+                               opt_utf8 = 1;
+                               break;
+
                        case 'c':
                                {
                                        int new_count;
@@ -614,11 +639,27 @@ static void time_calc (struct timespec *ts_dest, /* {{{ */
 } /* }}} void time_calc */
 
 #if USE_NCURSES
+static _Bool has_utf8() /* {{{ */
+{
+       if (!opt_utf8)
+       {
+               /* Automatically determine */
+               if (strcasecmp ("UTF-8", nl_langinfo (CODESET)) == 0)
+                       opt_utf8 = 2;
+               else
+                       opt_utf8 = 1;
+       }
+       return ((_Bool) (opt_utf8 - 1));
+} /* }}} _Bool has_utf8 */
+
 static int update_prettyping_graph (ping_context_t *ctx, /* {{{ */
                double latency, unsigned int sequence)
 {
        int color = OPING_RED;
        char const *symbol = "!";
+       int symbolc = '!';
+       size_t hist_symbols_num;
+       size_t index_symbols;
 
        int x_max;
        int x_pos;
@@ -626,12 +667,19 @@ static int update_prettyping_graph (ping_context_t *ctx, /* {{{ */
        x_max = getmaxx (ctx->window);
        x_pos = ((sequence - 1) % (x_max - 4)) + 2;
 
+       if (has_utf8())
+       {
+               hist_symbols_num = hist_symbols_utf8_num;
+       }
+       else {
+               hist_symbols_num = hist_symbols_acs_num;
+       }
+
        if (latency >= 0.0)
        {
                double ratio;
                size_t intensity;
                size_t index_colors;
-               size_t index_symbols;
 
                ratio = latency / PING_DEF_TTL;
                if (ratio > 1) {
@@ -643,19 +691,36 @@ static int update_prettyping_graph (ping_context_t *ctx, /* {{{ */
 
                index_colors = intensity / hist_symbols_num;
                assert (index_colors < hist_colors_num);
-               color = hist_colors[index_colors];
 
                index_symbols = intensity % hist_symbols_num;
-               symbol = hist_symbols[index_symbols];
-       }
+               if (has_utf8())
+               {
+                       color = hist_colors_utf8[index_colors];
+                       symbol = hist_symbols_utf8[index_symbols];
+               }
+               else
+               {
+                       color = hist_colors_acs[index_colors];
+                       symbolc = hist_symbols_acs[index_symbols] | A_ALTCHARSET;
+               }
+        }
        else /* if (!(latency >= 0.0)) */
                wattron (ctx->window, A_BOLD);
 
        wattron (ctx->window, COLOR_PAIR(color));
-       mvwprintw (ctx->window,
-                       /* y = */ 3,
-                       /* x = */ x_pos,
-                       symbol);
+       if (has_utf8())
+       {
+               mvwprintw (ctx->window,
+                          /* y = */ 3,
+                          /* x = */ x_pos,
+                          symbol);
+       }
+       else {
+               mvwaddch (ctx->window,
+                         /* y = */ 3,
+                         /* x = */ x_pos,
+                         symbolc);
+       }
        wattroff (ctx->window, COLOR_PAIR(color));
 
        /* Use negation here to handle NaN correctly. */