o3de/Gems/EMotionFX/Code/Source/Editor/SkeletonModelJointWidget.cpp

/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include <Editor/ColliderContainerWidget.h>
#include <Editor/SkeletonModel.h>
#include <Editor/SkeletonModelJointWidget.h>
#include <Editor/Plugins/SkeletonOutliner/SkeletonOutlinerBus.h>
#include <EMotionStudio/EMStudioSDK/Source/EMStudioManager.h>
#include <QLabel>
#include <QItemSelectionModel>
#include <QModelIndex>
#include <QVBoxLayout>


namespace EMotionFX
{
    int SkeletonModelJointWidget::s_jointLabelSpacing = 17;
    int SkeletonModelJointWidget::s_jointNameSpacing = 130;

    SkeletonModelJointWidget::SkeletonModelJointWidget(QWidget* parent)
        : QWidget(parent)
        , m_jointNameLabel(nullptr)
        , m_contentsWidget(nullptr)
        , m_noSelectionWidget(nullptr)
    {
    }

    void SkeletonModelJointWidget::CreateGUI()
    {
        QVBoxLayout* mainLayout = new QVBoxLayout();
        mainLayout->setAlignment(Qt::AlignTop);
        mainLayout->setMargin(0);
        mainLayout->setSizeConstraint(QLayout::SetMinimumSize);
        mainLayout->setSizeConstraint(QLayout::SetMinimumSize);

        // Contents widget
        m_contentsWidget = new QWidget();
        m_contentsWidget->setVisible(false);
        QVBoxLayout* contentsLayout = new QVBoxLayout();
        contentsLayout->setSpacing(ColliderContainerWidget::s_layoutSpacing);

        // Joint name
        QHBoxLayout* jointNameLayout = new QHBoxLayout();
        jointNameLayout->setAlignment(Qt::AlignLeft);
        jointNameLayout->setMargin(0);
        jointNameLayout->setSpacing(0);
        contentsLayout->addLayout(jointNameLayout);

        jointNameLayout->addSpacerItem(new QSpacerItem(s_jointLabelSpacing, 0, QSizePolicy::Fixed));
        QLabel* tempLabel = new QLabel("Joint name");
        tempLabel->setStyleSheet("font-weight: bold;");
        jointNameLayout->addWidget(tempLabel);

        jointNameLayout->addSpacerItem(new QSpacerItem(s_jointNameSpacing, 0, QSizePolicy::Fixed));
        m_jointNameLabel = new QLabel();
        jointNameLayout->addWidget(m_jointNameLabel);
        jointNameLayout->addSpacerItem(new QSpacerItem(0, 0, QSizePolicy::Ignored, QSizePolicy::Ignored));

        contentsLayout->addWidget(CreateContentWidget(m_contentsWidget));

        m_contentsWidget->setLayout(contentsLayout);
        mainLayout->addWidget(m_contentsWidget);

        // No selection widget
        m_noSelectionWidget = CreateNoSelectionWidget(m_contentsWidget);
        mainLayout->addWidget(m_noSelectionWidget);

        setLayout(mainLayout);

        AZ::Outcome<const QModelIndexList&> selectedRowIndicesOutcome;
        QModelIndexList selectedModelIndices;
        SkeletonOutlinerRequestBus::BroadcastResult(selectedRowIndicesOutcome, &SkeletonOutlinerRequests::GetSelectedRowIndices);
        if (selectedRowIndicesOutcome.IsSuccess())
        {
            selectedModelIndices = selectedRowIndicesOutcome.GetValue();
        }
        Reinit(selectedModelIndices);

        // Connect to the model.
        SkeletonModel* skeletonModel = nullptr;
        SkeletonOutlinerRequestBus::BroadcastResult(skeletonModel, &SkeletonOutlinerRequests::GetModel);
        if (skeletonModel)
        {
            connect(skeletonModel, &SkeletonModel::dataChanged, this, &SkeletonModelJointWidget::OnDataChanged);
            connect(skeletonModel, &SkeletonModel::modelReset, this, &SkeletonModelJointWidget::OnModelReset);
            connect(&skeletonModel->GetSelectionModel(), &QItemSelectionModel::selectionChanged, this, &SkeletonModelJointWidget::OnSelectionChanged);
        }
    }

    void SkeletonModelJointWidget::Reinit(const QModelIndexList& selectedModelIndices)
    {
        m_selectedModelIndices = selectedModelIndices;

        if (!EMStudio::GetManager()->GetIgnoreVisibility() && !isVisible())
        {
            return;
        }

        if (GetActor())
        {
            if (!m_selectedModelIndices.isEmpty())
            {
                if (m_selectedModelIndices.size() == 1)
                {
                    m_jointNameLabel->setText(GetNode()->GetName());
                }
                else
                {
                    m_jointNameLabel->setText(QString("%1 joints selected").arg(m_selectedModelIndices.size()));
                }

                m_noSelectionWidget->hide();
                InternalReinit();
                m_contentsWidget->show();
            }
            else
            {
                m_contentsWidget->hide();
                InternalReinit();
                m_noSelectionWidget->show();
            }
        }
        else
        {
            m_contentsWidget->hide();
            InternalReinit();
            m_noSelectionWidget->hide();
        }
    }

    void SkeletonModelJointWidget::showEvent(QShowEvent* event)
    {
        QWidget::showEvent(event);
        Reinit(m_selectedModelIndices);
    }

    void SkeletonModelJointWidget::OnSelectionChanged([[maybe_unused]] const QItemSelection& selected, [[maybe_unused]] const QItemSelection& deselected)
    {
        SkeletonModel* skeletonModel = nullptr;
        SkeletonOutlinerRequestBus::BroadcastResult(skeletonModel, &SkeletonOutlinerRequests::GetModel);
        if (skeletonModel)
        {
            const QModelIndexList selectedRows = skeletonModel->GetSelectionModel().selectedRows();
            Reinit(selectedRows);
        }
    }

    void SkeletonModelJointWidget::OnDataChanged([[maybe_unused]] const QModelIndex& topLeft, [[maybe_unused]] const QModelIndex& bottomRight, [[maybe_unused]] const QVector<int>& roles)
    {
        Reinit(m_selectedModelIndices);
    }

    void SkeletonModelJointWidget::OnModelReset()
    {
        Reinit(QModelIndexList());
    }

    Actor* SkeletonModelJointWidget::GetActor() const
    {
        Actor* actor = nullptr;
        if (!m_selectedModelIndices.empty())
        {
            actor = m_selectedModelIndices[0].data(SkeletonModel::ROLE_ACTOR_POINTER).value<Actor*>();
        }
        
        if (!actor)
        {
            SkeletonModel* skeletonModel = nullptr;
            SkeletonOutlinerRequestBus::BroadcastResult(skeletonModel, &SkeletonOutlinerRequests::GetModel);
            if (skeletonModel)
            {
                actor = skeletonModel->GetActor();
            }
        }
        return actor;
    }

    Node* SkeletonModelJointWidget::GetNode() const
    {
        Node* node = nullptr;
        if (!m_selectedModelIndices.empty())
        {
            node = m_selectedModelIndices[0].data(SkeletonModel::ROLE_POINTER).value<Node*>();
        }
        return node;
    }
} // namespace EMotionFX