Continued working on reimplementing action parser.

lib/pddlparse/src/pddlparse/detail/parsing/Action.cpp

@@ -1,6 +1,9 @@
 #include <pddlparse/detail/parsing/Action.h>
 
 #include <pddlparse/AST.h>
+#include <pddlparse/detail/ASTContext.h>
+#include <pddlparse/detail/VariableStack.h>
+#include <pddlparse/detail/parsing/Precondition.h>
 // TODO: remove
 #include <pddlparse/detail/parsing/Utils.h>
 #include <pddlparse/detail/parsing/VariableDeclaration.h>
@@ -16,50 +19,180 @@ namespace detail
 //
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-void parseAndAddAction(Context &context, ast::Domain &domain)
+ActionParser::ActionParser(Context &context, ast::Domain &domain)
+:	m_context{context},
+	m_domain{domain},
+	m_parametersPosition{-1},
+	m_preconditionPosition{-1},
+	m_effectPosition{-1},
+	m_varsPosition{-1}
 {
-	auto &tokenizer = context.tokenizer;
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+ast::ActionPointer ActionParser::parse()
+{
+	auto action = std::make_unique<ast::Action>();
+
+	findSections(*action);
+
+	auto &tokenizer = m_context.tokenizer;
+
+	if (m_parametersPosition != -1)
+	{
+		tokenizer.seek(m_parametersPosition);
+		parseParameterSection(*action);
+	}
+
+	// For compatibility with old PDDL versions, vars sections are parsed in addition to parameters
+	if (m_varsPosition != -1)
+	{
+		tokenizer.seek(m_varsPosition);
+		parseVarsSection(*action);
+	}
+
+	if (m_preconditionPosition != -1)
+	{
+		tokenizer.seek(m_preconditionPosition);
+		parsePreconditionSection(*action);
+	}
+
+	if (m_effectPosition != -1)
+	{
+		tokenizer.seek(m_effectPosition);
+		parseEffectSection(*action);
+	}
+
+	return action;
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void ActionParser::findSections(ast::Action &action)
+{
+	auto &tokenizer = m_context.tokenizer;
 
 	tokenizer.expect<std::string>("(");
 	tokenizer.expect<std::string>(":");
 	tokenizer.expect<std::string>("action");
 
-	auto action = std::make_unique<ast::Action>();
-	action->name = tokenizer.getIdentifier();
+	action.name = tokenizer.getIdentifier();
+
+	// TODO: rename parameters appropriately
+	const auto setSectionPosition =
+		[&](const std::string &sectionName, auto &sectionPosition, const auto value, bool unique = false)
+		{
+			if (unique && sectionPosition != -1)
+			{
+				tokenizer.seek(value);
+				throw tokenize::TokenizerException(tokenizer.location(), "only one “:" + sectionName + "” section allowed");
+			}
+
+			sectionPosition = value;
+		};
+
+	tokenizer.skipWhiteSpace();
+
+	while (tokenizer.currentCharacter() != ')')
+	{
+		const auto position = tokenizer.position();
+
+		tokenizer.expect<std::string>(":");
+
+		if (tokenizer.testIdentifierAndSkip("parameters"))
+			setSectionPosition("parameters", m_parametersPosition, position, true);
+		else if (tokenizer.testIdentifierAndSkip("precondition"))
+			setSectionPosition("precondition", m_preconditionPosition, position, true);
+		else if (tokenizer.testIdentifierAndSkip("effect"))
+			setSectionPosition("effect", m_effectPosition, position, true);
+		else if (tokenizer.testIdentifierAndSkip("vars"))
+			setSectionPosition("vars", m_varsPosition, position, true);
+		else
+		{
+			const auto sectionIdentifier = tokenizer.getIdentifier();
+
+			tokenizer.seek(position);
+			throw tokenize::TokenizerException(tokenizer.location(), "unknown action section “" + sectionIdentifier + "”");
+		}
+
+		tokenizer.expect<std::string>("(");
+
+		// Skip section for now and parse it later
+		skipSection(tokenizer);
+
+		tokenizer.skipWhiteSpace();
+	}
+
+	tokenizer.expect<std::string>(")");
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void ActionParser::parseParameterSection(ast::Action &action)
+{
+	auto &tokenizer = m_context.tokenizer;
 
 	tokenizer.expect<std::string>(":parameters");
 	tokenizer.expect<std::string>("(");
 
-	// Read parameters
-	action->parameters = parseVariableDeclarations(context, domain);
+	parseAndAddVariableDeclarations(m_context, m_domain, action.parameters);
 
 	tokenizer.expect<std::string>(")");
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void ActionParser::parsePreconditionSection(ast::Action &action)
+{
+	auto &tokenizer = m_context.tokenizer;
+
+	tokenizer.expect<std::string>(":precondition");
+
+	VariableStack variableStack;
+	variableStack.push(&action.parameters);
+
+	ASTContext astContext(m_domain);
+
+	action.precondition = parsePrecondition(m_context, astContext, variableStack);
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void ActionParser::parseEffectSection(ast::Action &)
+{
+	auto &tokenizer = m_context.tokenizer;
+
+	tokenizer.expect<std::string>(":effect");
+	tokenizer.expect<std::string>("(");
 
 	// TODO: reimplement
-	skipSection(tokenizer);
+	//VariableStack variableStack;
+	//variableStack.push(&action.parameters);
 
-	/*
-	// Parse preconditions and effects
-	while (!tokenizer.testAndReturn(')'))
-	{
-		tokenizer.expect<std::string>(":");
+	//ASTContext astContext(m_domain);
 
-		if (tokenizer.testIdentifierAndSkip("precondition"))
-			// TODO: reimplement
-			//action->precondition = parsePreconditionExpression(context, expressionContext);
-			skipSection(tokenizer);
-		else if (tokenizer.testIdentifierAndSkip("effect"))
-			// TODO: reimplement
-			//action->effect = parseEffectExpression(context, expressionContext);
-			skipSection(tokenizer);
-
-		tokenizer.skipWhiteSpace();
-	}*/
-
-	// Store new action
-	domain.actions.emplace_back(std::move(action));
+	//action.precondition = parseEffect(m_context, astContext, variableStack);
 
 	//tokenizer.expect<std::string>(")");
+
+	skipSection(tokenizer);
+}
+
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void ActionParser::parseVarsSection(ast::Action &action)
+{
+	auto &tokenizer = m_context.tokenizer;
+
+	tokenizer.expect<std::string>(":vars");
+	tokenizer.expect<std::string>("(");
+
+	m_context.warningCallback(tokenizer.location(), "“vars” section is not part of the PDDL 3.1 specification, treating it like additional “parameters” section");
+
+	parseAndAddVariableDeclarations(m_context, m_domain, action.parameters);
+
+	tokenizer.expect<std::string>(")");
 }
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////

lib/pddlparse/src/pddlparse/detail/parsing/Domain.cpp

@@ -294,7 +294,7 @@ void DomainParser::parsePredicateSection(ast::Domain &domain)
 
 void DomainParser::parseActionSection(ast::Domain &domain)
 {
-	parseAndAddAction(m_context, domain);
+	domain.actions.emplace_back(ActionParser(m_context, domain).parse());
 }
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////

lib/pddlparse/src/pddlparse/detail/parsing/VariableDeclaration.cpp

@@ -29,10 +29,8 @@ void parseAndAddUntypedVariableDeclaration(Context &context, ast::VariableDeclar
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-ast::VariableDeclarations parseVariableDeclarations(Context &context, ast::Domain &domain)
+void parseAndAddVariableDeclarations(Context &context, ast::Domain &domain, ast::VariableDeclarations &variableDeclarations)
 {
-	ast::VariableDeclarations variableDeclarations;
-
 	auto &tokenizer = context.tokenizer;
 	tokenizer.skipWhiteSpace();
 
@@ -58,8 +56,17 @@ ast::VariableDeclarations parseVariableDeclarations(Context &context, ast::Domai
 
 		tokenizer.skipWhiteSpace();
 	}
+}
 
-	return variableDeclarations;
+////////////////////////////////////////////////////////////////////////////////////////////////////
+
+ast::VariableDeclarations parseVariableDeclarations(Context &context, ast::Domain &domain)
+{
+	ast::VariableDeclarations result;
+
+	parseAndAddVariableDeclarations(context, domain, result);
+
+	return result;
 }
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////