1 /****************************************************************************
2 * Copyright 2018-2019,2020 Thomas E. Dickey *
3 * Copyright 2006-2017,2018 Free Software Foundation, Inc. *
5 * Permission is hereby granted, free of charge, to any person obtaining a *
6 * copy of this software and associated documentation files (the *
7 * "Software"), to deal in the Software without restriction, including *
8 * without limitation the rights to use, copy, modify, merge, publish, *
9 * distribute, distribute with modifications, sublicense, and/or sell *
10 * copies of the Software, and to permit persons to whom the Software is *
11 * furnished to do so, subject to the following conditions: *
13 * The above copyright notice and this permission notice shall be included *
14 * in all copies or substantial portions of the Software. *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS *
17 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF *
18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. *
19 * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, *
20 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR *
21 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR *
22 * THE USE OR OTHER DEALINGS IN THE SOFTWARE. *
24 * Except as contained in this notice, the name(s) of the above copyright *
25 * holders shall not be used in advertising or otherwise to promote the *
26 * sale, use or other dealings in this Software without prior written *
28 ****************************************************************************/
30 * $Id: savescreen.c,v 1.57 2020/02/02 23:34:34 tom Exp $
32 * Demonstrate save/restore functions from the curses library.
33 * Thomas Dickey - 2007/7/14
37 #include <test.priv.h>
38 #include <popup_msg.h>
39 #include <parse_rgb.h>
43 #include <sys/types.h>
49 #define MyMarker ACS_DIAMOND
54 static bool use_init = FALSE;
55 static bool keep_dumps = FALSE;
58 /* In HPUX curses, cchar_t is opaque; other implementations are not */
60 BaseChar(cchar_t data)
62 wchar_t my_wchar[CCHARW_MAX];
66 if (getcchar(&data, my_wchar, &my_attr, &my_pair, NULL) == OK)
73 fexists(const char *name)
76 return (stat(name, &sb) == 0 && (sb.st_mode & S_IFMT) == S_IFREG);
87 cleanup(char *files[])
92 for (n = 0; files[n] != 0; ++n) {
99 load_screen(char *filename)
104 if ((result = scr_init(filename)) != ERR)
105 result = scr_restore(filename);
107 result = scr_set(filename);
113 * scr_restore() or scr_set() operates on curscr. If we read a character using
114 * getch() that will refresh stdscr, wiping out the result. To avoid that,
115 * copy the data back from curscr to stdscr.
120 overwrite(curscr, stdscr);
125 show_what(int color, int which, int last)
134 printw("Color %d. Saved %d of %d (? for help)", color, which, last + 1);
136 now = time((time_t *) 0);
137 mytime = ctime(&now);
138 for (n = (int) strlen(mytime) - 1; n >= 0; --n) {
139 if (isspace(UChar(mytime[n]))) {
145 mvprintw(0, (COLS - n - 2), " %s", mytime);
153 get_command(int color, int which, int last)
160 show_what(color, which, last);
168 dump_screen(char **files, int color, int which, int last, bool use_colors)
170 #if USE_WIDEC_SUPPORT
174 char *filename = files[which];
179 show_what(color, ++which, last);
180 if (scr_dump(filename) == ERR) {
182 printf("Cannot write screen-dump %s\n", filename);
184 ExitProgram(EXIT_SUCCESS);
188 int pair = 1 + (which % MAX_ANSI);
190 * Change the background color, to make it more obvious. But that
191 * changes the existing text-color. Copy the old values from the
192 * currently displayed screen.
194 bkgd((chtype) COLOR_PAIR(pair));
195 for (cy = 1; cy < LINES; ++cy) {
196 for (cx = 0; cx < COLS; ++cx) {
197 wmove(curscr, cy, cx);
198 wmove(stdscr, cy, cx);
199 #if USE_WIDEC_SUPPORT
200 if (win_wch(curscr, &mycc) != ERR) {
201 myxx = wcwidth(BaseChar(mycc));
203 wadd_wchnstr(stdscr, &mycc, 1);
208 waddch(stdscr, winch(curscr));
220 static const char *msgs[] =
222 "You are now in the screen-editor, which allows you to make some",
223 "lines on the screen, as well as save copies of the screen to a",
228 " n run the screen-loader to show the saved screens",
229 " <space> dump a screen",
231 " a toggle between '#' and graphic symbol for drawing",
232 " c change color drawn by line to next in palette",
233 " h,j,k,l or arrows to move around the screen, drawing",
236 popup_msg(stdscr, msgs);
242 static const char *msgs[] =
244 "You are now in the screen-loader, which allows you to view",
245 "the dumped/restored screens.",
249 " <space> load the next screen",
250 " <backspace> load the previous screen",
253 popup_msg(stdscr, msgs);
259 static const char *msg[] =
261 "Usage: savescreen [-r] files",
264 " -f file fill/initialize screen using text from this file",
265 " -i use scr_init/scr_restore rather than scr_set",
266 " -k keep the restored dump-files rather than removing them",
267 " -r replay the screen-dump files"
270 for (n = 0; n < SIZEOF(msg); ++n) {
271 fprintf(stderr, "%s\n", msg[n]);
273 ExitProgram(EXIT_FAILURE);
277 main(int argc, char *argv[])
282 bool use_colors = FALSE;
283 bool replaying = FALSE;
287 #if USE_WIDEC_SUPPORT
293 static const wchar_t mywc[2] =
295 bool using_rgb = FALSE;
298 setlocale(LC_ALL, "");
300 while ((ch = getopt(argc, argv, "f:ikr")) != -1) {
320 files = argv + optind;
321 last = argc - optind - 1;
324 while (last >= 0 && !fexists(files[last]))
331 keypad(stdscr, TRUE);
334 if (has_colors() && (start_color() == OK) && COLORS >= MAX_ANSI) {
335 static const struct {
337 } table[MAX_ANSI] = {
338 #define DATA(fg,bg) { COLOR_##fg, COLOR_##bg }
343 DATA(MAGENTA, WHITE),
344 DATA(MAGENTA, BLACK),
354 * Discounting color-pair 0 (no color), make the next 8 color pairs
355 * useful for leaving a visually distinct trail of characters on the
358 for (n = 0; n < MAX_ANSI; ++n) {
359 init_pair((short) pair++, (short) table[n].fg, (short) table[n].bg);
362 * After that, use color pairs for constructing a test-pattern, e.g.,
363 * imitating xterm's scripts.
367 for (n = 0; n < COLORS; ++n)
368 init_pair((short) (n + MAX_ANSI), (short) n, (short) n);
370 #if HAVE_TIGETSTR && USE_WIDEC_SUPPORT
372 int r_max, g_max, b_max;
374 if (parse_rgb(&r_max, &g_max, &b_max) > 0) {
375 int rows = LINES - 1;
377 int b_delta = (b_max / rows);
378 int r_delta = (r_max / cols);
379 int g_delta = (g_max / cols);
384 while (row++ < rows) {
388 while (col++ < cols) {
389 int color = (((r * (g_max + 1)) + g) * (b_max + 1)
391 #if USE_EXTENDED_COLOR
392 init_extended_pair(pair, color, color);
394 init_pair(pair, color, color);
406 if ((fill_by == 0) && !replaying) {
408 * Originally (before wide-characters) ncurses supported 16 colors.
410 if (COLORS >= 16 && COLORS <= 256) {
411 mvprintw(2, 0, "System colors:\n");
412 for (n = 0; n < 16; ++n) {
414 addch((chtype) (' ' | COLOR_PAIR(pair)));
415 addch((chtype) (' ' | COLOR_PAIR(pair)));
416 if (((n + 1) % 8) == 0)
421 * Even with ncurses, you need wide-character support to have more
424 #if USE_WIDEC_SUPPORT
427 } else if (COLORS == 256) {
432 cube1 = cube0 + (cube * cube * cube);
435 printw("Color cube, %dx%dx%d:\n", cube, cube, cube);
436 for (g = 0; g < cube; g++) {
437 for (r = 0; r < cube; r++) {
438 for (b = 0; b < cube; b++) {
441 + (r * cube * cube) + (g * cube) + b;
442 setcchar(&mycc, mywc, 0, (short) pair, NULL);
451 printw("Grayscale ramp:\n");
452 for (n = cube1; n < COLORS; ++n) {
454 setcchar(&mycc, mywc, 0, (short) pair, NULL);
458 } else if ((COLORS > 256) && using_rgb) {
459 int rows = LINES - 1;
465 while (row++ < rows) {
467 while (col++ < cols) {
468 setcchar(&mycc, mywc, 0, (short) pair, &pair);
481 FILE *fp = fopen(fill_by, "r");
485 while ((ch = fgetc(fp)) != EOF) {
486 if (addch(UChar(ch)) == ERR) {
493 while (addch(' ') != ERR) {
500 fprintf(stderr, "Cannot open \"%s\"\n", fill_by);
501 ExitProgram(EXIT_FAILURE);
508 * Use the last file as the initial/current screen.
512 printf("No screen-dumps given\n");
513 ExitProgram(EXIT_FAILURE);
517 if (load_screen(files[which]) == ERR) {
519 printf("Cannot load screen-dump %s\n", files[which]);
520 ExitProgram(EXIT_FAILURE);
524 while (!done && (ch = getch()) != ERR) {
528 * If we got a "next" here, skip to the final screen before
529 * moving to the next process.
558 } else if (scr_restore(files[which]) == ERR) {
560 printf("Cannot load screen-dump %s\n", files[which]);
562 ExitProgram(EXIT_FAILURE);
573 bool dirty = use_colors || (fill_by != 0);
576 switch (get_command(color, which, last)) {
578 if (dirty && files[which]) {
579 dump_screen(files, color, which, last, use_colors);
589 if (dump_screen(files, color, which, last, use_colors)) {
590 which = (which + 1) % MAX_ANSI;
618 altchars = !altchars;
622 color = (color + 1) % MAX_ANSI;
633 chtype attr = A_REVERSE;
634 chtype ch2 = (altchars ? MyMarker : '#');
636 attr |= (chtype) COLOR_PAIR(color);
646 ExitProgram(EXIT_SUCCESS);
651 main(int argc GCC_UNUSED, char *argv[]GCC_UNUSED)
653 printf("This program requires the screen-dump functions\n");
654 ExitProgram(EXIT_FAILURE);