1 28994509 2004-04-21 devnull #include <u.h>
2 28994509 2004-04-21 devnull #include <libc.h>
3 28994509 2004-04-21 devnull #include <draw.h>
4 28994509 2004-04-21 devnull #include <memdraw.h>
5 28994509 2004-04-21 devnull #include <bio.h>
6 28994509 2004-04-21 devnull #include "imagefile.h"
8 28994509 2004-04-21 devnull #define MAXLINE 70
11 28994509 2004-04-21 devnull * Write data
15 28994509 2004-04-21 devnull writedata(Biobuf *fd, Image *image, Memimage *memimage)
17 28994509 2004-04-21 devnull char *err;
18 28994509 2004-04-21 devnull uchar *data;
19 28994509 2004-04-21 devnull int i, x, y, ndata, depth, col, pix, xmask, pmask;
20 28994509 2004-04-21 devnull ulong chan;
21 28994509 2004-04-21 devnull Rectangle r;
23 28994509 2004-04-21 devnull if(memimage != nil){
24 28994509 2004-04-21 devnull r = memimage->r;
25 28994509 2004-04-21 devnull depth = memimage->depth;
26 28994509 2004-04-21 devnull chan = memimage->chan;
28 28994509 2004-04-21 devnull r = image->r;
29 28994509 2004-04-21 devnull depth = image->depth;
30 28994509 2004-04-21 devnull chan = image->chan;
34 28994509 2004-04-21 devnull * Read image data into memory
35 28994509 2004-04-21 devnull * potentially one extra byte on each end of each scan line
37 28994509 2004-04-21 devnull ndata = Dy(r)*(2+Dx(r)*depth/8);
38 28994509 2004-04-21 devnull data = malloc(ndata);
39 28994509 2004-04-21 devnull if(data == nil)
40 28994509 2004-04-21 devnull return "WritePPM: malloc failed";
41 28994509 2004-04-21 devnull if(memimage != nil)
42 28994509 2004-04-21 devnull ndata = unloadmemimage(memimage, r, data, ndata);
44 28994509 2004-04-21 devnull ndata = unloadimage(image, r, data, ndata);
45 28994509 2004-04-21 devnull if(ndata < 0){
46 28994509 2004-04-21 devnull err = malloc(ERRMAX);
47 28994509 2004-04-21 devnull if(err == nil)
48 28994509 2004-04-21 devnull return "WritePPM: malloc failed";
49 28994509 2004-04-21 devnull snprint(err, ERRMAX, "WriteGIF: %r");
50 28994509 2004-04-21 devnull free(data);
51 28994509 2004-04-21 devnull return err;
54 28994509 2004-04-21 devnull /* Encode and emit the data */
56 28994509 2004-04-21 devnull switch(chan){
57 28994509 2004-04-21 devnull case GREY1:
58 28994509 2004-04-21 devnull case GREY2:
59 28994509 2004-04-21 devnull case GREY4:
60 28994509 2004-04-21 devnull pmask = (1<<depth)-1;
61 28994509 2004-04-21 devnull xmask = 7>>drawlog2[depth];
62 28994509 2004-04-21 devnull for(y=r.min.y; y<r.max.y; y++){
63 28994509 2004-04-21 devnull i = (y-r.min.y)*bytesperline(r, depth);
64 28994509 2004-04-21 devnull for(x=r.min.x; x<r.max.x; x++){
65 28994509 2004-04-21 devnull pix = (data[i]>>depth*((xmask-x)&xmask))&pmask;
66 28994509 2004-04-21 devnull if(((x+1)&xmask) == 0)
68 28994509 2004-04-21 devnull col += Bprint(fd, "%d ", pix);
69 28994509 2004-04-21 devnull if(col >= MAXLINE-(2+1)){
70 28994509 2004-04-21 devnull Bprint(fd, "\n");
73 28994509 2004-04-21 devnull col += Bprint(fd, " ");
77 28994509 2004-04-21 devnull case GREY8:
78 28994509 2004-04-21 devnull for(i=0; i<ndata; i++){
79 28994509 2004-04-21 devnull col += Bprint(fd, "%d ", data[i]);
80 28994509 2004-04-21 devnull if(col >= MAXLINE-(4+1)){
81 28994509 2004-04-21 devnull Bprint(fd, "\n");
84 28994509 2004-04-21 devnull col += Bprint(fd, " ");
87 28994509 2004-04-21 devnull case RGB24:
88 28994509 2004-04-21 devnull for(i=0; i<ndata; i+=3){
89 28994509 2004-04-21 devnull col += Bprint(fd, "%d %d %d", data[i+2], data[i+1], data[i]);
90 28994509 2004-04-21 devnull if(col >= MAXLINE-(4+4+4+1)){
91 28994509 2004-04-21 devnull Bprint(fd, "\n");
94 28994509 2004-04-21 devnull col += Bprint(fd, " ");
98 28994509 2004-04-21 devnull return "WritePPM: can't handle channel type";
101 28994509 2004-04-21 devnull return nil;
106 28994509 2004-04-21 devnull writeppm0(Biobuf *fd, Image *image, Memimage *memimage, Rectangle r, int chan, char *comment)
108 28994509 2004-04-21 devnull char *err;
110 28994509 2004-04-21 devnull switch(chan){
111 28994509 2004-04-21 devnull case GREY1:
112 28994509 2004-04-21 devnull Bprint(fd, "P1\n");
114 28994509 2004-04-21 devnull case GREY2:
115 28994509 2004-04-21 devnull case GREY4:
116 28994509 2004-04-21 devnull case GREY8:
117 28994509 2004-04-21 devnull Bprint(fd, "P2\n");
119 28994509 2004-04-21 devnull case RGB24:
120 28994509 2004-04-21 devnull Bprint(fd, "P3\n");
122 28994509 2004-04-21 devnull default:
123 28994509 2004-04-21 devnull return "WritePPM: can't handle channel type";
126 28994509 2004-04-21 devnull if(comment!=nil && comment[0]!='\0'){
127 28994509 2004-04-21 devnull Bprint(fd, "# %s", comment);
128 28994509 2004-04-21 devnull if(comment[strlen(comment)-1] != '\n')
129 28994509 2004-04-21 devnull Bprint(fd, "\n");
131 28994509 2004-04-21 devnull Bprint(fd, "%d %d\n", Dx(r), Dy(r));
133 28994509 2004-04-21 devnull /* maximum pixel value */
134 28994509 2004-04-21 devnull switch(chan){
135 28994509 2004-04-21 devnull case GREY2:
136 28994509 2004-04-21 devnull Bprint(fd, "%d\n", 3);
138 28994509 2004-04-21 devnull case GREY4:
139 28994509 2004-04-21 devnull Bprint(fd, "%d\n", 15);
141 28994509 2004-04-21 devnull case GREY8:
142 28994509 2004-04-21 devnull case RGB24:
143 28994509 2004-04-21 devnull Bprint(fd, "%d\n", 255);
147 28994509 2004-04-21 devnull err = writedata(fd, image, memimage);
149 28994509 2004-04-21 devnull Bprint(fd, "\n");
150 28994509 2004-04-21 devnull Bflush(fd);
151 28994509 2004-04-21 devnull return err;
155 28994509 2004-04-21 devnull writeppm(Biobuf *fd, Image *image, char *comment)
157 28994509 2004-04-21 devnull return writeppm0(fd, image, nil, image->r, image->chan, comment);
161 28994509 2004-04-21 devnull memwriteppm(Biobuf *fd, Memimage *memimage, char *comment)
163 28994509 2004-04-21 devnull return writeppm0(fd, nil, memimage, memimage->r, memimage->chan, comment);