开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 游戏引擎 > 查看源码
savepng.cpp
资源名称：hge181.rar　[点击查看]
上传用户：jnfxsk
上传日期：2022-06-16
资源大小：3675k
文件大小：2k
源码类别：

游戏引擎

开发平台：

Visual C++

savepng.cpp：源码内容
  2. #include "fonted.h"
  3. #include "libPNGpng.h"
  5. class PNGError {};
  7. void WarningCallback(png_structp png_ptr,
  8.  png_const_charp msg)
  9. {
  10. hge->System_Log("LIBPNG Warning: %s", msg);
  11. }
  13. void ErrorCallback(png_structp png_ptr,
  14.    png_const_charp msg)
  15. {
  16. hge->System_Log("LIBPNG Error: %s", msg);
  17. throw PNGError();
  18. }
  20. bool Write32BitPNG(FILE* fp, void* pBits, bool bNeedAlpha, int nWidth, int nHeight)
  21. {
  22. png_structp png_ptr = 0;
  23. png_infop info_ptr = 0;
  24. try 
  25. {
  26. // create png structs
  27. png_ptr = png_create_write_struct
  28. (PNG_LIBPNG_VER_STRING, 0,
  29. ErrorCallback, WarningCallback);
  30. if (png_ptr == 0) return false;
  32. info_ptr = png_create_info_struct(png_ptr);
  33. if (info_ptr == 0) return false;
  35. // init i/o
  36. png_init_io(png_ptr, fp);
  38. // setup params
  39. if ( bNeedAlpha )
  40. {
  41. png_set_IHDR( png_ptr, info_ptr, nWidth, nHeight, 
  42. 8, 
  43. PNG_COLOR_TYPE_RGB_ALPHA, 
  44. PNG_INTERLACE_NONE,
  45. PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
  46. }
  47. else
  48. {
  49. png_set_IHDR( png_ptr, info_ptr, nWidth, nHeight, 
  50. 8, 
  51. PNG_COLOR_TYPE_RGB,
  52. PNG_INTERLACE_NONE,
  53. PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
  54. }
  56. png_write_info(png_ptr, info_ptr);
  58. if ( !bNeedAlpha )
  59. png_set_filler(png_ptr, 0, PNG_FILLER_AFTER); // strip alpha
  61. png_set_bgr(png_ptr); // switch to little-endian notation
  63. // writing
  64. unsigned char* pSrc = (unsigned char*)pBits;
  65. for ( int i = 0; i < nHeight; i++, pSrc+= nWidth*4 )
  66. {
  67. png_write_row( png_ptr, pSrc );
  68. }
  70. png_write_end(png_ptr, info_ptr);
  71. } 
  72. catch(PNGError)
  73. {
  74. png_destroy_write_struct(&png_ptr, (info_ptr == 0) ? NULL : &info_ptr);
  75. return false;
  76. }
  78. // cleanup
  79. png_destroy_write_struct( &png_ptr, (info_ptr == 0) ? NULL : &info_ptr);
  80. return true;
  81. }
  84. bool SavePNG(HTEXTURE tex, char *filename)
  85. {
  86. FILE *fp=0;
  87. DWORD *pTexData=0;
  88. int nWidth, nHeight;
  90. try
  91. {
  92. fp=fopen(filename,"wb");
  93. if(!fp) throw PNGError();
  95. nWidth=hge->Texture_GetWidth(tex);
  96. nHeight=hge->Texture_GetHeight(tex);
  97. pTexData=hge->Texture_Lock(tex, true);
  98. if(!pTexData) throw PNGError();
  100. if(!Write32BitPNG(fp, pTexData, true, nWidth, nHeight)) throw PNGError();
  101. }
  102. catch(PNGError)
  103. {
  104. if(fp) fclose(fp);
  105. if(pTexData) hge->Texture_Unlock(tex);
  106. remove(filename);
  107. return false;
  108. }
  110. fclose(fp);
  111. hge->Texture_Unlock(tex);
  112. return true;
  113. }