/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

#pragma once

#if __has_include("msdfgen.h")

#include "Scene/AtlasMetadata.hpp"
#include "FontAtlasGenerator.hpp"
#include "Scene/Texture.hpp"
#include <msdfgen.h>
#include <msdf-atlas-gen/msdf-atlas-gen.h>
#include <string>
#include <optional>
#include <map>
#include <variant>
#define MSDFGEN_AVAILABLE 1

namespace OpenVulkano::Scene
{

	struct MsdfFontAtlasGeneratorConfig
	{
		int glyphSize = 42;
		double miterLimit = 1.0;
		msdfgen::Range pixelRange = 5;
	};

	class MsdfFontAtlasGenerator : public FontAtlasGenerator
	{
	public:
		using SdfGenerator = msdf_atlas::ImmediateAtlasGenerator<float, 1, msdf_atlas::sdfGenerator,
		                                                         msdf_atlas::BitmapAtlasStorage<msdfgen::byte, 1>>;
		using Config = MsdfFontAtlasGeneratorConfig;
		static msdf_atlas::Charset LoadAllGlyphs(const std::variant<std::string, Array<char>>& data);
		void GenerateAtlas(const std::string& fontFile, const std::set<uint32_t>& charset,
		                   const std::optional<std::string>& pngOutput = std::nullopt) override;
		void GenerateAtlas(const Array<char>& fontData, int length, const std::set<uint32_t>& charset,
		                   const std::optional<std::string>& pngOutput = std::nullopt) override;
		void GenerateAtlas(const std::string& fontFile, const msdf_atlas::Charset& charset = msdf_atlas::Charset::ASCII,
		                   const std::optional<std::string>& pngOutput = std::nullopt);
		void GenerateAtlas(const msdfgen::byte* fontData, int length,
		                   const msdf_atlas::Charset& charset = msdf_atlas::Charset::ASCII,
		                   const std::optional<std::string>& pngOutput = std::nullopt);
		void SaveAtlasMetadataInfo(const std::string& outputFile, bool packIntoSingleFile = true) const override;
		void SetGeneratorConfig(const Config& config) { m_config = config; }
		const Texture& GetAtlas() const override { return m_atlasTex; }
		std::map<uint32_t, GlyphInfo>& GetGlyphsInfo() override { return m_symbols; }
		AtlasMetadata& GetAtlasMetadata() override { return m_meta; }
		SdfGenerator& GetFontAtlasGenerator() { return m_generator; }
		Config& GetGeneratorConfig() { return m_config; }
	private:
		void InitFreetypeFromFile(msdfgen::FreetypeHandle*& ft, msdfgen::FontHandle*& font, const std::string& file);
		void InitFreetypeFromBuffer(msdfgen::FreetypeHandle*& ft, msdfgen::FontHandle*& font,
		                            const msdfgen::byte* fontData, int length);
		void Generate(msdfgen::FreetypeHandle* ft, msdfgen::FontHandle* font, const msdf_atlas::Charset& chset,
		              const std::optional<std::string>& pngOutput);
		void SavePng(const msdfgen::BitmapConstRef<msdfgen::byte, 1>& storage, const std::string& output,
		             int channels) const;

	private:
		SdfGenerator m_generator;
		Texture m_atlasTex;
		AtlasMetadata m_meta;
		Config m_config;
		std::map<uint32_t, GlyphInfo> m_symbols;
	};
}
#endif