6b4093e745b7369bd2d0bdbc151dc70e5d35e3ac
[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.15 2012/10/27 19:31:42 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     limit = LINES - 5;
245     if (level > 0) {
246         look = newwin(limit, COLS - (2 * (level - 1)), 0, level - 1);
247         work = newwin(limit - 2, COLS - (2 * level), 1, level);
248         show = newwin(4, COLS, limit + 1, 0);
249         box(look, 0, 0);
250         wnoutrefresh(look);
251         limit -= 2;
252     } else {
253         work = stdscr;
254         show = derwin(stdscr, 4, COLS, limit + 1, 0);
255     }
256     keypad(work, TRUE);
257
258     for (col = margin + 1; col < COLS; col += MY_TABSIZE)
259         MvWVLine(work, row, col, '.', limit - 2);
260
261     MvWVLine(work, row, margin, ACS_VLINE, limit - 2);
262     MvWVLine(work, row, margin + 1, ACS_VLINE, limit - 2);
263     limit /= 2;
264
265     MvWAddChStr(work, 1, 2, ChStr("String"));
266     MvWAddChStr(work, limit + 1, 2, ChStr("Chars"));
267     wnoutrefresh(work);
268
269     buffer[length = 0] = '\0';
270     legend(show, level, option, buffer, length);
271     wnoutrefresh(show);
272
273     doupdate();
274
275     /*
276      * Show the characters added in color, to distinguish from those that
277      * are shifted.
278      */
279     if (has_colors()) {
280         start_color();
281         init_pair(1, COLOR_WHITE, COLOR_BLUE);
282         show_attr = (attr_t) COLOR_PAIR(1);
283         wbkgdset(work, show_attr | ' ');
284     } else {
285         show_attr = A_STANDOUT;
286     }
287
288     while ((ch = read_linedata(work)) != ERR && !isQUIT(ch)) {
289         wmove(work, row, margin + 1);
290         switch (ch) {
291         case key_RECUR:
292             test_adds(level + 1);
293
294             touchwin(look);
295             touchwin(work);
296             touchwin(show);
297
298             wnoutrefresh(look);
299             wnoutrefresh(work);
300             wnoutrefresh(show);
301
302             doupdate();
303             break;
304         case key_NEWLINE:
305             if (row < limit) {
306                 ++row;
307                 /* put the whole string in, all at once */
308                 col2 = margin + 1;
309                 switch (option) {
310                 case oDefault:
311                     if (n_opt > 1) {
312                         for (col = 0; col < length; col += n_opt) {
313                             col2 = ColOf(buffer, col, margin);
314                             if (move(row, col2) != ERR) {
315                                 AddNStr(ChStr2(buffer + col), LEN(col));
316                             }
317                         }
318                     } else {
319                         if (move(row, col2) != ERR) {
320                             AddStr(ChStr2(buffer));
321                         }
322                     }
323                     break;
324                 case oMove:
325                     if (n_opt > 1) {
326                         for (col = 0; col < length; col += n_opt) {
327                             col2 = ColOf(buffer, col, margin);
328                             MvAddNStr(row, col2, ChStr2(buffer + col), LEN(col));
329                         }
330                     } else {
331                         MvAddStr(row, col2, ChStr2(buffer));
332                     }
333                     break;
334                 case oWindow:
335                     if (n_opt > 1) {
336                         for (col = 0; col < length; col += n_opt) {
337                             col2 = ColOf(buffer, col, margin);
338                             if (wmove(work, row, col2) != ERR) {
339                                 WAddNStr(work, ChStr2(buffer + col), LEN(col));
340                             }
341                         }
342                     } else {
343                         if (wmove(work, row, col2) != ERR) {
344                             WAddStr(work, ChStr2(buffer));
345                         }
346                     }
347                     break;
348                 case oMoveWindow:
349                     if (n_opt > 1) {
350                         for (col = 0; col < length; col += n_opt) {
351                             col2 = ColOf(buffer, col, margin);
352                             MvWAddNStr(work, row, col2, ChStr2(buffer + col),
353                                        LEN(col));
354                         }
355                     } else {
356                         MvWAddStr(work, row, col2, ChStr2(buffer));
357                     }
358                     break;
359                 }
360
361                 /* do the corresponding single-character add */
362                 row2 = limit + row;
363                 for (col = 0; col < length; ++col) {
364                     col2 = ColOf(buffer, col, margin);
365                     switch (option) {
366                     case oDefault:
367                         if (move(row2, col2) != ERR) {
368                             AddCh(UChar(buffer[col]));
369                         }
370                         break;
371                     case oMove:
372                         MvAddCh(row2, col2, UChar(buffer[col]));
373                         break;
374                     case oWindow:
375                         if (wmove(work, row2, col2) != ERR) {
376                             WAddCh(work, UChar(buffer[col]));
377                         }
378                         break;
379                     case oMoveWindow:
380                         MvWAddCh(work, row2, col2, UChar(buffer[col]));
381                         break;
382                     }
383                 }
384             } else {
385                 beep();
386             }
387             break;
388         case KEY_BACKSPACE:
389             ch = '\b';
390             /* FALLTHRU */
391         default:
392             if (ch <= 0 || ch > 255) {
393                 beep();
394                 break;
395             }
396             buffer[length++] = (char) ch;
397             buffer[length] = '\0';
398
399             /* put the string in, one character at a time */
400             col = ColOf(buffer, length - 1, margin);
401             switch (option) {
402             case oDefault:
403                 if (move(row, col) != ERR) {
404                     AddStr(ChStr2(buffer + length - 1));
405                 }
406                 break;
407             case oMove:
408                 MvAddStr(row, col, ChStr2(buffer + length - 1));
409                 break;
410             case oWindow:
411                 if (wmove(work, row, col) != ERR) {
412                     WAddStr(work, ChStr2(buffer + length - 1));
413                 }
414                 break;
415             case oMoveWindow:
416                 MvWAddStr(work, row, col, ChStr2(buffer + length - 1));
417                 break;
418             }
419
420             /* do the corresponding single-character add */
421             switch (option) {
422             case oDefault:
423                 if (move(limit + row, col) != ERR) {
424                     AddCh(UChar(ch));
425                 }
426                 break;
427             case oMove:
428                 MvAddCh(limit + row, col, UChar(ch));
429                 break;
430             case oWindow:
431                 if (wmove(work, limit + row, col) != ERR) {
432                     WAddCh(work, UChar(ch));
433                 }
434                 break;
435             case oMoveWindow:
436                 MvWAddCh(work, limit + row, col, UChar(ch));
437                 break;
438             }
439
440             wnoutrefresh(work);
441
442             legend(show, level, option, buffer, length);
443             wnoutrefresh(show);
444
445             doupdate();
446             break;
447         }
448     }
449     if (level > 0) {
450         delwin(show);
451         delwin(work);
452         delwin(look);
453     }
454 }
455
456 static void
457 usage(void)
458 {
459     static const char *tbl[] =
460     {
461         "Usage: test_addchstr [options]"
462         ,""
463         ,"Options:"
464         ,"  -f FILE read data from given file"
465         ,"  -n NUM  limit string-adds to NUM bytes on ^N replay"
466         ,"  -m      perform wmove/move separately from add-functions"
467         ,"  -p      pass-thru control characters without using unctrl()"
468         ,"  -w      use window-parameter even when stdscr would be implied"
469     };
470     unsigned n;
471     for (n = 0; n < SIZEOF(tbl); ++n)
472         fprintf(stderr, "%s\n", tbl[n]);
473     ExitProgram(EXIT_FAILURE);
474 }
475
476 int
477 main(int argc GCC_UNUSED, char *argv[]GCC_UNUSED)
478 {
479     int ch;
480
481     setlocale(LC_ALL, "");
482
483     while ((ch = getopt(argc, argv, "f:mn:pw")) != -1) {
484         switch (ch) {
485         case 'f':
486             init_linedata(optarg);
487             break;
488         case 'm':
489             m_opt = TRUE;
490             break;
491         case 'n':
492             n_opt = atoi(optarg);
493             if (n_opt == 0)
494                 n_opt = -1;
495             break;
496         case 'p':
497             pass_ctls = TRUE;
498             break;
499         case 'w':
500             w_opt = TRUE;
501             break;
502         default:
503             usage();
504             break;
505         }
506     }
507     if (optind < argc)
508         usage();
509
510     test_adds(0);
511     endwin();
512     ExitProgram(EXIT_SUCCESS);
513 }