Remove 156 files
[xonotic/netradiant.git] / plugins / imagepng / plugin.cpp
1 /*
2    Copyright (C) 1999-2006 Id Software, Inc. and contributors.
3    For a list of contributors, see the accompanying CONTRIBUTORS file.
4
5    This file is part of GtkRadiant.
6
7    GtkRadiant is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11
12    GtkRadiant is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16
17    You should have received a copy of the GNU General Public License
18    along with GtkRadiant; if not, write to the Free Software
19    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
20  */
21
22 #include "debugging/debugging.h"
23
24 #include "ifilesystem.h"
25 #include "iimage.h"
26
27 #include "imagelib.h"
28
29 // ====== PNG loader functionality ======
30
31 #include "png.h"
32 #include <stdlib.h>
33
34 void user_warning_fn( png_structp png_ptr, png_const_charp warning_msg ){
35         globalErrorStream() << "libpng warning: " << warning_msg << "\n";
36 }
37
38 void user_error_fn( png_structp png_ptr, png_const_charp error_msg ){
39         globalErrorStream() << "libpng error: " << error_msg << "\n";
40         longjmp( png_jmpbuf(png_ptr), 0 );
41 }
42
43 void user_read_data( png_structp png_ptr, png_bytep data, png_uint_32 length ){
44         png_bytep *p_p_fbuffer = (png_bytep*)png_get_io_ptr( png_ptr );
45         memcpy( data, *p_p_fbuffer, length );
46         *p_p_fbuffer += length;
47 }
48
49 Image* LoadPNGBuff( unsigned char* fbuffer ){
50         png_byte** row_pointers;
51         png_bytep p_fbuffer;
52
53         p_fbuffer = fbuffer;
54
55         // the reading glue
56         // http://www.libpng.org/pub/png/libpng-manual.html
57
58         png_structp png_ptr = png_create_read_struct
59                                                           ( PNG_LIBPNG_VER_STRING, (png_voidp)NULL,
60                                                           user_error_fn, user_warning_fn );
61         if ( !png_ptr ) {
62                 globalErrorStream() << "libpng error: png_create_read_struct\n";
63                 return 0;
64         }
65
66         png_infop info_ptr = png_create_info_struct( png_ptr );
67         if ( !info_ptr ) {
68                 png_destroy_read_struct( &png_ptr,
69                                                                  (png_infopp)NULL, (png_infopp)NULL );
70                 globalErrorStream() << "libpng error: png_create_info_struct (info_ptr)\n";
71                 return 0;
72         }
73
74         png_infop end_info = png_create_info_struct( png_ptr );
75         if ( !end_info ) {
76                 png_destroy_read_struct( &png_ptr, &info_ptr,
77                                                                  (png_infopp)NULL );
78                 globalErrorStream() << "libpng error: png_create_info_struct (end_info)\n";
79                 return 0;
80         }
81
82         // configure the read function
83         png_set_read_fn( png_ptr, ( png_voidp ) & p_fbuffer, ( png_rw_ptr ) & user_read_data );
84
85         if ( setjmp( png_jmpbuf(png_ptr) ) ) {
86                 png_destroy_read_struct( &png_ptr, &info_ptr,
87                                                                  &end_info );
88                 return 0;
89         }
90
91         png_read_info( png_ptr, info_ptr );
92
93         int bit_depth = png_get_bit_depth( png_ptr, info_ptr );
94         int color_type = png_get_color_type( png_ptr, info_ptr );
95
96         // we want to treat all images the same way
97         //   The following code transforms grayscale images of less than 8 to 8 bits,
98         //   changes paletted images to RGB, and adds a full alpha channel if there is
99         //   transparency information in a tRNS chunk.
100
101         if ( color_type == PNG_COLOR_TYPE_GRAY || color_type == PNG_COLOR_TYPE_GRAY_ALPHA ) {
102                 png_set_gray_to_rgb( png_ptr );
103         }
104         else if ( color_type == PNG_COLOR_TYPE_PALETTE ) {
105                 png_set_palette_to_rgb( png_ptr );
106         }
107
108         if ( color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8 ) {
109                 png_set_expand_gray_1_2_4_to_8( png_ptr );
110         }
111
112         if ( png_get_valid( png_ptr, info_ptr, PNG_INFO_tRNS ) ) {
113                 png_set_tRNS_to_alpha( png_ptr );
114         }
115
116         if ( !( color_type & PNG_COLOR_MASK_ALPHA ) ) {
117                 // Set the background color to draw transparent and alpha images over.
118                 png_color_16 my_background, *image_background;
119
120                 if ( png_get_bKGD( png_ptr, info_ptr, &image_background ) ) {
121                         png_set_background( png_ptr, image_background,
122                                                                 PNG_BACKGROUND_GAMMA_FILE, 1, 1.0 );
123                 }
124                 else{
125                         png_set_background( png_ptr, &my_background,
126                                                                 PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0 );
127                 }
128
129                 // Add alpha byte after each RGB triplet
130                 png_set_filler( png_ptr, 0xff, PNG_FILLER_AFTER );
131         }
132
133         // read the sucker in one chunk
134         png_read_update_info( png_ptr, info_ptr );
135
136         color_type = png_get_color_type( png_ptr, info_ptr );
137         bit_depth = png_get_bit_depth( png_ptr, info_ptr );
138
139         int width = png_get_image_width( png_ptr, info_ptr );
140         int height = png_get_image_height( png_ptr, info_ptr );
141
142         // allocate the pixel buffer, and the row pointers
143         RGBAImage* image = new RGBAImage( width, height );
144
145         row_pointers = (png_byte**) malloc( ( height ) * sizeof( png_byte* ) );
146
147         int i;
148         for ( i = 0; i < ( height ); i++ )
149                 row_pointers[i] = (png_byte*)( image->getRGBAPixels() ) + i * 4 * ( width );
150
151         // actual read
152         png_read_image( png_ptr, row_pointers );
153
154         /* read rest of file, and get additional chunks in info_ptr - REQUIRED */
155         png_read_end( png_ptr, info_ptr );
156
157         /* free up the memory structure */
158         png_destroy_read_struct( &png_ptr, &info_ptr, NULL );
159
160         free( row_pointers );
161
162         return image;
163 }
164
165 Image* LoadPNG( ArchiveFile& file ){
166         ScopedArchiveBuffer buffer( file );
167         return LoadPNGBuff( buffer.buffer );
168 }
169
170
171 #include "modulesystem/singletonmodule.h"
172
173
174 class ImageDependencies : public GlobalFileSystemModuleRef
175 {
176 };
177
178 class ImagePNGAPI
179 {
180 _QERPlugImageTable m_imagepng;
181 public:
182 typedef _QERPlugImageTable Type;
183 STRING_CONSTANT( Name, "png" );
184
185 ImagePNGAPI(){
186         m_imagepng.loadImage = LoadPNG;
187 }
188 _QERPlugImageTable* getTable(){
189         return &m_imagepng;
190 }
191 };
192
193 typedef SingletonModule<ImagePNGAPI, ImageDependencies> ImagePNGModule;
194
195 ImagePNGModule g_ImagePNGModule;
196
197
198 extern "C" void RADIANT_DLLEXPORT Radiant_RegisterModules( ModuleServer& server ){
199         initialiseModule( server );
200
201         g_ImagePNGModule.selfRegister();
202 }