#include <mln/mtl/renderer_backend.hpp>

#include <mln/gfx/backend_scope.hpp>
#include <mln/gfx/shader_registry.hpp>
#include <mln/mtl/context.hpp>
#include <mln/shaders/background_layer_ubo.hpp>

#include <mln/shaders/mtl/shader_group.hpp>
#include <mln/shaders/mtl/shader_program.hpp>
#include <mln/shaders/shader_manifest.hpp>
#include <mln/util/logging.hpp>

// ... shader_manifest.hpp
#include <mln/shaders/mtl/background.hpp>
#include <mln/shaders/mtl/circle.hpp>
#include <mln/shaders/mtl/clipping_mask.hpp>
#include <mln/shaders/mtl/collision.hpp>
#include <mln/shaders/mtl/color_relief.hpp>
#include <mln/shaders/mtl/custom_geometry.hpp>
#include <mln/shaders/mtl/custom_symbol_icon.hpp>
#include <mln/shaders/mtl/debug.hpp>
#include <mln/shaders/mtl/fill.hpp>
#include <mln/shaders/mtl/fill_extrusion.hpp>
#include <mln/shaders/mtl/heatmap.hpp>
#include <mln/shaders/mtl/heatmap_texture.hpp>
#include <mln/shaders/mtl/hillshade.hpp>
#include <mln/shaders/mtl/hillshade_prepare.hpp>
#include <mln/shaders/mtl/line.hpp>
#include <mln/shaders/mtl/location_indicator.hpp>
#include <mln/shaders/mtl/raster.hpp>
#include <mln/shaders/mtl/symbol.hpp>
#include <mln/shaders/mtl/widevector.hpp>

#include <cassert>
#include <string>

namespace mln {
namespace mtl {

RendererBackend::RendererBackend(const gfx::ContextMode contextMode_)
    : gfx::RendererBackend(contextMode_),
      device(NS::TransferPtr(MTL::CreateSystemDefaultDevice())),
      commandQueue(NS::TransferPtr(device->newCommandQueue())) {
    assert(device);
    assert(commandQueue);
#if TARGET_OS_SIMULATOR
    baseVertexInstanceDrawingSupported = true;
#else
    baseVertexInstanceDrawingSupported = device->supportsFamily(MTL::GPUFamilyApple3);
#endif
}

RendererBackend::~RendererBackend() = default;

std::unique_ptr<gfx::Context> RendererBackend::createContext() {
    return std::make_unique<mtl::Context>(*this);
}

PremultipliedImage RendererBackend::readFramebuffer(const Size& size) {
    return PremultipliedImage(size);
}

/// @brief Register a list of types with a shader registry instance
/// @tparam ...ShaderID Pack of BuiltIn:: shader IDs
/// @param registry A shader registry instance
/// @param programParameters ProgramParameters used to initialize each instance
template <shaders::BuiltIn... ShaderID>
void registerTypes(gfx::ShaderRegistry& registry, const ProgramParameters& programParameters) {
    /// The following fold expression will create a shader for every type
    /// in the parameter pack and register it with the shader registry.

    /// Registration calls are wrapped in a lambda that throws on registration
    /// failure, we shouldn't expect registration to fail unless the shader
    /// registry instance provided already has conflicting programs present.
    (
        [&]() {
            using namespace std::string_literals;
            using ShaderClass = shaders::ShaderSource<ShaderID, gfx::Backend::Type::Metal>;
            auto group = std::make_shared<ShaderGroup<ShaderID>>(programParameters);
            if (!registry.registerShaderGroup(std::move(group), ShaderClass::name)) {
                assert(!"duplicate shader group");
                throw std::runtime_error("Failed to register "s + ShaderClass::name + " with shader registry!");
            }
        }(),
        ...);
}

void RendererBackend::initShaders(gfx::ShaderRegistry& shaders, const ProgramParameters& programParameters) {
    registerTypes<shaders::BuiltIn::BackgroundShader,
                  shaders::BuiltIn::BackgroundPatternShader,
                  shaders::BuiltIn::CircleShader,
                  shaders::BuiltIn::ClippingMaskProgram,
                  shaders::BuiltIn::CollisionBoxShader,
                  shaders::BuiltIn::CollisionCircleShader,
                  shaders::BuiltIn::CustomGeometryShader,
                  shaders::BuiltIn::CustomSymbolIconShader,
                  shaders::BuiltIn::DebugShader,
                  shaders::BuiltIn::FillShader,
                  shaders::BuiltIn::FillOutlineShader,
                  shaders::BuiltIn::FillPatternShader,
                  shaders::BuiltIn::FillOutlinePatternShader,
                  shaders::BuiltIn::FillOutlineTriangulatedShader,
                  shaders::BuiltIn::FillExtrusionShader,
                  shaders::BuiltIn::FillExtrusionInstancedShader,
                  shaders::BuiltIn::FillExtrusionPatternShader,
                  shaders::BuiltIn::FillExtrusionPatternInstancedShader,
                  shaders::BuiltIn::HeatmapShader,
                  shaders::BuiltIn::HeatmapTextureShader,
                  shaders::BuiltIn::HillshadeShader,
                  shaders::BuiltIn::HillshadePrepareShader,
                  shaders::BuiltIn::ColorReliefShader,
                  shaders::BuiltIn::LineShader,
                  shaders::BuiltIn::LineGradientShader,
                  shaders::BuiltIn::LineSDFShader,
                  shaders::BuiltIn::LinePatternShader,
                  shaders::BuiltIn::LocationIndicatorShader,
                  shaders::BuiltIn::LocationIndicatorTexturedShader,
                  shaders::BuiltIn::RasterShader,
                  shaders::BuiltIn::SymbolIconShader,
                  shaders::BuiltIn::SymbolSDFShader,
                  shaders::BuiltIn::SymbolTextAndIconShader,
                  shaders::BuiltIn::WideVectorShader>(shaders, programParameters);
}

} // namespace mtl
} // namespace mln