/*
 * 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
 *
 */

#pragma once

#include <RHI/BasicRHIComponent.h>

#include <AzCore/Component/TickBus.h>

namespace AtomSampleViewer
{
    // This sample has the simple purpose of testing the texture array feature. This sample stores multiple image views
    // within a single binding slot within a SRG. This allows the shader to index within this texture array to sample from a texture. 
    class TextureArrayExampleComponent final
        : public BasicRHIComponent
        , public AZ::TickBus::Handler
    {
        using Base = BasicRHIComponent;

    public:
        AZ_COMPONENT(TextureArrayExampleComponent, "{11EAD34A-B4BD-4CBA-B20B-E5AB33F4F49D}", AZ::Component);
        AZ_DISABLE_COPY(TextureArrayExampleComponent);

        TextureArrayExampleComponent();
        ~TextureArrayExampleComponent() override = default;

        static void Reflect(AZ::ReflectContext* context);

        //AZ::Component
        void Activate();
        void Deactivate();

    private:
        // AZ::TickBus::Handler overrides...
        void OnTick(float deltaTime, AZ::ScriptTimePoint scriptTime);

        AZ::RHI::Ptr<AZ::RHI::BufferPool> m_inputAssemblyBufferPool;
        AZ::RHI::Ptr<AZ::RHI::Buffer> m_rectangleInputAssemblyBuffer;
        AZStd::array<AZ::RHI::StreamBufferView, 2> m_rectangleStreamBufferViews;
        AZ::RHI::InputStreamLayout m_rectangleInputStreamLayout;
        AZ::RHI::ConstPtr<AZ::RHI::PipelineState> m_pipelineState;
        AZ::RHI::GeometryView m_geometryView{ AZ::RHI::MultiDevice::AllDevices };

        // Srg related resources
        AZ::Data::Instance<AZ::RPI::Shader> m_shader;
        AZ::Data::Instance<AZ::RPI::ShaderResourceGroup> m_textureArraySrg;
        AZ::Data::Instance<AZ::RPI::ShaderResourceGroup> m_textureIndexSrg;
        AZ::RHI::ShaderInputImageIndex m_textureArray;
        AZ::RHI::ShaderInputConstantIndex m_textureIndex;
    };
} // namespace AtomSampleViewer