o3de/Gems/SceneProcessing/Code/Source/Config/SettingsObjects/FileSoftNameSetting.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 <AzCore/Component/ComponentApplicationBus.h>
#include <AzCore/Serialization/EditContext.h>
#include <AzFramework/StringFunc/StringFunc.h>
#include <SceneAPI/SceneCore/Containers/Scene.h>
#include <SceneAPI/SceneCore/DataTypes/IGraphObject.h>
#include <SceneAPI/SceneCore/Utilities/PatternMatcher.h>
#include <SceneAPI/SceneCore/Utilities/Reporting.h>
#include <Config/SettingsObjects/FileSoftNameSetting.h>

namespace AZ
{
    namespace SceneProcessingConfig
    {
        FileSoftNameSetting::GraphType::GraphType()
            : m_cachedId(Uuid::CreateNull())
        {
        }

        FileSoftNameSetting::GraphType::GraphType(const AZStd::string& name)
            : m_name(name)
            , m_cachedId(Uuid::CreateNull())
        {
        }

        FileSoftNameSetting::GraphType::GraphType(AZStd::string&& name)
            : m_name(AZStd::move(name))
            , m_cachedId(Uuid::CreateNull())
        {
        }

        const AZStd::string& FileSoftNameSetting::GraphType::GetName() const
        {
            return m_name;
        }

        const Uuid& FileSoftNameSetting::GraphType::GetId() const
        {
            if (m_cachedId.IsNull())
            {
                SerializeContext* context = nullptr;
                ComponentApplicationBus::BroadcastResult(context, &ComponentApplicationBus::Events::GetSerializeContext);
                AZ_Assert(context, "Unable to find valid serialize context.");

                context->EnumerateDerived<SceneAPI::DataTypes::IGraphObject>(
                    [this](const SerializeContext::ClassData* data, const Uuid& typeId) -> bool
                    {
                        AZ_UNUSED(typeId);
                        if (AzFramework::StringFunc::Equal(data->m_name, m_name.c_str()))
                        {
                            m_cachedId = data->m_typeId;
                            return false;
                        }
                        return true;
                    });

                if (m_cachedId.IsNull())
                {
                    AZ_TracePrintf(SceneAPI::Utilities::WarningWindow, "Unable to find '%s' in the serialize context.", m_name.c_str());
                }
            }
            return m_cachedId;
        }

        void FileSoftNameSetting::GraphType::Reflect(ReflectContext* context)
        {
            SerializeContext* serialize = azrtti_cast<SerializeContext*>(context);
            if (serialize)
            {
                serialize->Class<GraphType>()
                    ->Version(1)
                    ->Field("name", &GraphType::m_name);
            }
        }

        FileSoftNameSetting::GraphTypeContainer::GraphTypeContainer(std::initializer_list<GraphType> graphTypes)
            : m_types(graphTypes)
        {
        }

        AZStd::vector<FileSoftNameSetting::GraphType>& FileSoftNameSetting::GraphTypeContainer::GetGraphTypes()
        {
            return m_types;
        }

        const AZStd::vector<FileSoftNameSetting::GraphType>& FileSoftNameSetting::GraphTypeContainer::GetGraphTypes() const
        {
            return m_types;
        }

        void FileSoftNameSetting::GraphTypeContainer::Reflect(ReflectContext* context)
        {
            SerializeContext* serialize = azrtti_cast<SerializeContext*>(context);
            if (serialize)
            {
                serialize->Class<GraphTypeContainer>()
                    ->Version(1)
                    ->Field("types", &GraphTypeContainer::m_types);
            }
        }


        FileSoftNameSetting::FileSoftNameSetting(const char* pattern, SceneAPI::SceneCore::PatternMatcher::MatchApproach approach,
            const char* virtualType, bool inclusive, std::initializer_list<GraphType> graphTypes)
            : SoftNameSetting(pattern, approach, virtualType)
            , m_inclusiveList(inclusive)
            , m_graphTypes(graphTypes)
            , m_cachedScene(nullptr)
        {
        }

        bool FileSoftNameSetting::IsVirtualType(const SceneAPI::Containers::Scene& scene, SceneAPI::Containers::SceneGraph::NodeIndex node) const
        {
            bool nameMatch = false;
            if (m_cachedScene == &scene)
            {
                nameMatch = m_cachedNameMatch;
            }
            else
            {
                switch (m_pattern.GetMatchApproach())
                {
                case SceneAPI::SceneCore::PatternMatcher::MatchApproach::PreFix:
                    nameMatch = m_pattern.MatchesPattern(scene.GetName());
                    break;
                case SceneAPI::SceneCore::PatternMatcher::MatchApproach::PostFix:
                    nameMatch = m_pattern.MatchesPattern(scene.GetName());
                    break;
                case SceneAPI::SceneCore::PatternMatcher::MatchApproach::Regex:
                    nameMatch = m_pattern.MatchesPattern(scene.GetSourceFilename());
                    break;
                default:
                    AZ_Assert(false, "Unknown option '%i' for pattern matcher.", m_pattern.GetMatchApproach());
                    nameMatch = false;
                }

                m_cachedNameMatch = nameMatch;
                m_cachedScene = &scene;
            }

            if (nameMatch)
            {
                AZStd::shared_ptr<const SceneAPI::DataTypes::IGraphObject> object = scene.GetGraph().GetNodeContent(node);
                for (const GraphType& type : m_graphTypes.GetGraphTypes())
                {
                    if (object->RTTI_IsTypeOf(type.GetId()))
                    {
                        return m_inclusiveList;
                    }
                }
                return !m_inclusiveList;
            }
            else
            {
                return false;
            }
        }

        void FileSoftNameSetting::Reflect(ReflectContext* context)
        {
            GraphType::Reflect(context);
            GraphTypeContainer::Reflect(context);

            SerializeContext* serialize = azrtti_cast<SerializeContext*>(context);
            if (serialize)
            {
                serialize->Class<FileSoftNameSetting, SoftNameSetting>()
                    ->Version(1)
                    ->Field("graphTypes", &FileSoftNameSetting::m_graphTypes)
                    ->Field("inclusiveList", &FileSoftNameSetting::m_inclusiveList);

                EditContext* editContext = serialize->GetEditContext();
                if (editContext)
                {
                    editContext->Class<FileSoftNameSetting>("File name setting", "Applies the pattern to the name of the scene file.")
                        ->ClassElement(Edit::ClassElements::EditorData, "")
                            ->Attribute(Edit::Attributes::AutoExpand, true)
                        ->DataElement(AZ_CRC("GraphTypeSelector", 0x362ac245), &FileSoftNameSetting::m_graphTypes, "Graph type",
                            "The graph types that are the soft name applies to.")
                            ->Attribute(Edit::Attributes::AutoExpand, true)
                        ->DataElement(Edit::UIHandlers::Default, &FileSoftNameSetting::m_inclusiveList, "Inclusive", 
                            "If true the types in the list will marked as the virtual type, otherwise any types that are NOT in the list.");
                }
            }
        }
    } // namespace SceneProcessingConfig
} // namespace AZ