123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <draw.h>
- #include <event.h>
- #include "imagefile.h"
- int cflag = 0;
- int dflag = 0;
- int eflag = 0;
- int jflag = 0;
- int fflag = 0;
- int Fflag = 0;
- int nineflag = 0;
- int threeflag = 0;
- int colorspace = CYCbCr; /* default for 8-bit displays: combine color rotation with dither */
- int output = 0;
- ulong outchan = CMAP8;
- Image *image;
- int defaultcolor = 1;
- enum{
- Border = 2,
- Edge = 5
- };
- char *show(int, char*, int);
- void
- eresized(int new)
- {
- Rectangle r;
- if(new && getwindow(display, Refnone) < 0){
- fprint(2, "jpg: can't reattach to window\n");
- exits("resize");
- }
- if(image == nil)
- return;
- r = insetrect(screen->clipr, Edge+Border);
- r.max.x = r.min.x+Dx(image->r);
- r.max.y = r.min.y+Dy(image->r);
- border(screen, r, -Border, nil, ZP);
- drawop(screen, r, image, nil, image->r.min, S);
- flushimage(display, 1);
- }
- void
- main(int argc, char *argv[])
- {
- int fd, i, yflag;
- char *err;
- char buf[12+1];
- yflag = 0;
- ARGBEGIN{
- case 'c': /* produce encoded, compressed, bitmap file; no display by default */
- cflag++;
- dflag++;
- output++;
- if(defaultcolor)
- outchan = CMAP8;
- break;
- case 'd': /* suppress display of image */
- dflag++;
- break;
- case 'e': /* disable floyd-steinberg error diffusion */
- eflag++;
- break;
- case 'F':
- Fflag++; /* make a movie */
- fflag++; /* merge two fields per image */
- break;
- case 'f':
- fflag++; /* merge two fields per image */
- break;
- case 'J': /* decode jpeg only; no display or remap (for debugging, etc.) */
- jflag++;
- break;
- case 'k': /* force black and white */
- defaultcolor = 0;
- outchan = GREY8;
- break;
- case 'r':
- colorspace = CRGB;
- 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 'y': /* leave it in CYCbCr; for debugging only */
- yflag = 1;
- colorspace = CYCbCr;
- 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: jpg -39cdefFkJrtv [file.jpg ...]\n");
- exits("usage");
- }ARGEND;
- if(yflag==0 && dflag==0 && colorspace==CYCbCr){ /* see if we should convert right to RGB */
- fd = open("/dev/screen", OREAD);
- if(fd > 0){
- buf[12] = '\0';
- if(read(fd, buf, 12)==12 && chantodepth(strtochan(buf))>8)
- colorspace = CRGB;
- close(fd);
- }
- }
- err = nil;
- if(argc == 0)
- err = show(0, "<stdin>", outchan);
- else{
- for(i=0; i<argc; i++){
- fd = open(argv[i], OREAD);
- if(fd < 0){
- fprint(2, "jpg: can't open %s: %r\n", argv[i]);
- err = "open";
- }else{
- err = show(fd, argv[i], outchan);
- close(fd);
- }
- if((nineflag || cflag) && argc>1 && err==nil){
- fprint(2, "jpg: exiting after one file\n");
- break;
- }
- }
- }
- exits(err);
- }
- Rawimage**
- vidmerge(Rawimage **aa1, Rawimage **aa2)
- {
- Rawimage **aao, *ao, *a1, *a2;
- int i, c, row, col;
- aao = nil;
- for (i = 0; aa1[i]; i++) {
- a1 = aa1[i];
- a2 = aa2[i];
- if (a2 == nil){
- fprint(2, "jpg: vidmerge: unequal lengths\n");
- return nil;
- }
- aao = realloc(aao, (i+2)*sizeof(Rawimage *));
- if (aao == nil){
- fprint(2, "jpg: vidmerge: realloc\n");
- return nil;
- }
- aao[i+1] = nil;
- ao = aao[i] = malloc(sizeof(Rawimage));
- if (ao == nil){
- fprint(2, "jpg: vidmerge: realloc\n");
- return nil;
- }
- memcpy(ao, a1, sizeof(Rawimage));
- if (!eqrect(a1->r , a2->r)){
- fprint(2, "jpg: vidmerge: rects different in img %d\n", i);
- return nil;
- }
- if (a1->cmaplen != a2->cmaplen){
- fprint(2, "jpg: vidmerge: cmaplen different in img %d\n", i);
- return nil;
- }
- if (a1->nchans != a2->nchans){
- fprint(2, "jpg: vidmerge: nchans different in img %d\n", i);
- return nil;
- }
- if (a1->fields != a2->fields){
- fprint(2, "jpg: vidmerge: fields different in img %d\n", i);
- return nil;
- }
- ao->r.max.y += Dy(ao->r);
- ao->chanlen += ao->chanlen;
- if (ao->chanlen != Dx(ao->r)*Dy(ao->r)){
- fprint(2, "jpg: vidmerge: chanlen wrong %d != %d*%d\n",
- ao->chanlen, Dx(ao->r), Dy(ao->r));
- return nil;
- }
- row = Dx(a1->r);
- for (c = 0; c < ao->nchans; c++) {
- uchar *po, *p1, *p2;
- ao->chans[c] = malloc(ao->chanlen);
- po = ao->chans[c];
- p1 = a1->chans[c];
- p2 = a2->chans[c];
- for (col = 0; col < Dy(a1->r); col++) {
- memcpy(po, p1, row);
- po += row, p1 += row;
- memcpy(po, p2, row);
- po += row, p2 += row;
- }
- free(a1->chans[c]);
- free(a2->chans[c]);
- }
- if(a2->cmap != nil)
- free(a2->cmap);
- free(a1);
- free(a2);
- }
- if (aa2[i] != nil)
- fprint(2, "jpg: vidmerge: unequal lengths\n");
- free(aa1);
- free(aa2);
- return aao;
- }
- char*
- show(int fd, char *name, int outc)
- {
- Rawimage **array, *r, *c;
- static int inited;
- Image *i;
- int j, ch, outchan;
- Biobuf b;
- char buf[32];
- if(Binit(&b, fd, OREAD) < 0)
- return nil;
- outchan = outc;
- rpt: array = Breadjpg(&b, colorspace);
- if(array == nil || array[0]==nil){
- fprint(2, "jpg: decode %s failed: %r\n", name);
- return "decode";
- }
- if (fflag) {
- Rawimage **a;
- a = Breadjpg(&b, colorspace);
- if(a == nil || a[0]==nil){
- fprint(2, "jpg: decode %s-2 failed: %r\n", name);
- return "decode";
- }
- array = vidmerge(a, array);
- } else
- Bterm(&b);
- r = array[0];
- c = nil;
- if(jflag)
- goto Return;
- if(!dflag){
- if (!inited) {
- if(initdraw(0, 0, 0) < 0){
- fprint(2, "jpg: initdraw failed: %r\n");
- return "initdraw";
- }
- if(Fflag == 0)
- einit(Ekeyboard|Emouse);
- inited++;
- }
- if(defaultcolor && screen->depth>8 && outchan==CMAP8)
- outchan = RGB24;
- }
- if(outchan == CMAP8)
- c = torgbv(r, !eflag);
- else{
- if(outchan==GREY8 || (r->chandesc==CY && threeflag==0)){
- c = totruecolor(r, CY);
- outchan = GREY8;
- }else
- c = totruecolor(r, CRGB24);
- }
- if(c == nil){
- fprint(2, "jpg: conversion of %s failed: %r\n", name);
- return "torgbv";
- }
- if(!dflag){
- if(c->chandesc == CY)
- i = allocimage(display, c->r, GREY8, 0, 0);
- else
- i = allocimage(display, c->r, outchan, 0, 0);
- if(i == nil){
- fprint(2, "jpg: allocimage %s failed: %r\n", name);
- return "allocimage";
- }
- if(loadimage(i, i->r, c->chans[0], c->chanlen) < 0){
- fprint(2, "jpg: loadimage %s failed: %r\n", name);
- return "loadimage";
- }
- image = i;
- eresized(0);
- if (Fflag) {
- freeimage(i);
- for(j=0; j<r->nchans; j++)
- free(r->chans[j]);
- free(r->cmap);
- free(r);
- free(array);
- goto rpt;
- }
- if((ch=ekbd())=='q' || ch==0x7F || ch==0x04)
- exits(nil);
- draw(screen, screen->clipr, display->white, nil, ZP);
- image = nil;
- freeimage(i);
- }
- if(nineflag){
- chantostr(buf, outchan);
- print("%11s %11d %11d %11d %11d ", buf,
- c->r.min.x, c->r.min.y, c->r.max.x, c->r.max.y);
- if(write(1, c->chans[0], c->chanlen) != c->chanlen){
- fprint(2, "jpg: %s: write error %r\n", name);
- return "write";
- }
- }else if(cflag){
- if(writerawimage(1, c) < 0){
- fprint(2, "jpg: %s: write error: %r\n", name);
- return "write";
- }
- }
- Return:
- for(j=0; j<r->nchans; j++)
- free(r->chans[j]);
- free(r->cmap);
- free(r);
- free(array);
- if(c){
- free(c->chans[0]);
- free(c);
- }
- if (Fflag) goto rpt;
- return nil;
- }
|