/*
 * 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 <string>
#include <optional>
#include <map>
#include <variant>
#include "Scene/AtlasMetadata.hpp"
#include "FontAtlasGenerator.hpp"
#include "Scene/Texture.hpp"
#include "msdfgen.h"
#include "msdfgen-ext.h"
#include "msdf-atlas-gen/msdf-atlas-gen.h"
#define MSDFGEN_AVAILABLE

namespace OpenVulkano::Scene
{
	using namespace msdfgen;
	using namespace msdf_atlas;
	using namespace OpenVulkano::Scene;

	class MsdfFontAtlasGenerator : public FontAtlasGenerator
	{
	public:
		using SdfGenerator = ImmediateAtlasGenerator<float, 1, sdfGenerator, BitmapAtlasStorage<msdfgen::byte, 1>>;
		static 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 Charset& charset = Charset::ASCII,
		                   const std::optional<std::string>& pngOutput = std::nullopt);
		void GenerateAtlas(const msdfgen::byte* fontData, int length, const Charset& charset = Charset::ASCII,
		                   const std::optional<std::string>& pngOutput = std::nullopt);
		void SaveAtlasMetadataInfo(const std::string& outputFile, bool packIntoSingleFile = true) const override;
		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; }
	private:
		void InitFreetypeFromFile(FreetypeHandle*& ft, FontHandle*& font, const std::string& file);
		void InitFreetypeFromBuffer(FreetypeHandle*& ft, FontHandle*& font, const msdfgen::byte* fontData, int length);
		void Generate(FreetypeHandle* ft, FontHandle* font, const Charset& chset,
		              const std::optional<std::string>& pngOutput);
		void SavePng(const BitmapConstRef<byte, 1>& storage, const std::string& output, int channels) const;
	private:
		SdfGenerator m_generator;
		Texture m_atlasTex;
		AtlasMetadata m_meta;
		std::map<uint32_t, GlyphInfo> m_symbols;
	};
}
#endif