CoolVLViewer/indra/llgltf/llgltfbufferutil.h

/**
 * @file llgltfbufferutil.h
 * @brief LL GLTF Implementation
 *
 * $LicenseInfo:firstyear=2024&license=viewergpl$
 *
 * Copyright (c) 2024, Linden Research, Inc.
 *
 * Second Life Viewer Source Code
 * The source code in this file ("Source Code") is provided by Linden Lab
 * to you under the terms of the GNU General Public License, version 2.0
 * ("GPL"), unless you have obtained a separate licensing agreement
 * ("Other License"), formally executed by you and Linden Lab.  Terms of
 * the GPL can be found in doc/GPL-license.txt in this distribution, or
 * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
 *
 * There are special exceptions to the terms and conditions of the GPL as
 * it is applied to this Source Code. View the full text of the exception
 * in the file doc/FLOSS-exception.txt in this software distribution, or
 * online at
 * http://secondlifegrid.net/programs/open_source/licensing/flossexception
 *
 * By copying, modifying or distributing this software, you acknowledge
 * that you have read and understood your obligations described above,
 * and agree to abide by those obligations.
 *
 * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
 * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
 * COMPLETENESS OR PERFORMANCE.
 * $/LicenseInfo$
 */

// Inline template implementations for copying data out of GLTF buffers.
// DO NOT include from header files to avoid the need to rebuild the whole
// project whenever we add support for more types.
// Also, make sure to include after the other llgltf*.h headers, to that the
// proper math types used in the corresponding module are #include'd already.

#ifndef LL_LLGLTFBUFFERUTIL_H
#define LL_LLGLTFBUFFERUTIL_H

#include "llstrider.h"

// Suppress unused function warning: clang complains here but these
// specializations are definitely used.
#if LL_CLANG
# pragma clang diagnostic push
# pragma clang diagnostic ignored "-Wunused-function"
#endif

namespace LLGLTF
{
	// Copy one Scalar from src to dst
	template<class S, class T>
	static void copyScalar(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one vec2 from src to dst
	template<class S, class T>
	static void copyVec2(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one vec3 from src to dst
	template<class S, class T>
	static void copyVec3(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one vec4 from src to dst
	template<class S, class T>
	static void copyVec4(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one mat2 from src to dst
	template<class S, class T>
	static void copyMat2(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one mat3 from src to dst
	template<class S, class T>
	static void copyMat3(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	// Copy one mat4 from src to dst
	template<class S, class T>
	static void copyMat4(S* src, T& dst)
	{
		llerrs << "TODO: implement " << LL_FUNC << llendl;
	}

	//=========================================================================
	// Concrete implementations for different types of source and destination
	//=========================================================================

	template<>
	LL_INLINE void copyScalar<F32, F32>(F32* src, F32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U32, U32>(U32* src, U32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S32, S32>(S32* src, S32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U32, U16>(U32* src, U16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S32, S16>(S32* src, S16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U16, U16>(U16* src, U16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S16, S16>(S16* src, S16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U16, U32>(U16* src, U32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S16, S32>(S16* src, S32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U8, U16>(U8* src, U16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S8, S16>(S8* src, S16& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<U8, U32>(U8* src, U32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyScalar<S8, S32>(S8* src, S32& dst)
	{
		dst = *src;
	}

	template<>
	LL_INLINE void copyVec2<F32, LLVector2>(F32* src, LLVector2& dst)
	{
		dst.set(src[0], src[1]);
	}

	template<>
	LL_INLINE void copyVec3<F32, LLVector3>(F32* src, LLVector3& dst)
	{
		dst.set(src[0], src[1], src[2]);
	}

	template<>
	LL_INLINE void copyVec3<F32, LLVector4a>(F32* src, LLVector4a& dst)
	{
		dst.load3(src);
	}

	template<>
	LL_INLINE void copyVec3<U16, LLColor4U>(U16* src, LLColor4U& dst)
	{
		dst.set(src[0], src[1], src[2], 255);
	}

	template<>
	LL_INLINE void copyVec4<U8, LLColor4U>(U8* src, LLColor4U& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<U16, LLColor4U>(U16* src, LLColor4U& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<F32, LLColor4U>(F32* src, LLColor4U& dst)
	{
		dst.set(src[0]*255, src[1]*255, src[2]*255, src[3]*255);
	}

	template<>
	LL_INLINE void copyVec4<F32, LLVector4a>(F32* src, LLVector4a& dst)
	{
		dst.loadua(src);
	}

	template<>
	LL_INLINE void copyVec4<U32, LLVector4a>(U32* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<S32, LLVector4a>(S32* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<U16, LLVector4a>(U16* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<S16, LLVector4a>(S16* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<U8, LLVector4a>(U8* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<S8, LLVector4a>(S8* src, LLVector4a& dst)
	{
		dst.set(src[0], src[1], src[2], src[3]);
	}

	template<>
	LL_INLINE void copyVec4<F32, LLQuaternion>(F32* src, LLQuaternion& dst)
	{
		dst.setRaw(src);
	}

	template<>
	LL_INLINE void copyMat4<F32, LLMatrix4>(F32* src, LLMatrix4& dst)
	{
		for (U32 i = 0; i < 4; ++i)
		{
			for (U32 j = 0; j < 4; ++j)
			{
				dst.mMatrix[i][j] = *src++;
			}
		}
	}

#if 1	// *TODO: get rid of glh (again) !  HB
	template<>
	LL_INLINE void copyVec3<F32, glh::vec3f>(F32* src, glh::vec3f& dst)
	{
		dst.set_value(src[0], src[1], src[2]);
	}

	template<>
	LL_INLINE void copyVec4<F32, glh::quaternionf>(F32* src, glh::quaternionf& dst)
	{
		dst.set_value(src);
	}

	template<>
	LL_INLINE void copyMat4<F32, glh::matrix4f>(F32* src, glh::matrix4f& dst)
	{
		dst.set_value(src);
	}
#endif

	//=========================================================================

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyScalar(S* src, LLStrider<T> dst, S32 stride,
									 S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyScalar(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyVec2(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyVec2(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyVec3(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyVec3(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyVec4(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyVec4(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyMat2(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyMat2(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyMat3(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyMat3(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	// Copy from src to dst, stride is the number of bytes between each element
	// in src, count is number of elements to copy.
	template<class S, class T>
	LL_INLINE static void copyMat4(S* src, LLStrider<T> dst, S32 stride,
								   S32 count)
	{
		for (S32 i = 0; i < count; ++i)
		{
			copyMat4(src, *dst++);
			src = (S*)((U8*)src + stride);
		}
	}

	template<class S, class T>
	LL_INLINE static void copy(Asset& asset, Accessor& accessor, const S* src,
							   LLStrider<T>& dst, S32 bstride)
	{
		if (accessor.mType == (S32)Accessor::Type::SCALAR)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 1 : bstride;
			copyScalar((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::VEC2)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 2 : bstride;
			copyVec2((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::VEC3)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 3 : bstride;
			copyVec3((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::VEC4)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 4 : bstride;
			copyVec4((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::MAT2)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 4 : bstride;
			copyMat2((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::MAT3)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 9 : bstride;
			copyMat3((S*)src, dst, stride, accessor.mCount);
		}
		else if (accessor.mType == (S32)Accessor::Type::MAT4)
		{
			S32 stride = bstride == 0 ? sizeof(S) * 16 : bstride;
			copyMat4((S*)src, dst, stride, accessor.mCount);
		}
		else
		{
			llerrs << "Unsupported accessor type" << llendl;
		}
	}

	// Copy data from accessor to strider
	template<class T>
	LL_INLINE static void copy(Asset& asset, Accessor& accessor, LLStrider<T>& dst)
	{
		const BufferView& buffviewp = asset.mBufferViews[accessor.mBufferView];
		const Buffer& buffer = asset.mBuffers[buffviewp.mBuffer];
		const U8* src = buffer.mData.data() + buffviewp.mByteOffset +
						accessor.mByteOffset;

		if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_FLOAT)
		{
			LLGLTF::copy(asset, accessor, (const F32*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_UNSIGNED_SHORT)
		{
			LLGLTF::copy(asset, accessor, (const U16*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_UNSIGNED_INT)
		{
			LLGLTF::copy(asset, accessor, (const U32*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_UNSIGNED_BYTE)
		{
			LLGLTF::copy(asset, accessor, (const U8*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_SHORT)
		{
			LLGLTF::copy(asset, accessor, (const S16*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_BYTE)
		{
			LLGLTF::copy(asset, accessor, (const S8*)src, dst, buffviewp.mByteStride);
		}
		else if (accessor.mComponentType == TINYGLTF_COMPONENT_TYPE_DOUBLE)
		{
			LLGLTF::copy(asset, accessor, (const F64*)src, dst, buffviewp.mByteStride);
		}
		else
		{
			llerrs << "Unsupported component type" << llendl;
		}
	}

	// Copy data from accessor to vector
	template<class T>
	LL_INLINE static void copy(Asset& asset, Accessor& accessor, std::vector<T>& dst)
	{
		dst.resize(accessor.mCount);
		LLStrider<T> strider = dst.data();
		copy(asset, accessor, strider);
	}
}

#if LL_CLANG
# pragma clang diagnostic pop
#endif

#endif	// LL_LLGLTFBUFFERUTIL_H