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