#include "../Platform/stdafx.h"

#include "WstringLookup.h"

WstringLookup::WstringLookup() { numIDs = 0; }

std::wstring WstringLookup::lookup(unsigned int id) {
    // TODO
    // if (id > currentMaxID)
    //	throw error

    return int2str.at(id);
}

unsigned int WstringLookup::lookup(std::wstring str) {
    if (str2int.find(str) == str2int.end()) {
        std::pair<std::wstring, unsigned int> p =
            std::pair<std::wstring, unsigned int>(str, numIDs);

        str2int.insert(p);
        int2str.push_back(str);

        return numIDs++;
    } else {
        return str2int.at(str);
    }
}

void WstringLookup::getTable(std::wstring** lookup, unsigned int* len) {
    // Outputs
    std::wstring* out_lookup;
    unsigned int out_len;

    // Fill lookup.
    out_lookup = new std::wstring[int2str.size()];
    for (unsigned int i = 0; i < numIDs; i++) out_lookup[i] = int2str.at(i);

    out_len = numIDs;

    // Return.
    *lookup = out_lookup;
    *len = out_len;
    return;
}