/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef GRAPHICS_CONTEXT_H
#define GRAPHICS_CONTEXT_H

#include <3d/intf_graphics_context.h>
#include <base/containers/unique_ptr.h>
#include <base/containers/unordered_map.h>
#include <base/containers/vector.h>
#include <core/namespace.h>
#include <core/plugin/intf_class_factory.h>
#include <core/plugin/intf_class_register.h>
#include <core/plugin/intf_plugin.h>
#include <render/resource_handle.h>

CORE_BEGIN_NAMESPACE()
class IEcs;
CORE_END_NAMESPACE()

RENDER_BEGIN_NAMESPACE()
class IRenderContext;
class IRenderDataStore;
RENDER_END_NAMESPACE()

CORE3D_BEGIN_NAMESPACE()
class IMeshUtil;
class MeshUtil;
class IGltf2;
class Gltf2;
class ISceneUtil;
class SceneUtil;
class IRenderUtil;
class RenderUtil;
struct I3DPlugin;

class GraphicsContext final : public IGraphicsContext,
                              virtual CORE_NS::IClassRegister,
                              virtual CORE_NS::IClassFactory,
                              CORE_NS::IPluginRegister::ITypeInfoListener {
public:
    explicit GraphicsContext(struct Agp3DPluginState&, RENDER_NS::IRenderContext& context);
    ~GraphicsContext() override;

    GraphicsContext(const GraphicsContext&) = delete;
    GraphicsContext& operator=(const GraphicsContext&) = delete;

    void Init() override;

    RENDER_NS::IRenderContext& GetRenderContext() const override;

    BASE_NS::array_view<const RENDER_NS::RenderHandleReference> GetRenderNodeGraphs(
        const CORE_NS::IEcs& ecs) const override;

    ISceneUtil& GetSceneUtil() const override;

    IMeshUtil& GetMeshUtil() const override;

    IGltf2& GetGltf() const override;

    IRenderUtil& GetRenderUtil() const override;

    // IInterface
    const CORE_NS::IInterface* GetInterface(const BASE_NS::Uid& uid) const override;
    CORE_NS::IInterface* GetInterface(const BASE_NS::Uid& uid) override;
    void Ref() override;
    void Unref() override;

private:
    // IClassRegister
    void RegisterInterfaceType(const CORE_NS::InterfaceTypeInfo& interfaceInfo) override;
    void UnregisterInterfaceType(const CORE_NS::InterfaceTypeInfo& interfaceInfo) override;
    BASE_NS::array_view<const CORE_NS::InterfaceTypeInfo* const> GetInterfaceMetadata() const override;
    const CORE_NS::InterfaceTypeInfo& GetInterfaceMetadata(const BASE_NS::Uid& uid) const override;
    IInterface* GetInstance(const BASE_NS::Uid& uid) const override;

    // IClassFactory
    IInterface::Ptr CreateInstance(const BASE_NS::Uid& uid) override;

    // IPluginRegister::ITypeInfoListener
    void OnTypeInfoEvent(EventType type, BASE_NS::array_view<const CORE_NS::ITypeInfo* const> typeInfos) override;

    struct Agp3DPluginState& factory_;
    RENDER_NS::IRenderContext& context_;

    BASE_NS::vector<BASE_NS::pair<CORE_NS::PluginToken, const I3DPlugin*>> plugins_;
    BASE_NS::vector<const CORE_NS::InterfaceTypeInfo*> interfaceTypeInfos_;

    BASE_NS::unique_ptr<SceneUtil> sceneUtil_;
    BASE_NS::unique_ptr<MeshUtil> meshUtil_;
    BASE_NS::unique_ptr<Gltf2> gltf2_;
    BASE_NS::unique_ptr<RenderUtil> renderUtil_;
    bool initialized_ { false };
    uint32_t refcnt_ { 0 };
};

inline constexpr BASE_NS::string_view GetName(const IGraphicsContext*)
{
    return "IGraphicsContext3D";
}
CORE3D_END_NAMESPACE()

#endif // GRAPHICS_CONTEXT_H