image.c
上传用户:xk288cn
上传日期:2007-05-28
资源大小:4876k
文件大小:5k
源码类别:

GIS编程

开发平台:

Visual C++

  1. /* Lifted from libtk/image.c */
  2. /* This code lets your read in SGI .rgb files. */
  3. #include <stdio.h>
  4. #include <stdlib.h> 
  5. #include <string.h>
  6. #include "image.h"
  7. #ifndef SEEK_SET
  8. #  define SEEK_SET 0
  9. #endif
  10. /******************************************************************************/
  11. typedef unsigned int GLuint;  /* Should be 32-bit value. */
  12. typedef int GLint;  /* Should be 32-bit value. */
  13. typedef struct _rawImageRec {
  14.     unsigned short imagic;
  15.     unsigned short type;
  16.     unsigned short dim;
  17.     unsigned short sizeX, sizeY, sizeZ;
  18.     unsigned long min, max;
  19.     unsigned long wasteBytes;
  20.     char name[80];
  21.     unsigned long colorMap;
  22.     FILE *file;
  23.     unsigned char *tmp, *tmpR, *tmpG, *tmpB;
  24.     unsigned long rleEnd;
  25.     GLuint *rowStart;
  26.     GLint *rowSize;
  27. } rawImageRec;
  28. /******************************************************************************/
  29. static void ConvertShort(unsigned short *array, unsigned int length)
  30. {
  31.     unsigned short b1, b2;
  32.     unsigned char *ptr;
  33.     ptr = (unsigned char *)array;
  34.     while (length--) {
  35. b1 = *ptr++;
  36. b2 = *ptr++;
  37. *array++ = (b1 << 8) | (b2);
  38.     }
  39. }
  40. static void ConvertUint(GLuint *array, unsigned int length)
  41. {
  42.     unsigned int b1, b2, b3, b4;
  43.     unsigned char *ptr;
  44.     ptr = (unsigned char *)array;
  45.     while (length--) {
  46. b1 = *ptr++;
  47. b2 = *ptr++;
  48. b3 = *ptr++;
  49. b4 = *ptr++;
  50. *array++ = (b1 << 24) | (b2 << 16) | (b3 << 8) | (b4);
  51.     }
  52. }
  53. static rawImageRec *RawImageOpen(char *fileName)
  54. {
  55.     union {
  56. int testWord;
  57. char testByte[4];
  58.     } endianTest;
  59.     rawImageRec *raw;
  60.     int swapFlag;
  61.     int x;
  62.     endianTest.testWord = 1;
  63.     if (endianTest.testByte[0] == 1) {
  64. swapFlag = 1;
  65.     } else {
  66. swapFlag = 0;
  67.     }
  68.     raw = (rawImageRec *)malloc(sizeof(rawImageRec));
  69.     if (raw == NULL) {
  70. fprintf(stderr, "Out of memory!n");
  71. exit(1);
  72.     }
  73.     if ((raw->file = fopen(fileName, "rb")) == NULL) {
  74. perror(fileName);
  75. exit(1);
  76.     }
  77.     fread(raw, 1, 12, raw->file);
  78.     if (swapFlag) {
  79. ConvertShort(&raw->imagic, 6);
  80.     }
  81.     raw->tmp = (unsigned char *)malloc(raw->sizeX*256);
  82.     raw->tmpR = (unsigned char *)malloc(raw->sizeX*256);
  83.     raw->tmpG = (unsigned char *)malloc(raw->sizeX*256);
  84.     raw->tmpB = (unsigned char *)malloc(raw->sizeX*256);
  85.     if (raw->tmp == NULL || raw->tmpR == NULL || raw->tmpG == NULL ||
  86. raw->tmpB == NULL) {
  87. fprintf(stderr, "Out of memory!n");
  88. exit(1);
  89.     }
  90.     if ((raw->type & 0xFF00) == 0x0100) {
  91. x = raw->sizeY * raw->sizeZ * (int) sizeof(GLuint);
  92. raw->rowStart = (GLuint *)malloc(x);
  93. raw->rowSize = (GLint *)malloc(x);
  94. if (raw->rowStart == NULL || raw->rowSize == NULL) {
  95.     fprintf(stderr, "Out of memory!n");
  96.     exit(1);
  97. }
  98. raw->rleEnd = 512 + (2 * x);
  99. fseek(raw->file, 512, SEEK_SET);
  100. fread(raw->rowStart, 1, x, raw->file);
  101. fread(raw->rowSize, 1, x, raw->file);
  102. if (swapFlag) {
  103.     ConvertUint(raw->rowStart,
  104.         (unsigned int) (x/sizeof(GLuint)));
  105.     ConvertUint((GLuint *)raw->rowSize,
  106.         (unsigned int) (x/sizeof(GLint)));
  107. }
  108.     }
  109.     return raw;
  110. }
  111. static void RawImageClose(rawImageRec *raw)
  112. {
  113.     fclose(raw->file);
  114.     free(raw->tmp);
  115.     free(raw->tmpR);
  116.     free(raw->tmpG);
  117.     free(raw->tmpB);
  118.     free(raw);
  119. }
  120. static void RawImageGetRow(rawImageRec *raw, unsigned char *buf, int y, int z)
  121. {
  122.     unsigned char *iPtr, *oPtr, pixel;
  123.     int count;
  124.     if ((raw->type & 0xFF00) == 0x0100) {
  125. fseek(raw->file, (long) raw->rowStart[y+z*raw->sizeY], SEEK_SET);
  126. fread(raw->tmp, 1, (unsigned int)raw->rowSize[y+z*raw->sizeY],
  127.       raw->file);
  128. iPtr = raw->tmp;
  129. oPtr = buf;
  130. for (;;) {
  131.     pixel = *iPtr++;
  132.     count = (int)(pixel & 0x7F);
  133.     if (!count) {
  134. return;
  135.     }
  136.     if (pixel & 0x80) {
  137. while (count--) {
  138.     *oPtr++ = *iPtr++;
  139. }
  140.     } else {
  141. pixel = *iPtr++;
  142. while (count--) {
  143.     *oPtr++ = pixel;
  144. }
  145.     }
  146. }
  147.     } else {
  148. fseek(raw->file, 512+(y*raw->sizeX)+(z*raw->sizeX*raw->sizeY),
  149.       SEEK_SET);
  150. fread(buf, 1, raw->sizeX, raw->file);
  151.     }
  152. }
  153. static void RawImageGetData(rawImageRec *raw, RGBImageRec *final)
  154. {
  155.     unsigned char *ptr;
  156.     int i, j;
  157.     final->data = (unsigned char *)malloc((raw->sizeX+1)*(raw->sizeY+1)*4);
  158.     if (final->data == NULL) {
  159. fprintf(stderr, "Out of memory!n");
  160. exit(1);
  161.     }
  162.     ptr = final->data;
  163.     for (i = 0; i < (int)(raw->sizeY); i++) {
  164. RawImageGetRow(raw, raw->tmpR, i, 0);
  165. RawImageGetRow(raw, raw->tmpG, i, 1);
  166. RawImageGetRow(raw, raw->tmpB, i, 2);
  167. for (j = 0; j < (int)(raw->sizeX); j++) {
  168.     *ptr++ = *(raw->tmpR + j);
  169.     *ptr++ = *(raw->tmpG + j);
  170.     *ptr++ = *(raw->tmpB + j);
  171. }
  172.     }
  173. }
  174. RGBImageRec *RGBImageLoad(char *fileName)
  175. {
  176.     rawImageRec *raw;
  177.     RGBImageRec *final;
  178.     raw = RawImageOpen(fileName);
  179.     final = (RGBImageRec *)malloc(sizeof(RGBImageRec));
  180.     if (final == NULL) {
  181. fprintf(stderr, "Out of memory!n");
  182. exit(1);
  183.     }
  184.     final->sizeX = raw->sizeX;
  185.     final->sizeY = raw->sizeY;
  186.     RawImageGetData(raw, final);
  187.     RawImageClose(raw);
  188.     return final;
  189. }
  190. /******************************************************************************/