/* headerwriter.cpp - Copyright (C) 2015-2024 a dinosaur (zlib, see COPYING.txt) */

#include "headerwriter.hpp"
#include <algorithm>


template <typename T> static constexpr std::string_view DatType();
template <> constexpr std::string_view DatType<uint8_t>() { return "unsigned char"; }
template <> constexpr std::string_view DatType<uint16_t>() { return "unsigned short"; }
template <> constexpr std::string_view DatType<uint32_t>() { return "unsigned int"; }

void HeaderWriter::WriteSize(int width, int height)
{
	stream << std::endl;
	WriteDefine(mName + "Width", width);
	WriteDefine(mName + "Height", height);
}

void HeaderWriter::WriteCharacterMap(const std::span<uint16_t> charData)
{
	stream << std::endl;
	WriteDefine(mName + "TilesLen", charData.size() * 2);
	WriteSymbol(mName + "Tiles", DatType<uint16_t>(), charData.size());
}

void HeaderWriter::WriteCollision(const std::span<uint8_t> collisionData)
{
	stream << std::endl;
	WriteDefine(mName + "CollisionLen", collisionData.size());
	WriteSymbol(mName + "Collision", DatType<uint8_t>(), collisionData.size());
}

void HeaderWriter::WriteObjects(const std::span<uint32_t> objData)
{
	stream << std::endl;
	WriteDefine(mName + "ObjCount", objData.size() / 3);
	WriteDefine(mName + "ObjdatLen", objData.size() * sizeof(int));
	WriteSymbol(mName + "Objdat", DatType<uint32_t>(), objData.size());
}


static std::string GuardName(std::string label)
{
	std::transform(label.begin(), label.end(), label.begin(), ::toupper);
	return "TMX2GBA_" + label;
}


void HeaderWriter::WriteGuardStart()
{
	const std::string guard = GuardName(mName);
	stream << "#ifndef " << guard << std::endl;
	stream << "#define " << guard << std::endl;
}

void HeaderWriter::WriteGuardEnd()
{
	const std::string guard = GuardName(mName);
	stream << std::endl << "#endif//" << guard << std::endl;
}


HeaderWriter::~HeaderWriter()
{
	if (stream.is_open())
	{
		WriteGuardEnd();
		stream.close();
	}
}


bool HeaderWriter::Open(const std::filesystem::path& path, const std::string_view name)
{
	mName = name;
	stream.open(path);
	if (!stream.is_open())
		return false;

	WriteGuardStart();
	return true;
}

void HeaderWriter::WriteDefine(const std::string_view name, const std::string_view value)
{
	stream << "#define " << name << " " << value << std::endl;
}

void HeaderWriter::WriteSymbol(const std::string_view name, const std::string_view type, std::size_t count)
{
	stream << "extern const " << type << " " << name << "[" << count << "];" << std::endl;
}