You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Code/Tools/SceneAPI/SceneBuilder/Tests/TestFbxMesh.h

87 lines
3.2 KiB
C++

#pragma once
/*
* Copyright (c) Contributors to the Open 3D Engine Project.
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
*
* SPDX-License-Identifier: Apache-2.0 OR MIT
*
*/
#include <vector>
#include <unordered_map>
#include <AzCore/Math/Vector3.h>
#include <SceneAPI/FbxSDKWrapper/FbxMeshWrapper.h>
#include <SceneAPI/FbxSDKWrapper/FbxSkinWrapper.h>
namespace AZ
{
namespace FbxSDKWrapper
{
struct TestFbxPolygon
{
size_t m_startVertexIndex;
size_t m_vertexCount;
TestFbxPolygon(size_t startVertexIndex, size_t vertexCount)
: m_startVertexIndex(startVertexIndex)
, m_vertexCount(vertexCount)
{
}
};
// TestFbxMesh
// FbxMesh Test Data creation
class TestFbxMesh
: public FbxMeshWrapper
{
public:
TestFbxMesh();
~TestFbxMesh() override = default;
int GetDeformerCount() const override;
AZStd::shared_ptr<const FbxSkinWrapper> GetSkin(int index) const override;
bool GetMaterialIndices(FbxLayerElementArrayTemplate<int>** lockableArray) const override;
int GetControlPointsCount() const;
AZStd::vector<Vector3> GetControlPoints() const override;
int GetPolygonCount() const override;
int GetPolygonSize(int polygonIndex) const override;
int* GetPolygonVertices() const override;
int GetPolygonVertexIndex(int polygonIndex) const;
FbxUVWrapper GetElementUV(int index = 0) override;
int GetElementUVCount() const override;
FbxVertexColorWrapper GetElementVertexColor(int index = 0) override;
int GetElementVertexColorCount() const override;
bool GetPolygonVertexNormal(int polyIndex, int vertexIndex, Vector3& normal) const override;
// Create test data APIs
void CreateMesh(std::vector<AZ::Vector3>& points, std::vector<std::vector<int> >& polygonVertexIndices);
void CreateExpectMeshInfo(std::vector<std::vector<int> >& expectedFaceVertexIndices);
void SetSkin(const AZStd::shared_ptr<FbxSkinWrapper>& skin);
size_t GetExpectedVetexCount() const;
size_t GetExpectedFaceCount() const;
AZ::Vector3 GetExpectedFaceVertexPosition(unsigned int faceIndex, unsigned int vertexIndex) const;
protected:
AZStd::vector<Vector3> m_vertexControlPoints; // vertex positions
size_t m_vertexCount;
int* m_polygonVertexIndices; // store all polygons' vertex indices in sequence. Each index maps to a control point.
FbxLayerElementArrayTemplate<int>* m_materialIndices;
std::unordered_map<int, TestFbxPolygon> m_polygonInfo;
FbxUVWrapper m_uvElements;
FbxVertexColorWrapper m_vertexColorElements;
AZStd::shared_ptr<FbxSkinWrapper> m_skin;
// Expected converted data
unsigned int m_expectedVertexCount;
std::vector<std::vector<int> > m_expectedFaceVertexIndices;
};
}
}