#include <plasp/pddl/expressions/PredicateDeclaration.h>

#include <plasp/pddl/Context.h>
#include <plasp/pddl/Domain.h>
#include <plasp/pddl/ExpressionContext.h>
#include <plasp/pddl/expressions/Constant.h>
#include <plasp/pddl/expressions/Variable.h>

namespace plasp
{
namespace pddl
{
namespace expressions
{

////////////////////////////////////////////////////////////////////////////////////////////////////
//
// PredicateDeclaration
//
////////////////////////////////////////////////////////////////////////////////////////////////////

PredicateDeclaration::PredicateDeclaration()
:	m_isDeclared{false}
{
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void PredicateDeclaration::parse(Context &context, Domain &domain)
{
	auto &tokenizer = context.tokenizer;

	tokenizer.expect<std::string>("(");

	auto predicate = PredicateDeclarationPointer(new PredicateDeclaration);

	predicate->m_name = tokenizer.getIdentifier();

	// Flag predicate as correctly declared in the types section
	predicate->setDeclared();

	tokenizer.skipWhiteSpace();

	ExpressionContext expressionContext(domain);
	expressionContext.variables.push(&predicate->m_parameters);

	// Parse parameters
	Variable::parseTypedDeclarations(context, expressionContext, predicate->m_parameters);

	tokenizer.expect<std::string>(")");

	domain.predicates().emplace_back(std::move(predicate));
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void PredicateDeclaration::setDeclared()
{
	m_isDeclared = true;
}

////////////////////////////////////////////////////////////////////////////////////////////////////

bool PredicateDeclaration::isDeclared() const
{
	return m_isDeclared;
}

////////////////////////////////////////////////////////////////////////////////////////////////////

const std::string &PredicateDeclaration::name() const
{
	return m_name;
}

////////////////////////////////////////////////////////////////////////////////////////////////////

const Variables &PredicateDeclaration::parameters() const
{
	return m_parameters;
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void PredicateDeclaration::normalizeParameterNames()
{
	for (size_t i = 0; i < m_parameters.size(); i++)
		m_parameters[i]->setName("X" + std::to_string(i));
}

////////////////////////////////////////////////////////////////////////////////////////////////////

void PredicateDeclaration::print(std::ostream &ostream) const
{
	// TODO: implement correctly
	ostream << "(<predicate declaration>)";
}

////////////////////////////////////////////////////////////////////////////////////////////////////

}
}
}