/**************************************************************************** * Copyright (c) 2017 Free Software Foundation, Inc. * * * * Permission is hereby granted, free of charge, to any person obtaining a * * copy of this software and associated documentation files (the * * "Software"), to deal in the Software without restriction, including * * without limitation the rights to use, copy, modify, merge, publish, * * distribute, distribute with modifications, sublicense, and/or sell * * copies of the Software, and to permit persons to whom the Software is * * furnished to do so, subject to the following conditions: * * * * The above copyright notice and this permission notice shall be included * * in all copies or substantial portions of the Software. * * * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. * * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, * * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR * * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR * * THE USE OR OTHER DEALINGS IN THE SOFTWARE. * * * * Except as contained in this notice, the name(s) of the above copyright * * holders shall not be used in advertising or otherwise to promote the * * sale, use or other dealings in this Software without prior written * * authorization. * ****************************************************************************/ /* * $Id: picsmap.c,v 1.114 2017/12/16 23:49:40 tom Exp $ * * Author: Thomas E. Dickey * * A little more interesting than "dots", read a simple image into memory and * measure the time taken to paint it normally vs randomly. * * TODO improve use of rgb-names using tsearch. * * TODO add option to dump picture in non-optimized mode, e.g., like tput. * TODO write cells/second to stderr (or log) * TODO write picture left-to-right/top-to-bottom * TODO write picture randomly * TODO add one-shot option vs repeat-count before exiting * TODO add option "-xc" for init_color vs init_extended_color * TODO add option "-xa" for init_pair vs alloc_pair * TODO use pad to allow pictures larger than screen * TODO add option to just use convert (which can scale) vs builtin xbm/xpm. * TODO add scr_dump and scr_restore calls * TODO add option for assume_default_colors */ #include #include #include #ifdef HAVE_STDINT_H #include #define my_intptr_t intptr_t #else #define my_intptr_t long #endif #if HAVE_TSEARCH #include #endif #undef CUR /* use only the curses interface */ #define L_BLOCK '[' #define R_BLOCK ']' #define L_CURLY '{' #define R_CURLY '}' #define okCOLOR(n) ((n) >= 0 && (n) < COLORS) #define okRGB(n) ((n) >= 0 && (n) <= 1000) #define Scaled256(n) (NCURSES_COLOR_T) (int)(((n) * 1000.0) / 256) #define ScaledColor(n) (NCURSES_COLOR_T) (int)(((n) * 1000.0) / scale) #ifndef RGB_PATH #define RGB_PATH "/etc/X11/rgb.txt" #endif #include typedef struct { size_t file; size_t name; size_t list; size_t data; size_t head; size_t pair; size_t cell; } HOW_MUCH; #undef MAX #define MAX(a,b) ((a)>(b)?(a):(b)) /* * tfind will return null on failure, so we map subscripts starting at one. */ #define P2I(n) (((int)(my_intptr_t)(n)) - 1) #define I2P(n) (void *)(my_intptr_t)((n) + 1) #define stop_curses() if (in_curses) endwin() #define debugmsg if (debugging) logmsg #define debugmsg2 if (debugging) logmsg2 static void cleanup(int) GCC_NORETURN; static void giveup(const char *fmt,...) GCC_PRINTFLIKE(1, 2); static void logmsg(const char *fmt,...) GCC_PRINTFLIKE(1, 2); static void logmsg2(const char *fmt,...) GCC_PRINTFLIKE(1, 2); static void warning(const char *fmt,...) GCC_PRINTFLIKE(1, 2); static int gather_c_values(int); static FILE *logfp = 0; static double aspect_ratio = 0.6; static bool in_curses = FALSE; static bool debugging = FALSE; static bool quiet = FALSE; static int slow_time = -1; static RGB_NAME *rgb_table; static RGB_DATA *all_colors; static HOW_MUCH how_much; static int reading_last; static int reading_size; static FG_NODE *reading_ncols; #if HAVE_TSEARCH static void *reading_ntree; #endif #if HAVE_ALLOC_PAIR && HAVE_INIT_EXTENDED_COLOR #define USE_EXTENDED_COLORS 1 static bool use_extended_pairs = FALSE; static bool use_extended_colors = FALSE; #else #define USE_EXTENDED_COLORS 0 #endif static void logmsg(const char *fmt,...) { if (logfp != 0) { va_list ap; va_start(ap, fmt); vfprintf(logfp, fmt, ap); va_end(ap); fputc('\n', logfp); fflush(logfp); } } static void logmsg2(const char *fmt,...) { if (logfp != 0) { va_list ap; va_start(ap, fmt); vfprintf(logfp, fmt, ap); va_end(ap); fflush(logfp); } } static void close_log(void) { if (logfp != 0) { logmsg("Allocations:"); logmsg("%8ld file", (long) how_much.file); logmsg("%8ld name", (long) how_much.name); logmsg("%8ld list", (long) how_much.list); logmsg("%8ld data", (long) how_much.data); logmsg("%8ld head", (long) how_much.head); logmsg("%8ld pair", (long) how_much.pair); logmsg("%8ld cell", (long) how_much.cell); logmsg("%8ld window", LINES * COLS * (long) sizeof(NCURSES_CH_T)); fclose(logfp); logfp = 0; } } static void cleanup(int code) { stop_curses(); close_log(); ExitProgram(code); /* NOTREACHED */ } static void failed(const char *msg) { int save = errno; perror(msg); logmsg("failed with %s", strerror(save)); cleanup(EXIT_FAILURE); } static void warning(const char *fmt,...) { if (logfp != 0) { va_list ap; va_start(ap, fmt); vfprintf(logfp, fmt, ap); va_end(ap); fputc('\n', logfp); fflush(logfp); } else { va_list ap; va_start(ap, fmt); vfprintf(stderr, fmt, ap); va_end(ap); fputc('\n', stderr); cleanup(EXIT_FAILURE); } } static void free_data(char **data) { if (data != 0) { free(data[0]); free(data); } } static PICS_HEAD * free_pics_head(PICS_HEAD * pics) { if (pics != 0) { free(pics->fgcol); free(pics->cells); free(pics->name); free(pics); pics = 0; } return pics; } static void begin_c_values(int size) { reading_last = 0; reading_size = size; reading_ncols = typeCalloc(FG_NODE, size + 1); how_much.pair += (sizeof(FG_NODE) * (size_t) size); /* black is always the first slot, to work around P2I/I2P logic */ gather_c_values(0); } #if HAVE_TSEARCH static int compare_c_values(const void *p, const void *q) { const int a = P2I(p); const int b = P2I(q); return (reading_ncols[a].fgcol - reading_ncols[b].fgcol); } #ifdef DEBUG_TSEARCH static void check_c_values(int ln) { static int oops = 5; FG_NODE **ft; int n; for (n = 0; n < reading_last; ++n) { ft = tfind(I2P(n), &reading_ntree, compare_c_values); if (ft != 0) { int q = P2I(*ft); if (reading_ncols[q].fgcol != reading_ncols[n].fgcol) { logmsg("@%d, %d:%d (%d) %d %d fgcol %06X %06X", ln, n, reading_last - 1, reading_size, q, n, reading_ncols[n].fgcol, reading_ncols[q].fgcol); } } else { logmsg("@%d, %d:%d (%d) ? %d null %06X", ln, n, reading_last - 1, reading_size, n, reading_ncols[n].fgcol); if (oops-- <= 0) return; } } } #else #define check_c_values(n) /* nothing */ #endif #endif static int gather_c_values(int fg) { int found = -1; #if HAVE_TSEARCH FG_NODE **ft; int next = reading_last; reading_ncols[next].fgcol = fg; reading_ncols[next].count = 0; check_c_values(__LINE__); if ((ft = tfind(I2P(next), &reading_ntree, compare_c_values)) != 0) { found = P2I(*ft); } else { if (reading_last + 2 >= reading_size) { int more = ((MAX(reading_last, reading_size) + 2) * 3) / 2; int last = reading_last + 1; FG_NODE *p = typeRealloc(FG_NODE, more, reading_ncols); if (p == 0) goto done; reading_size = more; reading_ncols = p; memset(reading_ncols + last, 0, sizeof(FG_NODE) * (size_t) (more - last)); check_c_values(__LINE__); } ++reading_last; how_much.pair += sizeof(FG_NODE); if ((ft = tsearch(I2P(next), &reading_ntree, compare_c_values)) != 0) { found = P2I(*ft); if (found != next) logmsg("OOPS expected slot %d, got %d", next, found); debugmsg("allocated color #%d as #%06X", next, fg); check_c_values(__LINE__); } } #else int n; for (n = 0; n < reading_last; ++n) { if (reading_ncols[n].fgcol == fg) { found = n; break; } } if (found < 0) { if (reading_last + 2 >= reading_size) { int more = ((reading_last + 2) * 3) / 2; FG_NODE *p = typeRealloc(FG_NODE, more, reading_ncols); if (p == 0) goto done; how_much.pair -= (sizeof(FG_NODE) * (size_t) reading_size); how_much.pair += (sizeof(FG_NODE) * (size_t) more); reading_size = more; reading_ncols = p; memset(reading_ncols + reading_last, 0, sizeof(FG_NODE) * (size_t) (more - reading_last)); } reading_ncols[reading_last].fgcol = fg; found = reading_last++; } #endif done: return found; } static void finish_c_values(PICS_HEAD * head) { head->colors = reading_last; head->fgcol = reading_ncols; reading_last = 0; reading_size = 0; reading_ncols = 0; } #if HAVE_TSEARCH && HAVE_TDESTROY static void never_free(void *node GCC_UNUSED) { } #endif static void dispose_c_values(void) { #if HAVE_TSEARCH if (reading_ntree != 0) { #if HAVE_TDESTROY tdestroy(reading_ntree, never_free); #else int n; for (n = 0; n < reading_last; ++n) { tdelete(I2P(n), &reading_ntree, compare_c_values); } #endif reading_ntree = 0; } #endif if (reading_ncols != 0) { free(reading_ncols); reading_ncols = 0; } reading_last = 0; reading_size = 0; } static int is_file(const char *filename, struct stat *sb) { int result = 0; if (stat(filename, sb) == 0 && (sb->st_mode & S_IFMT) == S_IFREG && sb->st_size != 0) { result = 1; } debugmsg("is_file(%s) %d", filename, result); return result; } /* * Simplify reading xbm/xpm files by first making an array of lines. Blank * lines are filtered out. */ static char ** read_file(const char *filename) { char **result = 0; struct stat sb; if (!quiet) { stop_curses(); printf("** %s\n", filename); } if (is_file(filename, &sb)) { size_t size = (size_t) sb.st_size; char *blob = typeCalloc(char, size + 1); bool had_line = TRUE; bool binary = FALSE; unsigned j; unsigned k = 0; result = typeCalloc(char *, size + 1); how_much.file += ((size + 1) * 2); if (blob != 0 && result != 0) { FILE *fp = fopen(filename, "r"); if (fp != 0) { logmsg("opened %s", filename); if (fread(blob, sizeof(char), size, fp) == size) { for (j = 0; (size_t) j < size; ++j) { if (blob[j] == '\0' || (UChar(blob[j]) < 32 && !isspace(UChar(blob[j]))) || (UChar(blob[j]) >= 128 && UChar(blob[j]) < 160)) { binary = TRUE; } if (blob[j] == '\n') { blob[j] = '\0'; if (k && !binary) { debugmsg2("[%5d] %s\n", k, result[k - 1]); } had_line = TRUE; } else if (had_line) { had_line = FALSE; result[k++] = blob + j; } } result[k] = 0; if (k && !binary) { debugmsg2("[%5d] %s\n", k, result[k - 1]); } } fclose(fp); } else { logmsg("cannot open %s", filename); } } if (k == 0) { debugmsg("...file is empty"); free(blob); free(result); result = 0; } else if (binary) { debugmsg("...file is non-text"); } } return result; } static void usage(void) { static const char *msg[] = { "Usage: picsmap [options] [imagefile [...]]" ,"Read/display one or more xbm/xpm files (possibly use \"convert\")" ,"" ,"Options:" ," -a ratio aspect-ratio correction for ImageMagick" #if HAVE_USE_DEFAULT_COLORS ," -d invoke use_default_colors" #endif ," -L add debugging information to logfile" ," -l logfile write informational messages to logfile" ," -p palette color-palette file (default \"$TERM.dat\")" ," -q less verbose" ," -r rgb-path xpm uses X rgb color-names (default \"" RGB_PATH "\")" ," -s SECS pause for SECS seconds after display vs getch" #if USE_EXTENDED_COLORS ," -x [pc] use extension (p=extended-pairs, c=extended-colors)" ," Either/both extension may be given" #endif }; size_t n; stop_curses(); fflush(stdout); for (n = 0; n < SIZEOF(msg); n++) fprintf(stderr, "%s\n", msg[n]); cleanup(EXIT_FAILURE); } static void giveup(const char *fmt,...) { va_list ap; stop_curses(); fflush(stdout); va_start(ap, fmt); vfprintf(stderr, fmt, ap); fputc('\n', stderr); va_end(ap); if (logfp) { va_start(ap, fmt); vfprintf(logfp, fmt, ap); fputc('\n', logfp); va_end(ap); fflush(logfp); } usage(); } /* * Palette files are named for $TERM values. However, there are fewer palette * files than $TERM's. Although there are known problems (some cannot even get * black and white correct), for the purpose of comparison, pretending that * those map into "xterm" is useful. */ static char ** read_palette(const char *filename) { static const char *data_dir = DATA_DIR; char **result = 0; char *full_name = malloc(strlen(data_dir) + 20 + strlen(filename)); char *s; struct stat sb; if (full_name != 0) { int tries; for (tries = 0; tries < 8; ++tries) { *(s = full_name) = '\0'; if (tries & 1) { if (strchr(filename, '/') == 0) { sprintf(full_name, "%s/", data_dir); } else { continue; } } s += strlen(s); strcpy(s, filename); if (tries & 4) { char *t = s; int num; char chr; int found = 0; while (*t != '\0') { if (*t == '-') { if (sscanf(t, "-%d%c", &num, &chr) == 2 && chr == 'c' && !strncmp(strchr(t, chr), "color", 5)) { found = 1; } break; } ++t; } if (found && (t != s) && strncmp(s, "xterm", (size_t) (t - s))) { sprintf(s, "xterm%s", filename + (t - s)); } else { continue; } } s += strlen(s); if (tries & 2) { int len = (int) strlen(filename); if (len <= 4 || strcmp(filename + len - 4, ".dat")) { strcpy(s, ".dat"); } else { continue; } } if (is_file(full_name, &sb)) goto ok; } goto failed; ok: result = read_file(full_name); failed: free(full_name); } return result; } static void init_palette(const char *palette_file) { if (palette_file != 0) { char **data = read_palette(palette_file); all_colors = typeMalloc(RGB_DATA, (unsigned) COLORS); how_much.data += (sizeof(RGB_DATA) * (unsigned) COLORS); #if HAVE_COLOR_CONTENT { int cp; for (cp = 0; cp < COLORS; ++cp) { color_content((short) cp, &all_colors[cp].red, &all_colors[cp].green, &all_colors[cp].blue); } } #else memset(all_colors, 0, sizeof(RGB_DATA) * (size_t) COLORS); #endif if (data != 0) { int n; int red, green, blue; int scale = 1000; int c; for (n = 0; data[n] != 0; ++n) { if (sscanf(data[n], "scale:%d", &c) == 1) { scale = c; } else if (sscanf(data[n], "%d:%d %d %d", &c, &red, &green, &blue) == 4 && okCOLOR(c) && okRGB(red) && okRGB(green) && okRGB(blue)) { /* *INDENT-EQLS* */ all_colors[c].red = ScaledColor(red); all_colors[c].green = ScaledColor(green); all_colors[c].blue = ScaledColor(blue); } } } free_data(data); /* *INDENT-EQLS* */ } else if (COLORS > 1) { int power2 = 1; int shift = 0; while (power2 < COLORS) { ++shift; power2 <<= 1; } if ((power2 != COLORS) || ((shift % 3) != 0)) { if (all_colors == 0) { init_palette(getenv("TERM")); } if (all_colors == 0) { giveup("With %d colors, you need a palette-file", COLORS); } } } } /* * Map the 24-bit RGB value to a color index if using a palette, otherwise to a * direct color value. */ static int map_color(int value) { int result = value; if (result < 0) { result = -1; } else { /* *INDENT-EQLS* */ int red = (value & 0xff0000) >> 16; int green = (value & 0x00ff00) >> 8; int blue = (value & 0x0000ff) >> 0; if (all_colors != 0) { #define Diff2(n,m) ((m) - all_colors[n].m) * ((m) - all_colors[n].m) #define Diff2S(n) Diff2(n,red) + Diff2(n,green) + Diff2(n,blue) int d2 = Diff2S(0); int n; /* *INDENT-EQLS* */ red = Scaled256(red); green = Scaled256(green); blue = Scaled256(blue); for (result = 0, n = 1; n < COLORS; ++n) { int d = Diff2(n, red) + Diff2(n, green) + Diff2(n, blue); if (d < d2) { d2 = d; result = n; } } } else { /* direct color */ int power2 = 1; int shifts = 8; while (power2 < COLORS) { power2 <<= 3; shifts--; } if (shifts > 0) { /* TODO: round up */ red >>= shifts; green >>= shifts; blue >>= shifts; result = ((red << (2 * (8 - shifts))) + (green << (8 - shifts)) + blue); } } } return result; } static int bytes_of(int value) { if (value & 7) { value |= 7; value++; } return value; } static int match_c(const char *, const char *,...) GCC_SCANFLIKE(2,3); static char * skip_s(char *s) { while (isspace(UChar(*s))) s++; return s; } static const char * skip_cs(const char *s) { while (isspace(UChar(*s))) s++; return s; } static char * skip_word(char *s) { s = skip_s(s); while (isgraph(UChar(*s))) s++; return s; } static int match_c(const char *source, const char *pattern,...) { int limit = (int) strlen(source); const char *last_s = source + limit; va_list ap; int ch; int *ip; char *cp; long lv; va_start(ap, pattern); limit = -1; while (*pattern != '\0') { ch = UChar(*pattern++); /* blank in the pattern matches zero-or-more blanks in source */ if (isspace(ch)) { source = skip_cs(source); continue; } /* %c, %d, %s are like sscanf except for special treatment of blanks */ if (ch == '%' && *pattern != '\0' && strchr("cdnsx", *pattern)) { bool found = FALSE; ch = *pattern++; switch (ch) { case 'c': cp = va_arg(ap, char *); do { *cp++ = *source++; } while (--limit > 0); break; case 'd': case 'x': limit = -1; ip = va_arg(ap, int *); lv = strtol(source, &cp, ch == 'd' ? 10 : 16); if (cp != 0 && cp != source) { *ip = (int) lv; source = cp; } else { goto finish; } break; case 'n': /* not really sscanf... */ limit = *va_arg(ap, int *); break; case 's': limit = -1; cp = va_arg(ap, char *); while (*source != '\0') { ch = UChar(*source); if (isspace(ch)) { break; } else if (found && (ch == *skip_cs(pattern))) { break; } else { *cp++ = *source++; found = TRUE; } } *cp = '\0'; break; } continue; } /* other characters are matched literally */ if (*source++ != ch) { break; } } finish: va_end(ap); if (source > last_s) source = last_s; return (*source || *pattern) ? 0 : 1; } static int match_colors(const char *source, int cpp, char *arg1, char *arg2, char *arg3) { int result = 0; /* most files use a quasi-fixed format */ if (match_c(source, " \"%n%c %s %s \" , ", &cpp, arg1, arg2, arg3)) { arg1[cpp] = '\0'; result = 1; } else { char *t; const char *s = skip_cs(source); size_t have = strlen(source); if (*s++ == '"' && have > ((size_t) cpp + 2)) { memcpy(arg1, s, (size_t) cpp); s += cpp; while (*s++ == '\t') { for (t = arg2; (*s != '\0') && strchr("\t\"", *s) == 0;) { if (*s == ' ') { s = skip_cs(s); break; } *t++ = *s++; *t = '\0'; } for (t = arg3; (*s != '\0') && strchr("\t\"", *s) == 0;) { *t++ = *s++; *t = '\0'; } if (!strcmp(arg2, "c")) { result = 1; break; } } } } return result; } static RGB_NAME * parse_rgb(char **data) { char buf[BUFSIZ]; int n; unsigned long r, g, b; char *s, *t; size_t item = 0; size_t need; RGB_NAME *result = 0; for (need = 0; data[need] != 0; ++need) ; result = typeCalloc(RGB_NAME, need + 2); how_much.name += (sizeof(RGB_NAME) * (need + 2)); for (n = 0; data[n] != 0; ++n) { if (strlen(t = data[n]) >= sizeof(buf) - 1) continue; if (*(s = skip_s(t)) == '!') continue; r = strtoul(s, &t, 10); s = skip_s(t); g = strtoul(s, &t, 10); s = skip_s(t); b = strtoul(s, &t, 10); s = skip_s(t); result[item].name = s; t = s + strlen(s); while (t-- != s && isspace(UChar(*t))) { *t = '\0'; } result[item].value = (int) ((r & 0xff) << 16 | (g & 0xff) << 8 | (b & 0xff)); ++item; } result[item].name = "none"; result[item].value = -1; return result; } static RGB_NAME * lookup_rgb(const char *name) { RGB_NAME *result = 0; if (rgb_table != 0) { int n; for (n = 0; rgb_table[n].name != 0; ++n) { if (!strcasecmp(name, rgb_table[n].name)) { result = &rgb_table[n]; break; } } } return result; } static PICS_HEAD * parse_xbm(char **data) { int n; int state = 0; char buf[BUFSIZ]; int num; char ch; char *s; char *t; PICS_HEAD *result; size_t which = 0; size_t cells = 0; debugmsg("called parse_xbm"); result = typeCalloc(PICS_HEAD, 1); how_much.head += sizeof(PICS_HEAD); begin_c_values(2); gather_c_values(0); gather_c_values(0xffffff); for (n = 0; data[n] != 0; ++n) { if (strlen(s = data[n]) >= sizeof(buf) - 1) continue; switch (state) { case 0: case 1: case 2: if (sscanf(s, "#define %s %d%c", buf, &num, &ch) >= 2) { if ((t = strstr(buf, "_width")) != 0) { state |= 1; result->wide = (short) bytes_of(num); } else if ((t = strstr(buf, "_height")) != 0) { state |= 2; result->high = (short) num; } *t = '\0'; if (result->name) { if (strcmp(result->name, buf)) { goto finish; } } else { result->name = strdup(buf); } } break; case 3: if (sscanf(s, "static char %[^_ ]_bits[]%c", buf, &ch) >= 1) { if (strcmp(result->name, buf)) { goto finish; } state = 4; cells = (size_t) (result->wide * result->high); result->cells = typeCalloc(PICS_CELL, cells); how_much.cell += (sizeof(PICS_CELL) * cells); if ((s = strchr(s, L_CURLY)) == 0) break; ++s; } else { break; } case 4: while (*s != '\0') { while (isspace(UChar(*s))) { ++s; } if (isdigit(UChar(*s))) { long value = strtol(s, &t, 0); int b; if (t != s || value > 255 || value < 0) { s = t; } else { state = -1; goto finish; } for (b = 0; b < 8; ++b) { if (((1L << b) & value) != 0) { result->cells[which].ch = '*'; result->cells[which].fg = 1; reading_ncols[1].count++; } else { result->cells[which].ch = ' '; result->cells[which].fg = 0; reading_ncols[0].count++; } if (++which > cells) { state = -1; goto finish; } } } if (*s == R_CURLY) { state = 5; goto finish; } else if (*s == ',') { ++s; } } break; default: break; } } finish: if (state < 4) { debugmsg("...state was only %d", state); if (result) { result = free_pics_head(result); } } else { finish_c_values(result); } return result; } static PICS_HEAD * parse_xpm(char **data) { int state = 0; PICS_HEAD *result; RGB_NAME *by_name; int n; int cells = 0; int cpp = 1; /* chars per pixel */ int num[6]; int found; int which = 0; int num_colors = 0; char ch; const char *cs; char *s; char buf[BUFSIZ]; char arg1[BUFSIZ]; char arg2[BUFSIZ]; char arg3[BUFSIZ]; char **list = 0; debugmsg("called parse_xpm"); result = typeCalloc(PICS_HEAD, 1); how_much.head += sizeof(PICS_HEAD); for (n = 0; data[n] != 0; ++n) { if (strlen(s = data[n]) >= sizeof(buf) - 1) continue; switch (state) { case 0: if (match_c(s, " /* XPM */ ")) { state = 1; } break; case 1: if (match_c(s, " static char * %s [] = %c ", arg1, &ch) && ch == L_CURLY) { result->name = strdup(arg1); state = 2; } break; case 2: if (match_c(s, " \" %d %d %d %d \" , ", num + 0, num + 1, num + 2, num + 3) || match_c(s, " \" %d %d %d %d %d %d \" , ", num + 0, num + 1, num + 2, num + 3, num + 4, num + 5)) { result->wide = (short) num[0]; result->high = (short) num[1]; result->colors = num[2]; begin_c_values(num[2]); cells = (result->wide * result->high); result->cells = typeCalloc(PICS_CELL, cells); how_much.cell += sizeof(PICS_CELL) * (size_t) cells; list = typeCalloc(char *, result->colors + 1); how_much.list += sizeof(char *) * (size_t) (result->colors + 1); cpp = num[3]; state = 3; } break; case 3: if (!match_colors(s, cpp, arg1, arg2, arg3)) { break; } num_colors++; free(list[reading_last]); list[reading_last] = strdup(arg1); if ((by_name = lookup_rgb(arg3)) != 0) { found = gather_c_values(by_name->value); } else if (*arg3 == '#') { char *rgb = arg3 + 1; unsigned long value = strtoul(rgb, &s, 16); switch ((int) strlen(rgb)) { case 6: break; case 12: value = (((value >> 24) & 0xff0000L) | ((value >> 16) & 0xff00L) | ((value >> 8) & 0xffL)); break; default: warning("unexpected rgb value %s", rgb); break; } found = gather_c_values((int) value); } else { found = gather_c_values(0); /* actually an error */ } debugmsg(" [%d:%d] %06X", num_colors, result->colors, reading_ncols[(found >= 0) ? found : 0].fgcol); if (num_colors >= result->colors) { finish_c_values(result); state = 4; if (list != 0 && list[0] == 0) list[0] = strdup("\033"); } break; case 4: if (*(cs = skip_cs(s)) == '"') { ++cs; while (*cs != '\0' && *cs != '"') { int c; /* FIXME - factor out */ for (c = 0; c < result->colors; ++c) { if (list[c] == 0) { /* should not happen... */ continue; } if (!strncmp(cs, list[c], (size_t) cpp)) { result->cells[which].ch = list[c][0]; result->cells[which].fg = c; result->fgcol[c].count++; break; } } if (result->cells[which].ch == 0) { result->cells[which].ch = '?'; result->cells[which].fg = 0; } if (++which >= cells) { state = 5; break; } for (c = cpp; c > 0; --c, ++cs) { if (*cs == '\0') break; } } } break; } } if (result && list) { for (n = 0; n < result->colors; ++n) free(list[n]); free(list); } if (state < 5) { debugmsg("...state was only %d", state); result = free_pics_head(result); } if (result) { debugmsg("...allocated %d colors", result->colors); } return result; } /* * The obscurely-named "convert" is provided by ImageMagick */ static PICS_HEAD * parse_img(const char *filename) { char *cmd = malloc(strlen(filename) + 256); FILE *pp; char buffer[BUFSIZ]; char dummy[BUFSIZ]; bool okay = TRUE; PICS_HEAD *result; int pic_x = 0; int pic_y = 0; int width = in_curses ? COLS : 80; sprintf(cmd, "identify \"%s\"", filename); if (quiet) strcat(cmd, " 2>/dev/null"); logmsg("...opening pipe to %s", cmd); result = typeCalloc(PICS_HEAD, 1); how_much.head += sizeof(PICS_HEAD); if ((pp = popen(cmd, "r")) != 0) { if (fgets(buffer, sizeof(buffer), pp) != 0) { size_t n = strlen(filename); debugmsg2("...read %s", buffer); if (strlen(buffer) > n && !strncmp(buffer, filename, n) && isspace(UChar(buffer[n])) && sscanf(skip_word(buffer + n), " %dx%d ", &pic_x, &pic_y) == 2) { /* distort image to make it show normally on terminal */ pic_x = (int) ((double) pic_x / aspect_ratio); } else { pic_x = pic_y = 0; } } pclose(pp); } if (pic_x <= 0 || pic_y <= 0) goto finish; sprintf(cmd, "convert " "-resize %dx%d\\! " "-thumbnail %dx \"%s\" " "-define txt:compliance=SVG txt:-", pic_x, pic_y, width, filename); if (quiet) strcat(cmd, " 2>/dev/null"); logmsg("...opening pipe to %s", cmd); if ((pp = popen(cmd, "r")) != 0) { int count = 0; int col = 0; int row = 0; int len = 0; while (fgets(buffer, sizeof(buffer), pp) != 0) { debugmsg2("[%5d] %s", count + 1, buffer); if (strlen(buffer) > 160) { /* 80 columns would be enough */ okay = FALSE; break; } if (count++ == 0) { if (match_c(buffer, "# ImageMagick pixel enumeration: %d,%d,%d,%s ", &col, &row, &len, dummy)) { result->name = strdup(filename); result->wide = (short) col; result->high = (short) row; begin_c_values(256); result->cells = typeCalloc(PICS_CELL, (size_t) (col * row)); how_much.cell += (sizeof(PICS_CELL) * (size_t) (col * row)); } else { okay = FALSE; break; } } else { /* subsequent lines begin "col,row: (r,g,b,a) #RGB" */ int r, g, b, nocolor; unsigned check; int which, c; char *t; char *s = t = strchr(buffer, '#'); if (s != 0) { /* after the "#RGB", there are differences - just ignore */ while (*s != '\0' && !isspace(UChar(*s))) ++s; *++s = '\0'; } if (match_c(buffer, "%d,%d: (%d,%d,%d,%d) #%x ", &col, &row, &r, &g, &b, &nocolor, &check)) { if ((s - t) > 8) /* 6 hex digits vs 8 */ check /= 256; if (r > 255 || g > 255 || b > 255 || check != (unsigned) ((r << 16) | (g << 8) | b)) { okay = FALSE; break; } c = gather_c_values((int) check); which = col + (row * result->wide); result->cells[which].ch = ((in_curses || check == 0xffffff) ? ' ' : '#'); if (c >= 0 && c < reading_last) { result->cells[which].fg = c; reading_ncols[c].count++; } else { result->cells[which].fg = -1; } } else { okay = FALSE; break; } } } finish_c_values(result); pclose(pp); if (okay) { /* FIXME - is this trimming needed? */ for (len = result->colors; len > 3; len--) { if (result->fgcol[len - 1].fgcol == 0) { result->colors = len - 1; } else { break; } } } } finish: free(cmd); if (!okay) { result = free_pics_head(result); } return result; } static PICS_HEAD * read_picture(const char *filename, char **data) { PICS_HEAD *pics; if ((pics = parse_xbm(data)) == 0) { dispose_c_values(); if ((pics = parse_xpm(data)) == 0) { dispose_c_values(); if ((pics = parse_img(filename)) == 0) { dispose_c_values(); free_data(data); warning("unexpected file-format for \"%s\"", filename); } else if (pics->high == 0 || pics->wide == 0) { dispose_c_values(); free_data(data); pics = free_pics_head(pics); warning("no picture found in \"%s\"", filename); } } } return pics; } #define fg_color(pics,n) (pics->fgcol[n].fgcol) static void dump_picture(PICS_HEAD * pics) { int y, x; printf("Name %s\n", pics->name); printf("Size %dx%d\n", pics->high, pics->wide); printf("Color\n"); for (y = 0; y < pics->colors; ++y) { if (fg_color(pics, y) < 0) { printf(" %3d: %d\n", y, fg_color(pics, y)); } else { printf(" %3d: #%06x\n", y, fg_color(pics, y)); } } for (y = 0; y < pics->high; ++y) { for (x = 0; x < pics->wide; ++x) { putchar(pics->cells[y * pics->wide + x].ch); } putchar('\n'); } } #ifndef USE_DISPLAY_DRIVER static void init_display(const char *palette_path, int opt_d) { if (isatty(fileno(stdout))) { in_curses = TRUE; initscr(); cbreak(); noecho(); curs_set(0); if (has_colors()) { start_color(); #if HAVE_USE_DEFAULT_COLORS if (opt_d) use_default_colors(); #endif init_palette(palette_path); } scrollok(stdscr, FALSE); exit_curses(); } } static void show_picture(PICS_HEAD * pics) { int y, x; int n; int my_pair, my_color; debugmsg("called show_picture"); logmsg("...using %dx%d screen", LINES, COLS); #if USE_EXTENDED_COLORS reset_color_pairs(); #elif HAVE_CURSCR wclear(curscr); clear(); #endif if (has_colors()) { logmsg("...using %d colors", pics->colors); for (n = 0; n < pics->colors; ++n) { my_pair = (n + 1); my_color = map_color(fg_color(pics, n)); #if USE_EXTENDED_COLORS if (use_extended_pairs) { init_extended_pair(my_pair, my_color, my_color); } else #endif { my_pair &= 0x7fff; my_color &= 0x7fff; init_pair((short) my_pair, (short) my_color, (short) my_color); } } attrset(COLOR_PAIR(1)); erase(); } for (y = 0; y < pics->high; ++y) { if (y >= LINES) break; move(y, 0); for (x = 0; x < pics->wide; ++x) { if (x >= COLS) break; n = (y * pics->wide + x); my_pair = pics->cells[n].fg + 1; #if USE_EXTENDED_COLORS if (use_extended_pairs) { cchar_t temp; wchar_t wch[2]; wch[0] = (wchar_t) pics->cells[n].ch; wch[1] = 0; setcchar(&temp, wch, A_NORMAL, (short) my_pair, &my_pair); add_wch(&temp); } else #endif { attrset(COLOR_PAIR(my_pair)); addch((chtype) pics->cells[n].ch); } } } if (slow_time >= 0) { refresh(); if (slow_time > 0) { #ifdef NCURSES_VERSION napms(1000 * slow_time); #else sleep((unsigned) slow_time); #endif } } else { wmove(stdscr, 0, 0); getch(); } if (!quiet) endwin(); } #endif static int compare_fg_counts(const void *a, const void *b) { const FG_NODE *p = (const FG_NODE *) a; const FG_NODE *q = (const FG_NODE *) b; return (q->count - p->count); } static void report_colors(PICS_HEAD * pics) { int j, k; int high; int wide = 4; int accum; double level; int shift; int total; char buffer[256]; if (logfp == 0) return; qsort(pics->fgcol, (size_t) pics->colors, sizeof(FG_NODE), compare_fg_counts); /* * For debugging, show a (short) list of the colors used. */ if (debugging && (pics->colors < 1000)) { int digits = 0; for (j = pics->colors; j != 0; j /= 10) { ++digits; if (j < 10) ++digits; } logmsg("These colors were used:"); high = (pics->colors + wide - 1) / wide; for (j = 0; j < high && j < pics->colors; ++j) { char *s = buffer; *s = '\0'; for (k = 0; k < wide; ++k) { int n = j + (k * high); if (n >= pics->colors) break; if (k) { *s++ = ' '; if (digits < 8) { sprintf(s, "%*s", 8 - digits, " "); s += strlen(s); } } if (pics->fgcol[n].fgcol >= 0) { sprintf(s, "%3d #%06X %*d", n, pics->fgcol[n].fgcol, digits, pics->fgcol[n].count); } else { sprintf(s, "%3d (empty) %*d", n, digits, pics->fgcol[n].count); } s += strlen(s); if ((s - buffer) > 100) break; } logmsg("%s", buffer); } } /* * Given the list of colors sorted by the number of times they are used, * log a short report showing the number of colors for 90%, 99%, 99.9%, * etc. */ logmsg("Number of colors versus number of cells"); total = pics->high * pics->wide; accum = 0; level = 0.1; shift = 1; for (j = 0; j < pics->colors; ++j) { accum += pics->fgcol[j].count; if (accum >= (total * (1.0 - level))) { int after = (shift > 2) ? shift - 2 : 0; logmsg("%8d colors (%.1f%%) in %d cells (%.*f%%)", j + 1, (100.0 * (j + 1)) / pics->colors, accum, after, (100.0 * accum) / total); if (accum >= total) break; level /= 10.0; shift++; } } } int main(int argc, char *argv[]) { int n; int opt_d = FALSE; char ignore_ch; const char *palette_path = 0; const char *rgb_path = RGB_PATH; while ((n = getopt(argc, argv, "a:dLl:p:qr:s:x:")) != -1) { switch (n) { case 'a': if (sscanf(optarg, "%lf%c", &aspect_ratio, &ignore_ch) != 1 || aspect_ratio < 0.1 || aspect_ratio > 10.) { fprintf(stderr, "Expected a number in [0.1 to 10.]: %s\n", optarg); usage(); } break; #if HAVE_USE_DEFAULT_COLORS case 'd': opt_d = TRUE; break; #endif case 'L': debugging = TRUE; break; case 'l': if ((logfp = fopen(optarg, "a")) == 0) failed(optarg); break; case 'p': palette_path = optarg; break; case 'q': quiet = TRUE; break; case 'r': rgb_path = optarg; break; case 's': slow_time = atoi(optarg); break; #if USE_EXTENDED_COLORS case 'x': { char *s = optarg; while (*s) { switch (*s++) { case 'p': use_extended_pairs = TRUE; break; case 'c': use_extended_colors = TRUE; break; default: usage(); break; } } } break; #endif default: usage(); break; } } if (optind < argc) { char **rgb_data = read_file(rgb_path); if (rgb_data) rgb_table = parse_rgb(rgb_data); init_display(palette_path, opt_d); if (optind >= argc) giveup("expected at least one image filename"); for (n = optind; n < argc; ++n) { PICS_HEAD *pics; char **data = read_file(argv[n]); if (data == 0) { warning("cannot read \"%s\"", argv[n]); continue; } if ((pics = read_picture(argv[n], data)) != 0) { if (in_curses) { show_picture(pics); } else { dump_picture(pics); } report_colors(pics); dispose_c_values(); free_data(data); free_pics_head(pics); } } free_data(rgb_data); free(rgb_table); free(all_colors); } else { usage(); } cleanup(EXIT_SUCCESS); }