home *** CD-ROM | disk | FTP | other *** search
/ Mega Top 1 / os2_top1.zip / os2_top1 / APPS / RAYTRACE / RT / FIO.C < prev    next >
C/C++ Source or Header  |  1994-05-22  |  5KB  |  248 lines
  1. /*
  2.  
  3. FIO.C  New bitmap file IO
  4.  
  5. This module uses the Generalised Bitmap Module to do the I/O.
  6. This code is also portable, therefore the whole raytracer still is.
  7.  
  8. */
  9.  
  10. /*...sincludes:0:*/
  11. #include <stdio.h>
  12. #include <ctype.h>
  13. #include <stdlib.h>
  14. #include <stddef.h>
  15. #include <stdarg.h>
  16. #include <string.h>
  17. #include <memory.h>
  18. #include <malloc.h>
  19. #ifdef AIX
  20. #include <unistd.h>
  21. #else
  22. #include <io.h>
  23. #endif
  24. #include <fcntl.h>
  25. #include <sys/types.h>
  26. #include <sys/stat.h>
  27. #include "standard.h"
  28. #include "gbm.h"
  29.  
  30. #ifndef O_BINARY
  31. #define    O_BINARY 0
  32. #endif
  33. /*...e*/
  34.  
  35. typedef struct
  36.     {
  37.     GBM gbm;
  38.     GBMRGB gbmrgb[0x100];
  39.     byte *data;
  40.     int stride;
  41.     int ref_count;
  42.     } BITMAP;
  43.  
  44. /*...sfio_init:0:*/
  45. void fio_init(void)
  46.     {
  47.     gbm_init();
  48.     }
  49. /*...e*/
  50. /*...sfio_deinit:0:*/
  51. void fio_deinit(void)
  52.     {
  53.     gbm_deinit();
  54.     }
  55. /*...e*/
  56. /*...sfio_create_bitmap:0:*/
  57. BITMAP *fio_create_bitmap(int w, int h)
  58.     {
  59.     BITMAP *bitmap;
  60.  
  61.     if ( (bitmap = malloc(sizeof(BITMAP))) == NULL )
  62.         return NULL;
  63.  
  64.     bitmap->gbm.w   = w;
  65.     bitmap->gbm.h   = h;
  66.     bitmap->gbm.bpp = 24;
  67.     bitmap->stride  = ((w * 24 + 31)/32)*4;
  68.     if ( (bitmap->data = malloc(bitmap->stride * h)) == NULL )
  69.         { free(bitmap); return NULL; }
  70.  
  71.     /* Initialse surface to medium grey */
  72.     memset(bitmap->data, 0x80, bitmap->stride * h);
  73.  
  74.     bitmap->ref_count = 1;
  75.  
  76.     return bitmap;
  77.     }
  78. /*...e*/
  79. /*...sfio_copy_bitmap:0:*/
  80. BITMAP *fio_copy_bitmap(BITMAP *bitmap)
  81.     {
  82.     (bitmap->ref_count)++;
  83.     return bitmap;
  84.     }
  85. /*...e*/
  86. /*...sfio_destroy_bitmap:0:*/
  87. void fio_destroy_bitmap(BITMAP *bitmap)
  88.     {
  89.     if ( --(bitmap->ref_count) == 0 )
  90.         {
  91.         free(bitmap->data);
  92.         free(bitmap);
  93.         }
  94.     }
  95. /*...e*/
  96. /*...sfio_read_bitmap:0:*/
  97. BITMAP *fio_read_bitmap(char *fn)
  98.     {
  99.     BITMAP *bitmap;
  100.     char *opt;
  101.     int fd, ft;
  102.  
  103.     if ( (opt = strchr(fn, ',')) != NULL )
  104.         *opt++ = '\0';
  105.     else
  106.         opt = "";
  107.  
  108.     if ( gbm_guess_filetype(fn, &ft) != GBM_ERR_OK )
  109.         return NULL;
  110.  
  111.     if ( (fd = open(fn, O_RDONLY | O_BINARY)) == -1 )
  112.         return NULL;
  113.  
  114.     if ( (bitmap = malloc(sizeof(BITMAP))) == NULL )
  115.         { close(fd); return NULL; }
  116.  
  117.     if ( gbm_read_header(fn, fd, ft, &(bitmap->gbm), opt) != GBM_ERR_OK )
  118.         { free(bitmap); close(fd); return NULL; }
  119.  
  120.     if ( gbm_read_palette(fd, ft, &(bitmap->gbm), bitmap->gbmrgb) != GBM_ERR_OK )
  121.         { free(bitmap); close(fd); return NULL; }
  122.  
  123.     bitmap->stride = ((bitmap->gbm.w * bitmap->gbm.bpp + 31)/32)*4;
  124.     if ( (bitmap->data = malloc(bitmap->stride * bitmap->gbm.h)) == NULL )
  125.         { free(bitmap); close(fd); return NULL; }
  126.  
  127.     if ( gbm_read_data(fd, ft, &(bitmap->gbm), bitmap->data) )
  128.         { free(bitmap); close(fd); return NULL; }
  129.  
  130.     close(fd);
  131.  
  132.     bitmap->ref_count = 1;
  133.  
  134.     return bitmap;
  135.     }
  136. /*...e*/
  137. /*...sfio_write_bitmap:0:*/
  138. BOOLEAN fio_write_bitmap(BITMAP *bitmap, char *fn)
  139.     {
  140.     char *opt;
  141.     int fd, ft;
  142.     GBMFT gbmft;
  143.  
  144.     if ( (opt = strchr(fn, ',')) != NULL )
  145.         *opt++ = '\0';
  146.     else
  147.         opt = "";
  148.  
  149.     if ( gbm_guess_filetype(fn, &ft) != GBM_ERR_OK )
  150.         return FALSE;
  151.  
  152.     gbm_query_filetype(ft, &gbmft);
  153.  
  154.     if ( (gbmft.flags & GBM_FT_W24) == 0 )
  155.         return FALSE;
  156.  
  157.     if ( (fd = open(fn, O_CREAT|O_TRUNC|O_WRONLY|O_BINARY, S_IREAD|S_IWRITE)) == -1 )
  158.         return FALSE;
  159.  
  160.     if ( gbm_write(fn, fd, ft, &(bitmap->gbm), bitmap->gbmrgb,
  161.             bitmap->data, opt) != GBM_ERR_OK )
  162.         { close(fd); unlink(fn); return FALSE; }
  163.  
  164.     close(fd);
  165.  
  166.     return TRUE;
  167.     }
  168. /*...e*/
  169. /*...sfio_get_pixel:0:*/
  170. void fio_get_pixel(
  171.     BITMAP *bitmap,
  172.     int x, int y,
  173.     byte *r, byte *g, byte *b
  174.     )
  175.     {
  176.     byte inx, *data = bitmap->data + y * bitmap->stride;
  177.  
  178.     switch ( bitmap->gbm.bpp )
  179.         {
  180. /*...s1:16:*/
  181. case 1:
  182.     inx = data[x >> 3];
  183.     inx >>= ( 7 - (x & 7) );
  184.     inx &= 1;    
  185.     *b = bitmap->gbmrgb[inx].b;
  186.     *g = bitmap->gbmrgb[inx].g;
  187.     *r = bitmap->gbmrgb[inx].r;
  188.     break;
  189. /*...e*/
  190. /*...s4:16:*/
  191. case 4:
  192.     inx = data[x >> 1];
  193.     if ( x & 1 )
  194.         inx &= 0x0f;
  195.     else
  196.         inx >>= 4;
  197.     *b = bitmap->gbmrgb[inx].b;
  198.     *g = bitmap->gbmrgb[inx].g;
  199.     *r = bitmap->gbmrgb[inx].r;
  200.     break;
  201. /*...e*/
  202. /*...s8:16:*/
  203. case 8:
  204.     inx = data[x];
  205.     *b = bitmap->gbmrgb[inx].b;
  206.     *g = bitmap->gbmrgb[inx].g;
  207.     *r = bitmap->gbmrgb[inx].r;
  208.     break;
  209. /*...e*/
  210. /*...s24:16:*/
  211. case 24:
  212.     data += (x * 3);
  213.     *b = *data++;
  214.     *g = *data++;
  215.     *r = *data;
  216.     break;
  217. /*...e*/
  218.         }
  219.     }
  220. /*...e*/
  221. /*...sfio_set_pixel:0:*/
  222. void fio_set_pixel(
  223.     BITMAP *bitmap,
  224.     int x, int y,
  225.     byte r, byte g, byte b
  226.     )
  227.     {
  228.     int stride = bitmap->stride;
  229.     byte *data = bitmap->data + y * stride + x * 3;
  230.  
  231.     *data++ = b;
  232.     *data++ = g;
  233.     *data   = r;
  234.     }
  235. /*...e*/
  236. /*...sfio_width:0:*/
  237. int fio_width(BITMAP *bitmap)
  238.     {
  239.     return bitmap->gbm.w;
  240.     }
  241. /*...e*/
  242. /*...sfio_height:0:*/
  243. int fio_height(BITMAP *bitmap)
  244.     {
  245.     return bitmap->gbm.h;
  246.     }
  247. /*...e*/
  248.