test/test_addchstr.c

   1 /****************************************************************************
   2  * Copyright (c) 2009-2010,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_addchstr.c,v 1.18 2012/12/16 00:36:27 tom Exp $
  30  *
  31  * Demonstrate the waddchstr() and waddch functions.
  32  * Thomas Dickey - 2009/9/12
  33  */
  34
  35 #include <test.priv.h>
  36
  37 #include <linedata.h>
  38
  39 #undef MvAddStr
  40 #undef MvWAddStr
  41
  42 #define AddNStr    addchnstr
  43 #define AddStr     addchstr
  44 #define MvAddNStr  (void) mvaddchnstr
  45 #define MvAddStr   (void) mvaddchstr
  46 #define MvWAddNStr (void) mvwaddchnstr
  47 #define MvWAddStr  (void) mvwaddchstr
  48 #define WAddNStr   waddchnstr
  49 #define WAddStr    waddchstr
  50
  51 #define AddCh      addch
  52 #define WAddCh     waddch
  53
  54 #define MY_TABSIZE 8
  55
  56 typedef enum {
  57     oDefault = 0,
  58     oMove = 1,
  59     oWindow = 2,
  60     oMoveWindow = 3
  61 } Options;
  62
  63 static bool m_opt = FALSE;
  64 static bool pass_ctls = FALSE;
  65 static bool w_opt = FALSE;
  66 static int n_opt = -1;
  67
  68 static attr_t show_attr;
  69 static chtype *temp_buffer;
  70 static size_t temp_length;
  71
  72 #define TempBuffer(source_cast)
  73
  74 static size_t
  75 ChLen(const char *source)
  76 {
  77     size_t result = strlen(source);
  78
  79     if (!pass_ctls) {
  80         size_t adjust = 0;
  81         size_t n;
  82
  83         for (n = 0; n < result; ++n) {
  84             const char *s = unctrl(UChar(source[n]));
  85             if (s != 0) {
  86                 adjust += (strlen(s) - 1);
  87             }
  88         }
  89         result += adjust;
  90     }
  91     return result;
  92 }
  93
  94 static chtype *
  95 ChStr(const char *source)
  96 {
  97     if (source != 0) {
  98         size_t need = ChLen(source) + 1;
  99         int n = 0;
 100
 101         if (need > temp_length) {
 102             temp_length = need * 2;
 103             temp_buffer = typeRealloc(chtype, temp_length, temp_buffer);
 104             if (!temp_buffer)
 105                 failed("TempBuffer");
 106         }
 107         do {
 108             const char *s;
 109             chtype ch = UChar(*source++);
 110             if (!pass_ctls && (s = unctrl(ch)) != 0) {
 111                 while (*s != '\0') {
 112                     temp_buffer[n++] = UChar(*s++);
 113                 }
 114             } else {
 115                 temp_buffer[n++] = ch;
 116             }
 117         } while (source[0] != 0);
 118         temp_buffer[n] = 0;
 119     } else if (temp_buffer != 0) {
 120         free(temp_buffer);
 121         temp_buffer = 0;
 122         temp_length = 0;
 123     }
 124     return temp_buffer;
 125 }
 126
 127 /* color the strings drawn in the workspace */
 128 static chtype *
 129 ChStr2(const char *source)
 130 {
 131     size_t len = ChLen(source);
 132     size_t n;
 133     chtype *result = ChStr(source);
 134     for (n = 0; n < len; ++n) {
 135         result[n] |= show_attr;
 136     }
 137     return result;
 138 }
 139
 140 static void
 141 legend(WINDOW *win, int level, Options state, char *buffer, int length)
 142 {
 143     const char *showstate;
 144
 145     switch (state) {
 146     default:
 147     case oDefault:
 148         showstate = "";
 149         break;
 150     case oMove:
 151         showstate = " (mvXXX)";
 152         break;
 153     case oWindow:
 154         showstate = " (winXXX)";
 155         break;
 156     case oMoveWindow:
 157         showstate = " (mvwinXXX)";
 158         break;
 159     }
 160
 161     wmove(win, 0, 0);
 162     wprintw(win,
 163             "The Strings/Chars displays should match.  Enter any characters, except:\n");
 164     wprintw(win,
 165             "down-arrow or ^N to repeat on next line, ^W for inner window, ESC to exit.\n");
 166     wclrtoeol(win);
 167     wprintw(win, "Level %d,%s added %d characters <%s>", level,
 168             showstate, length, buffer);
 169 }
 170
 171 static int
 172 ColOf(char *buffer, int length, int margin)
 173 {
 174     int n;
 175     int result;
 176
 177     for (n = 0, result = margin + 1; n < length; ++n) {
 178         int ch = UChar(buffer[n]);
 179         switch (ch) {
 180         case '\n':
 181             /* actually newline should clear the remainder of the line
 182              * and move to the next line - but that seems a little awkward
 183              * in this example.
 184              */
 185         case '\r':
 186             result = 0;
 187             break;
 188         case '\b':
 189             if (result > 0)
 190                 --result;
 191             break;
 192         case '\t':
 193             result += (MY_TABSIZE - (result % MY_TABSIZE));
 194             break;
 195         case '\177':
 196             result += 2;
 197             break;
 198         default:
 199             ++result;
 200             if (ch < 32)
 201                 ++result;
 202             break;
 203         }
 204     }
 205     return result;
 206 }
 207
 208 #define LEN(n) ((length - (n) > n_opt) ? n_opt : (length - (n)))
 209 static void
 210 test_adds(int level)
 211 {
 212     static bool first = TRUE;
 213
 214     int ch;
 215     int limit;
 216     int row = 1;
 217     int col;
 218     int row2, col2;
 219     int length;
 220     char buffer[BUFSIZ];
 221     WINDOW *look = 0;
 222     WINDOW *work = 0;
 223     WINDOW *show = 0;
 224     int margin = (2 * MY_TABSIZE) - 1;
 225     Options option = (Options) ((unsigned) (m_opt
 226                                             ? oMove
 227                                             : oDefault)
 228                                 | (unsigned) ((w_opt || (level > 0))
 229                                               ? oWindow
 230                                               : oDefault));
 231
 232     if (first) {
 233         static char cmd[80];
 234         setlocale(LC_ALL, "");
 235
 236         putenv(strcpy(cmd, "TABSIZE=8"));
 237
 238         initscr();
 239         (void) cbreak();        /* take input chars one at a time, no wait for \n */
 240         (void) noecho();        /* don't echo input */
 241         keypad(stdscr, TRUE);
 242
 243         /*
 244          * Show the characters added in color, to distinguish from those that
 245          * are shifted.
 246          */
 247         if (has_colors()) {
 248             start_color();
 249             init_pair(1, COLOR_WHITE, COLOR_BLUE);
 250         }
 251     }
 252
 253     limit = LINES - 5;
 254     if (level > 0) {
 255         look = newwin(limit, COLS - (2 * (level - 1)), 0, level - 1);
 256         work = newwin(limit - 2, COLS - (2 * level), 1, level);
 257         show = newwin(4, COLS, limit + 1, 0);
 258         box(look, 0, 0);
 259         wnoutrefresh(look);
 260         limit -= 2;
 261     } else {
 262         work = stdscr;
 263         show = derwin(stdscr, 4, COLS, limit + 1, 0);
 264     }
 265     keypad(work, TRUE);
 266
 267     for (col = margin + 1; col < COLS; col += MY_TABSIZE)
 268         MvWVLine(work, row, col, '.', limit - 2);
 269
 270     MvWVLine(work, row, margin, ACS_VLINE, limit - 2);
 271     MvWVLine(work, row, margin + 1, ACS_VLINE, limit - 2);
 272     limit /= 2;
 273
 274     MvWAddChStr(work, 1, 2, ChStr("String"));
 275     MvWAddChStr(work, limit + 1, 2, ChStr("Chars"));
 276     wnoutrefresh(work);
 277
 278     buffer[length = 0] = '\0';
 279     legend(show, level, option, buffer, length);
 280     wnoutrefresh(show);
 281
 282     doupdate();
 283
 284     if (has_colors()) {
 285         show_attr = (attr_t) COLOR_PAIR(1);
 286         wbkgdset(work, show_attr | ' ');
 287     } else {
 288         show_attr = A_STANDOUT;
 289     }
 290
 291     while ((ch = read_linedata(work)) != ERR && !isQUIT(ch)) {
 292         wmove(work, row, margin + 1);
 293         switch (ch) {
 294         case key_RECUR:
 295             test_adds(level + 1);
 296
 297             if (look)
 298                 touchwin(look);
 299             touchwin(work);
 300             touchwin(show);
 301
 302             if (look)
 303                 wnoutrefresh(look);
 304             wnoutrefresh(work);
 305             wnoutrefresh(show);
 306
 307             doupdate();
 308             break;
 309         case key_NEWLINE:
 310             if (row < limit) {
 311                 ++row;
 312                 /* put the whole string in, all at once */
 313                 col2 = margin + 1;
 314                 switch (option) {
 315                 case oDefault:
 316                     if (n_opt > 1) {
 317                         for (col = 0; col < length; col += n_opt) {
 318                             col2 = ColOf(buffer, col, margin);
 319                             if (move(row, col2) != ERR) {
 320                                 AddNStr(ChStr2(buffer + col), LEN(col));
 321                             }
 322                         }
 323                     } else {
 324                         if (move(row, col2) != ERR) {
 325                             AddStr(ChStr2(buffer));
 326                         }
 327                     }
 328                     break;
 329                 case oMove:
 330                     if (n_opt > 1) {
 331                         for (col = 0; col < length; col += n_opt) {
 332                             col2 = ColOf(buffer, col, margin);
 333                             MvAddNStr(row, col2, ChStr2(buffer + col), LEN(col));
 334                         }
 335                     } else {
 336                         MvAddStr(row, col2, ChStr2(buffer));
 337                     }
 338                     break;
 339                 case oWindow:
 340                     if (n_opt > 1) {
 341                         for (col = 0; col < length; col += n_opt) {
 342                             col2 = ColOf(buffer, col, margin);
 343                             if (wmove(work, row, col2) != ERR) {
 344                                 WAddNStr(work, ChStr2(buffer + col), LEN(col));
 345                             }
 346                         }
 347                     } else {
 348                         if (wmove(work, row, col2) != ERR) {
 349                             WAddStr(work, ChStr2(buffer));
 350                         }
 351                     }
 352                     break;
 353                 case oMoveWindow:
 354                     if (n_opt > 1) {
 355                         for (col = 0; col < length; col += n_opt) {
 356                             col2 = ColOf(buffer, col, margin);
 357                             MvWAddNStr(work, row, col2, ChStr2(buffer + col),
 358                                        LEN(col));
 359                         }
 360                     } else {
 361                         MvWAddStr(work, row, col2, ChStr2(buffer));
 362                     }
 363                     break;
 364                 }
 365
 366                 /* do the corresponding single-character add */
 367                 row2 = limit + row;
 368                 for (col = 0; col < length; ++col) {
 369                     col2 = ColOf(buffer, col, margin);
 370                     switch (option) {
 371                     case oDefault:
 372                         if (move(row2, col2) != ERR) {
 373                             AddCh(UChar(buffer[col]));
 374                         }
 375                         break;
 376                     case oMove:
 377                         MvAddCh(row2, col2, UChar(buffer[col]));
 378                         break;
 379                     case oWindow:
 380                         if (wmove(work, row2, col2) != ERR) {
 381                             WAddCh(work, UChar(buffer[col]));
 382                         }
 383                         break;
 384                     case oMoveWindow:
 385                         MvWAddCh(work, row2, col2, UChar(buffer[col]));
 386                         break;
 387                     }
 388                 }
 389             } else {
 390                 beep();
 391             }
 392             break;
 393         case KEY_BACKSPACE:
 394             ch = '\b';
 395             /* FALLTHRU */
 396         default:
 397             if (ch <= 0 || ch > 255) {
 398                 beep();
 399                 break;
 400             }
 401             buffer[length++] = (char) ch;
 402             buffer[length] = '\0';
 403
 404             /* put the string in, one character at a time */
 405             col = ColOf(buffer, length - 1, margin);
 406             switch (option) {
 407             case oDefault:
 408                 if (move(row, col) != ERR) {
 409                     AddStr(ChStr2(buffer + length - 1));
 410                 }
 411                 break;
 412             case oMove:
 413                 MvAddStr(row, col, ChStr2(buffer + length - 1));
 414                 break;
 415             case oWindow:
 416                 if (wmove(work, row, col) != ERR) {
 417                     WAddStr(work, ChStr2(buffer + length - 1));
 418                 }
 419                 break;
 420             case oMoveWindow:
 421                 MvWAddStr(work, row, col, ChStr2(buffer + length - 1));
 422                 break;
 423             }
 424
 425             /* do the corresponding single-character add */
 426             switch (option) {
 427             case oDefault:
 428                 if (move(limit + row, col) != ERR) {
 429                     AddCh(UChar(ch));
 430                 }
 431                 break;
 432             case oMove:
 433                 MvAddCh(limit + row, col, UChar(ch));
 434                 break;
 435             case oWindow:
 436                 if (wmove(work, limit + row, col) != ERR) {
 437                     WAddCh(work, UChar(ch));
 438                 }
 439                 break;
 440             case oMoveWindow:
 441                 MvWAddCh(work, limit + row, col, UChar(ch));
 442                 break;
 443             }
 444
 445             wnoutrefresh(work);
 446
 447             legend(show, level, option, buffer, length);
 448             wnoutrefresh(show);
 449
 450             doupdate();
 451             break;
 452         }
 453     }
 454     if (level > 0) {
 455         delwin(work);
 456         delwin(look);
 457     }
 458     delwin(show);
 459 }
 460
 461 static void
 462 usage(void)
 463 {
 464     static const char *tbl[] =
 465     {
 466         "Usage: test_addchstr [options]"
 467         ,""
 468         ,"Options:"
 469         ,"  -f FILE read data from given file"
 470         ,"  -n NUM  limit string-adds to NUM bytes on ^N replay"
 471         ,"  -m      perform wmove/move separately from add-functions"
 472         ,"  -p      pass-thru control characters without using unctrl()"
 473         ,"  -w      use window-parameter even when stdscr would be implied"
 474     };
 475     unsigned n;
 476     for (n = 0; n < SIZEOF(tbl); ++n)
 477         fprintf(stderr, "%s\n", tbl[n]);
 478     ExitProgram(EXIT_FAILURE);
 479 }
 480
 481 int
 482 main(int argc GCC_UNUSED, char *argv[]GCC_UNUSED)
 483 {
 484     int ch;
 485
 486     setlocale(LC_ALL, "");
 487
 488     while ((ch = getopt(argc, argv, "f:mn:pw")) != -1) {
 489         switch (ch) {
 490         case 'f':
 491             init_linedata(optarg);
 492             break;
 493         case 'm':
 494             m_opt = TRUE;
 495             break;
 496         case 'n':
 497             n_opt = atoi(optarg);
 498             if (n_opt == 0)
 499                 n_opt = -1;
 500             break;
 501         case 'p':
 502             pass_ctls = TRUE;
 503             break;
 504         case 'w':
 505             w_opt = TRUE;
 506             break;
 507         default:
 508             usage();
 509             break;
 510         }
 511     }
 512     if (optind < argc)
 513         usage();
 514
 515     test_adds(0);
 516     endwin();
 517     ExitProgram(EXIT_SUCCESS);
 518 }