raw.c

   1 #include "image.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   unsigned char *inbuffer;
  58   unsigned char *ilbuffer;
  59   unsigned char *exbuffer;
  60
  61   int inbuflen,ilbuflen,exbuflen;
  62
  63   io_glue_commit_types(ig);
  64   mm_log((1, "i_readraw(ig %p,x %d,y %d,datachannels %d,storechannels %d,intrl %d)\n",
  65           ig, x, y, datachannels, storechannels, intrl));
  66
  67   im = i_img_empty_ch(NULL,x,y,storechannels);
  68
  69   inbuflen = im->xsize*datachannels;
  70   ilbuflen = inbuflen;
  71   exbuflen = im->xsize*storechannels;
  72   inbuffer = (unsigned char*)mymalloc(inbuflen);
  73   mm_log((1,"inbuflen: %d, ilbuflen: %d, exbuflen: %d.\n",inbuflen,ilbuflen,exbuflen));
  74
  75   if (intrl==0) ilbuffer = inbuffer;
  76   else ilbuffer=mymalloc(inbuflen);
  77
  78   if (datachannels==storechannels) exbuffer=ilbuffer;
  79   else exbuffer= mymalloc(exbuflen);
  80
  81   k=0;
  82   while( k<im->ysize ) {
  83     rc = ig->readcb(ig, inbuffer, inbuflen);
  84     if (rc != inbuflen) { fprintf(stderr,"Premature end of file.\n"); exit(2); }
  85     interleave(inbuffer,ilbuffer,im->xsize,datachannels);
  86     expandchannels(ilbuffer,exbuffer,im->xsize,datachannels,storechannels);
  87     /* FIXME: Do we ever want to save to a virtual image? */
  88     memcpy(&(im->idata[im->xsize*storechannels*k]),exbuffer,exbuflen);
  89     k++;
  90   }
  91
  92   myfree(inbuffer);
  93   if (intrl != 0) myfree(ilbuffer);
  94   if (datachannels != storechannels) myfree(exbuffer);
  95   return im;
  96 }
  97
  98
  99
 100 undef_int
 101 i_writeraw_wiol(i_img* im, io_glue *ig) {
 102   int rc;
 103
 104   io_glue_commit_types(ig);
 105   i_clear_error();
 106   mm_log((1,"writeraw(im %p,ig %p)\n", im, ig));
 107
 108   if (im == NULL) { mm_log((1,"Image is empty\n")); return(0); }
 109   if (!im->virtual) {
 110     rc = ig->writecb(ig,im->idata,im->bytes);
 111     if (rc != im->bytes) {
 112       i_push_error(errno, "Could not write to file");
 113       mm_log((1,"i_writeraw: Couldn't write to file\n"));
 114       return(0);
 115     }
 116   } else {
 117     int y;
 118
 119     if (im->type == i_direct_type) {
 120       /* just save it as 8-bits, maybe support saving higher bit count
 121          raw images later */
 122       int line_size = im->xsize * im->channels;
 123       unsigned char *data = mymalloc(line_size);
 124       if (data) {
 125         int y = 0;
 126         rc = line_size;
 127         while (rc == line_size && y < im->ysize) {
 128           i_gsamp(im, 0, im->xsize, y, data, NULL, im->channels);
 129           rc = ig->writecb(ig, data, line_size);
 130           ++y;
 131         }
 132       } else {
 133         i_push_error(0, "Out of memory");
 134         return 0;
 135       }
 136       if (rc != line_size) {
 137         i_push_error(errno, "write error");
 138         return 0;
 139       }
 140     } else {
 141       /* paletted image - assumes the caller puts the palette somewhere
 142          else
 143       */
 144       int line_size = sizeof(i_palidx) * im->xsize;
 145       i_palidx *data = mymalloc(sizeof(i_palidx) * im->xsize);
 146       if (data) {
 147         int y = 0;
 148         rc = line_size;
 149         while (rc == line_size && y < im->ysize) {
 150           i_gpal(im, 0, im->xsize, y, data);
 151           rc = ig->writecb(ig, data, line_size);
 152           ++y;
 153         }
 154         myfree(data);
 155       } else {
 156         i_push_error(0, "Out of memory");
 157         return 0;
 158       }
 159       if (rc != line_size) {
 160         i_push_error(errno, "write error");
 161         return 0;
 162       }
 163     }
 164   }
 165   return(1);
 166 }