o3de-extras/Gems/ROS2RobotImporter/Code/Source/RobotImporter/Pages/PrefabMakerPage.cpp

/*
 * 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 "PrefabMakerPage.h"
#include <AzCore/Component/Entity.h>
#include <AzCore/Debug/Trace.h>
#include <AzCore/EBus/Results.h>
#include <AzCore/Math/Transform.h>
#include <AzCore/std/smart_ptr/make_shared.h>
#include <AzCore/std/string/string.h>

#include <ROS2/Spawner/SpawnerBus.h>
#include <ROS2/Spawner/SpawnerInfo.h>
#include <RobotImporter/RobotImporterWidget.h>
#include <optional>
#include <qcombobox.h>
#include <qvariant.h>

namespace ROS2RobotImporter
{
    PrefabMakerPage::PrefabMakerPage([[maybe_unused]] RobotImporterWidget* parent)
        : QWizardPage(parent)
        , m_success(false)
    {
        AZ::EBusAggregateResults<AZStd::unordered_map<AZStd::string, ROS2::SpawnPointInfo>> allActiveSpawnPoints;
        ROS2::SpawnerRequestsBus::BroadcastResult(allActiveSpawnPoints, &ROS2::SpawnerRequestsBus::Events::GetAllSpawnPointInfos);

        m_spawnPointsComboBox = new QComboBox(this);

        m_spawnPointsComboBox->addItem(tr(zeroPoint.data()));

        for (const auto& spawnPointMap : allActiveSpawnPoints.values)
        {
            for (const auto& spawnPoint : spawnPointMap)
            {
                m_spawnPointsComboBox->addItem(spawnPoint.first.c_str());
                m_spawnPointsInfos.insert({ spawnPoint.first.c_str(), spawnPoint.second });
            }
        }

        m_prefabName = new QLineEdit(this);
        m_createButton = new QPushButton(tr("Create Prefab"), this);
        m_log = new QTextEdit(this);
        m_log->acceptRichText();
        m_log->setReadOnly(true);

        setTitle(tr("Prefab creation"));
        QVBoxLayout* layout = new QVBoxLayout;
        QHBoxLayout* layoutInner = new QHBoxLayout;
        layoutInner->addWidget(m_prefabName);
        layoutInner->addWidget(m_createButton);
        layout->addLayout(layoutInner);
        QLabel* spawnPointListLabel;
        if (allActiveSpawnPoints.values.size() == 0)
        {
            spawnPointListLabel = new QLabel(tr("Select spawn position (No spawn positions were detected)"), this);
        }
        else
        {
            spawnPointListLabel = new QLabel(tr("Select spawn position"), this);
        }
        layout->addWidget(spawnPointListLabel);
        layout->addWidget(m_spawnPointsComboBox);
        layout->addWidget(m_log);
        setLayout(layout);
        connect(m_createButton, &QPushButton::pressed, this, &PrefabMakerPage::onCreateButtonPressed);
    }

    void PrefabMakerPage::SetProposedPrefabName(const AZStd::string prefabName)
    {
        m_prefabName->setText(QString::fromUtf8(prefabName.data(), int(prefabName.size())));
    }

    AZStd::string PrefabMakerPage::GetPrefabName() const
    {
        return AZStd::string(m_prefabName->text().toUtf8().constData());
    }

    void PrefabMakerPage::ReportProgress(const AZStd::string& progressForUser)
    {
        m_log->setMarkdown(QString::fromUtf8(progressForUser.data(), int(progressForUser.size())));
    }

    void PrefabMakerPage::SetSuccess(bool success)
    {
        m_success = success;
        emit completeChanged();
    }

    bool PrefabMakerPage::isComplete() const
    {
        return m_success;
    }

    AZStd::optional<AZ::Transform> PrefabMakerPage::getSelectedSpawnPoint() const
    {
        if (!m_spawnPointsInfos.empty())
        {
            AZStd::string spawnPointName(m_spawnPointsComboBox->currentText().toStdString().c_str());
            if (IsZeroPoint(spawnPointName))
            {
                return AZStd::nullopt;
            }
            if (auto spawnInfo = m_spawnPointsInfos.find(spawnPointName); spawnInfo != m_spawnPointsInfos.end())
            {
                return spawnInfo->second.pose;
            }
        }
        return AZStd::nullopt;
    }

    bool PrefabMakerPage::IsZeroPoint(AZStd::string spawnPointName)
    {
        return spawnPointName == PrefabMakerPage::zeroPoint;
    }
} // namespace ROS2RobotImporter