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.
118 lines
5.4 KiB
C++
118 lines
5.4 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 <Editor/Source/ComponentModes/Joints/JointsSubComponentModeSnapRotation.h>
|
|
|
|
#include <AzCore/Component/TransformBus.h>
|
|
#include <AzCore/Math/Transform.h>
|
|
#include <AzCore/Memory/SystemAllocator.h>
|
|
#include <AzToolsFramework/Manipulators/LinearManipulator.h>
|
|
|
|
#include <Editor/Source/ComponentModes/Joints/JointsComponentModeCommon.h>
|
|
#include <PhysX/EditorJointBus.h>
|
|
#include <Source/Utils.h>
|
|
|
|
namespace PhysX
|
|
{
|
|
AZ_CLASS_ALLOCATOR_IMPL(JointsSubComponentModeSnapRotation, AZ::SystemAllocator, 0);
|
|
|
|
void JointsSubComponentModeSnapRotation::Setup(const AZ::EntityComponentIdPair& idPair)
|
|
{
|
|
JointsSubComponentModeSnap::Setup(idPair);
|
|
|
|
PhysX::EditorJointRequestBus::EventResult(
|
|
m_resetRotation, m_entityComponentId, &PhysX::EditorJointRequests::GetVector3Value, JointsComponentModeCommon::ParamaterNames::Rotation);
|
|
|
|
m_manipulator->InstallLeftMouseDownCallback(
|
|
[this]([[maybe_unused]] const AzToolsFramework::LinearManipulator::Action& action) mutable
|
|
{
|
|
if (!m_pickedEntity.IsValid())
|
|
{
|
|
return;
|
|
}
|
|
|
|
AZ::EntityId leadEntityId;
|
|
PhysX::EditorJointRequestBus::EventResult(
|
|
leadEntityId, m_entityComponentId, &PhysX::EditorJointRequests::GetEntityIdValue,
|
|
JointsComponentModeCommon::ParamaterNames::LeadEntity);
|
|
|
|
if (leadEntityId.IsValid() && m_pickedEntity == leadEntityId)
|
|
{
|
|
AZ_Warning(
|
|
"EditorsubComponentModeSnapRotation", false,
|
|
"The entity %s is the lead of the joint. Please snap rotation (or orientation) of joint to another entity that is "
|
|
"not the lead entity.",
|
|
GetPickedEntityName().c_str());
|
|
return;
|
|
}
|
|
|
|
AZ::Transform worldTransform = AZ::Transform::CreateIdentity();
|
|
AZ::TransformBus::EventResult(worldTransform, m_entityComponentId.GetEntityId(), &AZ::TransformInterface::GetWorldTM);
|
|
worldTransform.ExtractUniformScale();
|
|
|
|
AZ::Transform localTransform = AZ::Transform::CreateIdentity();
|
|
EditorJointRequestBus::EventResult(
|
|
localTransform, m_entityComponentId, &EditorJointRequests::GetTransformValue, JointsComponentModeCommon::ParamaterNames::Transform);
|
|
|
|
AZ::Transform pickedEntityTransform = AZ::Transform::CreateIdentity();
|
|
AZ::TransformBus::EventResult(pickedEntityTransform, m_pickedEntity, &AZ::TransformInterface::GetWorldTM);
|
|
|
|
const AZ::Transform worldTransformInv = worldTransform.GetInverse();
|
|
const AZ::Vector3 pickedLocalPosition =
|
|
worldTransformInv.TransformVector(pickedEntityTransform.GetTranslation()) - localTransform.GetTranslation();
|
|
|
|
if (AZStd::abs(pickedLocalPosition.GetLength()) < FLT_EPSILON)
|
|
{
|
|
AZ_Warning(
|
|
"EditorsubComponentModeSnapRotation", false,
|
|
"The entity %s is too close to the joint position. Please snap rotation to an entity that is not at the position "
|
|
"of the joint.",
|
|
GetPickedEntityName().c_str());
|
|
return;
|
|
}
|
|
|
|
const AZ::Vector3 targetDirection = pickedLocalPosition.GetNormalized();
|
|
const AZ::Vector3 sourceDirection = AZ::Vector3::CreateAxisX();
|
|
const AZ::Quaternion newLocalRotation = AZ::Quaternion::CreateShortestArc(sourceDirection, targetDirection);
|
|
|
|
PhysX::EditorJointRequestBus::Event(
|
|
m_entityComponentId, &PhysX::EditorJointRequests::SetVector3Value,
|
|
JointsComponentModeCommon::ParamaterNames::Rotation // using rotation parameter name to set the local rotation value
|
|
,
|
|
newLocalRotation.GetEulerDegrees());
|
|
});
|
|
}
|
|
|
|
void JointsSubComponentModeSnapRotation::ResetValues([[maybe_unused]] const AZ::EntityComponentIdPair& idPair)
|
|
{
|
|
PhysX::EditorJointRequestBus::Event(
|
|
m_entityComponentId, &PhysX::EditorJointRequests::SetVector3Value, JointsComponentModeCommon::ParamaterNames::Rotation, m_resetRotation);
|
|
}
|
|
|
|
void JointsSubComponentModeSnapRotation::DisplaySpecificSnapType(
|
|
[[maybe_unused]] const AzFramework::ViewportInfo& viewportInfo,
|
|
AzFramework::DebugDisplayRequests& debugDisplay,
|
|
const AZ::Vector3& jointPosition,
|
|
const AZ::Vector3& snapDirection,
|
|
float snapLength)
|
|
{
|
|
const float circleRadius = 0.5f;
|
|
const float iconGap = 1.0f;
|
|
|
|
const AZ::Vector3 iconPosition = jointPosition + (snapDirection * (snapLength + circleRadius * 2.0f + iconGap));
|
|
|
|
debugDisplay.SetColor(AZ::Colors::Red);
|
|
debugDisplay.DrawCircle(iconPosition, circleRadius, 0);
|
|
debugDisplay.SetColor(AZ::Colors::Green);
|
|
debugDisplay.DrawCircle(iconPosition, circleRadius, 1);
|
|
debugDisplay.SetColor(AZ::Colors::Blue);
|
|
debugDisplay.DrawCircle(iconPosition, circleRadius, 2);
|
|
}
|
|
|
|
} // namespace PhysX
|