GLTF: Allow specifying export image format including from extensions

1 files changed, 39 insertions, 0 deletions

diff --git a/modules/gltf/extensions/gltf_document_extension_texture_webp.cpp b/modules/gltf/extensions/gltf_document_extension_texture_webp.cpp
index 73c869be3b..f8bd6d57cf 100644
--- a/modules/gltf/extensions/gltf_document_extension_texture_webp.cpp
+++ b/modules/gltf/extensions/gltf_document_extension_texture_webp.cpp
@@ -68,3 +68,42 @@ Error GLTFDocumentExtensionTextureWebP::parse_texture_json(Ref<GLTFState> p_stat
 	r_gltf_texture->set_src_image(texture_webp["source"]);
 	return OK;
 }
+
+Vector<String> GLTFDocumentExtensionTextureWebP::get_saveable_image_formats() {
+	Vector<String> ret;
+	ret.push_back("Lossless WebP");
+	ret.push_back("Lossy WebP");
+	return ret;
+}
+
+PackedByteArray GLTFDocumentExtensionTextureWebP::serialize_image_to_bytes(Ref<GLTFState> p_state, Ref<Image> p_image, Dictionary p_image_dict, const String &p_image_format, float p_lossy_quality) {
+	if (p_image_format == "Lossless WebP") {
+		p_image_dict["mimeType"] = "image/webp";
+		return p_image->save_webp_to_buffer(false);
+	} else if (p_image_format == "Lossy WebP") {
+		p_image_dict["mimeType"] = "image/webp";
+		return p_image->save_webp_to_buffer(true, p_lossy_quality);
+	}
+	ERR_FAIL_V(PackedByteArray());
+}
+
+Error GLTFDocumentExtensionTextureWebP::save_image_at_path(Ref<GLTFState> p_state, Ref<Image> p_image, const String &p_file_path, const String &p_image_format, float p_lossy_quality) {
+	if (p_image_format == "Lossless WebP") {
+		p_image->save_webp(p_file_path, false);
+		return OK;
+	} else if (p_image_format == "Lossy WebP") {
+		p_image->save_webp(p_file_path, true, p_lossy_quality);
+		return OK;
+	}
+	return ERR_INVALID_PARAMETER;
+}
+
+Error GLTFDocumentExtensionTextureWebP::serialize_texture_json(Ref<GLTFState> p_state, Dictionary p_texture_json, Ref<GLTFTexture> p_gltf_texture, const String &p_image_format) {
+	Dictionary ext_texture_webp;
+	ext_texture_webp["source"] = p_gltf_texture->get_src_image();
+	Dictionary texture_extensions;
+	texture_extensions["EXT_texture_webp"] = ext_texture_webp;
+	p_texture_json["extensions"] = texture_extensions;
+	p_state->add_used_extension("EXT_texture_webp", true);
+	return OK;
+}