#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);

}