add github action for CI on OS X
[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 (not implemented)
27
28 */
29
30 static
31 void
32 interleave(unsigned char *inbuffer,unsigned char *outbuffer,i_img_dim rowsize,int channels) {
33   i_img_dim ind,i;
34   int ch;
35   i=0;
36   if (inbuffer == outbuffer) return; /* Check if data is already in interleaved format */
37   for (ind=0; ind<rowsize; ind++) 
38     for (ch=0; ch<channels; ch++) 
39       outbuffer[i++] = inbuffer[rowsize*ch+ind]; 
40 }
41
42 static
43 void
44 expandchannels(unsigned char *inbuffer, unsigned char *outbuffer, 
45                i_img_dim xsize, int datachannels, int storechannels) {
46   i_img_dim x;
47   int ch;
48   int copy_chans = storechannels > datachannels ? datachannels : storechannels;
49   if (inbuffer == outbuffer)
50     return; /* Check if data is already in expanded format */
51   for(x = 0; x < xsize; x++) {
52     for (ch = 0; ch < copy_chans; ch++) 
53       outbuffer[x*storechannels+ch] = inbuffer[x*datachannels+ch];
54     for (; ch < storechannels; ch++)
55       outbuffer[x*storechannels+ch] = 0;
56   }
57 }
58
59 i_img *
60 i_readraw_wiol(io_glue *ig, i_img_dim x, i_img_dim y, int datachannels, int storechannels, int intrl) {
61   i_img* im;
62   ssize_t rc;
63   i_img_dim k;
64
65   unsigned char *inbuffer;
66   unsigned char *ilbuffer;
67   unsigned char *exbuffer;
68   
69   size_t inbuflen,ilbuflen,exbuflen;
70
71   i_clear_error();
72   
73   mm_log((1, "i_readraw(ig %p,x %" i_DF ",y %" i_DF ",datachannels %d,storechannels %d,intrl %d)\n",
74           ig, i_DFc(x), i_DFc(y), datachannels, storechannels, intrl));
75
76   if (intrl != 0 && intrl != 1) {
77     i_push_error(0, "raw_interleave must be 0 or 1");
78     return NULL;
79   }
80   if (storechannels < 1 || storechannels > 4) {
81     i_push_error(0, "raw_storechannels must be between 1 and 4");
82     return NULL;
83   }
84   
85   im = i_img_empty_ch(NULL,x,y,storechannels);
86   if (!im)
87     return NULL;
88   
89   inbuflen = im->xsize*datachannels;
90   ilbuflen = inbuflen;
91   exbuflen = im->xsize*storechannels;
92   inbuffer = (unsigned char*)mymalloc(inbuflen);
93   mm_log((1,"inbuflen: %ld, ilbuflen: %ld, exbuflen: %ld.\n",
94           (long)inbuflen, (long)ilbuflen, (long)exbuflen));
95
96   if (intrl==0) ilbuffer = inbuffer; 
97   else ilbuffer=mymalloc(inbuflen);
98
99   if (datachannels==storechannels) exbuffer=ilbuffer; 
100   else exbuffer= mymalloc(exbuflen);
101   
102   k=0;
103   while( k<im->ysize ) {
104     rc = i_io_read(ig, inbuffer, inbuflen);
105     if (rc != inbuflen) { 
106       if (rc < 0)
107         i_push_error(0, "error reading file");
108       else
109         i_push_error(0, "premature end of file");
110       i_img_destroy(im);
111       myfree(inbuffer);
112       if (intrl != 0) myfree(ilbuffer);
113       if (datachannels != storechannels) myfree(exbuffer);
114       return NULL;
115     }
116     interleave(inbuffer,ilbuffer,im->xsize,datachannels);
117     expandchannels(ilbuffer,exbuffer,im->xsize,datachannels,storechannels);
118     /* FIXME: Do we ever want to save to a virtual image? */
119     memcpy(&(im->idata[im->xsize*storechannels*k]),exbuffer,exbuflen);
120     k++;
121   }
122
123   myfree(inbuffer);
124   if (intrl != 0) myfree(ilbuffer);
125   if (datachannels != storechannels) myfree(exbuffer);
126
127   i_tags_add(&im->tags, "i_format", 0, "raw", -1, 0);
128
129   return im;
130 }
131
132
133
134 undef_int
135 i_writeraw_wiol(i_img* im, io_glue *ig) {
136   ssize_t rc;
137
138   i_clear_error();
139   mm_log((1,"writeraw(im %p,ig %p)\n", im, ig));
140   
141   if (im == NULL) { mm_log((1,"Image is empty\n")); return(0); }
142   if (!im->virtual) {
143     rc = i_io_write(ig,im->idata,im->bytes);
144     if (rc != im->bytes) { 
145       i_push_error(errno, "Could not write to file");
146       mm_log((1,"i_writeraw: Couldn't write to file\n")); 
147       return(0);
148     }
149   } else {
150     if (im->type == i_direct_type) {
151       /* just save it as 8-bits, maybe support saving higher bit count
152          raw images later */
153       size_t line_size = im->xsize * im->channels;
154       unsigned char *data = mymalloc(line_size);
155
156       i_img_dim y = 0;
157       rc = line_size;
158       while (rc == line_size && y < im->ysize) {
159         i_gsamp(im, 0, im->xsize, y, data, NULL, im->channels);
160         rc = i_io_write(ig, data, line_size);
161         ++y;
162       }
163       if (rc != line_size) {
164         i_push_error(errno, "write error");
165         return 0;
166       }
167       myfree(data);
168     } else {
169       /* paletted image - assumes the caller puts the palette somewhere 
170          else
171       */
172       size_t line_size = sizeof(i_palidx) * im->xsize;
173       i_palidx *data = mymalloc(sizeof(i_palidx) * im->xsize);
174
175       i_img_dim y = 0;
176       rc = line_size;
177       while (rc == line_size && y < im->ysize) {
178         i_gpal(im, 0, im->xsize, y, data);
179         rc = i_io_write(ig, data, line_size);
180         ++y;
181       }
182       myfree(data);
183       if (rc != line_size) {
184         i_push_error(errno, "write error");
185         return 0;
186       }
187     }
188   }
189
190   if (i_io_close(ig))
191     return 0;
192
193   return(1);
194 }