transfer from internal tree r5311 branches/1.4-gpl
[xonotic/netradiant.git] / plugins / imagepng / plugin.cpp
1 /*\r
2 Copyright (C) 1999-2007 id Software, Inc. and contributors.\r
3 For a list of contributors, see the accompanying CONTRIBUTORS file.\r
4 \r
5 This file is part of GtkRadiant.\r
6 \r
7 GtkRadiant is free software; you can redistribute it and/or modify\r
8 it under the terms of the GNU General Public License as published by\r
9 the Free Software Foundation; either version 2 of the License, or\r
10 (at your option) any later version.\r
11 \r
12 GtkRadiant is distributed in the hope that it will be useful,\r
13 but WITHOUT ANY WARRANTY; without even the implied warranty of\r
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
15 GNU General Public License for more details.\r
16 \r
17 You should have received a copy of the GNU General Public License\r
18 along with GtkRadiant; if not, write to the Free Software\r
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA\r
20 */\r
21 \r
22 // =============================================================================\r
23 // global tables\r
24 \r
25 #include "plugin.h"\r
26 \r
27 _QERFuncTable_1 g_FuncTable;\r
28 _QERFileSystemTable g_FileSystemTable;\r
29 \r
30 // =============================================================================\r
31 // SYNAPSE\r
32 \r
33 #include "synapse.h"\r
34 \r
35 class CSynapseClientImage : public CSynapseClient\r
36 {\r
37 public:\r
38   // CSynapseClient API\r
39   bool RequestAPI(APIDescriptor_t *pAPI);\r
40   const char* GetInfo();\r
41   \r
42   CSynapseClientImage() { }\r
43   virtual ~CSynapseClientImage() { }\r
44 };\r
45 \r
46 CSynapseServer* g_pSynapseServer = NULL;\r
47 CSynapseClientImage g_SynapseClient;\r
48 \r
49 extern "C" CSynapseClient* SYNAPSE_DLL_EXPORT Synapse_EnumerateInterfaces (const char *version, CSynapseServer *pServer)\r
50 {\r
51   if (strcmp(version, SYNAPSE_VERSION))\r
52   {\r
53     Syn_Printf("ERROR: synapse API version mismatch: should be '" SYNAPSE_VERSION "', got '%s'\n", version);\r
54     return NULL;\r
55   }\r
56   g_pSynapseServer = pServer;\r
57   g_pSynapseServer->IncRef();\r
58   Set_Syn_Printf(g_pSynapseServer->Get_Syn_Printf());\r
59 \r
60   g_SynapseClient.AddAPI(IMAGE_MAJOR, "png", sizeof(_QERPlugImageTable));\r
61   g_SynapseClient.AddAPI(RADIANT_MAJOR, NULL, sizeof(_QERFuncTable_1), SYN_REQUIRE, &g_FuncTable);\r
62   // NOTE: if imagepng starts being used for non "VFS" "pk3" config, need to add a dynamic config chunk\r
63   // see:\r
64   // http://zerowing.idsoftware.com/bugzilla/show_bug.cgi?id=794\r
65   // http://zerowing.idsoftware.com/bugzilla/show_bug.cgi?id=800\r
66   g_SynapseClient.AddAPI(VFS_MAJOR, "pk3", sizeof(_QERFileSystemTable), SYN_REQUIRE, &g_FileSystemTable);\r
67 \r
68   return &g_SynapseClient;\r
69 }\r
70 \r
71 bool CSynapseClientImage::RequestAPI(APIDescriptor_t *pAPI)\r
72 {\r
73   if (!strcmp(pAPI->major_name, IMAGE_MAJOR))\r
74   {    \r
75     _QERPlugImageTable* pTable= static_cast<_QERPlugImageTable*>(pAPI->mpTable);\r
76     if (!strcmp(pAPI->minor_name, "png"))\r
77     {\r
78       pTable->m_pfnLoadImage = &LoadImage;\r
79       return true;\r
80     }\r
81   }\r
82 \r
83   Syn_Printf("ERROR: RequestAPI( '%s' ) not found in '%s'\n", pAPI->major_name, GetInfo());\r
84   return false;\r
85 }\r
86 \r
87 #include "version.h"\r
88 \r
89 const char* CSynapseClientImage::GetInfo()\r
90 {\r
91   return "PNG loader module built " __DATE__ " " RADIANT_VERSION;\r
92 }\r
93 \r
94 \r
95 \r
96 // ====== PNG loader functionality ======\r
97 \r
98 #include "png.h"\r
99 \r
100 #ifdef __APPLE__        //tigital\r
101 #include <stdlib.h>\r
102 #endif\r
103 \r
104 void user_warning_fn(png_structp png_ptr, png_const_charp warning_msg)\r
105 {\r
106   g_FuncTable.m_pfnSysPrintf ("libpng warning: %s\n", warning_msg);\r
107 }\r
108 \r
109 void user_error_fn(png_structp png_ptr, png_const_charp error_msg)\r
110 {\r
111   g_FuncTable.m_pfnSysPrintf ("libpng error: %s\n", error_msg);\r
112   longjmp(png_ptr->jmpbuf, 0);\r
113 }\r
114 \r
115 void user_read_data(png_structp png_ptr, png_bytep data, png_uint_32 length)\r
116 {\r
117   png_bytep *p_p_fbuffer = (png_bytep*)png_get_io_ptr(png_ptr);\r
118   memcpy(data, *p_p_fbuffer, length);\r
119   *p_p_fbuffer += length;\r
120 }\r
121 \r
122 void LoadImage (const char *filename, unsigned char **pic, int *width, int *height)\r
123 {\r
124   png_byte** row_pointers;\r
125   unsigned char *fbuffer = NULL;\r
126   png_bytep p_fbuffer;\r
127 \r
128   int nLen = g_FileSystemTable.m_pfnLoadFile( (char *)filename, (void **)&fbuffer, 0 );\r
129   if (nLen == -1)\r
130     return;\r
131 \r
132   p_fbuffer = fbuffer;\r
133         \r
134   // the reading glue\r
135   // http://www.libpng.org/pub/png/libpng-manual.html\r
136 \r
137   png_structp png_ptr = png_create_read_struct\r
138     (PNG_LIBPNG_VER_STRING, png_voidp_NULL,\r
139     user_error_fn, user_warning_fn);\r
140   if (!png_ptr)\r
141   {\r
142     g_FuncTable.m_pfnSysPrintf ("libpng error: png_create_read_struct\n");\r
143     return;\r
144   }\r
145                 \r
146   png_infop info_ptr = png_create_info_struct(png_ptr);\r
147   if (!info_ptr) {\r
148     png_destroy_read_struct(&png_ptr,\r
149       png_infopp_NULL, png_infopp_NULL);\r
150     g_FuncTable.m_pfnSysPrintf ("libpng error: png_create_info_struct (info_ptr)\n");\r
151     return;\r
152   }\r
153         \r
154   png_infop end_info = png_create_info_struct(png_ptr);\r
155   if (!end_info) {\r
156     png_destroy_read_struct(&png_ptr, &info_ptr,\r
157       png_infopp_NULL);\r
158     g_FuncTable.m_pfnSysPrintf ("libpng error: png_create_info_struct (end_info)\n");\r
159     return;\r
160   }\r
161 \r
162   // configure the read function\r
163   png_set_read_fn(png_ptr, (voidp)&p_fbuffer, (png_rw_ptr)&user_read_data);\r
164 \r
165   if (setjmp(png_ptr->jmpbuf)) {\r
166     png_destroy_read_struct(&png_ptr, &info_ptr,\r
167       &end_info);\r
168     if (*pic)\r
169     {\r
170       g_free(*pic);\r
171       free(row_pointers);\r
172     }\r
173     return;\r
174   }\r
175 \r
176   png_read_info(png_ptr, info_ptr);\r
177 \r
178   int bit_depth = png_get_bit_depth(png_ptr, info_ptr);\r
179   int color_type = png_get_color_type(png_ptr, info_ptr);\r
180 \r
181   // we want to treat all images the same way\r
182   //   The following code transforms grayscale images of less than 8 to 8 bits, \r
183   //   changes paletted images to RGB, and adds a full alpha channel if there is \r
184   //   transparency information in a tRNS chunk.\r
185   if (color_type == PNG_COLOR_TYPE_PALETTE)\r
186    png_set_palette_to_rgb(png_ptr);\r
187 \r
188   if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8)\r
189     png_set_gray_1_2_4_to_8(png_ptr);\r
190 \r
191   if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))\r
192     png_set_tRNS_to_alpha(png_ptr);\r
193 \r
194   if ( ! ( color_type & PNG_COLOR_MASK_ALPHA ) ) {\r
195     // Set the background color to draw transparent and alpha images over.\r
196     png_color_16 my_background, *image_background;\r
197 \r
198     if (png_get_bKGD(png_ptr, info_ptr, &image_background))\r
199       png_set_background(png_ptr, image_background, \r
200       PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);\r
201     else\r
202       png_set_background(png_ptr, &my_background,\r
203       PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);\r
204 \r
205     // Add alpha byte after each RGB triplet\r
206     png_set_filler(png_ptr, 0xff, PNG_FILLER_AFTER);\r
207   }\r
208 \r
209   // read the sucker in one chunk\r
210   png_read_update_info(png_ptr, info_ptr);\r
211 \r
212   color_type = png_get_color_type(png_ptr, info_ptr);\r
213   bit_depth = png_get_bit_depth(png_ptr, info_ptr);\r
214 \r
215   *width = png_get_image_width(png_ptr, info_ptr);\r
216   *height = png_get_image_height(png_ptr, info_ptr);\r
217 \r
218   // allocate the pixel buffer, and the row pointers\r
219   int size = (*width)*(*height)*4;\r
220   // still have to use that g_malloc heresy\r
221   // http://zerowing.idsoftware.com/bugzilla/show_bug.cgi?id=491\r
222   *pic = (unsigned char *)g_malloc(size);\r
223   row_pointers = (png_byte**) malloc((*height) * sizeof(png_byte*));\r
224 \r
225   int i;\r
226   for(i = 0; i < (*height); i++)\r
227     row_pointers[i] = (png_byte*)(*pic) + i * 4 * (*width);\r
228 \r
229   // actual read\r
230   png_read_image(png_ptr, row_pointers);\r
231 \r
232   /* read rest of file, and get additional chunks in info_ptr - REQUIRED */\r
233   png_read_end(png_ptr, info_ptr);\r
234 \r
235   /* free up the memory structure */\r
236   png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);\r
237 \r
238   free(row_pointers);\r
239   g_FileSystemTable.m_pfnFreeFile (fbuffer);\r
240 }\r
241 \r
242 \r