From f3298400e6704844aeb1d3e0951e84b4236d2302 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bert=20M=C3=BCnnich?= Date: Mon, 11 Feb 2013 23:05:26 +0100 Subject: [PATCH] Spawn and read from info script without blocking --- main.c | 144 +++++++++++++++++++++++++++++++++++-------------------- window.c | 59 ++++++++++++----------- window.h | 12 +++-- 3 files changed, 131 insertions(+), 84 deletions(-) diff --git a/main.c b/main.c index 9ff3c1e..f648dab 100644 --- a/main.c +++ b/main.c @@ -22,10 +22,14 @@ #include #include #include +#include #include +#include +#include #include #include #include +#include #include #include "types.h" @@ -38,8 +42,6 @@ #include "config.h" enum { - BAR_L_LEN = 512, - BAR_R_LEN = 64, FILENAME_CNT = 1024, TITLE_LEN = 256 }; @@ -70,12 +72,11 @@ int prefix; bool resized = false; const char * const INFO_SCRIPT = ".sxiv/exec/image-info"; -char *info_script; - struct { - char l[BAR_L_LEN]; - char r[BAR_R_LEN]; -} bar; + char *script; + int fd; + unsigned int i, lastsep; +} info; timeout_t timeouts[] = { { { 0, 0 }, false, redraw }, @@ -212,35 +213,66 @@ bool check_timeouts(struct timeval *t) return tmin > 0; } +void open_info(void) +{ + static pid_t pid; + int pfd[2]; + + win.bar.l[0] = '\0'; + + if (info.fd != -1) { + close(info.fd); + kill(pid, SIGTERM); + while (waitpid(-1, NULL, WNOHANG) > 0); + info.fd = -1; + } + if (info.script == NULL || pipe(pfd) < 0) + return; + + pid = fork(); + if (pid > 0) { + close(pfd[1]); + fcntl(pfd[0], F_SETFL, O_NONBLOCK); + info.fd = pfd[0]; + info.i = info.lastsep = 0; + } else if (pid == 0) { + close(pfd[0]); + dup2(pfd[1], 1); + execl(info.script, info.script, files[fileidx].name, NULL); + } +} + void read_info(void) { - char cmd[4096]; - FILE *outp; - int c, i = 0, n = sizeof(bar.l) - 1; - bool lastsep = false; - - if (info_script != NULL) { - snprintf(cmd, sizeof(cmd), "%s \"%s\"", info_script, files[fileidx].name); - outp = popen(cmd, "r"); - if (outp == NULL) + ssize_t i, n; + char buf[BAR_L_LEN]; + + while (true) { + n = read(info.fd, buf, sizeof(buf)); + if (n < 0 && errno == EAGAIN) + return; + else if (n == 0) goto end; - while (i < n && (c = fgetc(outp)) != EOF) { - if (c == '\n') { - if (!lastsep) { - bar.l[i++] = ' '; - lastsep = true; + for (i = 0; i < n; i++) { + if (buf[i] == '\n') { + if (info.lastsep == 0) { + win.bar.l[info.i++] = ' '; + info.lastsep = 1; } } else { - bar.l[i++] = c; - lastsep = false; + win.bar.l[info.i++] = buf[i]; + info.lastsep = 0; } + if (info.i + 1 == sizeof(win.bar.l)) + goto end; } - pclose(outp); } end: - if (lastsep) - i--; - bar.l[i] = '\0'; + info.i -= info.lastsep; + win.bar.l[info.i] = '\0'; + win_update_bar(&win); + info.fd = -1; + while (waitpid(-1, NULL, WNOHANG) > 0); } void load_image(int new) @@ -261,7 +293,7 @@ void load_image(int new) alternate = fileidx; fileidx = new; - read_info(); + open_info(); if (img.multi.cnt > 0 && img.multi.animate) set_timeout(animate, img.multi.frames[img.multi.sel].delay, true); @@ -271,9 +303,10 @@ void load_image(int new) void update_info(void) { - unsigned int i, fn, fw, n, len = sizeof(bar.r); int sel; - char *t = bar.r, title[TITLE_LEN]; + unsigned int i, fn, fw, n; + unsigned int llen = sizeof(win.bar.l), rlen = sizeof(win.bar.r); + char *lt = win.bar.l, *rt = win.bar.r, title[TITLE_LEN]; bool ow_info; for (fw = 0, i = filecnt; i > 0; fw++, i /= 10); @@ -283,39 +316,37 @@ void update_info(void) win_set_title(&win, "sxiv"); if (tns.cnt == filecnt) { - n = snprintf(t, len, "%0*d/%d", fw, sel + 1, filecnt); + n = snprintf(rt, rlen, "%0*d/%d", fw, sel + 1, filecnt); ow_info = true; } else { - snprintf(bar.l, sizeof(bar.l), "Loading... %0*d/%d", - fw, tns.cnt, filecnt); - bar.r[0] = '\0'; + snprintf(lt, llen, "Loading... %0*d/%d", fw, tns.cnt, filecnt); + rt[0] = '\0'; ow_info = false; } } else { snprintf(title, sizeof(title), "sxiv - %s", files[sel].name); win_set_title(&win, title); - n = snprintf(t, len, "%3d%% ", (int) (img.zoom * 100.0)); + n = snprintf(rt, rlen, "%3d%% ", (int) (img.zoom * 100.0)); if (img.multi.cnt > 0) { for (fn = 0, i = img.multi.cnt; i > 0; fn++, i /= 10); - n += snprintf(t + n, len - n, "(%0*d/%d) ", + n += snprintf(rt + n, rlen - n, "(%0*d/%d) ", fn, img.multi.sel + 1, img.multi.cnt); } - n += snprintf(t + n, len - n, "%0*d/%d", fw, sel + 1, filecnt); - ow_info = bar.l[0] == '\0'; + n += snprintf(rt + n, rlen - n, "%0*d/%d", fw, sel + 1, filecnt); + ow_info = info.script == NULL; } if (ow_info) { fn = strlen(files[sel].name); - if (fn < sizeof(bar.l) && + if (fn < llen && win_textwidth(files[sel].name, fn, true) + - win_textwidth(bar.r, n, true) < win.w) + win_textwidth(rt, n, true) < win.w) { - strncpy(bar.l, files[sel].name, sizeof(bar.l)); + strncpy(lt, files[sel].name, llen); } else { - strncpy(bar.l, files[sel].base, sizeof(bar.l)); + strncpy(lt, files[sel].base, llen); } } - win_set_bar_info(&win, bar.l, bar.r); } void redraw(void) @@ -458,8 +489,8 @@ void run(void) int xfd; fd_set fds; struct timeval timeout; + bool discard, to_set; XEvent ev, nextev; - bool discard; redraw(); @@ -482,12 +513,20 @@ void run(void) check_timeouts(NULL); } - while (XPending(win.env.dpy) == 0 && check_timeouts(&timeout)) { - /* wait for timeouts */ + while (XPending(win.env.dpy) == 0 + && ((to_set = check_timeouts(&timeout)) || info.fd != -1)) + { + /* check for timeouts & input */ xfd = ConnectionNumber(win.env.dpy); FD_ZERO(&fds); FD_SET(xfd, &fds); - select(xfd + 1, &fds, 0, 0, &timeout); + if (info.fd != -1) { + FD_SET(info.fd, &fds); + xfd = MAX(xfd, info.fd); + } + select(xfd + 1, &fds, 0, 0, to_set ? &timeout : NULL); + if (FD_ISSET(info.fd, &fds)) + read_info(); } do { @@ -641,13 +680,14 @@ int main(int argc, char **argv) warn("could not locate home directory"); } else { len = strlen(homedir) + strlen(INFO_SCRIPT) + 2; - info_script = (char*) s_malloc(len); - snprintf(info_script, len, "%s/%s", homedir, INFO_SCRIPT); - if (access(info_script, X_OK) != 0) { - free(info_script); - info_script = NULL; + info.script = (char*) s_malloc(len); + snprintf(info.script, len, "%s/%s", homedir, INFO_SCRIPT); + if (access(info.script, X_OK) != 0) { + free(info.script); + info.script = NULL; } } + info.fd = -1; if (options->thumb_mode) { mode = MODE_THUMB; diff --git a/window.c b/window.c index 0b1ce65..a51499e 100644 --- a/window.c +++ b/window.c @@ -414,31 +414,28 @@ void win_draw_bar(win_t *win) XSetForeground(e->dpy, gc, win->bar.fgcol); XSetBackground(e->dpy, gc, win->bar.bgcol); - if (win->bar.r != NULL) { - len = strlen(win->bar.r); - if (len > 0) { - if ((tw = win_textwidth(win->bar.r, len, true)) > w) - return; - x = win->w - tw + H_TEXT_PAD; - w -= tw; - if (font.set) - XmbDrawString(e->dpy, win->pm, font.set, gc, x, y, win->bar.r, len); - else - XDrawString(e->dpy, win->pm, gc, x, y, win->bar.r, len); - } + if ((len = strlen(win->bar.r)) > 0) { + if ((tw = win_textwidth(win->bar.r, len, true)) > w) + return; + x = win->w - tw + H_TEXT_PAD; + w -= tw; + if (font.set) + XmbDrawString(e->dpy, win->pm, font.set, gc, x, y, win->bar.r, len); + else + XDrawString(e->dpy, win->pm, gc, x, y, win->bar.r, len); } - if (win->bar.l != NULL) { - olen = len = strlen(win->bar.l); + if ((len = strlen(win->bar.l)) > 0) { + olen = len; while (len > 0 && (tw = win_textwidth(win->bar.l, len, true)) > w) len--; if (len > 0) { - if (len != olen) { - w = strlen(dots); - if (len <= w) - return; - memcpy(rest, win->bar.l + len - w, w); - memcpy(win->bar.l + len - w, dots, w); - } + if (len != olen) { + w = strlen(dots); + if (len <= w) + return; + memcpy(rest, win->bar.l + len - w, w); + memcpy(win->bar.l + len - w, dots, w); + } x = H_TEXT_PAD; if (font.set) XmbDrawString(e->dpy, win->pm, font.set, gc, x, y, win->bar.l, len); @@ -480,6 +477,18 @@ void win_draw_rect(win_t *win, Pixmap pm, int x, int y, int w, int h, XDrawRectangle(win->env.dpy, pm, gc, x, y, w, h); } +void win_update_bar(win_t *win) +{ + if (win == NULL || win->xwin == None || win->pm == None) + return; + + if (win->bar.h > 0) { + win_draw_bar(win); + XCopyArea(win->env.dpy, win->pm, win->xwin, gc, + 0, win->h, win->w, win->bar.h, 0, win->h); + } +} + int win_textwidth(const char *text, unsigned int len, bool with_padding) { XRectangle r; @@ -514,14 +523,6 @@ void win_set_title(win_t *win, const char *title) PropModeReplace, (unsigned char *) title, strlen(title)); } -void win_set_bar_info(win_t *win, char *linfo, char *rinfo) -{ - if (win != NULL) { - win->bar.l = linfo; - win->bar.r = rinfo; - } -} - void win_set_cursor(win_t *win, cursor_t cursor) { if (win == NULL || win->xwin == None) diff --git a/window.h b/window.h index dfb39aa..19a7171 100644 --- a/window.h +++ b/window.h @@ -24,6 +24,11 @@ #include "types.h" +enum { + BAR_L_LEN = 512, + BAR_R_LEN = 64 +}; + typedef struct { Display *dpy; int scr; @@ -55,8 +60,8 @@ typedef struct { struct { unsigned int h; - char *l; - char *r; + char l[BAR_L_LEN]; + char r[BAR_R_LEN]; unsigned long bgcol; unsigned long fgcol; } bar; @@ -80,10 +85,11 @@ void win_draw(win_t*); void win_draw_rect(win_t*, Pixmap, int, int, int, int, bool, int, unsigned long); +void win_update_bar(win_t*); + int win_textwidth(const char*, unsigned int, bool); void win_set_title(win_t*, const char*); -void win_set_bar_info(win_t*, char*, char*); void win_set_cursor(win_t*, cursor_t); #endif /* WINDOW_H */