#include "SetObjectList.h"

#include "Globals.h"
#include "Utils/BitConverter.h"
#include "Utils/StringHelper.h"
#include "ZFile.h"
#include "ZRoom/ZNames.h"
#include "ZRoom/ZRoom.h"

SetObjectList::SetObjectList(ZFile* nParent) : ZRoomCommand(nParent)
{
}

void SetObjectList::ParseRawData()
{
	ZRoomCommand::ParseRawData();
	uint8_t objectCnt = parent->GetRawData().at(rawDataIndex + 1);
	uint32_t currentPtr = segmentOffset;

	objects.reserve(objectCnt);
	for (uint8_t i = 0; i < objectCnt; i++)
	{
		uint16_t objectIndex = BitConverter::ToInt16BE(parent->GetRawData(), currentPtr);
		objects.push_back(objectIndex);
		currentPtr += 2;
	}
}

void SetObjectList::DeclareReferences(const std::string& prefix)
{
	if (!objects.empty())
	{
		std::string declaration;

		for (size_t i = 0; i < objects.size(); i++)
		{
			uint16_t objectIndex = objects[i];
			declaration +=
				StringHelper::Sprintf("    %s,", ZNames::GetObjectName(objectIndex).c_str());

			if (i < objects.size() - 1)
				declaration += "\n";
		}

		parent->AddDeclarationArray(
			segmentOffset, DeclarationAlignment::Align4, objects.size() * 2, "s16",
			StringHelper::Sprintf("%sObjectList_%06X", prefix.c_str(), segmentOffset),
			objects.size(), declaration);
	}
}

std::string SetObjectList::GetBodySourceCode() const
{
	std::string listName;
	Globals::Instance->GetSegmentedPtrName(cmdArg2, parent, "s16", listName, parent->workerID);
	return StringHelper::Sprintf("SCENE_CMD_OBJECT_LIST(%i, %s)", objects.size(), listName.c_str());
}

std::string SetObjectList::GetCommandCName() const
{
	return "SCmdObjectList";
}

RoomCommand SetObjectList::GetRoomCommand() const
{
	return RoomCommand::SetObjectList;
}