update joints to use cluster viewport ui. (#4352)

also some joints Component mode cleanup. Signed-off-by: amzn-sean <75276488+amzn-sean@users.noreply.github.com>
2021-09-30 13:39:22 +01:00
parent 9eb9a13d99
commit 43748c37fd
77 changed files with 3006 additions and 3026 deletions
@@ -0,0 +1,91 @@
+/*
+ * 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/JointsSubComponentModeSnapPosition.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(JointsSubComponentModeSnapPosition, AZ::SystemAllocator, 0);
+
+    void JointsSubComponentModeSnapPosition::Setup(const AZ::EntityComponentIdPair& idPair)
+    {
+        JointsSubComponentModeSnap::Setup(idPair);
+
+        PhysX::EditorJointRequestBus::EventResult(
+            m_resetPosition, m_entityComponentId, &PhysX::EditorJointRequests::GetVector3Value, JointsComponentModeCommon::ParamaterNames::Position);
+
+        PhysX::EditorJointRequestBus::EventResult(
+            m_resetLeadEntity, m_entityComponentId, &PhysX::EditorJointRequests::GetEntityIdValue,
+            JointsComponentModeCommon::ParamaterNames::LeadEntity);
+
+        m_manipulator->InstallLeftMouseDownCallback(
+            [this](const AzToolsFramework::LinearManipulator::Action& /*action*/) mutable
+            {
+                if (!m_pickedEntity.IsValid())
+                {
+                    return;
+                }
+
+                const AZ::Vector3 newLocalPosition = PhysX::Utils::ComputeJointLocalTransform(
+                                                         PhysX::Utils::GetEntityWorldTransformWithScale(m_pickedEntity),
+                                                         PhysX::Utils::GetEntityWorldTransformWithScale(m_entityComponentId.GetEntityId()))
+                                                         .GetTranslation();
+
+                PhysX::EditorJointRequestBus::Event(
+                    m_entityComponentId, &PhysX::EditorJointRequests::SetVector3Value, JointsComponentModeCommon::ParamaterNames::Position,
+                    newLocalPosition);
+
+                const bool selectedEntityIsNotJointEntity = m_pickedEntity != m_entityComponentId.GetEntityId();
+
+                AZ_Error(
+                    "EditorSubComponentModeSnapPosition", selectedEntityIsNotJointEntity,
+                    "Joint's lead entity cannot be the same as the entity in which the joint resides. Select lead entity on snap failed.");
+
+                if (selectedEntityIsNotJointEntity)
+                {
+                    PhysX::EditorJointRequestBus::Event(
+                        m_entityComponentId, &PhysX::EditorJointRequests::SetEntityIdValue, JointsComponentModeCommon::ParamaterNames::LeadEntity,
+                        m_pickedEntity);
+                }
+            });
+    }
+
+    void JointsSubComponentModeSnapPosition::ResetValues([[maybe_unused]]const AZ::EntityComponentIdPair& idPair)
+    {
+        PhysX::EditorJointRequestBus::Event(
+            m_entityComponentId, &PhysX::EditorJointRequests::SetVector3Value, JointsComponentModeCommon::ParamaterNames::Position, m_resetPosition);
+        PhysX::EditorJointRequestBus::Event(
+            m_entityComponentId, &PhysX::EditorJointRequests::SetEntityIdValue, JointsComponentModeCommon::ParamaterNames::LeadEntity, m_resetLeadEntity);
+    }
+
+    void JointsSubComponentModeSnapPosition::DisplaySpecificSnapType(
+        [[maybe_unused]] const AzFramework::ViewportInfo& viewportInfo,
+        AzFramework::DebugDisplayRequests& debugDisplay,
+        const AZ::Vector3& jointPosition,
+        const AZ::Vector3& snapDirection,
+        float snapLength)
+    {
+        const float arrowLength = 1.0f;
+        const float iconGap = 1.0f;
+        const AZ::Vector3 iconPosition = jointPosition + (snapDirection * (snapLength + arrowLength + iconGap));
+
+        debugDisplay.SetColor(AZ::Colors::Red);
+        debugDisplay.DrawArrow(iconPosition, iconPosition + AZ::Vector3(arrowLength, 0.0f, 0.0f));
+        debugDisplay.SetColor(AZ::Colors::Green);
+        debugDisplay.DrawArrow(iconPosition, iconPosition + AZ::Vector3(0.2f, arrowLength, 0.2f));
+        debugDisplay.SetColor(AZ::Colors::Blue);
+        debugDisplay.DrawArrow(iconPosition, iconPosition + AZ::Vector3(0.0f, 0.0f, arrowLength));
+    }
+
+} // namespace PhysX