diff options
Diffstat (limited to 'core/image.h')
-rw-r--r-- | core/image.h | 336 |
1 files changed, 336 insertions, 0 deletions
diff --git a/core/image.h b/core/image.h new file mode 100644 index 0000000000..4ab2870c23 --- /dev/null +++ b/core/image.h @@ -0,0 +1,336 @@ +/*************************************************************************/ +/* image.h */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* http://www.godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ +/* */ +/* Permission is hereby granted, free of charge, to any person obtaining */ +/* a copy of this software and associated documentation files (the */ +/* "Software"), to deal in the Software without restriction, including */ +/* without limitation the rights to use, copy, modify, merge, publish, */ +/* distribute, sublicense, and/or sell copies of the Software, and to */ +/* permit persons to whom the Software is furnished to do so, subject to */ +/* the following conditions: */ +/* */ +/* The above copyright notice and this permission notice shall be */ +/* included in all copies or substantial portions of the Software. */ +/* */ +/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ +/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ +/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ +/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ +/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ +/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ +/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ +/*************************************************************************/ +#ifndef IMAGE_H +#define IMAGE_H + +#include "dvector.h" +#include "color.h" +#include "math_2d.h" +/** + * @author Juan Linietsky <reduzio@gmail.com> + * + * Image storage class. This is used to store an image in user memory, as well as + * providing some basic methods for image manipulation. + * Images can be loaded from a file, or registered into the Render object as textures. +*/ + + + +class Image { + + enum { + MAX_WIDTH=4096, // force a limit somehow + MAX_HEIGHT=4096 // force a limit somehow + }; +public: + + enum Format { + FORMAT_GRAYSCALE, ///< one byte per pixel, 0-255 + FORMAT_INTENSITY, ///< one byte per pixel, 0-255 + FORMAT_GRAYSCALE_ALPHA, ///< two bytes per pixel, 0-255. alpha 0-255 + FORMAT_RGB, ///< one byte R, one byte G, one byte B + FORMAT_RGBA, ///< one byte R, one byte G, one byte B, one byte A + FORMAT_INDEXED, ///< index byte 0-256, and after image end, 256*3 bytes of palette + FORMAT_INDEXED_ALPHA, ///< index byte 0-256, and after image end, 256*4 bytes of palette (alpha) + FORMAT_YUV_422, + FORMAT_YUV_444, + FORMAT_BC1, // DXT1 + FORMAT_BC2, // DXT3 + FORMAT_BC3, // DXT5 + FORMAT_BC4, // ATI1 + FORMAT_BC5, // ATI2 + FORMAT_PVRTC2, + FORMAT_PVRTC2_ALPHA, + FORMAT_PVRTC4, + FORMAT_PVRTC4_ALPHA, + FORMAT_ETC, // regular ETC, no transparency + /*FORMAT_ETC2_R, for the future.. + FORMAT_ETC2_RG, + FORMAT_ETC2_RGB, + FORMAT_ETC2_RGBA1, + FORMAT_ETC2_RGBA,*/ + FORMAT_CUSTOM, + + FORMAT_MAX + }; + + enum Interpolation { + + INTERPOLATE_NEAREST, + INTERPOLATE_BILINEAR, + /* INTERPOLATE GAUSS */ + }; + + static Image (*_png_mem_loader_func)(const uint8_t* p_png); + static void (*_image_compress_bc_func)(Image *); + static void (*_image_compress_pvrtc2_func)(Image *); + static void (*_image_compress_pvrtc4_func)(Image *); + static void (*_image_compress_etc_func)(Image *); + static void (*_image_decompress_pvrtc)(Image *); + static void (*_image_decompress_bc)(Image *); + static void (*_image_decompress_etc)(Image *); + + static DVector<uint8_t> (*lossy_packer)(const Image& p_image,float p_quality); + static Image (*lossy_unpacker)(const DVector<uint8_t>& p_buffer); + static DVector<uint8_t> (*lossless_packer)(const Image& p_image); + static Image (*lossless_unpacker)(const DVector<uint8_t>& p_buffer); +private: + + //internal byte based color + struct BColor { + union { + uint8_t col[4]; + struct { + uint8_t r,g,b,a; + }; + }; + + bool operator==(const BColor& p_color) const { for(int i=0;i<4;i++) {if (col[i]!=p_color.col[i]) return false; } return true; } + _FORCE_INLINE_ uint8_t gray() const { return (uint16_t(col[0])+uint16_t(col[1])+uint16_t(col[2]))/3; } + _FORCE_INLINE_ BColor() {} + BColor(uint8_t p_r,uint8_t p_g,uint8_t p_b,uint8_t p_a=255) { col[0]=p_r; col[1]=p_g; col[2]=p_b; col[3]=p_a; } + }; + + //median cut classes + + struct BColorPos { + + uint32_t index; + BColor color; + struct SortR { + + bool operator()(const BColorPos& ca, const BColorPos& cb) const { return ca.color.r < cb.color.r; } + }; + + struct SortG { + + bool operator()(const BColorPos& ca, const BColorPos& cb) const { return ca.color.g < cb.color.g; } + }; + + struct SortB { + + bool operator()(const BColorPos& ca, const BColorPos& cb) const { return ca.color.b < cb.color.b; } + }; + + struct SortA { + + bool operator()(const BColorPos& ca, const BColorPos& cb) const { return ca.color.a < cb.color.a; } + }; + }; + + struct SPTree { + + bool leaf; + uint8_t split_plane; + uint8_t split_value; + union { + int left; + int color; + }; + int right; + SPTree() { leaf=true; left=-1; right=-1;} + }; + + struct MCBlock { + + BColorPos min_color,max_color; + BColorPos *colors; + int sp_idx; + int color_count; + int get_longest_axis_index() const; + int get_longest_axis_length() const; + bool operator<(const MCBlock& p_block) const; + void shrink(); + MCBlock(); + MCBlock(BColorPos *p_colors,int p_color_count); + }; + + Format format; + DVector<uint8_t> data; + int width,height,mipmaps; + + + + _FORCE_INLINE_ BColor _get_pixel(int p_x,int p_y,const unsigned char *p_data,int p_data_size) const; + _FORCE_INLINE_ BColor _get_pixelw(int p_x,int p_y,int p_width,const unsigned char *p_data,int p_data_size) const; + _FORCE_INLINE_ void _put_pixelw(int p_x,int p_y, int p_width, const BColor& p_color, unsigned char *p_data); + _FORCE_INLINE_ void _put_pixel(int p_x,int p_y, const BColor& p_color, unsigned char *p_data); + _FORCE_INLINE_ void _get_mipmap_offset_and_size(int p_mipmap,int &r_offset, int &r_width, int &r_height) const; //get where the mipmap begins in data + _FORCE_INLINE_ static void _get_format_min_data_size(Format p_format,int &r_w, int &r_h); + + static int _get_dst_image_size(int p_width, int p_height, Format p_format,int &r_mipmaps,int p_mipmaps=-1); + bool _can_modify(Format p_format) const; + + + +public: + + + + int get_width() const; ///< Get image width + int get_height() const; ///< Get image height + int get_mipmaps() const; + + /** + * Get a pixel from the image. for grayscale or indexed formats, use Color::gray to obtain the actual + * value. + */ + Color get_pixel(int p_x,int p_y,int p_mipmap=0) const; + /** + * Set a pixel into the image. for grayscale or indexed formats, a suitable Color constructor. + */ + void put_pixel(int p_x,int p_y, const Color& p_color,int p_mipmap=0); /* alpha and index are averaged */ + + /** + * Convert the image to another format, as close as it can be done. + */ + void convert( Format p_new_format ); + + /** + * Get the current image format. + */ + Format get_format() const; + + int get_mipmap_offset(int p_mipmap) const; //get where the mipmap begins in data + void get_mipmap_offset_and_size(int p_mipmap,int &r_ofs, int &r_size) const; //get where the mipmap begins in data + + /** + * Resize the image, using the prefered interpolation method. + * Indexed-Color images always use INTERPOLATE_NEAREST. + */ + + void resize_to_po2(bool p_square=false); + void resize( int p_width, int p_height, Interpolation p_interpolation=INTERPOLATE_BILINEAR ); + Image resized( int p_width, int p_height, int p_interpolation=INTERPOLATE_BILINEAR ); + /** + * Crop the image to a specific size, if larger, then the image is filled by black + */ + void crop( int p_width, int p_height ); + + + void flip_x(); + void flip_y(); + /** + * Generate a mipmap to an image (creates an image 1/4 the size, with averaging of 4->1) + */ + Error generate_mipmaps(int p_amount=-1,bool p_keep_existing=false); + + void clear_mipmaps(); + + + /** + * Generate a normal map from a grayscale image + */ + + void make_normalmap(float p_height_scale=1.0); + + /** + * Create a new image of a given size and format. Current image will be lost + */ + void create(int p_width, int p_height, bool p_use_mipmaps, Format p_format); + void create(int p_width, int p_height, int p_mipmaps, Format p_format, const DVector<uint8_t>& p_data); + + void create( const char ** p_xpm ); + /** + * returns true when the image is empty (0,0) in size + */ + bool empty() const; + + DVector<uint8_t> get_data() const; + + Error load(const String& p_path); + + /** + * create an empty image + */ + Image(); + /** + * create an empty image of a specific size and format + */ + Image(int p_width, int p_height, bool p_use_mipmaps, Format p_format); + /** + * import an image of a specific size and format from a pointer + */ + Image(int p_width, int p_height, int p_mipmaps, Format p_format, const DVector<uint8_t>& p_data); + + enum AlphaMode { + ALPHA_NONE, + ALPHA_BIT, + ALPHA_BLEND + }; + + AlphaMode detect_alpha() const; + + void put_indexed_pixel(int p_x, int p_y, uint8_t p_idx,int p_mipmap=0); + uint8_t get_indexed_pixel(int p_x, int p_y,int p_mipmap=0) const; + void set_pallete(const DVector<uint8_t>& p_data); + + + static int get_format_pixel_size(Format p_format); + static int get_format_pixel_rshift(Format p_format); + static int get_format_pallete_size(Format p_format); + static int get_image_data_size(int p_width, int p_height, Format p_format,int p_mipmaps=0); + static int get_image_required_mipmaps(int p_width, int p_height, Format p_format); + + + + + bool operator==(const Image& p_image) const; + + void quantize(); + + enum CompressMode { + COMPRESS_BC, + COMPRESS_PVRTC2, + COMPRESS_PVRTC4, + COMPRESS_ETC + }; + + Error compress(CompressMode p_mode=COMPRESS_BC); + Image compressed(int p_mode); /* from the Image::CompressMode enum */ + void decompress(); + + void fix_alpha_edges(); + + void blit_rect(const Image& p_src, const Rect2& p_src_rect,const Point2& p_dest); + void brush_transfer(const Image& p_src, const Image& p_brush, const Point2& p_dest); + Image brushed(const Image& p_src, const Image& p_brush, const Point2& p_dest) const; + + Rect2 get_used_rect() const; + Image get_rect(const Rect2& p_area) const; + + static void set_compress_bc_func(void (*p_compress_func)(Image *)); + Image(const uint8_t* p_mem_png); + Image(const char **p_xpm); + ~Image(); + +}; + + +#endif |