o3de/Gems/Atom/Tools/AtomToolsFramework/Code/Source/PreviewRenderer/PreviewRendererCaptureState.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 <PreviewRenderer/PreviewRenderer.h>
#include <PreviewRenderer/PreviewRendererCaptureState.h>

namespace AtomToolsFramework
{
    PreviewRendererCaptureState::PreviewRendererCaptureState(PreviewRenderer* renderer)
        : PreviewRendererState(renderer)
    {
        m_renderer->PoseContent();
    }

    PreviewRendererCaptureState::~PreviewRendererCaptureState()
    {
        AZ::Render::FrameCaptureNotificationBus::Handler::BusDisconnect();
        m_renderer->EndCapture();
    }

    void PreviewRendererCaptureState::Update()
    {
        if (m_captureComplete)
        {
            AZ::Render::FrameCaptureNotificationBus::Handler::BusDisconnect();
            m_renderer->CompleteCaptureRequest();
            return;
        }

        if (AZStd::chrono::steady_clock::now() > m_abortTime)
        {
            AZ::Render::FrameCaptureNotificationBus::Handler::BusDisconnect();
            m_renderer->CancelCaptureRequest();
            return;
        }

        if (AZStd::chrono::steady_clock::now() > m_captureTime)
        {
            if (!AZ::Render::FrameCaptureNotificationBus::Handler::BusIsConnected())
            {
                // if the start capture call fails the capture will be retried next tick.
                const AZ::Render::FrameCaptureId frameCaptureId = m_renderer->StartCapture();
                if (frameCaptureId != AZ::Render::InvalidFrameCaptureId)
                {
                    AZ::Render::FrameCaptureNotificationBus::Handler::BusConnect(frameCaptureId);
                }
            }
        }
    }

    void PreviewRendererCaptureState::OnFrameCaptureFinished(
        [[maybe_unused]] AZ::Render::FrameCaptureResult result, [[maybe_unused]] const AZStd::string& info)
    {
        AZ::Render::FrameCaptureNotificationBus::Handler::BusDisconnect();
        m_captureComplete = true;
    }
} // namespace AtomToolsFramework