#include "toonz/namebuilder.h"
#include "tconvert.h"

//-------------------------------------------------------------------

NameBuilder *NameBuilder::getBuilder(std::wstring levelName)
{
	if (levelName == L"")
		return new NameCreator();
	else
		return new NameModifier(levelName);
}

//-------------------------------------------------------------------

std::wstring NameCreator::getNext()
{
	if (m_s.empty()) {
		m_s.push_back(0);
		return L"A";
	}
	int i = 0;
	int n = m_s.size();
	while (i < n) {
		m_s[i]++;
		if (m_s[i] <= 'Z' - 'A')
			break;
		m_s[i] = 0;
		i++;
	}
	if (i >= n) {
		n++;
		m_s.push_back(0);
	}
	std::wstring s;
	for (i = n - 1; i >= 0; i--)
		s.append(1, (wchar_t)(L'A' + m_s[i]));
	return s;
}

//-------------------------------------------------------------------

NameModifier::NameModifier(std::wstring name)
	: m_nameBase(name), m_index(0)
{
	int j = name.find_last_not_of(L"0123456789");
	if (j != (int)std::wstring::npos &&
		j + 1 < (int)name.length() &&
		name[j] == '_') {
		m_index = std::stoi(name.substr(j + 1));
		m_nameBase = name.substr(0, j);
	}
}

//-------------------------------------------------------------------

std::wstring NameModifier::getNext()
{
	int index = m_index++;
	if (index < 1)
		return m_nameBase;
	else
		return m_nameBase + L"_" + std::to_wstring(index);
}