- reading a raw image no longer exits on a short read or read error,
[imager.git] / raw.c
1 #include "imager.h"
2 #include <stdio.h>
3 #include "iolayer.h"
4 #ifndef _MSC_VER
5 #include <unistd.h>
6 #endif
7 #include <string.h>
8 #include <errno.h>
9
10
11
12 /*
13
14  Image loader for raw files.
15
16  This is a barebones raw loader...
17
18              fd: filedescriptor
19               x: xsize
20               y: ysize
21    datachannels: the number of channels the file contains
22   storechannels: the bitmap of channels we will read
23           intrl: interlace flag,
24                        0 = sample interleaving
25                        1 = line interleaving
26                        2 = image interleaving
27
28 */
29
30 static
31 void
32 interleave(unsigned char *inbuffer,unsigned char *outbuffer,int rowsize,int channels) {
33   int ch,ind,i;
34   i=0;
35   if (inbuffer == outbuffer) return; /* Check if data is already in interleaved format */
36   for (ind=0; ind<rowsize; ind++) 
37     for (ch=0; ch<channels; ch++) 
38       outbuffer[i++] = inbuffer[rowsize*ch+ind]; 
39 }
40
41 static
42 void
43 expandchannels(unsigned char *inbuffer, unsigned char *outbuffer, 
44                int chunks, int datachannels, int storechannels) {
45   int ch,i;
46   if (inbuffer == outbuffer) return; /* Check if data is already in expanded format */
47   for(ch=0; ch<chunks; ch++) 
48     for (i=0; i<storechannels; i++) 
49       outbuffer[ch*storechannels+i] = inbuffer[ch*datachannels+i];
50 }
51
52 i_img *
53 i_readraw_wiol(io_glue *ig, int x, int y, int datachannels, int storechannels, int intrl) {
54   i_img* im;
55   int rc,k;
56
57   i_clear_error();
58   
59   unsigned char *inbuffer;
60   unsigned char *ilbuffer;
61   unsigned char *exbuffer;
62   
63   int inbuflen,ilbuflen,exbuflen;
64
65   io_glue_commit_types(ig);
66   mm_log((1, "i_readraw(ig %p,x %d,y %d,datachannels %d,storechannels %d,intrl %d)\n",
67           ig, x, y, datachannels, storechannels, intrl));
68   
69   im = i_img_empty_ch(NULL,x,y,storechannels);
70   if (!im)
71     return NULL;
72   
73   inbuflen = im->xsize*datachannels;
74   ilbuflen = inbuflen;
75   exbuflen = im->xsize*storechannels;
76   inbuffer = (unsigned char*)mymalloc(inbuflen);
77   mm_log((1,"inbuflen: %d, ilbuflen: %d, exbuflen: %d.\n",inbuflen,ilbuflen,exbuflen));
78
79   if (intrl==0) ilbuffer = inbuffer; 
80   else ilbuffer=mymalloc(inbuflen);
81
82   if (datachannels==storechannels) exbuffer=ilbuffer; 
83   else exbuffer= mymalloc(exbuflen);
84   
85   k=0;
86   while( k<im->ysize ) {
87     rc = ig->readcb(ig, inbuffer, inbuflen);
88     if (rc != inbuflen) { 
89       if (rc < 0)
90         i_push_error(0, "error reading file");
91       else
92         i_push_error(0, "premature end of file");
93       i_img_destroy(im);
94       myfree(inbuffer);
95       if (intrl != 0) myfree(ilbuffer);
96       if (datachannels != storechannels) myfree(exbuffer);
97       return NULL;
98     }
99     interleave(inbuffer,ilbuffer,im->xsize,datachannels);
100     expandchannels(ilbuffer,exbuffer,im->xsize,datachannels,storechannels);
101     /* FIXME: Do we ever want to save to a virtual image? */
102     memcpy(&(im->idata[im->xsize*storechannels*k]),exbuffer,exbuflen);
103     k++;
104   }
105
106   myfree(inbuffer);
107   if (intrl != 0) myfree(ilbuffer);
108   if (datachannels != storechannels) myfree(exbuffer);
109
110   i_tags_add(&im->tags, "i_format", 0, "raw", -1, 0);
111
112   return im;
113 }
114
115
116
117 undef_int
118 i_writeraw_wiol(i_img* im, io_glue *ig) {
119   int rc;
120
121   io_glue_commit_types(ig);
122   i_clear_error();
123   mm_log((1,"writeraw(im %p,ig %p)\n", im, ig));
124   
125   if (im == NULL) { mm_log((1,"Image is empty\n")); return(0); }
126   if (!im->virtual) {
127     rc = ig->writecb(ig,im->idata,im->bytes);
128     if (rc != im->bytes) { 
129       i_push_error(errno, "Could not write to file");
130       mm_log((1,"i_writeraw: Couldn't write to file\n")); 
131       return(0);
132     }
133   } else {
134     if (im->type == i_direct_type) {
135       /* just save it as 8-bits, maybe support saving higher bit count
136          raw images later */
137       int line_size = im->xsize * im->channels;
138       unsigned char *data = mymalloc(line_size);
139
140       int y = 0;
141       rc = line_size;
142       while (rc == line_size && y < im->ysize) {
143         i_gsamp(im, 0, im->xsize, y, data, NULL, im->channels);
144         rc = ig->writecb(ig, data, line_size);
145         ++y;
146       }
147       if (rc != line_size) {
148         i_push_error(errno, "write error");
149         return 0;
150       }
151       myfree(data);
152     } else {
153       /* paletted image - assumes the caller puts the palette somewhere 
154          else
155       */
156       int line_size = sizeof(i_palidx) * im->xsize;
157       i_palidx *data = mymalloc(sizeof(i_palidx) * im->xsize);
158
159       int y = 0;
160       rc = line_size;
161       while (rc == line_size && y < im->ysize) {
162         i_gpal(im, 0, im->xsize, y, data);
163         rc = ig->writecb(ig, data, line_size);
164         ++y;
165       }
166       myfree(data);
167       if (rc != line_size) {
168         i_push_error(errno, "write error");
169         return 0;
170       }
171     }
172   }
173
174   ig->closecb(ig);
175
176   return(1);
177 }