/*
 * 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/Memory/SystemAllocator.h>
#include <AzCore/RTTI/RTTI.h>
#include <AzCore/Module/Module.h>

#include <CommonFeaturesSystemComponent.h>
#include <CoreLights/AreaLightComponent.h>
#include <CoreLights/DirectionalLightComponent.h>
#include <CubeMapCapture/CubeMapCaptureComponent.h>
#include <Debug/RayTracingDebugComponent.h>
#include <Debug/RenderDebugComponent.h>
#include <Decals/DecalComponent.h>
#include <Grid/GridComponent.h>
#include <ImageBasedLights/ImageBasedLightComponent.h>
#include <Material/MaterialComponent.h>
#include <Mesh/MeshComponent.h>
#include <ReflectionProbe/ReflectionProbeComponent.h>
#include <SpecularReflections/SpecularReflectionsComponent.h>
#include <OcclusionCullingPlane/OcclusionCullingPlaneComponent.h>
#include <PostProcess/PostFxLayerComponent.h>
#include <PostProcess/Bloom/BloomComponent.h>
#include <PostProcess/ColorGrading/HDRColorGradingComponent.h>
#include <PostProcess/DepthOfField/DepthOfFieldComponent.h>
#include <PostProcess/DisplayMapper/DisplayMapperComponent.h>
#include <PostProcess/ExposureControl/ExposureControlComponent.h>
#include <PostProcess/Ssao/SsaoComponent.h>
#include <PostProcess/LookModification/LookModificationComponent.h>
#include <PostProcess/RadiusWeightModifier/RadiusWeightModifierComponent.h>
#include <PostProcess/ShapeWeightModifier/ShapeWeightModifierComponent.h>
#include <PostProcess/GradientWeightModifier/GradientWeightModifierComponent.h>
#include <PostProcess/ChromaticAberration/ChromaticAberrationComponent.h>
#include <PostProcess/PaniniProjection/PaniniProjectionComponent.h>
#include <PostProcess/FilmGrain/FilmGrainComponent.h>
#include <PostProcess/WhiteBalance/WhiteBalanceComponent.h>
#include <PostProcess/Vignette/VignetteComponent.h>
#include <ScreenSpace/DeferredFogComponent.h>
#include <SkyAtmosphere/SkyAtmosphereComponent.h>
#include <SkyBox/HDRiSkyboxComponent.h>
#include <SkyBox/PhysicalSkyComponent.h>
#include <Scripting/EntityReferenceComponent.h>
#include <SurfaceData/SurfaceDataMeshComponent.h>
#include <Animation/AttachmentComponent.h>

#ifdef ATOMLYINTEGRATION_FEATURE_COMMON_EDITOR
#include <EditorCommonFeaturesSystemComponent.h>
#include <PostProcess/EditorPostFxSystemComponent.h>
#include <CoreLights/EditorAreaLightComponent.h>
#include <CoreLights/EditorDirectionalLightComponent.h>
#include <CubeMapCapture/EditorCubeMapCaptureComponent.h>
#include <Debug/RayTracingDebugEditorComponent.h>
#include <Debug/RenderDebugEditorComponent.h>
#include <Decals/EditorDecalComponent.h>
#include <Grid/EditorGridComponent.h>
#include <ImageBasedLights/EditorImageBasedLightComponent.h>
#include <Material/EditorMaterialComponent.h>
#include <Material/EditorMaterialSystemComponent.h>
#include <Mesh/EditorMeshComponent.h>
#include <Mesh/EditorMeshSystemComponent.h>
#include <ReflectionProbe/EditorReflectionProbeComponent.h>
#include <SpecularReflections/EditorSpecularReflectionsComponent.h>
#include <OcclusionCullingPlane/EditorOcclusionCullingPlaneComponent.h>
#include <PostProcess/EditorPostFxLayerComponent.h>
#include <PostProcess/Bloom/EditorBloomComponent.h>
#include <PostProcess/ColorGrading/EditorHDRColorGradingComponent.h>
#include <PostProcess/DepthOfField/EditorDepthOfFieldComponent.h>
#include <PostProcess/DisplayMapper/EditorDisplayMapperComponent.h>
#include <PostProcess/ExposureControl/EditorExposureControlComponent.h>
#include <PostProcess/Ssao/EditorSsaoComponent.h>
#include <PostProcess/LookModification/EditorLookModificationComponent.h>
#include <PostProcess/RadiusWeightModifier/EditorRadiusWeightModifierComponent.h>
#include <PostProcess/ShapeWeightModifier/EditorShapeWeightModifierComponent.h>
#include <PostProcess/GradientWeightModifier/EditorGradientWeightModifierComponent.h>
#include <PostProcess/ChromaticAberration/EditorChromaticAberrationComponent.h>
#include <PostProcess/PaniniProjection/EditorPaniniProjectionComponent.h>
#include <PostProcess/FilmGrain/EditorFilmGrainComponent.h>
#include <PostProcess/WhiteBalance/EditorWhiteBalanceComponent.h>
#include <PostProcess/Vignette/EditorVignetteComponent.h>
#include <ScreenSpace/EditorDeferredFogComponent.h>
#include <SkyAtmosphere/EditorSkyAtmosphereComponent.h>
#include <SkyBox/EditorHDRiSkyboxComponent.h>
#include <SkyBox/EditorPhysicalSkyComponent.h>
#include <Scripting/EditorEntityReferenceComponent.h>
#include <SurfaceData/EditorSurfaceDataMeshComponent.h>
#include <Animation/EditorAttachmentComponent.h>
#endif

namespace AZ
{
    namespace Render
    {
        class AtomLyIntegrationCommonFeaturesModule
            : public AZ::Module
        {
        public:
            AZ_RTTI(AtomLyIntegrationCommonFeaturesModule, "{E6FF4862-9355-4B23-AE00-B936F0C6E6C9}", AZ::Module);
            AZ_CLASS_ALLOCATOR(AtomLyIntegrationCommonFeaturesModule, AZ::SystemAllocator);

            AtomLyIntegrationCommonFeaturesModule()
                : AZ::Module()
            {
                m_descriptors.insert(m_descriptors.end(), {
                        AtomLyIntegrationCommonFeaturesSystemComponent::CreateDescriptor(),
                        AreaLightComponent::CreateDescriptor(),
                        DecalComponent::CreateDescriptor(),
                        DirectionalLightComponent::CreateDescriptor(),
                        BloomComponent::CreateDescriptor(),
                        HDRColorGradingComponent::CreateDescriptor(),
                        DisplayMapperComponent::CreateDescriptor(),
                        DepthOfFieldComponent::CreateDescriptor(),
                        ExposureControlComponent::CreateDescriptor(),
                        SsaoComponent::CreateDescriptor(),
                        LookModificationComponent::CreateDescriptor(),
                        GridComponent::CreateDescriptor(),
                        HDRiSkyboxComponent::CreateDescriptor(),
                        SkyAtmosphereComponent::CreateDescriptor(),
                        ImageBasedLightComponent::CreateDescriptor(),
                        MaterialComponent::CreateDescriptor(),
                        MeshComponent::CreateDescriptor(),
                        PhysicalSkyComponent::CreateDescriptor(),
                        PostFxLayerComponent::CreateDescriptor(),
                        ReflectionProbeComponent::CreateDescriptor(),
                        SpecularReflectionsComponent::CreateDescriptor(),
                        RayTracingDebugComponent::CreateDescriptor(),
                        RenderDebugComponent::CreateDescriptor(),
                        RadiusWeightModifierComponent::CreateDescriptor(),
                        ShapeWeightModifierComponent::CreateDescriptor(),
                        EntityReferenceComponent::CreateDescriptor(),
                        GradientWeightModifierComponent::CreateDescriptor(),
                        DeferredFogComponent::CreateDescriptor(),
                        SurfaceData::SurfaceDataMeshComponent::CreateDescriptor(),
                        AttachmentComponent::CreateDescriptor(),
                        OcclusionCullingPlaneComponent::CreateDescriptor(),
                        ChromaticAberrationComponent::CreateDescriptor(),
                        PaniniProjectionComponent::CreateDescriptor(),
                        FilmGrainComponent::CreateDescriptor(),
                        WhiteBalanceComponent::CreateDescriptor(),
                        VignetteComponent::CreateDescriptor(),
                        CubeMapCaptureComponent::CreateDescriptor(),

#ifdef ATOMLYINTEGRATION_FEATURE_COMMON_EDITOR
                        EditorAreaLightComponent::CreateDescriptor(),
                        EditorCommonFeaturesSystemComponent::CreateDescriptor(),
                        EditorPostFxSystemComponent::CreateDescriptor(),
                        EditorDecalComponent::CreateDescriptor(),
                        EditorDirectionalLightComponent::CreateDescriptor(),
                        EditorBloomComponent::CreateDescriptor(),
                        EditorHDRColorGradingComponent::CreateDescriptor(),
                        EditorDepthOfFieldComponent::CreateDescriptor(),
                        EditorDisplayMapperComponent::CreateDescriptor(),
                        EditorExposureControlComponent::CreateDescriptor(),
                        EditorSsaoComponent::CreateDescriptor(),
                        EditorLookModificationComponent::CreateDescriptor(),
                        EditorGridComponent::CreateDescriptor(),
                        EditorHDRiSkyboxComponent::CreateDescriptor(),
                        EditorSkyAtmosphereComponent::CreateDescriptor(),
                        EditorImageBasedLightComponent::CreateDescriptor(),
                        EditorMaterialComponent::CreateDescriptor(),
                        EditorMaterialSystemComponent::CreateDescriptor(),
                        EditorMeshSystemComponent::CreateDescriptor(),
                        EditorMeshComponent::CreateDescriptor(),
                        EditorPhysicalSkyComponent::CreateDescriptor(),
                        EditorPostFxLayerComponent::CreateDescriptor(),
                        EditorReflectionProbeComponent::CreateDescriptor(),
                        EditorSpecularReflectionsComponent::CreateDescriptor(),
                        RayTracingDebugEditorComponent::CreateDescriptor(),
                        RenderDebugEditorComponent::CreateDescriptor(),
                        EditorRadiusWeightModifierComponent::CreateDescriptor(),
                        EditorShapeWeightModifierComponent::CreateDescriptor(),
                        EditorEntityReferenceComponent::CreateDescriptor(),
                        EditorGradientWeightModifierComponent::CreateDescriptor(),
                        EditorDeferredFogComponent::CreateDescriptor(),
                        SurfaceData::EditorSurfaceDataMeshComponent::CreateDescriptor(),
                        EditorAttachmentComponent::CreateDescriptor(),
                        EditorOcclusionCullingPlaneComponent::CreateDescriptor(),
                        EditorChromaticAberrationComponent::CreateDescriptor(),
                        EditorPaniniProjectionComponent::CreateDescriptor(),
                        EditorFilmGrainComponent::CreateDescriptor(),
                        EditorWhiteBalanceComponent::CreateDescriptor(),
                        EditorVignetteComponent::CreateDescriptor(),
                        EditorCubeMapCaptureComponent::CreateDescriptor(),
#endif
                    });
            }

            AZ::ComponentTypeList GetRequiredSystemComponents() const override
            {
                return AZ::ComponentTypeList{
                    azrtti_typeid<AtomLyIntegrationCommonFeaturesSystemComponent>(),
#ifdef ATOMLYINTEGRATION_FEATURE_COMMON_EDITOR
                    azrtti_typeid<EditorMaterialSystemComponent>(),
                    azrtti_typeid<EditorMeshSystemComponent>(),
                    azrtti_typeid<EditorCommonFeaturesSystemComponent>(),
                    azrtti_typeid<EditorPostFxSystemComponent>(),
#endif
                };
            }
        };
    } // namespace Render
} // namespace AZ

#if defined(O3DE_GEM_NAME)
AZ_DECLARE_MODULE_CLASS(AZ_JOIN(Gem_, O3DE_GEM_NAME), AZ::Render::AtomLyIntegrationCommonFeaturesModule)
#else
AZ_DECLARE_MODULE_CLASS(Gem_CommonFeaturesAtom, AZ::Render::AtomLyIntegrationCommonFeaturesModule)
#endif