/*
 * All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
 * its licensors.
 *
 * For complete copyright and license terms please see the LICENSE at the root of this
 * distribution (the "License"). All use of this software is governed by the License,
 * or, if provided, by the license below or the license accompanying this file. Do not
 * remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *
 */

#pragma once

struct ImGuiContext;

////////////////////////////////////////////////////////////////////////////////////////////////////
namespace ImGui
{
    ////////////////////////////////////////////////////////////////////////////////////////////////
    //! Utility class to set a new current ImGui context on construction,
    //! then reset the previous ImGui context on destruction.
    class ImGuiContextScope
    {
    public:
        ////////////////////////////////////////////////////////////////////////////////////////////
        explicit ImGuiContextScope([[maybe_unused]] ImGuiContext* newContext)
            : m_previousContext(nullptr)
        {
#if defined(IMGUI_ENABLED)
            m_previousContext = ImGui::GetCurrentContext();
            ImGui::SetCurrentContext(newContext);
#endif
        }

        ////////////////////////////////////////////////////////////////////////////////////////////
        ~ImGuiContextScope()
        {
#if defined(IMGUI_ENABLED)
            ImGui::SetCurrentContext(m_previousContext);
#endif
        }

    private:
        ////////////////////////////////////////////////////////////////////////////////////////////
        ImGuiContext* m_previousContext = nullptr;
    };
}