#include "STDInclude.hpp"

namespace Assets
{
    void IMenuList::load(Game::XAssetHeader* header, const std::string& name, Components::ZoneBuilder::Zone* builder)
    {
        Utils::Memory::Allocator* allocator = builder->getAllocator();

        // actually gets the whole list
        std::vector<Game::menuDef_t*> menus = Components::Menus::LoadMenu(name);
        if (menus.empty()) return;

        // Allocate new menu list
        Game::MenuList* newList = allocator->allocate<Game::MenuList>();
        if (!newList) return;

        newList->menus = allocator->allocateArray<Game::menuDef_t*>(menus.size());
        if (!newList->menus)
        {
            allocator->free(newList);
            return;
        }

        newList->name = allocator->duplicateString(name);
        newList->menuCount = menus.size();

        // Copy new menus
        for (unsigned int i = 0; i < menus.size(); ++i)
        {
            newList->menus[i] = menus[i];
        }

        header->menuList = newList;
    }
	void IMenuList::mark(Game::XAssetHeader header, Components::ZoneBuilder::Zone* builder)
	{
		Game::MenuList *asset = header.menuList;

		for (int i = 0; i < asset->menuCount; ++i)
		{
			if (asset->menus[i])
			{
				builder->loadAsset(Game::XAssetType::ASSET_TYPE_MENU, asset->menus[i]);
			}
		}
	}
	void IMenuList::save(Game::XAssetHeader header, Components::ZoneBuilder::Zone* builder)
	{
		AssertSize(Game::MenuList, 12);

		Utils::Stream* buffer = builder->getBuffer();
		Game::MenuList* asset = header.menuList;
		Game::MenuList* dest = buffer->dest<Game::MenuList>();

		buffer->save(asset);

		buffer->pushBlock(Game::XFILE_BLOCK_VIRTUAL);

		if (asset->name)
		{
			buffer->saveString(builder->getAssetName(this->getType(), asset->name));
			Utils::Stream::ClearPointer(&dest->name);
		}

		if (asset->menus)
		{
			buffer->align(Utils::Stream::ALIGN_4);

			Game::menuDef_t **destMenus = buffer->dest<Game::menuDef_t*>();
			buffer->saveArray(asset->menus, asset->menuCount);

			for (int i = 0; i < asset->menuCount; ++i)
			{
				destMenus[i] = builder->saveSubAsset(Game::XAssetType::ASSET_TYPE_MENU, asset->menus[i]).menu;
			}
		}

		buffer->popBlock();
	}
}