Initial revision
[people/mcb30/gpxe.git] / contrib / ppmtoansi / ppmtoansi.c
1 #include <getopt.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4
5 static int palette[8][3] = {
6 /* black,        red,          green,        yellow,      */ 
7   {  0,  0,  0},{255,  0,  0},{  0,255,  0},{255,255,  0},
8   {  0,  0,255},{255,  0,255},{  0,255,255},{255,255,255}};
9 /* blue,         magenta,      cyan,         white        */
10
11 static struct trans {
12   struct trans *next;
13   int          idx,r,g,b;
14 } *trans = NULL;
15
16 static int skipcomment(FILE *fp)
17 {
18   int ch;
19   
20   for (;;) {
21     ch = getc(fp);
22     if (ch != '#')
23       return(ch);
24     while (ch != '\n' && ch != EOF)
25       ch = getc(fp); }
26 }
27
28 static int readentry(FILE *fp,int format,int depth)
29 {
30   int ch,i = 0;
31   
32   if (format == '3') {
33     while ((ch = getc(fp)) == ' ' || ch == '\t' || ch == '\r' || ch == '\n');
34     if (ch < '0' || ch > '9') {
35     error:
36       fprintf(stderr,"Format error in input file\n");
37       exit(1); }
38     for (; ch >= '0' && ch <= '9'; ch = getc(fp))
39       i = 10*i + ch - '0'; }
40   else {
41     if ((i = getc(fp)) > depth || i < 0)
42       goto error; }
43   return((i*256)/(depth+1));
44 }
45
46 static void packpixel(char *data,int c)
47 {
48   int i = 0, n = 0;
49
50   while (c--) {
51     i = (i << 3) | (*data++ & 0x7);
52     if ((n += 3) >= 8)
53       putchar((i >> (n -= 8)) & 0xFF); }
54   if (n)
55     putchar(i << (8 - n));
56   return;
57 }
58
59 static int dg(int i)
60 {
61   int d;
62
63   for (d = 0; i; d++, i /= 10);
64   return(d);
65 }
66
67 static char *i2s(char *buf,int i)
68 {
69 /*  if (!i)
70     *buf = '\000';
71   else*/
72     sprintf(buf,"%d",i);
73   return(buf);
74 }
75
76 static void flushdata(int x,int y,int c,char *data)
77 {
78   char b1[10],b2[10],b3[10],b4[10];
79   int i,j,rle,v;
80
81   for (i = j = v = 0; i < c; ) {
82     for (rle = 0; i+rle < c && data[i] == data[i+rle]; rle++);
83     if (rle > (i != j ? (v ? 4 : 6) : 0) +
84               ((v || (i != j)) ? 4+dg(rle)+dg(data[i])
85                                : 6+dg(x+i)+dg(y)+dg(rle)+dg(data[i]))) {
86       if (i != j) {
87         if (v)
88           printf("\e[%s-",i2s(b1,i-j));
89         else
90           printf("\e[%s;%s;%s-",i2s(b1,x+j),i2s(b2,y),i2s(b3,i-j));
91         packpixel(data+j,i-j); }
92       if (v++ || (i != j))
93         printf("\e[%s;%s+",i2s(b1,rle),i2s(b2,data[i]));
94       else
95         printf("\e[%s;%s;%s;%s+",i2s(b1,x+i),i2s(b2,y),
96                i2s(b3,rle),i2s(b4,data[i]));
97       j = i += rle; }
98     else
99       i++; }
100   if (j != c) {
101     if (v)
102       printf("\e[%s-",i2s(b1,c-j));
103     else
104       printf("\e[%s;%s;%s-",i2s(b1,x+j),i2s(b2,y),i2s(b3,c-j));
105     packpixel(data+j,c-j); }
106   return;
107 }
108
109 int main(int argc,char *argv[])
110 {
111   extern int optind;
112   extern char *optarg;
113   FILE *infile = NULL;
114   int  ch,i,j,dist,idx;
115   int  format,width,height,depth;
116   int  bg = 0,bgred = 0,bggreen = 0,bgblue = 0;
117   int  xoffset = 0,yoffset = 0;
118   int  w,h,r,g,b,c;
119   struct trans *tp;
120   char *buf;
121   
122   while ((i = getopt(argc,argv,"b:t:x:y:")) >= 0) switch(i) {
123   case 'b':
124     bg++;
125     for (i = bgred = 0; optarg[i] >= '0' && optarg[i] <= '9';
126          bgred = 10*bgred + optarg[i++] - '0');
127     if (optarg[i++] != '/')
128       goto usage;
129     for (bggreen = 0; optarg[i] >= '0' && optarg[i] <= '9';
130          bggreen = 10*bggreen + optarg[i++] - '0');
131     if (optarg[i++] != '/')
132       goto usage;
133     for (bgblue = 0; optarg[i] >= '0' && optarg[i] <= '9';
134          bgblue = 10*bgblue + optarg[i++] - '0');
135     if (optarg[i])
136     goto usage;
137     break;
138   case 't':
139     if ((tp = malloc(sizeof(struct trans))) == NULL)
140       goto usage;
141     for (i = tp->r = 0; optarg[i] >= '0' && optarg[i] <= '9';
142          tp->r = 10*tp->r + optarg[i++] - '0');
143     if (optarg[i++] != '/')
144       goto usage;
145     for (tp->g = 0; optarg[i] >= '0' && optarg[i] <= '9';
146          tp->g = 10*tp->g + optarg[i++] - '0');
147     if (optarg[i++] != '/')
148       goto usage;
149     for (tp->b = 0; optarg[i] >= '0' && optarg[i] <= '9';
150          tp->b = 10*tp->b + optarg[i++] - '0');
151     if (optarg[i++] != ':')
152       goto usage;
153     if (optarg[i] == '-') {
154       j = -1; i++; }
155     else j = 1;
156     for (tp->idx = 0; optarg[i] >= '0' && optarg[i] <= '9';
157          tp->idx = 10*tp->idx + optarg[i++] - '0');
158     tp->idx *= j;
159     if (tp->idx < -1 || tp->idx >= 8)
160       goto usage;
161     if (optarg[i]) 
162       goto usage;
163     tp->next = trans;
164     trans    = tp;
165     break;
166   case 'x':
167     for (i = xoffset = 0; optarg[i] >= '0' && optarg[i] <= '9';
168          xoffset = 10*xoffset + optarg[i++] - '0');
169     if (optarg[i])
170       goto usage;
171     break;
172   case 'y':
173     for (i = yoffset = 0; optarg[i] >= '0' && optarg[i] <= '9';
174          yoffset = 10*yoffset + optarg[i++] - '0');
175     if (optarg[i])
176       goto usage;
177     break;
178   default:
179   usage:
180     fprintf(stderr,"Usage: %s [-b r/g/b] [-t r/g/b:idx] "
181             "[-x offset] [-y offset] [ppmfile]\n",argv[0]);
182     exit(1); }
183   if (argc-optind == 0)
184     infile = stdin;
185   else if (argc-optind == 1)
186     infile = fopen(argv[optind],"r");
187   if (!infile)
188     goto usage;
189   if ((ch = skipcomment(infile)) != 'P' ||
190       ((format = getc(infile)) != '3' && format != '6') ||
191       ((ch = getc(infile)) != '\n' && ch != '\r' && getc(infile) != '\n'))
192     goto usage;
193   for (width = 0; (ch = skipcomment(infile)) >= '0' && ch <= '9';
194        width = 10*width + ch - '0');
195   while (ch == ' ') ch = getc(infile);
196   for (height = 0; ch >= '0' && ch <= '9'; ch = getc(infile))
197     height = 10*height + ch - '0'; 
198   if (ch != '\n' && ch != '\r' && getc(infile) != '\n')
199     goto usage;
200   for (depth = 0; (ch = skipcomment(infile)) >= '0' && ch <= '9';
201        depth = 10*depth + ch - '0');
202   if (ch != '\n' && ch != '\r' && getc(infile) != '\n')
203     goto usage;
204   if (!width || !height || !depth /* || depth > 255 */)
205     goto usage;
206   if ((buf = malloc(width)) == NULL)
207     goto usage;
208   for (h = 0; h < height; h++) {
209     for (w = c = 0; w < width; w++) {
210       r = readentry(infile,format,depth);
211       g = readentry(infile,format,depth);
212       b = readentry(infile,format,depth);
213       idx = 255;
214       if (bg && bgred == r &&
215           bggreen == g && bgblue == b)
216         idx = -1;
217       else for (tp = trans; tp; tp = tp->next)
218         if (tp->r == r && tp->g == g && tp->b == b) {
219           idx = tp->idx;
220           break; }
221       if (idx == 255)
222         for (idx = -1, dist = 3*255*255, i = 8; i--;)
223           if ((j = (r-palette[i][0])*(r-palette[i][0]) +
224                    (g-palette[i][1])*(g-palette[i][1]) +
225                    (b-palette[i][2])*(b-palette[i][2])) < dist) {
226             dist = j; idx = i; }
227       if (idx >= 0)
228         buf[c++] = idx;
229       else if (c) {
230         flushdata(w-c+xoffset,h+yoffset,c,buf);
231         c = 0; } }
232     if (c)
233       flushdata(w-c+xoffset,h+yoffset,c,buf); }
234   exit(0);
235 }