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 }