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.
119 lines
5.3 KiB
C++
119 lines
5.3 KiB
C++
/*
|
|
* 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 <SceneAPI/SceneBuilder/Importers/AssImpTransformImporter.h>
|
|
|
|
#include <AzCore/std/smart_ptr/make_shared.h>
|
|
#include <AzCore/Serialization/SerializeContext.h>
|
|
#include <AzToolsFramework/Debug/TraceContext.h>
|
|
#include <SceneAPI/SceneBuilder/SceneSystem.h>
|
|
#include <SceneAPI/SceneBuilder/Importers/ImporterUtilities.h>
|
|
#include <SceneAPI/SceneBuilder/Importers/Utilities/RenamedNodesMap.h>
|
|
#include <SceneAPI/SceneCore/Utilities/Reporting.h>
|
|
#include <SceneAPI/SceneData/GraphData/TransformData.h>
|
|
#include <SceneAPI/SDKWrapper/AssImpTypeConverter.h>
|
|
#include <SceneAPI/SDKWrapper/AssImpNodeWrapper.h>
|
|
#include <SceneAPI/SDKWrapper/AssImpSceneWrapper.h>
|
|
#include <assimp/scene.h>
|
|
#include <SceneAPI/SceneBuilder/Importers/AssImpImporterUtilities.h>
|
|
|
|
namespace AZ
|
|
{
|
|
namespace SceneAPI
|
|
{
|
|
namespace SceneBuilder
|
|
{
|
|
const char* AssImpTransformImporter::s_transformNodeName = "transform";
|
|
|
|
AssImpTransformImporter::AssImpTransformImporter()
|
|
{
|
|
BindToCall(&AssImpTransformImporter::ImportTransform);
|
|
}
|
|
|
|
void AssImpTransformImporter::Reflect(ReflectContext* context)
|
|
{
|
|
SerializeContext* serializeContext = azrtti_cast<SerializeContext*>(context);
|
|
if (serializeContext)
|
|
{
|
|
serializeContext->Class<AssImpTransformImporter, SceneCore::LoadingComponent>()->Version(1);
|
|
}
|
|
}
|
|
|
|
Events::ProcessingResult AssImpTransformImporter::ImportTransform(AssImpSceneNodeAppendedContext& context)
|
|
{
|
|
AZ_TraceContext("Importer", "transform");
|
|
const aiNode* currentNode = context.m_sourceNode.GetAssImpNode();
|
|
const aiScene* scene = context.m_sourceScene.GetAssImpScene();
|
|
|
|
if (currentNode == scene->mRootNode || IsPivotNode(currentNode->mName))
|
|
{
|
|
return Events::ProcessingResult::Ignored;
|
|
}
|
|
|
|
DataTypes::MatrixType localTransform = GetLocalSpaceBindPoseTransform(scene, currentNode);
|
|
|
|
// Don't bother adding a node with the identity matrix
|
|
if (localTransform == DataTypes::MatrixType::Identity())
|
|
{
|
|
return Events::ProcessingResult::Ignored;
|
|
}
|
|
|
|
context.m_sourceSceneSystem.SwapTransformForUpAxis(localTransform);
|
|
context.m_sourceSceneSystem.ConvertUnit(localTransform);
|
|
|
|
AZStd::shared_ptr<SceneData::GraphData::TransformData> transformData =
|
|
AZStd::make_shared<SceneData::GraphData::TransformData>(localTransform);
|
|
AZ_Error(SceneAPI::Utilities::ErrorWindow, transformData, "Failed to allocate transform data.");
|
|
if (!transformData)
|
|
{
|
|
return Events::ProcessingResult::Failure;
|
|
}
|
|
|
|
// If it is non-endpoint data populated node, add a transform attribute
|
|
if (context.m_scene.GetGraph().HasNodeContent(context.m_currentGraphPosition))
|
|
{
|
|
if (!context.m_scene.GetGraph().IsNodeEndPoint(context.m_currentGraphPosition))
|
|
{
|
|
AZStd::string nodeName = s_transformNodeName;
|
|
RenamedNodesMap::SanitizeNodeName(nodeName, context.m_scene.GetGraph(), context.m_currentGraphPosition);
|
|
AZ_TraceContext("Transform node name", nodeName);
|
|
|
|
Containers::SceneGraph::NodeIndex newIndex =
|
|
context.m_scene.GetGraph().AddChild(context.m_currentGraphPosition, nodeName.c_str());
|
|
|
|
AZ_Error(SceneAPI::Utilities::ErrorWindow, newIndex.IsValid(), "Failed to create SceneGraph node for attribute.");
|
|
if (!newIndex.IsValid())
|
|
{
|
|
return Events::ProcessingResult::Failure;
|
|
}
|
|
|
|
Events::ProcessingResult transformAttributeResult;
|
|
AssImpSceneAttributeDataPopulatedContext dataPopulated(context, transformData, newIndex, nodeName);
|
|
transformAttributeResult = Events::Process(dataPopulated);
|
|
|
|
if (transformAttributeResult != Events::ProcessingResult::Failure)
|
|
{
|
|
transformAttributeResult = AddAttributeDataNodeWithContexts(dataPopulated);
|
|
}
|
|
|
|
return transformAttributeResult;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bool addedData = context.m_scene.GetGraph().SetContent(context.m_currentGraphPosition, transformData);
|
|
|
|
AZ_Error(SceneAPI::Utilities::ErrorWindow, addedData, "Failed to add node data");
|
|
return addedData ? Events::ProcessingResult::Success : Events::ProcessingResult::Failure;
|
|
}
|
|
|
|
return Events::ProcessingResult::Ignored;
|
|
}
|
|
} // namespace SceneBuilder
|
|
} // namespace SceneAPI
|
|
} // namespace AZ
|