test/test_add_wchstr.c

   1 /****************************************************************************
   2  * Copyright (c) 2010-2011,2012 Free Software Foundation, Inc.                   *
   3  *                                                                          *
   4  * Permission is hereby granted, free of charge, to any person obtaining a  *
   5  * copy of this software and associated documentation files (the            *
   6  * "Software"), to deal in the Software without restriction, including      *
   7  * without limitation the rights to use, copy, modify, merge, publish,      *
   8  * distribute, distribute with modifications, sublicense, and/or sell       *
   9  * copies of the Software, and to permit persons to whom the Software is    *
  10  * furnished to do so, subject to the following conditions:                 *
  11  *                                                                          *
  12  * The above copyright notice and this permission notice shall be included  *
  13  * in all copies or substantial portions of the Software.                   *
  14  *                                                                          *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS  *
  16  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF               *
  17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.   *
  18  * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,   *
  19  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR    *
  20  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR    *
  21  * THE USE OR OTHER DEALINGS IN THE SOFTWARE.                               *
  22  *                                                                          *
  23  * Except as contained in this notice, the name(s) of the above copyright   *
  24  * holders shall not be used in advertising or otherwise to promote the     *
  25  * sale, use or other dealings in this Software without prior written       *
  26  * authorization.                                                           *
  27  ****************************************************************************/
  28 /*
  29  * $Id: test_add_wchstr.c,v 1.20 2012/12/16 00:12:04 tom Exp $
  30  *
  31  * Demonstrate the waddwchstr() and wadd_wch functions.
  32  * Thomas Dickey - 2009/9/12
  33  *
  34  * Note: to provide inputs for *add_wch(), we use setcchar().  A quirk of the
  35  * X/Open definition for that function is that the string contains no
  36  * characters with negative width.  Any control character (such as tab) falls
  37  * into that category.  So it follows that *add_wch() cannot render a tab
  38  * character because there is no legal way to construct a cchar_t containing
  39  * one.  X/Open does not document this, and it would be logical to assume that
  40  * *addwchstr() has the same limitation, but it uses a wchar_t string directly,
  41  * and does not document how tabs are handled.
  42  */
  43
  44 #include <test.priv.h>
  45
  46 #if USE_WIDEC_SUPPORT
  47
  48 #define WIDE_LINEDATA
  49 #include <linedata.h>
  50
  51 #undef MvAddCh
  52 #undef MvAddStr
  53 #undef MvWAddCh
  54 #undef MvWAddStr
  55
  56 /* definitions to make it simpler to compare with test_addstr.c */
  57 #define AddNStr    add_wchnstr
  58 #define AddStr     add_wchstr
  59 #define MvAddNStr  (void) mvadd_wchnstr
  60 #define MvAddStr   (void) mvadd_wchstr
  61 #define MvWAddNStr (void) mvwadd_wchnstr
  62 #define MvWAddStr  (void) mvwadd_wchstr
  63 #define WAddNStr   wadd_wchnstr
  64 #define WAddStr    wadd_wchstr
  65
  66 #define MY_TABSIZE 8
  67
  68 typedef enum {
  69     oDefault = 0,
  70     oMove = 1,
  71     oWindow = 2,
  72     oMoveWindow = 3
  73 } Options;
  74
  75 static bool m_opt = FALSE;
  76 static bool pass_ctls = FALSE;
  77 static bool w_opt = FALSE;
  78 static int n_opt = -1;
  79
  80 static cchar_t *temp_buffer;
  81 static size_t temp_length;
  82
  83 #define TempBuffer(source_len, source_cast) \
  84     if (source != 0) { \
  85         const char *temp; \
  86         size_t need = source_len + 1; \
  87         wchar_t have[2]; \
  88         int n = 0; \
  89  \
  90         if (need > temp_length) { \
  91             temp_length = need * 2; \
  92             temp_buffer = typeRealloc(cchar_t, temp_length, temp_buffer); \
  93             if (!temp_buffer) \
  94                 failed("TempBuffer"); \
  95         } \
  96         have[0] = 0; \
  97         have[1] = 0; \
  98         do { \
  99             have[0] = source_cast; \
 100             if (!pass_ctls \
 101              && have[0] != 0 \
 102              && have[0] < 256 \
 103              && (temp = unctrl((chtype) have[0])) != 0 \
 104              && strlen(temp) > 1) { \
 105                 while (*temp != '\0') { \
 106                     have[0] = *temp++; \
 107                     setcchar(&temp_buffer[n++], have, A_NORMAL, 0, NULL); \
 108                 } \
 109             } else { \
 110                 setcchar(&temp_buffer[n++], have, A_NORMAL, 0, NULL); \
 111             } \
 112         } while (have[0] != 0); \
 113     } else if (temp_buffer != 0) { \
 114         free(temp_buffer); \
 115         temp_buffer = 0; \
 116         temp_length = 0; \
 117     } \
 118     return temp_buffer;
 119
 120 static size_t
 121 ChWLen(const wchar_t *source)
 122 {
 123     size_t result = wcslen(source);
 124
 125     if (!pass_ctls) {
 126         size_t adjust = 0;
 127         size_t n;
 128         const char *s;
 129
 130         for (n = 0; n < result; ++n) {
 131             if (source[n] < 256 && (s = unctrl((chtype) source[n])) != 0) {
 132                 adjust += (strlen(s) - 1);
 133             }
 134         }
 135         result += adjust;
 136     }
 137     return result;
 138 }
 139
 140 static cchar_t *
 141 ChStr(const char *source)
 142 {
 143     TempBuffer(strlen(source), UChar(*source++));
 144 }
 145
 146 static cchar_t *
 147 ChWStr(const wchar_t *source)
 148 {
 149     TempBuffer(ChWLen(source), *source++);
 150 }
 151
 152 static void
 153 legend(WINDOW *win, int level, Options state, wchar_t *buffer, int length)
 154 {
 155     const char *showstate;
 156
 157     switch (state) {
 158     default:
 159     case oDefault:
 160         showstate = "";
 161         break;
 162     case oMove:
 163         showstate = " (mvXXX)";
 164         break;
 165     case oWindow:
 166         showstate = " (winXXX)";
 167         break;
 168     case oMoveWindow:
 169         showstate = " (mvwinXXX)";
 170         break;
 171     }
 172
 173     wmove(win, 0, 0);
 174     wprintw(win,
 175             "The Strings/Chars displays should match.  Enter any characters, except:\n");
 176     wprintw(win,
 177             "down-arrow or ^N to repeat on next line, ^W for inner window, ESC to exit.\n");
 178     wclrtoeol(win);
 179     wprintw(win, "Level %d,%s added %d characters <", level,
 180             showstate, length);
 181     waddwstr(win, buffer);
 182     waddstr(win, ">");
 183 }
 184
 185 static int
 186 ColOf(wchar_t *buffer, int length, int margin)
 187 {
 188     int n;
 189     int result;
 190
 191     for (n = 0, result = margin + 1; n < length; ++n) {
 192         int ch = buffer[n];
 193         switch (ch) {
 194         case '\n':
 195             /* actually newline should clear the remainder of the line
 196              * and move to the next line - but that seems a little awkward
 197              * in this example.
 198              */
 199         case '\r':
 200             result = 0;
 201             break;
 202         case '\b':
 203             if (result > 0)
 204                 --result;
 205             break;
 206         case '\t':
 207             result += (MY_TABSIZE - (result % MY_TABSIZE));
 208             break;
 209         case '\177':
 210             result += 2;
 211             break;
 212         default:
 213             result += wcwidth(ch);
 214             if (ch < 32)
 215                 ++result;
 216             break;
 217         }
 218     }
 219     return result;
 220 }
 221
 222 static int
 223 ConvertCh(chtype source, cchar_t *target)
 224 {
 225     wchar_t tmp_wchar[2];
 226
 227     tmp_wchar[0] = (wchar_t) source;
 228     tmp_wchar[1] = 0;
 229     if (setcchar(target, tmp_wchar, A_NORMAL, 0, (void *) 0) == ERR) {
 230         beep();
 231         return FALSE;
 232     }
 233     return TRUE;
 234 }
 235
 236 static int
 237 MvWAddCh(WINDOW *win, int y, int x, chtype ch)
 238 {
 239     int code;
 240     cchar_t tmp_cchar;
 241
 242     if (ConvertCh(ch, &tmp_cchar)) {
 243         code = mvwadd_wch(win, y, x, &tmp_cchar);
 244     } else {
 245         code = mvwaddch(win, y, x, ch);
 246     }
 247     return code;
 248 }
 249
 250 static int
 251 MvAddCh(int y, int x, chtype ch)
 252 {
 253     int code;
 254     cchar_t tmp_cchar;
 255
 256     if (ConvertCh(ch, &tmp_cchar)) {
 257         code = mvadd_wch(y, x, &tmp_cchar);
 258     } else {
 259         code = mvaddch(y, x, ch);
 260     }
 261     return code;
 262 }
 263
 264 static int
 265 WAddCh(WINDOW *win, chtype ch)
 266 {
 267     int code;
 268     cchar_t tmp_cchar;
 269
 270     if (ConvertCh(ch, &tmp_cchar)) {
 271         code = wadd_wch(win, &tmp_cchar);
 272     } else {
 273         code = waddch(win, ch);
 274     }
 275     return code;
 276 }
 277
 278 static int
 279 AddCh(chtype ch)
 280 {
 281     int code;
 282     cchar_t tmp_cchar;
 283
 284     if (ConvertCh(ch, &tmp_cchar)) {
 285         code = add_wch(&tmp_cchar);
 286     } else {
 287         code = addch(ch);
 288     }
 289     return code;
 290 }
 291
 292 #define LEN(n) ((length - (n) > n_opt) ? n_opt : (length - (n)))
 293 static void
 294 test_add_wchstr(int level)
 295 {
 296     static bool first = TRUE;
 297
 298     int ch;
 299     int limit;
 300     int row = 1;
 301     int col;
 302     int row2, col2;
 303     int length;
 304     wchar_t buffer[BUFSIZ];
 305     WINDOW *look = 0;
 306     WINDOW *work = 0;
 307     WINDOW *show = 0;
 308     int margin = (2 * MY_TABSIZE) - 1;
 309     Options option = ((m_opt ? oMove : oDefault)
 310                       | ((w_opt || (level > 0)) ? oWindow : oDefault));
 311
 312     if (first) {
 313         static char cmd[80];
 314         setlocale(LC_ALL, "");
 315
 316         putenv(strcpy(cmd, "TABSIZE=8"));
 317
 318         initscr();
 319         (void) cbreak();        /* take input chars one at a time, no wait for \n */
 320         (void) noecho();        /* don't echo input */
 321         keypad(stdscr, TRUE);
 322
 323         /*
 324          * Show the characters added in color, to distinguish from those that
 325          * are shifted.
 326          */
 327         if (has_colors()) {
 328             start_color();
 329             init_pair(1, COLOR_WHITE, COLOR_BLUE);
 330         }
 331     }
 332
 333     limit = LINES - 5;
 334     if (level > 0) {
 335         look = newwin(limit, COLS - (2 * (level - 1)), 0, level - 1);
 336         work = newwin(limit - 2, COLS - (2 * level), 1, level);
 337         show = newwin(4, COLS, limit + 1, 0);
 338         box(look, 0, 0);
 339         wnoutrefresh(look);
 340         limit -= 2;
 341     } else {
 342         work = stdscr;
 343         show = derwin(stdscr, 4, COLS, limit + 1, 0);
 344     }
 345     keypad(work, TRUE);
 346
 347     for (col = margin + 1; col < COLS; col += MY_TABSIZE)
 348         MvWVLine(work, row, col, '.', limit - 2);
 349
 350     MvWVLine(work, row, margin, ACS_VLINE, limit - 2);
 351     MvWVLine(work, row, margin + 1, ACS_VLINE, limit - 2);
 352     limit /= 2;
 353
 354     (void) mvwadd_wchstr(work, 1, 2, ChStr("String"));
 355     (void) mvwadd_wchstr(work, limit + 1, 2, ChStr("Chars"));
 356     wnoutrefresh(work);
 357
 358     buffer[length = 0] = '\0';
 359     legend(show, level, option, buffer, length);
 360     wnoutrefresh(show);
 361
 362     doupdate();
 363
 364     if (has_colors()) {
 365         wbkgdset(work, (chtype) (COLOR_PAIR(1) | ' '));
 366     }
 367
 368     while ((ch = read_linedata(work)) != ERR && !isQUIT(ch)) {
 369         wmove(work, row, margin + 1);
 370         switch (ch) {
 371         case key_RECUR:
 372             test_add_wchstr(level + 1);
 373
 374             if (look)
 375                 touchwin(look);
 376             touchwin(work);
 377             touchwin(show);
 378
 379             if (look)
 380                 wnoutrefresh(look);
 381             wnoutrefresh(work);
 382             wnoutrefresh(show);
 383
 384             doupdate();
 385             break;
 386         case key_NEWLINE:
 387             if (row < limit) {
 388                 ++row;
 389                 /* put the whole string in, all at once */
 390                 col2 = margin + 1;
 391                 switch (option) {
 392                 case oDefault:
 393                     if (n_opt > 1) {
 394                         for (col = 0; col < length; col += n_opt) {
 395                             col2 = ColOf(buffer, col, margin);
 396                             if (move(row, col2) != ERR) {
 397                                 AddNStr(ChWStr(buffer + col), LEN(col));
 398                             }
 399                         }
 400                     } else {
 401                         if (move(row, col2) != ERR) {
 402                             AddStr(ChWStr(buffer));
 403                         }
 404                     }
 405                     break;
 406                 case oMove:
 407                     if (n_opt > 1) {
 408                         for (col = 0; col < length; col += n_opt) {
 409                             col2 = ColOf(buffer, col, margin);
 410                             MvAddNStr(row, col2, ChWStr(buffer + col), LEN(col));
 411                         }
 412                     } else {
 413                         MvAddStr(row, col2, ChWStr(buffer));
 414                     }
 415                     break;
 416                 case oWindow:
 417                     if (n_opt > 1) {
 418                         for (col = 0; col < length; col += n_opt) {
 419                             col2 = ColOf(buffer, col, margin);
 420                             if (wmove(work, row, col2) != ERR) {
 421                                 WAddNStr(work, ChWStr(buffer + col), LEN(col));
 422                             }
 423                         }
 424                     } else {
 425                         if (wmove(work, row, col2) != ERR) {
 426                             WAddStr(work, ChWStr(buffer));
 427                         }
 428                     }
 429                     break;
 430                 case oMoveWindow:
 431                     if (n_opt > 1) {
 432                         for (col = 0; col < length; col += n_opt) {
 433                             col2 = ColOf(buffer, col, margin);
 434                             MvWAddNStr(work, row, col2, ChWStr(buffer +
 435                                                                col), LEN(col));
 436                         }
 437                     } else {
 438                         MvWAddStr(work, row, col2, ChWStr(buffer));
 439                     }
 440                     break;
 441                 }
 442
 443                 /* do the corresponding single-character add */
 444                 row2 = limit + row;
 445                 for (col = 0; col < length; ++col) {
 446                     col2 = ColOf(buffer, col, margin);
 447                     switch (option) {
 448                     case oDefault:
 449                         if (move(row2, col2) != ERR) {
 450                             AddCh((chtype) buffer[col]);
 451                         }
 452                         break;
 453                     case oMove:
 454                         MvAddCh(row2, col2, (chtype) buffer[col]);
 455                         break;
 456                     case oWindow:
 457                         if (wmove(work, row2, col2) != ERR) {
 458                             WAddCh(work, (chtype) buffer[col]);
 459                         }
 460                         break;
 461                     case oMoveWindow:
 462                         MvWAddCh(work, row2, col2, (chtype) buffer[col]);
 463                         break;
 464                     }
 465                 }
 466             } else {
 467                 beep();
 468             }
 469             break;
 470         default:
 471             buffer[length++] = ch;
 472             buffer[length] = '\0';
 473
 474             /* put the string in, one character at a time */
 475             col = ColOf(buffer, length - 1, margin);
 476             switch (option) {
 477             case oDefault:
 478                 if (move(row, col) != ERR) {
 479                     AddStr(ChWStr(buffer + length - 1));
 480                 }
 481                 break;
 482             case oMove:
 483                 MvAddStr(row, col, ChWStr(buffer + length - 1));
 484                 break;
 485             case oWindow:
 486                 if (wmove(work, row, col) != ERR) {
 487                     WAddStr(work, ChWStr(buffer + length - 1));
 488                 }
 489                 break;
 490             case oMoveWindow:
 491                 MvWAddStr(work, row, col, ChWStr(buffer + length - 1));
 492                 break;
 493             }
 494
 495             /* do the corresponding single-character add */
 496             switch (option) {
 497             case oDefault:
 498                 if (move(limit + row, col) != ERR) {
 499                     AddCh((chtype) ch);
 500                 }
 501                 break;
 502             case oMove:
 503                 MvAddCh(limit + row, col, (chtype) ch);
 504                 break;
 505             case oWindow:
 506                 if (wmove(work, limit + row, col) != ERR) {
 507                     WAddCh(work, (chtype) ch);
 508                 }
 509                 break;
 510             case oMoveWindow:
 511                 MvWAddCh(work, limit + row, col, (chtype) ch);
 512                 break;
 513             }
 514
 515             wnoutrefresh(work);
 516
 517             legend(show, level, option, buffer, length);
 518             wnoutrefresh(show);
 519
 520             doupdate();
 521             break;
 522         }
 523     }
 524     delwin(show);
 525     if (level > 0) {
 526         delwin(work);
 527         delwin(look);
 528     }
 529 }
 530
 531 static void
 532 usage(void)
 533 {
 534     static const char *tbl[] =
 535     {
 536         "Usage: test_add_wchstr [options]"
 537         ,""
 538         ,"Options:"
 539         ,"  -f FILE read data from given file"
 540         ,"  -n NUM  limit string-adds to NUM bytes on ^N replay"
 541         ,"  -m      perform wmove/move separately from add-functions"
 542         ,"  -p      pass-thru control characters without using unctrl()"
 543         ,"  -w      use window-parameter even when stdscr would be implied"
 544     };
 545     unsigned n;
 546     for (n = 0; n < SIZEOF(tbl); ++n)
 547         fprintf(stderr, "%s\n", tbl[n]);
 548     ExitProgram(EXIT_FAILURE);
 549 }
 550
 551 int
 552 main(int argc GCC_UNUSED, char *argv[]GCC_UNUSED)
 553 {
 554     int ch;
 555
 556     setlocale(LC_ALL, "");
 557
 558     while ((ch = getopt(argc, argv, "f:mn:pw")) != -1) {
 559         switch (ch) {
 560         case 'f':
 561             init_linedata(optarg);
 562             break;
 563         case 'm':
 564             m_opt = TRUE;
 565             break;
 566         case 'n':
 567             n_opt = atoi(optarg);
 568             if (n_opt == 0)
 569                 n_opt = -1;
 570             break;
 571         case 'p':
 572             pass_ctls = TRUE;
 573             break;
 574         case 'w':
 575             w_opt = TRUE;
 576             break;
 577         default:
 578             usage();
 579             break;
 580         }
 581     }
 582     if (optind < argc)
 583         usage();
 584
 585     test_add_wchstr(0);
 586     endwin();
 587     ExitProgram(EXIT_SUCCESS);
 588 }
 589 #else
 590 int
 591 main(void)
 592 {
 593     printf("This program requires the wide-ncurses library\n");
 594     ExitProgram(EXIT_FAILURE);
 595 }
 596 #endif