/*
    Copyright (C) 2011 Adam Wardyński, Tomasz Śniatowski, Paweł Kędzia,
    Adam Radziszewski, Bartosz Broda
    Part of the WCCL project

    This program is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published by the Free
Software Foundation; either version 3 of the License, or (at your option)
any later version.

    This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. 

    See the LICENSE, COPYING.LESSER and COPYING files for more details.
*/

#include <libwccl/lexicon/lexicon.h>
#include <libpwrutils/util.h>
#include <libwccl/exception.h>
#include <boost/foreach.hpp>

#include <boost/make_shared.hpp>
#include <unicode/unistr.h>

namespace Wccl {

const UnicodeString& Lexicon::translate(const UnicodeString &key) const
{
	static UnicodeString empty;
	map_t::const_iterator i = map_.find(key);
	if (i == map_.end()) {
		return empty;
	}
	return i->second;
}

boost::shared_ptr<StrSet> Lexicon::translate(const StrSet& set) const
{
	boost::shared_ptr<StrSet> ret_set = boost::make_shared<StrSet>();
	BOOST_FOREACH (const UnicodeString& s, set.get_value()) {
		const UnicodeString& v = translate(s);
		if (!v.isEmpty()) {
			ret_set->insert(v);
		}
	}
	return ret_set;
}

void Lexicon::insert(const UnicodeString& key, const UnicodeString& value)
{
	if (has_key(key)) {
		throw InvalidArgument("key", "Duplicated key in lexicon: " + PwrNlp::to_utf8(key));
	}
	if (key.isEmpty()) {
		throw InvalidArgument("key", "Empty key string in lexicon.");
	}
	if (value.isEmpty()) {
		throw InvalidArgument("value", "Empty value string in lexicon.");
	}
	map_[key] = value;
}

} /* end ns Wccl */