BansheeEngine/Source/BansheeFBXImporter/Source/BsFBXImporter.cpp

//********************************** Banshee Engine (www.banshee3d.com) **************************************************//
//**************** Copyright (c) 2016 Marko Pintera ([email protected]). All rights reserved. **********************//
#include "BsFBXImporter.h"
#include "BsResource.h"
#include "BsCoreApplication.h"
#include "BsDebug.h"
#include "BsDataStream.h"
#include "BsMeshData.h"
#include "BsMesh.h"
#include "BsVector2.h"
#include "BsVector3.h"
#include "BsVector4.h"
#include "BsVertexDataDesc.h"
#include "BsFBXUtility.h"
#include "BsMeshUtility.h"
#include "BsRendererMeshData.h"
#include "BsMeshImportOptions.h"
#include "BsPhysicsMesh.h"
#include "BsAnimationCurve.h"
#include "BsAnimationClip.h"
#include "BsAnimationUtility.h"
#include "BsSkeleton.h"
#include "BsMorphShapes.h"
#include "BsPhysics.h"

namespace bs
{
	Matrix4 FBXToNativeType(const FbxAMatrix& value)
	{
		Matrix4 native;
		for (UINT32 row = 0; row < 4; row++)
			for (UINT32 col = 0; col < 4; col++)
			native[row][col] = (float)value[col][row];

		return native;
	}

	Vector3 FBXToNativeType(const FbxVector4& value)
	{
		Vector3 native;
		native.x = (float)value[0];
		native.y = (float)value[1];
		native.z = (float)value[2];

		return native;
	}

	Vector3 FBXToNativeType(const FbxDouble3& value)
	{
		Vector3 native;
		native.x = (float)value[0];
		native.y = (float)value[1];
		native.z = (float)value[2];

		return native;
	}

	Vector2 FBXToNativeType(const FbxVector2& value)
	{
		Vector2 native;
		native.x = (float)value[0];
		native.y = (float)value[1];

		return native;
	}

	RGBA FBXToNativeType(const FbxColor& value)
	{
		Color native;
		native.r = (float)value[0];
		native.g = (float)value[1];
		native.b = (float)value[2];
		native.a = (float)value[3];

		return native.getAsRGBA();
	}

	FbxSurfaceMaterial* FBXToNativeType(FbxSurfaceMaterial* const& value)
	{
		return value;
	}

	int FBXToNativeType(const int & value)
	{
		return value;
	}

	FBXImporter::FBXImporter()
		:SpecificImporter(), mFBXManager(nullptr)
	{
		mExtensions.push_back(L"fbx");
		mExtensions.push_back(L"obj");
		mExtensions.push_back(L"dae");
	}

	FBXImporter::~FBXImporter() 
	{

	}

	bool FBXImporter::isExtensionSupported(const WString& ext) const
	{
		WString lowerCaseExt = ext;
		StringUtil::toLowerCase(lowerCaseExt);

		return find(mExtensions.begin(), mExtensions.end(), lowerCaseExt) != mExtensions.end();
	}

	bool FBXImporter::isMagicNumberSupported(const UINT8* magicNumPtr, UINT32 numBytes) const
	{
		return true; // FBX files can be plain-text so I don't even check for magic number
	}

	SPtr<ImportOptions> FBXImporter::createImportOptions() const
	{
		return bs_shared_ptr_new<MeshImportOptions>();
	}

	SPtr<Resource> FBXImporter::import(const Path& filePath, SPtr<const ImportOptions> importOptions)
	{
		MESH_DESC desc;

		Vector<FBXAnimationClipData> dummy;
		SPtr<RendererMeshData> rendererMeshData = importMeshData(filePath, importOptions, desc.subMeshes, dummy, 
			desc.skeleton, desc.morphShapes);

		const MeshImportOptions* meshImportOptions = static_cast<const MeshImportOptions*>(importOptions.get());

		desc.usage = MU_STATIC;
		if (meshImportOptions->getCPUCached())
			desc.usage |= MU_CPUCACHED;

		if (meshImportOptions->getCPUReadable())
			desc.usage |= MU_CPUREADABLE;

		SPtr<Mesh> mesh = Mesh::_createPtr(rendererMeshData->getData(), desc);

		WString fileName = filePath.getWFilename(false);
		mesh->setName(fileName);

		return mesh;
	}

	Vector<SubResourceRaw> FBXImporter::importAll(const Path& filePath, SPtr<const ImportOptions> importOptions)
	{
		MESH_DESC desc;

		Vector<FBXAnimationClipData> animationClips;
		SPtr<RendererMeshData> rendererMeshData = importMeshData(filePath, importOptions, desc.subMeshes, animationClips, 
			desc.skeleton, desc.morphShapes);

		const MeshImportOptions* meshImportOptions = static_cast<const MeshImportOptions*>(importOptions.get());

		desc.usage = MU_STATIC;
		if (meshImportOptions->getCPUCached())
			desc.usage |= MU_CPUCACHED;

		if (meshImportOptions->getCPUReadable())
			desc.usage |= MU_CPUREADABLE;

		SPtr<Mesh> mesh = Mesh::_createPtr(rendererMeshData->getData(), desc);

		WString fileName = filePath.getWFilename(false);
		mesh->setName(fileName);

		Vector<SubResourceRaw> output;
		if(mesh != nullptr)
		{
			output.push_back({ L"primary", mesh });

			CollisionMeshType collisionMeshType = meshImportOptions->getCollisionMeshType();
			if(collisionMeshType != CollisionMeshType::None)
			{
				if(Physics::isStarted())
				{
					PhysicsMeshType type = collisionMeshType == CollisionMeshType::Convex ? 
						PhysicsMeshType::Convex : PhysicsMeshType::Triangle;

					SPtr<PhysicsMesh> physicsMesh = PhysicsMesh::_createPtr(rendererMeshData->getData(), type);

					output.push_back({ L"collision", physicsMesh });
				}
				else
				{
					LOGWRN("Cannot generate a collision mesh as the physics module was not started.");
				}
			}

			Vector<ImportedAnimationEvents> events = meshImportOptions->getAnimationEvents();
			for(auto& entry : animationClips)
			{
				SPtr<AnimationClip> clip = AnimationClip::_createPtr(entry.curves, entry.isAdditive, entry.sampleRate, 
					entry.rootMotion);
				
				for(auto& eventsEntry : events)
				{
					if(entry.name == eventsEntry.name)
					{
						clip->setEvents(eventsEntry.events);
						break;
					}
				}

				output.push_back({ toWString(entry.name), clip });
			}
		}

		return output;
	}

	SPtr<RendererMeshData> FBXImporter::importMeshData(const Path& filePath, SPtr<const ImportOptions> importOptions, 
		Vector<SubMesh>& subMeshes, Vector<FBXAnimationClipData>& animation, SPtr<Skeleton>& skeleton, 
		SPtr<MorphShapes>& morphShapes)
	{
		FbxScene* fbxScene = nullptr;

		if (!startUpSdk(fbxScene))
			return nullptr;

		if (!loadFBXFile(fbxScene, filePath))
			return nullptr;

		const MeshImportOptions* meshImportOptions = static_cast<const MeshImportOptions*>(importOptions.get());
		FBXImportOptions fbxImportOptions;
		fbxImportOptions.importNormals = meshImportOptions->getImportNormals();
		fbxImportOptions.importTangents = meshImportOptions->getImportTangents();
		fbxImportOptions.importAnimation = meshImportOptions->getImportAnimation();
		fbxImportOptions.importBlendShapes = meshImportOptions->getImportBlendShapes();
		fbxImportOptions.importSkin = meshImportOptions->getImportSkin();
		fbxImportOptions.importScale = meshImportOptions->getImportScale();

		FBXImportScene importedScene;
		bakeTransforms(fbxScene);
		parseScene(fbxScene, fbxImportOptions, importedScene);

		if (fbxImportOptions.importBlendShapes)
			importBlendShapes(importedScene, fbxImportOptions);

		if (fbxImportOptions.importSkin)
			importSkin(importedScene, fbxImportOptions);

		if (fbxImportOptions.importAnimation)
			importAnimations(fbxScene, fbxImportOptions, importedScene);

		splitMeshVertices(importedScene);
		generateMissingTangentSpace(importedScene, fbxImportOptions);

		SPtr<RendererMeshData> rendererMeshData = generateMeshData(importedScene, fbxImportOptions, subMeshes);

		skeleton = createSkeleton(importedScene, subMeshes.size() > 1);
		morphShapes = createMorphShapes(importedScene);		

		// Import animation clips
		if (!importedScene.clips.empty())
		{
			Vector<AnimationSplitInfo> splits = meshImportOptions->getAnimationClipSplits();
			convertAnimations(importedScene.clips, splits, skeleton, meshImportOptions->getImportRootMotion(), animation);
		}

		// TODO - Later: Optimize mesh: Remove bad and degenerate polygons, weld nearby vertices, optimize for vertex cache

		shutDownSdk();

		return rendererMeshData;
	}

	SPtr<Skeleton> FBXImporter::createSkeleton(const FBXImportScene& scene, bool sharedRoot)
	{
		Vector<BONE_DESC> allBones;
		UnorderedMap<FBXImportNode*, UINT32> boneMap;

		for (auto& mesh : scene.meshes)
		{
			// Create bones
			UINT32 numBones = (UINT32)mesh->bones.size();
			for (auto& fbxBone : mesh->bones)
			{
				UINT32 boneIdx = (UINT32)allBones.size();
				boneMap[fbxBone.node] = boneIdx;

				allBones.push_back(BONE_DESC());
				BONE_DESC& bone = allBones.back();

				bone.name = fbxBone.node->name;
				bone.invBindPose = fbxBone.bindPose;
			}
		}

		// Generate skeleton
		if (allBones.size() > 0)
		{
			// Find bone parents
			UINT32 numProcessedBones = 0;

			// Generate common root bone for all meshes
			UINT32 rootBoneIdx = (UINT32)-1;
			if (sharedRoot)
			{
				rootBoneIdx = (UINT32)allBones.size();

				allBones.push_back(BONE_DESC());
				BONE_DESC& bone = allBones.back();

				bone.name = "MultiMeshRoot";
				bone.invBindPose = Matrix4::IDENTITY;
				bone.parent = (UINT32)-1;

				numProcessedBones++;
			}

			Stack<std::pair<FBXImportNode*, UINT32>> todo;
			todo.push({ scene.rootNode, rootBoneIdx });

			while (!todo.empty())
			{
				auto entry = todo.top();
				todo.pop();

				FBXImportNode* node = entry.first;
				UINT32 parentBoneIdx = entry.second;

				auto boneIter = boneMap.find(node);
				if (boneIter != boneMap.end())
				{
					UINT32 boneIdx = boneIter->second;
					allBones[boneIdx].parent = parentBoneIdx;
					numProcessedBones++;

					parentBoneIdx = boneIdx;
				}

				for (auto& child : node->children)
					todo.push({ child, parentBoneIdx });
			}

			UINT32 numAllBones = (UINT32)allBones.size();
			if (numProcessedBones == numAllBones)
				return Skeleton::create(allBones.data(), numAllBones);

			LOGERR("Not all bones were found in the node hierarchy. Skeleton invalid.");
		}

		return nullptr;
	}

	SPtr<MorphShapes> FBXImporter::createMorphShapes(const FBXImportScene& scene)
	{
		// Combine morph shapes from all sub-meshes, and transform them
		struct RawMorphShape
		{
			String name;
			float weight;
			Vector<MorphVertex> vertices;
		};

		UnorderedMap<String, UnorderedMap<String, RawMorphShape>> allRawMorphShapes;
		UINT32 totalNumVertices = 0;

		// Note: Order in which we combine meshes must match the order in MeshData::combine
		for (auto& mesh : scene.meshes)
		{
			UINT32 numVertices = (UINT32)mesh->positions.size();
			UINT32 numNormals = (UINT32)mesh->normals.size();
			bool hasNormals = numVertices == numNormals;

			for (auto& node : mesh->referencedBy)
			{
				Matrix4 worldTransform = node->worldTransform * scene.globalScale;
				Matrix4 worldTransformIT = worldTransform.inverse();
				worldTransformIT = worldTransformIT.transpose();

				// Copy & transform positions
				for(auto& blendShape : mesh->blendShapes)
				{
					UnorderedMap<String, RawMorphShape>& channelShapes = allRawMorphShapes[blendShape.name];

					for(auto& blendFrame : blendShape.frames)
					{
						RawMorphShape& shape = channelShapes[blendFrame.name];
						shape.name = blendFrame.name;
						shape.weight = blendFrame.weight;

						UINT32 frameNumVertices = (UINT32)blendFrame.positions.size();
						if (frameNumVertices == numVertices)
						{
							for (UINT32 i = 0; i < numVertices; i++)
							{
								Vector3 meshPosition = worldTransform.multiplyAffine(mesh->positions[i]);
								Vector3 blendPosition = worldTransform.multiplyAffine(blendFrame.positions[i]);

								Vector3 positionDelta = blendPosition - meshPosition;
								Vector3 normalDelta;
								if (hasNormals)
								{
									Vector3 blendNormal = worldTransformIT.multiplyDirection(blendFrame.normals[i]);
									blendNormal = Vector3::normalize(blendNormal);

									Vector3 meshNormal = worldTransformIT.multiplyDirection(mesh->normals[i]);
									meshNormal = Vector3::normalize(meshNormal);

									normalDelta = blendNormal - meshNormal;
								}
								else
									normalDelta = Vector3::ZERO;

								if (positionDelta.squaredLength() > 0.000001f || normalDelta.squaredLength() > 0.0001f)
									shape.vertices.push_back(MorphVertex(positionDelta, normalDelta, totalNumVertices + i));
							}
						}
						else
						{
							LOGERR("Corrupt blend shape frame. Number of vertices doesn't match the number of mesh vertices.");
						}
					}
				}

				totalNumVertices += numVertices;
			}
		}

		// Create morph shape object from combined shape data
		SPtr<MorphShapes> morphShapes;
		Vector<SPtr<MorphChannel>> allChannels;
		for (auto& channel : allRawMorphShapes)
		{
			Vector<SPtr<MorphShape>> channelShapes;
			for (auto& entry : channel.second)
			{
				RawMorphShape& shape = entry.second;
				shape.vertices.shrink_to_fit();

				SPtr<MorphShape> morphShape = MorphShape::create(shape.name, shape.weight, shape.vertices);
				channelShapes.push_back(morphShape);
			}

			if(channelShapes.size() > 0)
			{
				SPtr<MorphChannel> morphChannel = MorphChannel::create(channel.first, channelShapes);
				allChannels.push_back(morphChannel);
			}
		}

		if (!allChannels.empty())
			return MorphShapes::create(allChannels, totalNumVertices);

		return morphShapes;
	}

	bool FBXImporter::startUpSdk(FbxScene*& scene)
	{
		mFBXManager = FbxManager::Create();
		if (mFBXManager == nullptr)
		{
			LOGERR("FBX import failed: FBX SDK failed to initialize. FbxManager::Create() failed.");
			return false;
		}

		FbxIOSettings* ios = FbxIOSettings::Create(mFBXManager, IOSROOT);
		mFBXManager->SetIOSettings(ios);

		scene = FbxScene::Create(mFBXManager, "Import Scene");
		if (scene == nullptr)
		{
			LOGWRN("FBX import failed: Failed to create FBX scene.");
			return false;
		}

		return true;
	}

	void FBXImporter::shutDownSdk()
	{
		mFBXManager->Destroy();
		mFBXManager = nullptr;
	}

	bool FBXImporter::loadFBXFile(FbxScene* scene, const Path& filePath)
	{
		int lFileMajor, lFileMinor, lFileRevision;
		int lSDKMajor,  lSDKMinor,  lSDKRevision;
		FbxManager::GetFileFormatVersion(lSDKMajor, lSDKMinor, lSDKRevision);

		FbxImporter* importer = FbxImporter::Create(mFBXManager, "");
		bool importStatus = importer->Initialize(filePath.toString().c_str(), -1, mFBXManager->GetIOSettings());
		
		importer->GetFileVersion(lFileMajor, lFileMinor, lFileRevision);

		if(!importStatus)
		{
			LOGERR("FBX import failed: Call to FbxImporter::Initialize() failed.\n" +
				String("Error returned: %s\n\n") + String(importer->GetStatus().GetErrorString()));
			return false;
		}

		mFBXManager->GetIOSettings()->SetBoolProp(IMP_FBX_TEXTURE, false);
		mFBXManager->GetIOSettings()->SetBoolProp(IMP_FBX_GOBO, false);

		importStatus = importer->Import(scene);

		if(!importStatus)
		{
			importer->Destroy();
			
			LOGERR("FBX import failed: Call to FbxImporter::Import() failed.\n" +
				String("Error returned: %s\n\n") + String(importer->GetStatus().GetErrorString()));
			return false;
		}

		FbxAxisSystem fileCoordSystem = scene->GetGlobalSettings().GetAxisSystem();
		FbxAxisSystem bsCoordSystem(FbxAxisSystem::eYAxis, FbxAxisSystem::eParityOdd, FbxAxisSystem::eRightHanded);
		if (fileCoordSystem != bsCoordSystem)
			bsCoordSystem.ConvertScene(scene);

		importer->Destroy();
		return true;
	}

	void FBXImporter::parseScene(FbxScene* scene, const FBXImportOptions& options, FBXImportScene& outputScene)
	{
		float importScale = 1.0f;
		if (options.importScale > 0.0001f)
			importScale = options.importScale;

		FbxSystemUnit units = scene->GetGlobalSettings().GetSystemUnit();
		FbxSystemUnit bsScaledUnits(100.0f);

		outputScene.scaleFactor = (float)units.GetConversionFactorTo(bsScaledUnits) * importScale;
		outputScene.globalScale = Matrix4::scaling(outputScene.scaleFactor);
		outputScene.rootNode = createImportNode(outputScene, scene->GetRootNode(), nullptr);

		Stack<FbxNode*> todo;
		todo.push(scene->GetRootNode());

		while(!todo.empty())
		{
			FbxNode* curNode = todo.top();
			FBXImportNode* curImportNode = outputScene.nodeMap[curNode];
			todo.pop();

			FbxNodeAttribute* attrib = curNode->GetNodeAttribute();
			if(attrib != nullptr)
			{
				FbxNodeAttribute::EType attribType = attrib->GetAttributeType();

				switch(attribType)
				{
				case FbxNodeAttribute::eNurbs:
				case FbxNodeAttribute::eNurbsSurface:
				case FbxNodeAttribute::ePatch:
				{
					FbxGeometryConverter geomConverter(mFBXManager);
					attrib = geomConverter.Triangulate(attrib, true);

					if (attrib->GetAttributeType() == FbxNodeAttribute::eMesh)
					{
						FbxMesh* mesh = static_cast<FbxMesh*>(attrib);
						mesh->RemoveBadPolygons();

						parseMesh(mesh, curImportNode, options, outputScene);
					}
				}
					break;
				case FbxNodeAttribute::eMesh:
					{
						FbxMesh* mesh = static_cast<FbxMesh*>(attrib);
						mesh->RemoveBadPolygons();

						if(!mesh->IsTriangleMesh())
						{
							FbxGeometryConverter geomConverter(mFBXManager);
							geomConverter.Triangulate(mesh, true);
							attrib = curNode->GetNodeAttribute();
							mesh = static_cast<FbxMesh*>(attrib);
						}

						parseMesh(mesh, curImportNode, options, outputScene);
					}
					break;
				default:
					break;
				}
			}

			for (int i = 0; i < curNode->GetChildCount(); i++)
			{
				FbxNode* childNode = curNode->GetChild(i);
				createImportNode(outputScene, childNode, curImportNode);

				todo.push(childNode);
			}
		}
	}

	FBXImportNode* FBXImporter::createImportNode(FBXImportScene& scene, FbxNode* fbxNode, FBXImportNode* parent)
	{
		FBXImportNode* node = bs_new<FBXImportNode>();

		Vector3 translation = FBXToNativeType(fbxNode->LclTranslation.Get());
		Vector3 rotationEuler = FBXToNativeType(fbxNode->LclRotation.Get());
		Vector3 scale = FBXToNativeType(fbxNode->LclScaling.Get());

		Quaternion rotation((Radian)rotationEuler.x, (Radian)rotationEuler.y, (Radian)rotationEuler.z);

		node->localTransform.setTRS(translation, rotation, scale);
		node->name = fbxNode->GetNameWithoutNameSpacePrefix().Buffer();
		node->fbxNode = fbxNode;

		if (parent != nullptr)
		{
			node->worldTransform = node->localTransform * parent->worldTransform;

			parent->children.push_back(node);
		}
		else
			node->worldTransform = node->localTransform;

		scene.nodeMap.insert(std::make_pair(fbxNode, node));

		return node;
	}

	void FBXImporter::splitMeshVertices(FBXImportScene& scene)
	{
		Vector<FBXImportMesh*> splitMeshes;

		for (auto& mesh : scene.meshes)
		{
			FBXImportMesh* splitMesh = bs_new<FBXImportMesh>();
			splitMesh->fbxMesh = mesh->fbxMesh;
			splitMesh->referencedBy = mesh->referencedBy;
			splitMesh->bones = mesh->bones;

			FBXUtility::splitVertices(*mesh, *splitMesh);
			FBXUtility::flipWindingOrder(*splitMesh);
			splitMeshes.push_back(splitMesh);

			bs_delete(mesh);
		}

		scene.meshes = splitMeshes;
	}

	void FBXImporter::convertAnimations(const Vector<FBXAnimationClip>& clips, const Vector<AnimationSplitInfo>& splits,
		const SPtr<Skeleton>& skeleton, bool importRootMotion, Vector<FBXAnimationClipData>& output)
	{
		UnorderedSet<String> names;

		String rootBoneName;
		if (skeleton == nullptr)
			importRootMotion = false;
		else
		{
			UINT32 rootBoneIdx = skeleton->getRootBoneIndex();
			if (rootBoneIdx == (UINT32)-1)
				importRootMotion = false;
			else
				rootBoneName = skeleton->getBoneInfo(rootBoneIdx).name;
		}

		bool isFirstClip = true;
		for (auto& clip : clips)
		{
			SPtr<AnimationCurves> curves = bs_shared_ptr_new<AnimationCurves>();
			SPtr<RootMotion> rootMotion;

			// Find offset so animations start at time 0
			float animStart = std::numeric_limits<float>::infinity();

			for (auto& bone : clip.boneAnimations)
			{
				if(bone.translation.getNumKeyFrames() > 0)
					animStart = std::min(bone.translation.getKeyFrame(0).time, animStart);

				if (bone.rotation.getNumKeyFrames() > 0)
					animStart = std::min(bone.rotation.getKeyFrame(0).time, animStart);

				if (bone.scale.getNumKeyFrames() > 0)
					animStart = std::min(bone.scale.getKeyFrame(0).time, animStart);
			}

			for (auto& anim : clip.blendShapeAnimations)
			{
				if (anim.curve.getNumKeyFrames() > 0)
					animStart = std::min(anim.curve.getKeyFrame(0).time, animStart);
			}

			AnimationCurveFlags blendShapeFlags = AnimationCurveFlag::ImportedCurve | AnimationCurveFlag::MorphFrame;
			if (animStart != 0.0f && animStart != std::numeric_limits<float>::infinity())
			{
				for (auto& bone : clip.boneAnimations)
				{
					TAnimationCurve<Vector3> translation = AnimationUtility::offsetCurve(bone.translation, -animStart);
					TAnimationCurve<Quaternion> rotation = AnimationUtility::offsetCurve(bone.rotation, -animStart);
					TAnimationCurve<Vector3> scale = AnimationUtility::offsetCurve(bone.scale, -animStart);

					if(importRootMotion && bone.node->name == rootBoneName)
						rootMotion = bs_shared_ptr_new<RootMotion>(translation, rotation);
					else
					{
						curves->position.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, translation });
						curves->rotation.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, rotation });
						curves->scale.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, scale });
					}
				}

				for (auto& anim : clip.blendShapeAnimations)
				{
					TAnimationCurve<float> curve = AnimationUtility::offsetCurve(anim.curve, -animStart);
					curves->generic.push_back({ anim.blendShape, blendShapeFlags, curve });
				}
			}
			else
			{
				for (auto& bone : clip.boneAnimations)
				{
					if (importRootMotion && bone.node->name == rootBoneName)
						rootMotion = bs_shared_ptr_new<RootMotion>(bone.translation, bone.rotation);
					else
					{
						curves->position.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, bone.translation });
						curves->rotation.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, bone.rotation });
						curves->scale.push_back({ bone.node->name, AnimationCurveFlag::ImportedCurve, bone.scale });
					}
				}

				for (auto& anim : clip.blendShapeAnimations)
					curves->generic.push_back({ anim.blendShape, blendShapeFlags, anim.curve });
			}

			// See if any splits are required. We only split the first clip as it is assumed if FBX has multiple clips the
			// user has the ability to split them externally.
			if(isFirstClip && !splits.empty())
			{
				float secondsPerFrame = 1.0f / clip.sampleRate;

				for(auto& split : splits)
				{
					SPtr<AnimationCurves> splitClipCurve = bs_shared_ptr_new<AnimationCurves>();
					SPtr<RootMotion> splitRootMotion;

					auto splitCurves = [&](auto& inCurves, auto& outCurves)
					{
						UINT32 numCurves = (UINT32)inCurves.size();
						outCurves.resize(numCurves);

						for (UINT32 i = 0; i < numCurves; i++)
						{
							auto& animCurve = inCurves[i].curve;
							outCurves[i].name = inCurves[i].name;

							UINT32 numFrames = animCurve.getNumKeyFrames();
							if (numFrames == 0)
								continue;

							float startTime = split.startFrame * secondsPerFrame;
							float endTime = split.endFrame * secondsPerFrame;

							outCurves[i].curve = inCurves[i].curve.split(startTime, endTime);

							if (split.isAdditive)
								outCurves[i].curve.makeAdditive();
						}
					};

					splitCurves(curves->position, splitClipCurve->position);
					splitCurves(curves->rotation, splitClipCurve->rotation);
					splitCurves(curves->scale, splitClipCurve->scale);
					splitCurves(curves->generic, splitClipCurve->generic);

					if(rootMotion != nullptr)
					{
						auto splitCurve = [&](auto& inCurve, auto& outCurve)
						{
							UINT32 numFrames = inCurve.getNumKeyFrames();
							if (numFrames > 0)
							{
								float startTime = split.startFrame * secondsPerFrame;
								float endTime = split.endFrame * secondsPerFrame;

								outCurve = inCurve.split(startTime, endTime);

								if (split.isAdditive)
									outCurve.makeAdditive();
							}
						};

						splitRootMotion = bs_shared_ptr_new<RootMotion>();
						splitCurve(rootMotion->position, splitRootMotion->position);
						splitCurve(rootMotion->rotation, splitRootMotion->rotation);
					}

					// Search for a unique name
					String name = split.name;
					UINT32 attemptIdx = 0;
					while (names.find(name) != names.end())
					{
						name = clip.name + "_" + toString(attemptIdx);
						attemptIdx++;
					}

					names.insert(name);
					output.push_back(FBXAnimationClipData(name, split.isAdditive, clip.sampleRate, splitClipCurve,
						splitRootMotion));
				}
			}
			else
			{
				// Search for a unique name
				String name = clip.name;
				UINT32 attemptIdx = 0;
				while(names.find(name) != names.end())
				{
					name = clip.name + "_" + toString(attemptIdx);
					attemptIdx++;
				}

				names.insert(name);
				output.push_back(FBXAnimationClipData(name, false, clip.sampleRate, curves, rootMotion));
			}

			isFirstClip = false;
		}
	}

	SPtr<RendererMeshData> FBXImporter::generateMeshData(const FBXImportScene& scene, const FBXImportOptions& options, 
		Vector<SubMesh>& outputSubMeshes)
	{
		Vector<SPtr<MeshData>> allMeshData;
		Vector<Vector<SubMesh>> allSubMeshes;
		Vector<BONE_DESC> allBones;
		UnorderedMap<FBXImportNode*, UINT32> boneMap;
		UINT32 boneIndexOffset = 0;

		for (auto& mesh : scene.meshes)
		{
			Vector<Vector<UINT32>> indicesPerMaterial;
			for (UINT32 i = 0; i < (UINT32)mesh->indices.size(); i++)
			{
				while ((UINT32)mesh->materials[i] >= (UINT32)indicesPerMaterial.size())
					indicesPerMaterial.push_back(Vector<UINT32>());

				indicesPerMaterial[mesh->materials[i]].push_back(mesh->indices[i]);
			}

			UINT32* orderedIndices = (UINT32*)bs_alloc((UINT32)mesh->indices.size() * sizeof(UINT32));
			Vector<SubMesh> subMeshes;
			UINT32 currentIndex = 0;

			for (auto& subMeshIndices : indicesPerMaterial)
			{
				UINT32 indexCount = (UINT32)subMeshIndices.size();
				UINT32* dest = orderedIndices + currentIndex;
				memcpy(dest, subMeshIndices.data(), indexCount * sizeof(UINT32));

				subMeshes.push_back(SubMesh(currentIndex, indexCount, DOT_TRIANGLE_LIST));

				currentIndex += indexCount;
			}

			UINT32 vertexLayout = (UINT32)VertexLayout::Position;

			size_t numVertices = mesh->positions.size();
			bool hasColors = mesh->colors.size() == numVertices;
			bool hasNormals = mesh->normals.size() == numVertices;
			bool hasBoneInfluences = mesh->boneInfluences.size() == numVertices;

			if (hasColors)
				vertexLayout |= (UINT32)VertexLayout::Color;

			bool hasTangents = false;
			if (hasNormals)
			{
				vertexLayout |= (UINT32)VertexLayout::Normal;

				if (mesh->tangents.size() == numVertices &&
					mesh->bitangents.size() == numVertices)
				{
					vertexLayout |= (UINT32)VertexLayout::Tangent;
					hasTangents = true;
				}
			}

			if (hasBoneInfluences)
				vertexLayout |= (UINT32)VertexLayout::BoneWeights;

			for (UINT32 i = 0; i < FBX_IMPORT_MAX_UV_LAYERS; i++)
			{
				if (mesh->UV[i].size() == numVertices)
				{
					if (i == 0)
						vertexLayout |= (UINT32)VertexLayout::UV0;
					else if (i == 1)
						vertexLayout |= (UINT32)VertexLayout::UV1;
				}
			}

			UINT32 numIndices = (UINT32)mesh->indices.size();
			for (auto& node : mesh->referencedBy)
			{
				Matrix4 worldTransform = node->worldTransform * scene.globalScale;
				Matrix4 worldTransformIT = worldTransform.inverse();
				worldTransformIT = worldTransformIT.transpose();

				SPtr<RendererMeshData> meshData = RendererMeshData::create((UINT32)numVertices, numIndices, (VertexLayout)vertexLayout);

				// Copy indices
				meshData->setIndices((UINT32*)mesh->indices.data(), numIndices * sizeof(UINT32));

				// Copy & transform positions
				UINT32 positionsSize = sizeof(Vector3) * (UINT32)numVertices;
				Vector3* transformedPositions = (Vector3*)bs_stack_alloc(positionsSize);

				for (UINT32 i = 0; i < (UINT32)numVertices; i++)
					transformedPositions[i] = worldTransform.multiplyAffine((Vector3)mesh->positions[i]);

				meshData->setPositions(transformedPositions, positionsSize);
				bs_stack_free(transformedPositions);

				// Copy & transform normals
				if (hasNormals)
				{
					UINT32 normalsSize = sizeof(Vector3) * (UINT32)numVertices;
					Vector3* transformedNormals = (Vector3*)bs_stack_alloc(normalsSize);

					// Copy, convert & transform tangents & bitangents
					if (hasTangents)
					{
						UINT32 tangentsSize = sizeof(Vector4) * (UINT32)numVertices;
						Vector4* transformedTangents = (Vector4*)bs_stack_alloc(tangentsSize);

						for (UINT32 i = 0; i < (UINT32)numVertices; i++)
						{
							Vector3 normal = (Vector3)mesh->normals[i];
							normal = worldTransformIT.multiplyDirection(normal);
							transformedNormals[i] = Vector3::normalize(normal);

							Vector3 tangent = (Vector3)mesh->tangents[i];
							tangent = Vector3::normalize(worldTransformIT.multiplyDirection(tangent));

							Vector3 bitangent = (Vector3)mesh->bitangents[i];
							bitangent = worldTransformIT.multiplyDirection(bitangent);

							Vector3 engineBitangent = Vector3::cross(normal, tangent);
							float sign = Vector3::dot(engineBitangent, bitangent);

							transformedTangents[i] = Vector4(tangent.x, tangent.y, tangent.z, sign > 0 ? 1.0f : -1.0f);
						}

						meshData->setTangents(transformedTangents, tangentsSize);
						bs_stack_free(transformedTangents);
					}
					else // Just normals
					{
						for (UINT32 i = 0; i < (UINT32)numVertices; i++)
							transformedNormals[i] = Vector3::normalize(worldTransformIT.multiplyDirection((Vector3)mesh->normals[i]));
					}

					meshData->setNormals(transformedNormals, normalsSize);
					bs_stack_free(transformedNormals);
				}

				// Copy colors
				if (hasColors)
				{
					meshData->setColors(mesh->colors.data(), sizeof(UINT32) * (UINT32)numVertices);
				}

				// Copy UV
				int writeUVIDx = 0;
				for (auto& uvLayer : mesh->UV)
				{
					if (uvLayer.size() == numVertices)
					{
						UINT32 size = sizeof(Vector2) * (UINT32)numVertices;
						Vector2* transformedUV = (Vector2*)bs_stack_alloc(size);

						UINT32 i = 0;
						for (auto& uv : uvLayer)
						{
							transformedUV[i] = uv;
							transformedUV[i].y = 1.0f - uv.y;

							i++;
						}

						if (writeUVIDx == 0)
							meshData->setUV0(transformedUV, size);
						else if (writeUVIDx == 1)
							meshData->setUV1(transformedUV, size);

						bs_stack_free(transformedUV);

						writeUVIDx++;
					}
				}

				// Copy bone influences
				if(hasBoneInfluences)
				{
					UINT32 bufferSize = sizeof(BoneWeight) * (UINT32)numVertices;
					BoneWeight* weights = (BoneWeight*)bs_stack_alloc(bufferSize);
					for(UINT32 i = 0; i < (UINT32)numVertices; i++)
					{
						weights[i].index0 = mesh->boneInfluences[i].indices[0] + boneIndexOffset;
						weights[i].index1 = mesh->boneInfluences[i].indices[1] + boneIndexOffset;
						weights[i].index2 = mesh->boneInfluences[i].indices[2] + boneIndexOffset;
						weights[i].index3 = mesh->boneInfluences[i].indices[3] + boneIndexOffset;

						weights[i].weight0 = mesh->boneInfluences[i].weights[0];
						weights[i].weight1 = mesh->boneInfluences[i].weights[1];
						weights[i].weight2 = mesh->boneInfluences[i].weights[2];
						weights[i].weight3 = mesh->boneInfluences[i].weights[3];
					}

					meshData->setBoneWeights(weights, bufferSize);
					bs_stack_free(weights);
				}

				allMeshData.push_back(meshData->getData());
				allSubMeshes.push_back(subMeshes);
			}

			UINT32 numBones = (UINT32)mesh->bones.size();
			boneIndexOffset += numBones;
		}

		if (allMeshData.size() > 1)
		{
			return RendererMeshData::create(MeshData::combine(allMeshData, allSubMeshes, outputSubMeshes));
		}
		else if (allMeshData.size() == 1)
		{
			outputSubMeshes = allSubMeshes[0];
			return RendererMeshData::create(allMeshData[0]);
		}

		return nullptr;
	}

	template<class TFBX, class TNative>
	class FBXDirectIndexer
	{
	public:
		FBXDirectIndexer(const FbxLayerElementTemplate<TFBX>& layer)
			:mElementArray(layer.GetDirectArray()),
			mElementCount(mElementArray.GetCount())
		{}

		bool get(int index, TNative& output) const
		{
			if (index < 0 || index >= mElementCount)
				return false;

			output = FBXToNativeType(mElementArray.GetAt(index));
			return true;
		}

		bool isEmpty() const
		{
			return mElementCount == 0;
		}

	private:
		const FbxLayerElementArrayTemplate<TFBX>& mElementArray;
		int mElementCount;
	};

	template<class TFBX, class TNative>
	class FBXIndexIndexer
	{
	public:
		FBXIndexIndexer(const FbxLayerElementTemplate<TFBX>& layer)
			:mElementArray(layer.GetDirectArray()),
			mIndexArray(layer.GetIndexArray()),
			mElementCount(mElementArray.GetCount()),
			mIndexCount(mIndexArray.GetCount())
		{}

		bool get(int index, TNative& output) const
		{
			if (index < 0 || index >= mIndexCount)
				return false;

			int actualIndex = mIndexArray.GetAt(index);

			if (actualIndex < 0 || actualIndex >= mElementCount)
				return false;

			output = FBXToNativeType(mElementArray.GetAt(actualIndex));
			return true;
		}

		bool isEmpty() const
		{
			return mElementCount == 0 || mIndexCount == 0;
		}

	private:
		const FbxLayerElementArrayTemplate<TFBX>& mElementArray;
		const FbxLayerElementArrayTemplate<int>& mIndexArray;
		int mElementCount;
		int mIndexCount;
	};

	template<class TFBX, class TNative, class TIndexer>
	void readLayerData(FbxLayerElementTemplate<TFBX>& layer, Vector<TNative>& output, const Vector<int>& indices)
	{
		TIndexer indexer(layer);
		if (indexer.isEmpty())
			return;

		output.resize(indices.size());

		FbxLayerElement::EMappingMode mappingMode = layer.GetMappingMode();

		UINT32 indexCount = (UINT32)indices.size();
		switch (mappingMode)
		{
		case FbxLayerElement::eByControlPoint:
			for (UINT32 i = 0; i < indexCount; i++)
			{
				int index = indices[i];
				indexer.get(index, output[i]);
			}
			break;
		case FbxLayerElement::eByPolygonVertex:
			for (UINT32 i = 0; i < indexCount; i++)
				indexer.get(i, output[i]);
			break;
		case FbxLayerElement::eByPolygon:
			// We expect mesh to be triangulated here
		{
			UINT32 polygonCount = indexCount / 3;
			UINT32 index = 0;

			for (UINT32 i = 0; i < polygonCount; i++)
			{
				TNative value;
				indexer.get(i, value);

				output[index++] = value;
				output[index++] = value;
				output[index++] = value;
			}
		}
			break;
		case FbxLayerElement::eAllSame:
		{
			TNative value;
			indexer.get(0, value);

			for (UINT32 i = 0; i < indexCount; i++)
				output[i] = value;
		}
			break;
		default:
			LOGWRN("FBX Import: Unsupported layer mapping mode.");
			break;
		}
	}

	template<class TFBX, class TNative>
	void readLayerData(FbxLayerElementTemplate<TFBX>& layer, Vector<TNative>& output, const Vector<int>& indices)
	{
		FbxLayerElement::EReferenceMode refMode = layer.GetReferenceMode();

		if (refMode == FbxLayerElement::eDirect)
			readLayerData<TFBX, TNative, FBXDirectIndexer<TFBX, TNative> >(layer, output, indices);
		else if (refMode == FbxLayerElement::eIndexToDirect)
			readLayerData<TFBX, TNative, FBXIndexIndexer<TFBX, TNative> >(layer, output, indices);
		else
			LOGWRN("FBX Import: Unsupported layer reference mode.");
	}

	void FBXImporter::parseMesh(FbxMesh* mesh, FBXImportNode* parentNode, const FBXImportOptions& options, FBXImportScene& outputScene)
	{
		// Check if valid
		if (!mesh->IsTriangleMesh())
			return;

		UINT32 vertexCount = mesh->GetControlPointsCount();
		UINT32 triangleCount = mesh->GetPolygonCount();

		if (vertexCount == 0 || triangleCount == 0)
			return;

		// Register in global mesh array
		FBXImportMesh* importMesh = nullptr;

		auto iterFindMesh = outputScene.meshMap.find(mesh);
		if (iterFindMesh != outputScene.meshMap.end())
		{
			UINT32 meshIdx = iterFindMesh->second;
			outputScene.meshes[meshIdx]->referencedBy.push_back(parentNode);

			return;
		}
		else
		{
			importMesh = bs_new<FBXImportMesh>();
			outputScene.meshes.push_back(importMesh);

			importMesh->referencedBy.push_back(parentNode);
			importMesh->fbxMesh = mesh;

			outputScene.meshMap[mesh] = (UINT32)outputScene.meshes.size() - 1;
		}

		// Import vertices
		importMesh->positions.resize(vertexCount);
		FbxVector4* controlPoints = mesh->GetControlPoints();

		for (UINT32 i = 0; i < vertexCount; i++)
			importMesh->positions[i] = FBXToNativeType(controlPoints[i]);

		// Import triangles
		UINT32 indexCount = triangleCount * 3;
		importMesh->indices.resize(indexCount);

		int* fbxIndices = mesh->GetPolygonVertices();
		importMesh->indices.assign(fbxIndices, fbxIndices + indexCount);

		// Import UVs
		Vector<FbxLayerElementUV*> fbxUVLayers;

		//// Search the diffuse layers first
		for (UINT32 i = 0; i < FBX_IMPORT_MAX_UV_LAYERS; i++)
		{
			FbxLayer* layer = mesh->GetLayer(i, FbxLayerElement::eUV);
			if (layer == nullptr)
				continue;

			for (int j = FbxLayerElement::eTextureDiffuse; j < FbxLayerElement::eTypeCount; j++)
			{
				FbxLayerElementUV* uvLayer = layer->GetUVs((FbxLayerElement::EType)j);
				if (uvLayer == nullptr)
					continue;

				fbxUVLayers.push_back(uvLayer);

				if (fbxUVLayers.size() == FBX_IMPORT_MAX_UV_LAYERS)
					break;
			}

			if (fbxUVLayers.size() == FBX_IMPORT_MAX_UV_LAYERS)
				break;
		}

		//// If there's room, search all others too
		if (fbxUVLayers.size() < FBX_IMPORT_MAX_UV_LAYERS)
		{
			UINT32 numLayers = mesh->GetLayerCount();
			for (UINT32 i = 0; i < numLayers; i++)
			{
				FbxLayer* layer = mesh->GetLayer(i);
				if (layer == nullptr)
					continue;

				for (int j = FbxLayerElement::eTextureDiffuse; j < FbxLayerElement::eTypeCount; j++)
				{
					FbxLayerElementUV* uvLayer = layer->GetUVs((FbxLayerElement::EType)j);
					if (uvLayer == nullptr)
						continue;

					auto iterFind = std::find(fbxUVLayers.begin(), fbxUVLayers.end(), uvLayer);
					if (iterFind != fbxUVLayers.end())
						continue;

					fbxUVLayers.push_back(uvLayer);

					if (fbxUVLayers.size() == FBX_IMPORT_MAX_UV_LAYERS)
						break;
				}

				if (fbxUVLayers.size() == FBX_IMPORT_MAX_UV_LAYERS)
					break;
			}
		}

		for (size_t i = 0; i < fbxUVLayers.size(); i++)
			readLayerData(*fbxUVLayers[i], importMesh->UV[i], importMesh->indices);

		FbxLayer* mainLayer = mesh->GetLayer(0);
		if (mainLayer != nullptr)
		{
			// Import colors
			if (mainLayer->GetVertexColors() != nullptr)
				readLayerData(*mainLayer->GetVertexColors(), importMesh->colors, importMesh->indices);

			// Import normals
			if (options.importNormals)
			{
				bool hasNormals = mainLayer->GetNormals() != nullptr;

				if (!hasNormals)
				{
					if (mainLayer->GetSmoothing() != nullptr)
					{
						FbxLayerElementSmoothing* smoothing = mainLayer->GetSmoothing();

						if (smoothing->GetMappingMode() == FbxLayerElement::eByEdge)
						{
							FbxGeometryConverter converter(mFBXManager);
							converter.ComputePolygonSmoothingFromEdgeSmoothing(mesh, 0);
						}

						readLayerData(*smoothing, importMesh->smoothingGroups, importMesh->indices);

						if (!importMesh->smoothingGroups.empty())
						{
							FBXUtility::normalsFromSmoothing(importMesh->positions, importMesh->indices, 
								importMesh->smoothingGroups, importMesh->normals);
						}
					}
				}
				else
					readLayerData(*mainLayer->GetNormals(), importMesh->normals, importMesh->indices);
			}

			// Import tangents
			if (options.importTangents)
			{
				bool hasTangents = mainLayer->GetTangents() != nullptr && mainLayer->GetBinormals() != nullptr;

				if (!hasTangents)
				{
					if (fbxUVLayers.size() > 0)
						hasTangents = mesh->GenerateTangentsData(0, false);
				}

				if (hasTangents)
				{
					readLayerData(*mainLayer->GetTangents(), importMesh->tangents, importMesh->indices);
					readLayerData(*mainLayer->GetBinormals(), importMesh->bitangents, importMesh->indices);
				}
			}

			// Import material indexes
			if (mainLayer->GetMaterials() != nullptr)
			{
				Vector<FbxSurfaceMaterial*> fbxMaterials;

				readLayerData(*mainLayer->GetMaterials(), fbxMaterials, importMesh->indices);

				UnorderedMap<FbxSurfaceMaterial*, int> materialLookup;
				int nextMaterialIdx = 0;
				for (UINT32 i = 0; i < (UINT32)fbxMaterials.size(); i++)
				{
					auto iterFind = materialLookup.find(fbxMaterials[i]);

					int materialIdx = 0;
					if (iterFind != materialLookup.end())
						materialIdx = iterFind->second;
					else
					{
						materialIdx = nextMaterialIdx++;
						materialLookup[fbxMaterials[i]] = materialIdx;
					}

					importMesh->materials.push_back(materialIdx);
				}
			}
			else
			{
				importMesh->materials.resize(importMesh->indices.size(), 0);
			}
		}
	}

	void FBXImporter::importBlendShapes(FBXImportScene& scene, const FBXImportOptions& options)
	{
		for (auto& mesh : scene.meshes)
		{
			FbxMesh* fbxMesh = mesh->fbxMesh;

			UINT32 deformerCount = (UINT32)fbxMesh->GetDeformerCount(FbxDeformer::eBlendShape);
			for (UINT32 i = 0; i < deformerCount; i++)
			{
				FbxBlendShape* deformer = static_cast<FbxBlendShape*>(fbxMesh->GetDeformer(i, FbxDeformer::eBlendShape));

				UINT32 blendShapeChannelCount = (UINT32)deformer->GetBlendShapeChannelCount();
				for (UINT32 j = 0; j < blendShapeChannelCount; ++j)
				{
					FbxBlendShapeChannel* channel = deformer->GetBlendShapeChannel(j);
					double* weights = channel->GetTargetShapeFullWeights();

					UINT32 frameCount = channel->GetTargetShapeCount();
					if (frameCount == 0)
						continue;

					mesh->blendShapes.push_back(FBXBlendShape());
					FBXBlendShape& blendShape = mesh->blendShapes.back();
					blendShape.name = channel->GetName();
					blendShape.frames.resize(frameCount);

					// Get name without invalid characters
					blendShape.name = StringUtil::replaceAll(blendShape.name, ".", "_");
					blendShape.name = StringUtil::replaceAll(blendShape.name, "/", "_");

					for (UINT32 k = 0; k < frameCount; k++)
					{
						FbxShape* fbxShape = channel->GetTargetShape(k);

						FBXBlendShapeFrame& frame = blendShape.frames[k];
						frame.name = fbxShape->GetName();
						frame.weight = (float)(weights[k] / 100.0);

						// Get name without invalid characters
						frame.name = StringUtil::replaceAll(frame.name, ".", "_");
						frame.name = StringUtil::replaceAll(frame.name, "/", "_");
						
						importBlendShapeFrame(fbxShape, *mesh, options, frame);
					}
				}
			}
		}
	}

	void FBXImporter::importBlendShapeFrame(FbxShape* shape, const FBXImportMesh& mesh, const FBXImportOptions& options, FBXBlendShapeFrame& outFrame)
	{
		UINT32 vertexCount = (UINT32)shape->GetControlPointsCount();
		outFrame.positions.resize(vertexCount);
		FbxVector4* controlPoints = shape->GetControlPoints();

		for (UINT32 i = 0; i < vertexCount; i++)
			outFrame.positions[i] = FBXToNativeType(controlPoints[i]);

		FbxLayer* mainLayer = shape->GetLayer(0);
		if (options.importNormals)
		{
			bool hasNormals = mainLayer->GetNormals() != nullptr;

			if (!hasNormals)
			{
				if (!mesh.smoothingGroups.empty())
				{
					FBXUtility::normalsFromSmoothing(outFrame.positions, mesh.indices,
						mesh.smoothingGroups, outFrame.normals);
				}
			}
			else
				readLayerData(*mainLayer->GetNormals(), outFrame.normals, mesh.indices);
		}

		if (options.importTangents)
		{
			bool hasTangents = mainLayer->GetTangents() != nullptr && mainLayer->GetBinormals() != nullptr;

			if (hasTangents)
			{
				readLayerData(*mainLayer->GetTangents(), outFrame.tangents, mesh.indices);
				readLayerData(*mainLayer->GetBinormals(), outFrame.bitangents, mesh.indices);
			}
		}
	}

	void FBXImporter::importSkin(FBXImportScene& scene, const FBXImportOptions& options)
	{
		for (auto& mesh : scene.meshes)
		{
			FbxMesh* fbxMesh = mesh->fbxMesh;

			UINT32 deformerCount = (UINT32)fbxMesh->GetDeformerCount(FbxDeformer::eSkin);
			if (deformerCount > 0)
			{
				// We ignore other deformers if there's more than one
				FbxSkin* deformer = static_cast<FbxSkin*>(fbxMesh->GetDeformer(0, FbxDeformer::eSkin));
				UINT32 boneCount = (UINT32)deformer->GetClusterCount();

				if (boneCount == 0)
					continue;

				// If only one bone and it links to itself, ignore the bone
				if (boneCount == 1)
				{
					FbxCluster* cluster = deformer->GetCluster(0);
					if (mesh->referencedBy.size() == 1 && mesh->referencedBy[0]->fbxNode == cluster->GetLink())
						continue;
				}

				importSkin(scene, deformer, *mesh, options);
			}
		}
	}

	void FBXImporter::importSkin(FBXImportScene& scene, FbxSkin* skin, FBXImportMesh& mesh, const FBXImportOptions& options)
	{
		Vector<FBXBoneInfluence>& influences = mesh.boneInfluences;
		influences.resize(mesh.positions.size());

		Matrix4 invGlobalScale = scene.globalScale.inverseAffine();

		UnorderedSet<FbxNode*> existingBones;
		UINT32 boneCount = (UINT32)skin->GetClusterCount();
		for (UINT32 i = 0; i < boneCount; i++)
		{
			FbxCluster* cluster = skin->GetCluster(i);
			FbxNode* link = cluster->GetLink();

			// The bone node doesn't exist, skip it
			auto iterFind = scene.nodeMap.find(link);
			if (iterFind == scene.nodeMap.end())
				continue;

			mesh.bones.push_back(FBXBone());

			FBXBone& bone = mesh.bones.back();
			bone.node = iterFind->second;

			if(mesh.referencedBy.size() > 1)
			{
				// Note: If this becomes a relevant issue (unlikely), then I will have to duplicate skeleton bones for
				// each such mesh, since they will all require their own bind poses. Animation curves will also need to be
				// handled specially (likely by allowing them to be applied to multiple bones at once). The other option is
				// not to bake the node transform into mesh vertices and handle it on a Scene Object level.
				LOGWRN("Skinned mesh has multiple different instances. This is not supported.");
			}

			// Calculate bind pose
			FbxAMatrix clusterTransform;
			cluster->GetTransformMatrix(clusterTransform);

			FbxAMatrix linkTransform;
			cluster->GetTransformLinkMatrix(linkTransform);

			FbxAMatrix invLinkTransform = linkTransform.Inverse() * clusterTransform;
			bone.bindPose = FBXToNativeType(invLinkTransform);

			// Apply global scale to bind pose (we only apply the scale to translation portion because we scale the
			// translation animation curves)
			const Matrix4& nodeTfrm = iterFind->second->worldTransform;

			Matrix4 nodeTfrmScaledTranslation = nodeTfrm;
			nodeTfrmScaledTranslation[0][3] = nodeTfrmScaledTranslation[0][3] / scene.scaleFactor;
			nodeTfrmScaledTranslation[1][3] = nodeTfrmScaledTranslation[1][3] / scene.scaleFactor;
			nodeTfrmScaledTranslation[2][3] = nodeTfrmScaledTranslation[2][3] / scene.scaleFactor;

			Matrix4 nodeTfrmInv = nodeTfrm.inverseAffine();

			Matrix4 scaledTranslation = nodeTfrmInv * scene.globalScale * nodeTfrmScaledTranslation;
			bone.bindPose = scaledTranslation * bone.bindPose * invGlobalScale;

			bool isDuplicate = !existingBones.insert(link).second;
			bool isAdditive = cluster->GetLinkMode() == FbxCluster::eAdditive;

			// We avoid importing weights twice for duplicate bones and we don't
			// support additive link mode.
			bool importWeights = !isDuplicate && !isAdditive;
			if (!importWeights)
				continue;

			double* weights = cluster->GetControlPointWeights();
			INT32* indices = cluster->GetControlPointIndices();
			UINT32 numIndices = (UINT32)cluster->GetControlPointIndicesCount();
			INT32 numVertices = (INT32)influences.size();

			// Add new weights while keeping them in order and removing the smallest ones
			// if number of influences exceeds the set maximum value
			for (UINT32 j = 0; j < numIndices; j++)
			{
				INT32 vertexIndex = indices[j];
				float weight = (float)weights[j];

				for (INT32 k = 0; k < FBX_IMPORT_MAX_BONE_INFLUENCES; k++)
				{
					if (vertexIndex < 0 || vertexIndex >= numVertices)
						continue;

					if (weight >= influences[vertexIndex].weights[k])
					{
						for (INT32 l = FBX_IMPORT_MAX_BONE_INFLUENCES - 2; l >= k; l--)
						{
							influences[vertexIndex].weights[l + 1] = influences[vertexIndex].weights[l];
							influences[vertexIndex].indices[l + 1] = influences[vertexIndex].indices[l];
						}

						influences[vertexIndex].weights[k] = weight;
						influences[vertexIndex].indices[k] = i;
						break;
					}
				}
			}
		}

		if (mesh.bones.empty())
			mesh.boneInfluences.clear();

		UINT32 numBones = (UINT32)mesh.bones.size();
		if (numBones > 256)
			LOGWRN("A maximum of 256 bones per skeleton are supported. Imported skeleton has " + toString(numBones) + " bones");

		// Normalize weights
		UINT32 numInfluences = (UINT32)mesh.boneInfluences.size();
		for (UINT32 i = 0; i < numInfluences; i++)
		{
			float sum = 0.0f;
			for (UINT32 j = 0; j < FBX_IMPORT_MAX_BONE_INFLUENCES; j++)
				sum += influences[i].weights[j];

			float invSum = 1.0f / sum;
			for (UINT32 j = 0; j < FBX_IMPORT_MAX_BONE_INFLUENCES; j++)
				influences[i].weights[j] *= invSum;
		}
	}

	void FBXImporter::generateMissingTangentSpace(FBXImportScene& scene, const FBXImportOptions& options)
	{
		for (auto& mesh : scene.meshes)
		{
			UINT32 numVertices = (UINT32)mesh->positions.size();
			UINT32 numIndices = (UINT32)mesh->indices.size();

			if ((options.importNormals || options.importTangents) && mesh->normals.empty())
			{
				mesh->normals.resize(numVertices);

				MeshUtility::calculateNormals(mesh->positions.data(), (UINT8*)mesh->indices.data(), numVertices, numIndices, mesh->normals.data());
			}

			if (options.importTangents && !mesh->UV[0].empty() && (mesh->tangents.empty() || mesh->bitangents.empty()))
			{
				mesh->tangents.resize(numVertices);
				mesh->bitangents.resize(numVertices);

				MeshUtility::calculateTangents(mesh->positions.data(), mesh->normals.data(), mesh->UV[0].data(), (UINT8*)mesh->indices.data(), 
					numVertices, numIndices, mesh->tangents.data(), mesh->bitangents.data());
			}

			for (auto& shape : mesh->blendShapes)
			{
				for (auto& frame : shape.frames)
				{
					if ((options.importNormals || options.importTangents) && frame.normals.empty())
					{
						frame.normals.resize(numVertices);

						MeshUtility::calculateNormals(mesh->positions.data(), (UINT8*)mesh->indices.data(), numVertices, numIndices, frame.normals.data());
					}

					if (options.importTangents && !mesh->UV[0].empty() && (frame.tangents.empty() || frame.bitangents.empty()))
					{
						frame.tangents.resize(numVertices);
						frame.bitangents.resize(numVertices);

						MeshUtility::calculateTangents(mesh->positions.data(), frame.normals.data(), mesh->UV[0].data(), (UINT8*)mesh->indices.data(),
							numVertices, numIndices, frame.tangents.data(), frame.bitangents.data());
					}
				}
			}
		}
	}

	void FBXImporter::importAnimations(FbxScene* scene, FBXImportOptions& importOptions, FBXImportScene& importScene)
	{
		FbxNode* root = scene->GetRootNode();

		UINT32 numAnimStacks = (UINT32)scene->GetSrcObjectCount<FbxAnimStack>();
		for (UINT32 i = 0; i < numAnimStacks; i++)
		{
			FbxAnimStack* animStack = scene->GetSrcObject<FbxAnimStack>(i);

			importScene.clips.push_back(FBXAnimationClip());
			FBXAnimationClip& clip = importScene.clips.back();
			clip.name = animStack->GetName();

			FbxTimeSpan timeSpan = animStack->GetLocalTimeSpan();
			clip.start = (float)timeSpan.GetStart().GetSecondDouble();
			clip.end = (float)timeSpan.GetStop().GetSecondDouble();

			clip.sampleRate = (UINT32)FbxTime::GetFrameRate(scene->GetGlobalSettings().GetTimeMode());

			UINT32 layerCount = animStack->GetMemberCount<FbxAnimLayer>();
			if (layerCount > 1)
			{
				FbxAnimEvaluator* evaluator = scene->GetAnimationEvaluator();

				FbxTime startTime;
				startTime.SetSecondDouble(clip.start);

				FbxTime endTime;
				endTime.SetSecondDouble(clip.end);

				FbxTime sampleRate;

				if (importOptions.animResample)
					sampleRate.SetSecondDouble(importOptions.animSampleRate);
				else
				{
					FbxTime::EMode timeMode = scene->GetGlobalSettings().GetTimeMode();
					sampleRate.SetSecondDouble(1.0f / FbxTime::GetFrameRate(timeMode));
				}

				if (!animStack->BakeLayers(evaluator, startTime, endTime, sampleRate))
					continue;

				layerCount = animStack->GetMemberCount<FbxAnimLayer>();
			}

			if (layerCount == 1)
			{
				FbxAnimLayer* animLayer = animStack->GetMember<FbxAnimLayer>(0);

				importAnimations(animLayer, root, importOptions, clip, importScene);
			}
		}
	}

	void FBXImporter::importAnimations(FbxAnimLayer* layer, FbxNode* node, FBXImportOptions& importOptions,
		FBXAnimationClip& clip, FBXImportScene& importScene)
	{
		FbxAnimCurve* translation[3];
		translation[0] = node->LclTranslation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_X);
		translation[1] = node->LclTranslation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Y);
		translation[2] = node->LclTranslation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Z);

		FbxAnimCurve* rotation[3];
		rotation[0] = node->LclRotation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_X);
		rotation[1] = node->LclRotation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Y);
		rotation[2] = node->LclRotation.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Z);

		FbxAnimCurve* scale[3];
		scale[0] = node->LclScaling.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_X);
		scale[1] = node->LclScaling.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Y);
		scale[2] = node->LclScaling.GetCurve(layer, FBXSDK_CURVENODE_COMPONENT_Z);

		auto hasCurveValues = [](FbxAnimCurve* curves[3])
		{
			for (UINT32 i = 0; i < 3; i++)
			{
				if (curves[i] != nullptr && curves[i]->KeyGetCount() > 0)
					return true;
			}

			return false;
		};

		bool hasBoneAnimation = hasCurveValues(translation) || hasCurveValues(rotation) || hasCurveValues(scale);
		if (hasBoneAnimation)
		{
			clip.boneAnimations.push_back(FBXBoneAnimation());
			FBXBoneAnimation& boneAnim = clip.boneAnimations.back();
			boneAnim.node = importScene.nodeMap[node];

			boneAnim.translation = importCurve<Vector3, 3>(translation, importOptions, clip.start, clip.end);
			boneAnim.scale = importCurve<Vector3, 3>(scale, importOptions, clip.start, clip.end);

			TAnimationCurve<Vector3> eulerAnimation = importCurve<Vector3, 3>(rotation, importOptions, clip.start, clip.end);
			if(importOptions.reduceKeyframes)
			{
				boneAnim.translation = reduceKeyframes(boneAnim.translation);
				boneAnim.scale = reduceKeyframes(boneAnim.scale);
				eulerAnimation = reduceKeyframes(eulerAnimation);
			}

			boneAnim.translation = AnimationUtility::scaleCurve(boneAnim.translation, importScene.scaleFactor);
			boneAnim.rotation = AnimationUtility::eulerToQuaternionCurve(eulerAnimation);
		}

		if (importOptions.importBlendShapes)
		{
			FbxMesh* fbxMesh = node->GetMesh();
			if (fbxMesh != nullptr)
			{
				INT32 deformerCount = fbxMesh->GetDeformerCount(FbxDeformer::eBlendShape);
				for (INT32 i = 0; i < deformerCount; i++)
				{
					FbxBlendShape* deformer = static_cast<FbxBlendShape*>(fbxMesh->GetDeformer(i, FbxDeformer::eBlendShape));

					INT32 channelCount = deformer->GetBlendShapeChannelCount();
					for (INT32 j = 0; j < channelCount; j++)
					{
						FbxBlendShapeChannel* channel = deformer->GetBlendShapeChannel(j);

						FbxAnimCurve* curve = fbxMesh->GetShapeChannel(i, j, layer);
						if (curve != nullptr && curve->KeyGetCount() > 0)
						{
							clip.blendShapeAnimations.push_back(FBXBlendShapeAnimation());
							FBXBlendShapeAnimation& blendShapeAnim = clip.blendShapeAnimations.back();
							blendShapeAnim.blendShape = channel->GetName();

							// Get name without invalid characters
							blendShapeAnim.blendShape = StringUtil::replaceAll(blendShapeAnim.blendShape, ".", "_");
							blendShapeAnim.blendShape = StringUtil::replaceAll(blendShapeAnim.blendShape, "/", "_");

							FbxAnimCurve* curves[1] = { curve };
							blendShapeAnim.curve = importCurve<float, 1>(curves, importOptions, clip.start, clip.end);

							// FBX contains data in [0, 100] range, but we need it in [0, 1] range
							blendShapeAnim.curve = AnimationUtility::scaleCurve(blendShapeAnim.curve, 0.01f);
						}
					}
				}
			}
		}

		UINT32 childCount = (UINT32)node->GetChildCount();
		for (UINT32 i = 0; i < childCount; i++)
		{
			FbxNode* child = node->GetChild(i);
			importAnimations(layer, child, importOptions, clip, importScene);
		}
	}

	void FBXImporter::bakeTransforms(FbxScene* scene)
	{
		// FBX stores transforms in a more complex way than just translation-rotation-scale as used by Banshee.
		// Instead they also support rotations offsets and pivots, scaling pivots and more. We wish to bake all this data
		// into a standard transform so we can access it using node's local TRS properties (e.g. FbxNode::LclTranslation).

		double frameRate = FbxTime::GetFrameRate(scene->GetGlobalSettings().GetTimeMode());

		bs_frame_mark();
		{
			FrameStack<FbxNode*> todo;
			todo.push(scene->GetRootNode());

			while(todo.size() > 0)
			{
				FbxNode* node = todo.top();
				todo.pop();

				FbxVector4 zero(0, 0, 0);
				FbxVector4 one(1, 1, 1);

				// Activate pivot converting
				node->SetPivotState(FbxNode::eSourcePivot, FbxNode::ePivotActive);
				node->SetPivotState(FbxNode::eDestinationPivot, FbxNode::ePivotActive);

				// We want to set all these to 0 (1 for scale) and bake them into the transforms
				node->SetPostRotation(FbxNode::eDestinationPivot, zero);
				node->SetPreRotation(FbxNode::eDestinationPivot, zero);
				node->SetRotationOffset(FbxNode::eDestinationPivot, zero);
				node->SetScalingOffset(FbxNode::eDestinationPivot, zero);
				node->SetRotationPivot(FbxNode::eDestinationPivot, zero);
				node->SetScalingPivot(FbxNode::eDestinationPivot, zero);
				node->SetGeometricTranslation(FbxNode::eDestinationPivot, zero);
				node->SetGeometricRotation(FbxNode::eDestinationPivot, zero);
				node->SetGeometricScaling(FbxNode::eDestinationPivot, one);

				// Banshee assumes euler angles are in YXZ order
				node->SetRotationOrder(FbxNode::eDestinationPivot, FbxEuler::eOrderYXZ);

				// Keep interpolation as is
				node->SetQuaternionInterpolation(FbxNode::eDestinationPivot, node->GetQuaternionInterpolation(FbxNode::eSourcePivot));

				for (int i = 0; i < node->GetChildCount(); i++)
				{
					FbxNode* childNode = node->GetChild(i);
					todo.push(childNode);
				}
			}

			scene->GetRootNode()->ConvertPivotAnimationRecursive(nullptr, FbxNode::eDestinationPivot, frameRate, false);
		}
		bs_frame_clear();
	}

	TAnimationCurve<Vector3> FBXImporter::reduceKeyframes(TAnimationCurve<Vector3>& curve)
	{
		UINT32 keyCount = curve.getNumKeyFrames();

		Vector<TKeyframe<Vector3>> newKeyframes;

		bool lastWasEqual = false;
		for (UINT32 i = 0; i < keyCount; i++)
		{
			bool isEqual = true;

			const TKeyframe<Vector3>& curKey = curve.getKeyFrame(i);
			if (i > 0)
			{
				TKeyframe<Vector3>& prevKey = newKeyframes.back();

				isEqual = Math::approxEquals(prevKey.value, curKey.value) &&
					Math::approxEquals(prevKey.outTangent, curKey.inTangent) && isEqual;
			}
			else
				isEqual = false;

			// More than two keys in a row are equal, remove previous key by replacing it with this one
			if (lastWasEqual && isEqual)
			{
				TKeyframe<Vector3>& prevKey = newKeyframes.back();

				// Other properties are guaranteed unchanged
				prevKey.time = curKey.time;
				prevKey.outTangent = curKey.outTangent;

				continue;
			}

			newKeyframes.push_back(curKey);
			lastWasEqual = isEqual;
		}

		return TAnimationCurve<Vector3>(newKeyframes);
	}

	template<class T>
	void setKeyframeValues(TKeyframe<T>& keyFrame, int idx, float value, float inTangent, float outTangent)
	{
		keyFrame.value = value;
		keyFrame.inTangent = inTangent;
		keyFrame.outTangent = outTangent;
	}

	template<>
	void setKeyframeValues<Vector3>(TKeyframe<Vector3>& keyFrame, int idx, float value, float inTangent, float outTangent)
	{
		keyFrame.value[idx] = value;
		keyFrame.inTangent[idx] = inTangent;
		keyFrame.outTangent[idx] = outTangent;
	}

	template<class T, int C>
	TAnimationCurve<T> FBXImporter::importCurve(FbxAnimCurve*(&fbxCurve)[C], FBXImportOptions& importOptions,
		float start, float end)
	{
		// If curve key-counts don't match, we need to force resampling 
		bool forceResample = false;
		
		for(int i = 1; i < C; i++)
		{
			forceResample |= fbxCurve[i - 1]->KeyGetCount() != fbxCurve[i]->KeyGetCount();
			if (forceResample)
				break;
		}

		// Read keys directly
		if(!importOptions.animResample && !forceResample)
		{
			bool foundMismatch = false;
			int keyCount = fbxCurve[0]->KeyGetCount();
			Vector<TKeyframe<T>> keyframes;
			for (int i = 0; i < keyCount; i++)
			{
				FbxTime fbxTime = fbxCurve[0]->KeyGetTime(i);
				float time = (float)fbxTime.GetSecondDouble();

				// Ensure times from other curves match
				for (int j = 1; j < C; j++)
				{
					fbxTime = fbxCurve[j]->KeyGetTime(i);
					float otherTime = (float)fbxTime.GetSecondDouble();

					if (!Math::approxEquals(time, otherTime))
					{
						foundMismatch = true;
						break;
					}
				}

				if(foundMismatch)
					break;

				if (time < start || time > end)
					continue;

				keyframes.push_back(TKeyframe<T>());
				TKeyframe<T>& keyFrame = keyframes.back();

				keyFrame.time = time;

				for (int j = 0; j < C; j++)
				{
					setKeyframeValues(keyFrame, j,
						fbxCurve[j]->KeyGetValue(i),
						fbxCurve[j]->KeyGetLeftDerivative(i),
						fbxCurve[j]->KeyGetRightDerivative(i));
				}
			}

			if (!foundMismatch)
				return TAnimationCurve<T>(keyframes);
			else
				forceResample = true;
		}

		if (!importOptions.animResample && forceResample)
			LOGWRN("Animation has different keyframes for different curve components, forcing resampling.");

		// Resample keys
		float curveStart = std::numeric_limits<float>::infinity();
		float curveEnd = -std::numeric_limits<float>::infinity();

		for (INT32 i = 0; i < C; i++)
		{
			int keyCount = fbxCurve[i]->KeyGetCount();
			for (INT32 j = 0; j < keyCount; j++)
			{
				FbxTime fbxTime = fbxCurve[i]->KeyGetTime(j);
				float time = (float)fbxTime.GetSecondDouble();

				curveStart = std::min(time, curveStart);
				curveEnd = std::max(time, curveEnd);
			}
		}

		curveStart = Math::clamp(curveStart, start, end);
		curveEnd = Math::clamp(curveEnd, start, end);

		float curveLength = curveEnd - curveStart;
		INT32 numSamples = Math::ceilToInt(curveLength / importOptions.animSampleRate);

		// We don't use the exact provided sample rate but instead modify it slightly so it
		// completely covers the curve range including start/end points while maintaining
		// constant time step between keyframes.
		float dt = curveLength / (float)numSamples; 

		INT32 lastKeyframe[] = { 0, 0, 0 };
		INT32 lastLeftTangent[] = { 0, 0, 0 };
		INT32 lastRightTangent[] = { 0, 0, 0 };

		Vector<TKeyframe<T>> keyframes(numSamples);
		for (INT32 i = 0; i < numSamples; i++)
		{
			float sampleTime = std::min(curveStart + i * dt, curveEnd);
			FbxTime fbxSampleTime;
			fbxSampleTime.SetSecondDouble(sampleTime);

			TKeyframe<T>& keyFrame = keyframes[i];
			keyFrame.time = sampleTime;

			for (int j = 0; j < C; j++)
			{
				setKeyframeValues(keyFrame, j,
					fbxCurve[j]->Evaluate(fbxSampleTime, &lastKeyframe[j]),
					fbxCurve[j]->EvaluateLeftDerivative(fbxSampleTime, &lastLeftTangent[j]),
					fbxCurve[j]->EvaluateRightDerivative(fbxSampleTime, &lastRightTangent[j]));
			}
		}

		return TAnimationCurve<T>(keyframes);
	}
}