Blame | Last modification | View Log | RSS feed
/*
* graphics file reading for page
*/
#include <u.h>
#include <libc.h>
#include <draw.h>
#include <event.h>
#include <bio.h>
#include "page.h"
typedef struct Convert Convert;
typedef struct GfxInfo GfxInfo;
typedef struct Graphic Graphic;
struct Convert {
char *name;
char *cmd;
char *truecmd; /* cmd for true color */
};
struct GfxInfo {
Graphic *g;
};
struct Graphic {
int type;
char *name;
uchar *buf; /* if stdin */
int nbuf;
};
enum {
Ipic,
Itiff,
Ijpeg,
Igif,
Iinferno,
Ifax,
Icvt2pic,
Iplan9bm,
Iccittg4,
Ippm,
Ipng,
Iyuv,
Ibmp,
};
/*
* N.B. These commands need to read stdin if %a is replaced
* with an empty string.
*/
Convert cvt[] = {
[Ipic] { "plan9", "fb/3to1 rgbv %a |fb/pcp -tplan9" },
[Itiff] { "tiff", "fb/tiff2pic %a | fb/3to1 rgbv | fb/pcp -tplan9" },
[Iplan9bm] { "plan9bm", nil },
[Ijpeg] { "jpeg", "jpg -9 %a", "jpg -t9 %a" },
[Igif] { "gif", "gif -9 %a", "gif -t9 %a" },
[Iinferno] { "inferno", nil },
[Ifax] { "fax", "aux/g3p9bit -g %a" },
[Icvt2pic] { "unknown", "fb/cvt2pic %a |fb/3to1 rgbv" },
[Ippm] { "ppm", "ppm -9 %a", "ppm -t9 %a" },
/* ``temporary'' hack for hobby */
[Iccittg4] { "ccitt-g4", "cat %a|rx nslocum /usr/lib/ocr/bin/bcp -M|fb/pcp -tcompressed -l0" },
[Ipng] { "png", "png -9 %a", "png -t9 %a" },
[Iyuv] { "yuv", "yuv -9 %a", "yuv -t9 %a" },
[Ibmp] { "bmp", "bmp -9 %a", "bmp -t9 %a" },
};
static Image* convert(Graphic*);
static Image* gfxdrawpage(Document *d, int page);
static char* gfxpagename(Document*, int);
static int spawnrc(char*, uchar*, int);
static void waitrc(void);
static int spawnpost(int);
static int addpage(Document*, char*);
static int rmpage(Document*, int);
static int genaddpage(Document*, char*, uchar*, int);
static char*
gfxpagename(Document *doc, int page)
{
GfxInfo *gfx = doc->extra;
return gfx->g[page].name;
}
static Image*
gfxdrawpage(Document *doc, int page)
{
setlabel(gfxpagename(doc, page));
GfxInfo *gfx = doc->extra;
return convert(gfx->g+page);
}
Document*
initgfx(Biobuf*, int argc, char **argv, uchar *buf, int nbuf)
{
GfxInfo *gfx;
Document *doc;
int i;
doc = emalloc(sizeof(*doc));
gfx = emalloc(sizeof(*gfx));
gfx->g = nil;
doc->npage = 0;
doc->drawpage = gfxdrawpage;
doc->pagename = gfxpagename;
doc->addpage = addpage;
doc->rmpage = rmpage;
doc->extra = gfx;
doc->fwdonly = 0;
fprint(2, "reading through graphics...\n");
if(argc==0 && buf)
genaddpage(doc, nil, buf, nbuf);
else{
for(i=0; i<argc; i++)
if(addpage(doc, argv[i]) < 0)
fprint(2, "warning: not including %s: %r\n", argv[i]);
}
return doc;
}
static int
genaddpage(Document *doc, char *name, uchar *buf, int nbuf)
{
Graphic *g;
GfxInfo *gfx;
Biobuf *b;
uchar xbuf[32];
int i, l;
l = 0;
gfx = doc->extra;
assert((name == nil) ^ (buf == nil));
assert(name != nil || doc->npage == 0);
for(i=0; i<doc->npage; i++)
if(strcmp(gfx->g[i].name, name) == 0)
return i;
if(name){
l = strlen(name);
if((b = Bopen(name, OREAD)) == nil) {
werrstr("Bopen: %r");
return -1;
}
if(Bread(b, xbuf, sizeof xbuf) != sizeof xbuf) {
werrstr("short read: %r");
return -1;
}
Bterm(b);
buf = xbuf;
nbuf = sizeof xbuf;
}
gfx->g = erealloc(gfx->g, (doc->npage+1)*(sizeof(*gfx->g)));
g = &gfx->g[doc->npage];
memset(g, 0, sizeof *g);
if(memcmp(buf, "GIF", 3) == 0)
g->type = Igif;
else if(memcmp(buf, "\111\111\052\000", 4) == 0)
g->type = Itiff;
else if(memcmp(buf, "\115\115\000\052", 4) == 0)
g->type = Itiff;
else if(memcmp(buf, "\377\330\377", 3) == 0)
g->type = Ijpeg;
else if(memcmp(buf, "\211PNG\r\n\032\n", 3) == 0)
g->type = Ipng;
else if(memcmp(buf, "compressed\n", 11) == 0)
g->type = Iinferno;
else if(memcmp(buf, "\0PC Research, Inc", 17) == 0)
g->type = Ifax;
else if(memcmp(buf, "TYPE=ccitt-g31", 14) == 0)
g->type = Ifax;
else if(memcmp(buf, "II*", 3) == 0)
g->type = Ifax;
else if(memcmp(buf, "TYPE=ccitt-g4", 13) == 0)
g->type = Iccittg4;
else if(memcmp(buf, "TYPE=", 5) == 0)
g->type = Ipic;
else if(buf[0] == 'P' && '0' <= buf[1] && buf[1] <= '9')
g->type = Ippm;
else if(memcmp(buf, "BM", 2) == 0)
g->type = Ibmp;
else if(memcmp(buf, " ", 10) == 0 &&
'0' <= buf[10] && buf[10] <= '9' &&
buf[11] == ' ')
g->type = Iplan9bm;
else if(strtochan((char*)buf) != 0)
g->type = Iplan9bm;
else if (l > 4 && strcmp(name + l -4, ".yuv") == 0)
g->type = Iyuv;
else
g->type = Icvt2pic;
if(name)
g->name = estrdup(name);
else{
g->name = estrdup("stdin"); /* so it can be freed */
g->buf = buf;
g->nbuf = nbuf;
}
if(chatty) fprint(2, "classified \"%s\" as \"%s\"\n", g->name, cvt[g->type].name);
return doc->npage++;
}
static int
addpage(Document *doc, char *name)
{
return genaddpage(doc, name, nil, 0);
}
static int
rmpage(Document *doc, int n)
{
int i;
GfxInfo *gfx;
if(n < 0 || n >= doc->npage)
return -1;
gfx = doc->extra;
doc->npage--;
free(gfx->g[n].name);
for(i=n; i<doc->npage; i++)
gfx->g[i] = gfx->g[i+1];
if(n < doc->npage)
return n;
if(n == 0)
return 0;
return n-1;
}
static Image*
convert(Graphic *g)
{
int fd;
Convert c;
char *cmd;
char *name, buf[1000];
Image *im;
int rcspawned = 0;
Waitmsg *w;
c = cvt[g->type];
if(c.cmd == nil) {
if(chatty) fprint(2, "no conversion for bitmap \"%s\"...\n", g->name);
if(g->buf == nil){ /* not stdin */
fd = open(g->name, OREAD);
if(fd < 0) {
fprint(2, "cannot open file: %r\n");
wexits("open");
}
}else
fd = stdinpipe(g->buf, g->nbuf);
} else {
cmd = c.cmd;
if(truecolor && c.truecmd)
cmd = c.truecmd;
if(g->buf != nil) /* is stdin */
name = "";
else
name = g->name;
if(strlen(cmd)+strlen(name) > sizeof buf) {
fprint(2, "command too long\n");
wexits("convert");
}
snprint(buf, sizeof buf, cmd, name);
if(chatty) fprint(2, "using \"%s\" to convert \"%s\"...\n", buf, g->name);
fd = spawnrc(buf, g->buf, g->nbuf);
rcspawned++;
if(fd < 0) {
fprint(2, "cannot spawn converter: %r\n");
wexits("convert");
}
}
im = readimage(display, fd, 0);
if(im == nil) {
fprint(2, "warning: couldn't read image: %r\n");
}
close(fd);
/* for some reason rx doesn't work well with wait */
/* for some reason 3to1 exits on success with a non-null status of |3to1 */
if(rcspawned && g->type != Iccittg4) {
if((w=wait())!=nil && w->msg[0] && !strstr(w->msg, "3to1"))
fprint(2, "slave wait error: %s\n", w->msg);
free(w);
}
return im;
}
static int
spawnrc(char *cmd, uchar *stdinbuf, int nstdinbuf)
{
int pfd[2];
int pid;
if(chatty) fprint(2, "spawning(%s)...", cmd);
if(pipe(pfd) < 0)
return -1;
if((pid = fork()) < 0)
return -1;
if(pid == 0) {
close(pfd[1]);
if(stdinbuf)
dup(stdinpipe(stdinbuf, nstdinbuf), 0);
else
dup(open("/dev/null", OREAD), 0);
dup(pfd[0], 1);
//dup(pfd[0], 2);
execl("/bin/rc", "rc", "-c", cmd, nil);
wexits("exec");
}
close(pfd[0]);
return pfd[1];
}