123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250 |
- #include <u.h>
- #include <lib9.h>
- #include <bio.h>
- #include <draw.h>
- #include <event.h>
- #include <keyboard.h>
- #include "imagefile.h"
- extern int debug;
- int cflag = 0;
- int dflag = 0;
- int eflag = 0;
- int nineflag = 0;
- int threeflag = 0;
- int output = 0;
- uint32_t outchan = CMAP8;
- Image *image;
- int defaultcolor = 1;
- enum{
- Border = 2,
- Edge = 5
- };
- char *show(int, char*, int);
- Rectangle
- imager(Image *i)
- {
- Point p1, p2;
- p1 = addpt(divpt(subpt(i->r.max, i->r.min), 2), i->r.min);
- p2 = addpt(divpt(subpt(screen->clipr.max, screen->clipr.min), 2), screen->clipr.min);
- return rectaddpt(i->r, subpt(p2, p1));
- }
- void
- eresized(int new)
- {
- Rectangle r;
- if(new && getwindow(display, Refnone) < 0){
- fprint(2, "png: can't reattach to window\n");
- exits("resize");
- }
- if(image == nil)
- return;
- r = imager(image);
- border(screen, r, -Border, nil, ZP);
- draw(screen, r, image, nil, image->r.min);
- flushimage(display, 1);
- }
- void
- main(int argc, char *argv[])
- {
- int fd, i;
- char *err;
- ARGBEGIN{
- case 'c': /* produce encoded, compressed, bitmap file; no display by default */
- cflag++;
- dflag++;
- output++;
- if(defaultcolor)
- outchan = CMAP8;
- break;
- case 'D':
- debug++;
- break;
- case 'd': /* suppress display of image */
- dflag++;
- break;
- case 'e': /* disable floyd-steinberg error diffusion */
- eflag++;
- break;
- case 'k': /* force black and white */
- defaultcolor = 0;
- outchan = GREY8;
- break;
- case '3': /* produce encoded, compressed, three-color bitmap file; no display by default */
- threeflag++;
- /* fall through */
- case 't': /* produce encoded, compressed, true-color bitmap file; no display by default */
- cflag++;
- dflag++;
- output++;
- defaultcolor = 0;
- outchan = RGB24;
- break;
- case 'v': /* force RGBV */
- defaultcolor = 0;
- outchan = CMAP8;
- break;
- case '9': /* produce plan 9, uncompressed, bitmap file; no display by default */
- nineflag++;
- dflag++;
- output++;
- if(defaultcolor)
- outchan = CMAP8;
- break;
- default:
- fprint(2, "usage: png [-39cdekrtv] [file.png ...]\n");
- exits("usage");
- }ARGEND;
- err = nil;
- if(argc == 0)
- err = show(0, "<stdin>", outchan);
- else{
- for(i=0; i<argc; i++){
- fd = sys_open(argv[i], OREAD);
- if(fd < 0){
- fprint(2, "png: can't open %s: %r\n", argv[i]);
- err = "open";
- }else{
- err = show(fd, argv[i], outchan);
- sys_close(fd);
- }
- if((nineflag || cflag) && argc>1 && err==nil){
- fprint(2, "png: exiting after one file\n");
- break;
- }
- }
- }
- exits(err);
- }
- char*
- show(int fd, char *name, int outc)
- {
- Rawimage **array, *r, *c;
- Image *i, *i2;
- int j, ch, outchan;
- long len;
- Biobuf b;
- char buf[32];
- static int inited;
- if(Binit(&b, fd, OREAD) < 0)
- return nil;
- outchan = outc;
- array = Breadpng(&b, CRGB);
- if(array == nil || array[0]==nil){
- fprint(2, "png: decode %s failed: %r\n", name);
- return "decode";
- }
- Bterm(&b);
- r = array[0];
- if(!dflag){
- if (!inited) {
- if(initdraw(0, 0, 0) < 0){
- fprint(2, "png: initdraw failed: %r\n");
- return "initdraw";
- }
- einit(Ekeyboard|Emouse);
- inited++;
- }
- if(defaultcolor && screen->depth>8 && outchan==CMAP8)
- outchan = RGB24;
- }
- if(outchan == CMAP8)
- c = torgbv(r, !eflag);
- else{
- switch(r->chandesc){
- case CY:
- outchan = GREY8;
- break;
- case CYA16:
- outchan = CHAN2(CGrey, 8, CAlpha, 8);
- break;
- case CRGB24:
- outchan = RGB24;
- break;
- case CRGBA32:
- outchan = RGBA32;
- break;
- }
- c = r;
- }
- if(c == nil){
- fprint(2, "png: conversion of %s failed: %r\n", name);
- return "torgbv";
- }
- if(!dflag){
- i = allocimage(display, c->r, outchan, 0, 0);
- if(i == nil){
- fprint(2, "png: allocimage %s failed: %r\n", name);
- return "allocimage";
- }
- if(loadimage(i, i->r, c->chans[0], c->chanlen) < 0){
- fprint(2, "png: loadimage %s of %d bytes failed: %r\n",
- name, c->chanlen);
- return "loadimage";
- }
- i2 = allocimage(display, c->r, outchan, 0, 0);
- draw(i2, i2->r, display->black, nil, ZP);
- draw(i2, i2->r, i, nil, i->r.min);
- image = i2;
- eresized(0);
- if((ch=ekbd())=='q' || ch==Kdel || ch==Keof)
- exits(nil);
- draw(screen, screen->clipr, display->white, nil, ZP);
- image = nil;
- freeimage(i);
- }
- if(nineflag){
- chantostr(buf, outchan);
- len = (c->r.max.x - c->r.min.x) * (c->r.max.y - c->r.min.y);
- switch(c->chandesc){
- case CY:
- // len *= 1;
- break;
- case CYA16:
- len *= 2;
- break;
- case CRGB24:
- len *= 3;
- break;
- case CRGBA32:
- len *= 4;
- break;
- }
- if(c->chanlen != len)
- fprint(2, "%s: writing %d bytes for len %ld chan %s\n",
- argv0, c->chanlen, len, buf);
- print("%11s %11d %11d %11d %11d ", buf,
- c->r.min.x, c->r.min.y, c->r.max.x, c->r.max.y);
- if(jehanne_write(1, c->chans[0], c->chanlen) != c->chanlen){
- fprint(2, "png: %s: write error %r\n", name);
- return "write";
- }
- }else if(cflag){
- if(writerawimage(1, c) < 0){
- fprint(2, "png: %s: write error: %r\n", name);
- return "write";
- }
- }
- for(j=0; j<r->nchans; j++)
- free(r->chans[j]);
- free(r->cmap);
- free(r);
- free(array);
- if(c && c != r){
- free(c->chans[0]);
- free(c);
- }
- return nil;
- }
|