TGA support based on SaracenOne's work.

7 files changed, 491 insertions, 2 deletions

diff --git a/modules/hdr/image_loader_hdr.h b/modules/hdr/image_loader_hdr.h
index 9bc1fadd13..127833ebd0 100644
--- a/modules/hdr/image_loader_hdr.h
+++ b/modules/hdr/image_loader_hdr.h
@@ -27,8 +27,8 @@
 /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
 /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
 /*************************************************************************/
-#ifndef IMAGE_LOADER_TINYEXR_H
-#define IMAGE_LOADER_TINYEXR_H
+#ifndef IMAGE_LOADER_HDR_H
+#define IMAGE_LOADER_HDR_H
 
 #include "io/image_loader.h"
 
diff --git a/modules/tga/SCsub b/modules/tga/SCsub
new file mode 100644
index 0000000000..7e405f405c
--- /dev/null
+++ b/modules/tga/SCsub
@@ -0,0 +1,9 @@
+#!/usr/bin/env python
+
+Import('env')
+Import('env_modules')
+
+env_tga = env_modules.Clone()
+
+# Godot's own source files
+env_tga.add_source_files(env.modules_sources, "*.cpp")
diff --git a/modules/tga/config.py b/modules/tga/config.py
new file mode 100644
index 0000000000..fb920482f5
--- /dev/null
+++ b/modules/tga/config.py
@@ -0,0 +1,7 @@
+
+def can_build(platform):
+    return True
+
+
+def configure(env):
+    pass
diff --git a/modules/tga/image_loader_tga.cpp b/modules/tga/image_loader_tga.cpp
new file mode 100644
index 0000000000..5b8610b975
--- /dev/null
+++ b/modules/tga/image_loader_tga.cpp
@@ -0,0 +1,314 @@
+/*************************************************************************/
+/*  image_loader_jpegd.cpp                                               */
+/*************************************************************************/
+/*                       This file is part of:                           */
+/*                           GODOT ENGINE                                */
+/*                    http://www.godotengine.org                         */
+/*************************************************************************/
+/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
+/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md)    */
+/*                                                                       */
+/* 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.                */
+/*************************************************************************/
+#include "image_loader_tga.h"
+
+#include "os/os.h"
+#include "print_string.h"
+
+Error ImageLoaderTGA::decode_tga_rle(const uint8_t *p_compressed_buffer, size_t p_pixel_size, uint8_t *p_uncompressed_buffer, size_t p_output_size) {
+	Error error;
+
+	PoolVector<uint8_t> pixels;
+	error = pixels.resize(p_pixel_size);
+	if (error != OK)
+		return error;
+
+	PoolVector<uint8_t>::Write pixels_w = pixels.write();
+
+	size_t compressed_pos = 0;
+	size_t output_pos = 0;
+	size_t c = 0;
+	size_t count = 0;
+
+	while (output_pos < p_output_size) {
+		c = p_compressed_buffer[compressed_pos];
+		compressed_pos += 1;
+		count = (c & 0x7f) + 1;
+
+		if (c & 0x80) {
+			for (int i = 0; i < p_pixel_size; i++) {
+				pixels_w.ptr()[i] = p_compressed_buffer[compressed_pos];
+				compressed_pos += 1;
+			}
+			for (int i = 0; i < count; i++) {
+				for (int j = 0; j < p_pixel_size; j++) {
+					p_uncompressed_buffer[output_pos + j] = pixels_w.ptr()[j];
+				}
+				output_pos += p_pixel_size;
+			}
+		} else {
+			count *= p_pixel_size;
+			for (int i = 0; i < count; i++) {
+				p_uncompressed_buffer[output_pos] = p_compressed_buffer[compressed_pos];
+				compressed_pos += 1;
+				output_pos += 1;
+			}
+		}
+	}
+	return OK;
+}
+
+Error ImageLoaderTGA::convert_to_image(Ref<Image> p_image, const uint8_t *p_buffer, const tga_header_s &p_header, const uint8_t *p_palette, const bool p_is_monochrome) {
+
+#define TGA_PUT_PIXEL(r, g, b, a)             \
+	int image_data_ofs = ((y * width) + x);   \
+	image_data_w[image_data_ofs * 4 + 0] = r; \
+	image_data_w[image_data_ofs * 4 + 1] = g; \
+	image_data_w[image_data_ofs * 4 + 2] = b; \
+	image_data_w[image_data_ofs * 4 + 3] = a;
+
+	uint32_t width = p_header.image_width;
+	uint32_t height = p_header.image_height;
+	tga_origin_e origin = static_cast<tga_origin_e>((p_header.image_descriptor & TGA_ORIGIN_MASK) >> TGA_ORIGIN_SHIFT);
+
+	uint32_t x_start;
+	int32_t x_step;
+	uint32_t x_end;
+	uint32_t y_start;
+	int32_t y_step;
+	uint32_t y_end;
+
+	if (origin == TGA_ORIGIN_TOP_LEFT || origin == TGA_ORIGIN_TOP_RIGHT) {
+		y_start = 0;
+		y_step = 1;
+		y_end = height;
+	} else {
+		y_start = height - 1;
+		y_step = -1;
+		y_end = -1;
+	}
+
+	if (origin == TGA_ORIGIN_TOP_LEFT || origin == TGA_ORIGIN_BOTTOM_LEFT) {
+		x_start = 0;
+		x_step = 1;
+		x_end = width;
+	} else {
+		x_start = width - 1;
+		x_step = -1;
+		x_end = -1;
+	}
+
+	PoolVector<uint8_t> image_data;
+	image_data.resize(width * height * sizeof(uint32_t));
+	PoolVector<uint8_t>::Write image_data_w = image_data.write();
+
+	size_t i = 0;
+	uint32_t x = x_start;
+	uint32_t y = y_start;
+
+	if (p_header.pixel_depth == 8) {
+		if (p_is_monochrome) {
+			while (y != y_end) {
+				while (x != x_end) {
+					uint8_t shade = p_buffer[i];
+
+					TGA_PUT_PIXEL(shade, shade, shade, 0xff)
+
+					x += x_step;
+					i += 1;
+				}
+				x = x_start;
+				y += y_step;
+			}
+		} else {
+			while (y != y_end) {
+				while (x != x_end) {
+					uint8_t index = p_buffer[i];
+					uint8_t r = 0x00;
+					uint8_t g = 0x00;
+					uint8_t b = 0x00;
+					uint8_t a = 0xff;
+
+					if (p_header.color_map_depth == 24) {
+						r = (p_palette[(index * 3) + 0]);
+						g = (p_palette[(index * 3) + 1]);
+						b = (p_palette[(index * 3) + 2]);
+					} else {
+						return ERR_INVALID_DATA;
+					}
+
+					TGA_PUT_PIXEL(r, g, b, a)
+
+					x += x_step;
+					i += 1;
+				}
+				x = x_start;
+				y += y_step;
+			}
+		}
+	} else if (p_header.pixel_depth == 24) {
+		while (y != y_end) {
+			while (x != x_end) {
+				uint8_t r = p_buffer[i + 2];
+				uint8_t g = p_buffer[i + 1];
+				uint8_t b = p_buffer[i + 0];
+
+				TGA_PUT_PIXEL(r, g, b, 0xff)
+
+				x += x_step;
+				i += 3;
+			}
+			x = x_start;
+			y += y_step;
+		}
+	} else if (p_header.pixel_depth == 32) {
+		while (y != y_end) {
+			while (x != x_end) {
+				uint8_t a = p_buffer[i + 3];
+				uint8_t r = p_buffer[i + 2];
+				uint8_t g = p_buffer[i + 1];
+				uint8_t b = p_buffer[i + 0];
+
+				TGA_PUT_PIXEL(r, g, b, a)
+
+				x += x_step;
+				i += 4;
+			}
+			x = x_start;
+			y += y_step;
+		}
+	}
+
+	image_data_w = PoolVector<uint8_t>::Write();
+
+	p_image->create(width, height, 0, Image::FORMAT_RGBA8, image_data);
+
+	return OK;
+}
+
+Error ImageLoaderTGA::load_image(Ref<Image> p_image, FileAccess *f, bool p_force_linear) {
+
+	PoolVector<uint8_t> src_image;
+	int src_image_len = f->get_len();
+	ERR_FAIL_COND_V(src_image_len == 0, ERR_FILE_CORRUPT);
+	ERR_FAIL_COND_V(src_image_len < sizeof(tga_header_s), ERR_FILE_CORRUPT);
+	src_image.resize(src_image_len);
+
+	Error err = OK;
+
+	tga_header_s tga_header;
+	tga_header.id_length = f->get_8();
+	tga_header.color_map_type = f->get_8();
+	tga_header.image_type = static_cast<tga_type_e>(f->get_8());
+
+	tga_header.first_color_entry = f->get_16();
+	tga_header.color_map_length = f->get_16();
+	tga_header.color_map_depth = f->get_8();
+
+	tga_header.x_origin = f->get_16();
+	tga_header.y_origin = f->get_16();
+	tga_header.image_width = f->get_16();
+	tga_header.image_height = f->get_16();
+	tga_header.pixel_depth = f->get_8();
+	tga_header.image_descriptor = f->get_8();
+
+	bool is_encoded = (tga_header.image_type == TGA_TYPE_RLE_INDEXED || tga_header.image_type == TGA_TYPE_RLE_RGB || tga_header.image_type == TGA_TYPE_RLE_MONOCHROME);
+	bool has_color_map = (tga_header.image_type == TGA_TYPE_RLE_INDEXED || tga_header.image_type == TGA_TYPE_INDEXED);
+	bool is_monochrome = (tga_header.image_type == TGA_TYPE_RLE_MONOCHROME || tga_header.image_type == TGA_TYPE_MONOCHROME);
+
+	if (tga_header.image_type == TGA_TYPE_NO_DATA)
+		err = FAILED;
+
+	if (has_color_map) {
+		if (tga_header.color_map_length > 256 || (tga_header.color_map_depth != 24) || tga_header.color_map_type != 1) {
+			err = FAILED;
+		}
+	} else {
+		if (tga_header.color_map_type) {
+			err = FAILED;
+		}
+	}
+
+	if (tga_header.image_width <= 0 || tga_header.image_height <= 0)
+		err = FAILED;
+
+	if (tga_header.pixel_depth != 8 && tga_header.pixel_depth != 24 && tga_header.pixel_depth != 32)
+		err = FAILED;
+
+	if (err == OK) {
+		f->seek(f->get_pos() + tga_header.id_length);
+
+		PoolVector<uint8_t> palette;
+
+		if (has_color_map) {
+			size_t color_map_size = tga_header.color_map_length * (tga_header.color_map_depth >> 3);
+			err = palette.resize(color_map_size);
+			if (err == OK) {
+				PoolVector<uint8_t>::Write palette_w = palette.write();
+				f->get_buffer(&palette_w[0], color_map_size);
+			} else {
+				return OK;
+			}
+		}
+
+		PoolVector<uint8_t>::Write src_image_w = src_image.write();
+		f->get_buffer(&src_image_w[0], src_image_len - f->get_pos());
+
+		PoolVector<uint8_t>::Read src_image_r = src_image.read();
+
+		const size_t pixel_size = tga_header.pixel_depth >> 3;
+		const size_t buffer_size = (tga_header.image_width * tga_header.image_height) * pixel_size;
+
+		PoolVector<uint8_t> uncompressed_buffer;
+		uncompressed_buffer.resize(buffer_size);
+		PoolVector<uint8_t>::Write uncompressed_buffer_w = uncompressed_buffer.write();
+		PoolVector<uint8_t>::Read uncompressed_buffer_r;
+
+		const uint8_t *buffer = NULL;
+
+		if (is_encoded) {
+
+			err = decode_tga_rle(src_image_r.ptr(), pixel_size, uncompressed_buffer_w.ptr(), buffer_size);
+
+			if (err == OK) {
+				uncompressed_buffer_r = uncompressed_buffer.read();
+				buffer = uncompressed_buffer_r.ptr();
+			}
+		} else {
+			buffer = src_image_r.ptr();
+		};
+
+		if (err == OK) {
+			PoolVector<uint8_t>::Read palette_r = palette.read();
+			err = convert_to_image(p_image, buffer, tga_header, palette_r.ptr(), is_monochrome);
+		}
+	}
+
+	f->close();
+	return err;
+}
+
+void ImageLoaderTGA::get_recognized_extensions(List<String> *p_extensions) const {
+
+	p_extensions->push_back("tga");
+}
+
+ImageLoaderTGA::ImageLoaderTGA() {
+}
diff --git a/modules/tga/image_loader_tga.h b/modules/tga/image_loader_tga.h
new file mode 100644
index 0000000000..11329ec68a
--- /dev/null
+++ b/modules/tga/image_loader_tga.h
@@ -0,0 +1,83 @@
+/*************************************************************************/
+/*  image_loader_jpegd.h                                                 */
+/*************************************************************************/
+/*                       This file is part of:                           */
+/*                           GODOT ENGINE                                */
+/*                    http://www.godotengine.org                         */
+/*************************************************************************/
+/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
+/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md)    */
+/*                                                                       */
+/* 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_LOADER_TGA_H
+#define IMAGE_LOADER_TGA_H
+
+#include "io/image_loader.h"
+
+/**
+	@author SaracenOne
+*/
+class ImageLoaderTGA : public ImageFormatLoader {
+	enum tga_type_e {
+		TGA_TYPE_NO_DATA = 0,
+		TGA_TYPE_INDEXED = 1,
+		TGA_TYPE_RGB = 2,
+		TGA_TYPE_MONOCHROME = 3,
+		TGA_TYPE_RLE_INDEXED = 9,
+		TGA_TYPE_RLE_RGB = 10,
+		TGA_TYPE_RLE_MONOCHROME = 11
+	};
+
+	enum tga_origin_e {
+		TGA_ORIGIN_BOTTOM_LEFT = 0x00,
+		TGA_ORIGIN_BOTTOM_RIGHT = 0x01,
+		TGA_ORIGIN_TOP_LEFT = 0x02,
+		TGA_ORIGIN_TOP_RIGHT = 0x03,
+		TGA_ORIGIN_SHIFT = 0x04,
+		TGA_ORIGIN_MASK = 0x30
+	};
+
+	struct tga_header_s {
+		uint8_t id_length;
+		uint8_t color_map_type;
+		tga_type_e image_type;
+
+		uint16_t first_color_entry;
+		uint16_t color_map_length;
+		uint8_t color_map_depth;
+
+		uint16_t x_origin;
+		uint16_t y_origin;
+		uint16_t image_width;
+		uint16_t image_height;
+		uint8_t pixel_depth;
+		uint8_t image_descriptor;
+	};
+	static Error decode_tga_rle(const uint8_t *p_compressed_buffer, size_t p_pixel_size, uint8_t *p_uncompressed_buffer, size_t p_output_size);
+	static Error convert_to_image(Ref<Image> p_image, const uint8_t *p_buffer, const tga_header_s &p_header, const uint8_t *p_palette, const bool p_is_monochrome);
+
+public:
+	virtual Error load_image(Ref<Image> p_image, FileAccess *f, bool p_force_linear);
+	virtual void get_recognized_extensions(List<String> *p_extensions) const;
+	ImageLoaderTGA();
+};
+
+#endif
diff --git a/modules/tga/register_types.cpp b/modules/tga/register_types.cpp
new file mode 100644
index 0000000000..6e120fa3bf
--- /dev/null
+++ b/modules/tga/register_types.cpp
@@ -0,0 +1,45 @@
+/*************************************************************************/
+/*  register_types.cpp                                                   */
+/*************************************************************************/
+/*                       This file is part of:                           */
+/*                           GODOT ENGINE                                */
+/*                    http://www.godotengine.org                         */
+/*************************************************************************/
+/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
+/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md)    */
+/*                                                                       */
+/* 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.                */
+/*************************************************************************/
+#include "register_types.h"
+
+#include "image_loader_tga.h"
+
+static ImageLoaderTGA *image_loader_tga = NULL;
+
+void register_tga_types() {
+
+	image_loader_tga = memnew(ImageLoaderTGA);
+	ImageLoader::add_image_format_loader(image_loader_tga);
+}
+
+void unregister_tga_types() {
+
+	memdelete(image_loader_tga);
+}
diff --git a/modules/tga/register_types.h b/modules/tga/register_types.h
new file mode 100644
index 0000000000..079b7bf291
--- /dev/null
+++ b/modules/tga/register_types.h
@@ -0,0 +1,31 @@
+/*************************************************************************/
+/*  register_types.h                                                     */
+/*************************************************************************/
+/*                       This file is part of:                           */
+/*                           GODOT ENGINE                                */
+/*                    http://www.godotengine.org                         */
+/*************************************************************************/
+/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
+/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md)    */
+/*                                                                       */
+/* 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.                */
+/*************************************************************************/
+void register_tga_types();
+void unregister_tga_types();