/*
 * 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 <AssetBuilderSDK/AssetBuilderSDK.h>

#include <RHI.Builders/ShaderPlatformInterfaceSystemComponent.h>

#include <Atom/RHI.Edit/ShaderPlatformInterfaceBus.h>
#include <AzCore/Serialization/EditContext.h>
#include <AzCore/Serialization/SerializeContext.h>

#if defined(AZ_TOOLS_EXPAND_FOR_RESTRICTED_PLATFORMS)
#   if defined(TOOLS_SUPPORT_JASPER)
#       include AZ_RESTRICTED_FILE_EXPLICIT(RHI.Builders/ShaderPlatformInterface, Jasper)
#   endif
#   if defined(TOOLS_SUPPORT_PROVO)
#       include AZ_RESTRICTED_FILE_EXPLICIT(RHI.Builders/ShaderPlatformInterface, Provo)
#   endif
#   if defined(TOOLS_SUPPORT_SALEM)
#       include AZ_RESTRICTED_FILE_EXPLICIT(RHI.Builders/ShaderPlatformInterface, Salem)
#   endif
#endif

#include <Atom/RHI.Reflect/DX12/Base.h>

namespace AZ
{
    namespace DX12
    {
        void ShaderPlatformInterfaceSystemComponent::Reflect(AZ::ReflectContext* context)
        {
            if (SerializeContext* serialize = azrtti_cast<SerializeContext*>(context))
            {
                serialize->Class<ShaderPlatformInterfaceSystemComponent, Component>()
                    ->Version(0)
                    ->Attribute(Edit::Attributes::SystemComponentTags, AZStd::vector<Crc32>({ AssetBuilderSDK::ComponentTags::AssetBuilder }))
                    ;
            }
        }

        void ShaderPlatformInterfaceSystemComponent::GetDependentServices(AZ::ComponentDescriptor::DependencyArrayType& dependent)
        {
            dependent.push_back(AZ_CRC("AzslShaderBuilderService", 0x09315a40));
        }

        void ShaderPlatformInterfaceSystemComponent::Activate()
        {
            m_shaderPlatformInterfaces.emplace_back(AZStd::make_unique<DX12::ShaderPlatformInterface>(APIUniqueIndex));

#if defined(AZ_TOOLS_EXPAND_FOR_RESTRICTED_PLATFORMS)
#define AZ_RESTRICTED_PLATFORM_EXPANSION(CodeName, CODENAME, codename, PrivateName, PRIVATENAME, privatename, PublicName, PUBLICNAME, publicname, PublicAuxName1, PublicAuxName2, PublicAuxName3)\
            CreateShaderPlatformInterface##CodeName(m_shaderPlatformInterfaces);
            AZ_TOOLS_EXPAND_FOR_RESTRICTED_PLATFORMS
#undef AZ_RESTRICTED_PLATFORM_EXPANSION
#endif

            for (auto& shaderPlatformInterface : m_shaderPlatformInterfaces)
            {
                RHI::ShaderPlatformInterfaceRegisterBus::Broadcast(
                    &RHI::ShaderPlatformInterfaceRegister::RegisterShaderPlatformHandler,
                    shaderPlatformInterface.get());
            }
        }

        void ShaderPlatformInterfaceSystemComponent::Deactivate()
        {
            for (auto& shaderPlatformInterface : m_shaderPlatformInterfaces)
            {
                RHI::ShaderPlatformInterfaceRegisterBus::Broadcast(
                    &RHI::ShaderPlatformInterfaceRegister::UnregisterShaderPlatformHandler,
                    shaderPlatformInterface.get());
                shaderPlatformInterface = nullptr;
            }
        }

    } // namespace DX12
} // namespace AZ