173 lines
5.2 KiB
C++
173 lines
5.2 KiB
C++
#ifndef __PLASP__PDDL__TRANSLATION__EFFECT_H
|
|
#define __PLASP__PDDL__TRANSLATION__EFFECT_H
|
|
|
|
#include <colorlog/Formatting.h>
|
|
|
|
#include <pddl/NormalizedAST.h>
|
|
|
|
#include <plasp/TranslatorException.h>
|
|
|
|
#include <plasp/pddl/translation/ConditionalEffect.h>
|
|
#include <plasp/pddl/translation/Precondition.h>
|
|
#include <plasp/pddl/translation/Predicate.h>
|
|
#include <plasp/pddl/translation/Primitives.h>
|
|
#include <plasp/pddl/translation/Variables.h>
|
|
#include <plasp/pddl/translation/VariableStack.h>
|
|
|
|
namespace plasp
|
|
{
|
|
namespace pddl
|
|
{
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Effect
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
template<typename PrintObjectName>
|
|
inline void translateEffect(colorlog::ColorStream &outputStream,
|
|
const ::pddl::normalizedAST::Effect &effect, PrintObjectName printObjectName,
|
|
VariableStack &variableStack, size_t &numberOfConditionalEffects, VariableIDMap &variableIDs)
|
|
{
|
|
const auto handlePredicate =
|
|
[&](const ::pddl::normalizedAST::PredicatePointer &predicate, bool isPositive = true)
|
|
{
|
|
outputStream << std::endl << colorlog::Function("postcondition") << "(";
|
|
printObjectName();
|
|
outputStream
|
|
<< ", " << colorlog::Keyword("effect") << "("
|
|
<< colorlog::Reserved("unconditional") << ")"
|
|
<< ", ";
|
|
translatePredicateToVariable(outputStream, *predicate, variableIDs, isPositive);
|
|
outputStream << ") :- " << colorlog::Function("action") << "(";
|
|
printObjectName();
|
|
outputStream << ")";
|
|
|
|
if (!variableStack.layers.empty())
|
|
for (const auto &layer : variableStack.layers)
|
|
{
|
|
if (!layer->empty())
|
|
outputStream << ", ";
|
|
|
|
translateVariablesForRuleBody(outputStream, *layer, variableIDs);
|
|
}
|
|
|
|
outputStream << ".";
|
|
};
|
|
|
|
const auto handleNegatedPredicate =
|
|
[&](const ::pddl::normalizedAST::PredicatePointer &predicate)
|
|
{
|
|
handlePredicate(predicate, false);
|
|
};
|
|
|
|
const auto handleDerivedPredicate =
|
|
[&](const ::pddl::normalizedAST::DerivedPredicatePointer &, bool = true)
|
|
{
|
|
// TODO: refactor as to exclude derived predicates from the Effect variant
|
|
throw TranslatorException("derived predicates should not appear in effects, please report to the bug tracker");
|
|
};
|
|
|
|
const auto handleNegatedDerivedPredicate =
|
|
[&](const ::pddl::normalizedAST::DerivedPredicatePointer &derivedPredicate)
|
|
{
|
|
handleDerivedPredicate(derivedPredicate, false);
|
|
};
|
|
|
|
const auto handleAtomicFormula =
|
|
[&](const ::pddl::normalizedAST::AtomicFormula &atomicFormula)
|
|
{
|
|
atomicFormula.match(handlePredicate, handleDerivedPredicate);
|
|
};
|
|
|
|
const auto handleNot =
|
|
[&](const ::pddl::normalizedAST::NotPointer<::pddl::normalizedAST::AtomicFormula> ¬_)
|
|
{
|
|
not_->argument.match(handleNegatedPredicate, handleNegatedDerivedPredicate);
|
|
};
|
|
|
|
const auto handleLiteral =
|
|
[&](const ::pddl::normalizedAST::Literal &literal)
|
|
{
|
|
literal.match(handleAtomicFormula, handleNot);
|
|
};
|
|
|
|
const auto handleAnd =
|
|
[&](const ::pddl::normalizedAST::AndPointer<::pddl::normalizedAST::Effect> &and_)
|
|
{
|
|
for (const auto &argument : and_->arguments)
|
|
translateEffect(outputStream, argument, printObjectName, variableStack,
|
|
numberOfConditionalEffects, variableIDs);
|
|
};
|
|
|
|
const auto handleForAll =
|
|
[&](const ::pddl::normalizedAST::ForAllPointer<::pddl::normalizedAST::Effect> &forAll)
|
|
{
|
|
variableStack.push(&forAll->parameters);
|
|
|
|
translateEffect(outputStream, forAll->argument, printObjectName,
|
|
variableStack, numberOfConditionalEffects, variableIDs);
|
|
|
|
variableStack.pop();
|
|
};
|
|
|
|
const auto handleWhen =
|
|
[&](const ::pddl::normalizedAST::WhenPointer<::pddl::normalizedAST::Precondition,
|
|
::pddl::normalizedAST::ConditionalEffect> &when)
|
|
{
|
|
numberOfConditionalEffects++;
|
|
|
|
const auto printConditionalEffectIdentifier =
|
|
[&]()
|
|
{
|
|
outputStream << colorlog::Keyword("effect") << "("
|
|
<< colorlog::Number<size_t>(numberOfConditionalEffects) << ")";
|
|
};
|
|
|
|
const auto printPreconditionRuleBody =
|
|
[&]()
|
|
{
|
|
outputStream << " :- " << colorlog::Function("action") << "(";
|
|
printObjectName();
|
|
outputStream << ")";
|
|
|
|
for (const auto &layer : variableStack.layers)
|
|
{
|
|
if (!layer->empty())
|
|
outputStream << ", ";
|
|
|
|
translateVariablesForRuleBody(outputStream, *layer, variableIDs);
|
|
}
|
|
};
|
|
|
|
translatePrecondition(outputStream, when->argumentLeft,
|
|
printConditionalEffectIdentifier, printPreconditionRuleBody, variableIDs);
|
|
|
|
translateConditionalEffect(outputStream, when->argumentRight, printObjectName,
|
|
variableStack, numberOfConditionalEffects, variableIDs);
|
|
};
|
|
|
|
effect.match(handleAnd, handleForAll, handleLiteral, handleWhen);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
template<typename PrintObjectName>
|
|
inline void translateEffect(colorlog::ColorStream &outputStream,
|
|
const ::pddl::normalizedAST::Effect &effect, PrintObjectName printObjectName,
|
|
size_t &numberOfConditionalEffects, VariableIDMap &variableIDs)
|
|
{
|
|
VariableStack variableStack;
|
|
|
|
translateEffect(outputStream, effect, printObjectName, variableStack,
|
|
numberOfConditionalEffects, variableIDs);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|