pddl_parser/domain/
typed_parameter.rs

1use nom::combinator::opt;
2use nom::multi::{many0, many1};
3use nom::sequence::{pair, preceded};
4use nom::IResult;
5use serde::{Deserialize, Serialize};
6
7use super::typing::Type;
8use crate::error::ParserError;
9use crate::lexer::{Token, TokenStream};
10use crate::tokens::var;
11
12/// A parameter with a type.
13#[derive(Debug, Deserialize, Serialize, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
14pub struct TypedParameter {
15    /// The name of the parameter.
16    pub name: String,
17    /// The type of the parameter. If not specified, the type is `object`.
18    #[serde(rename = "type")]
19    #[serde(default)]
20    pub type_: Type,
21}
22
23impl TypedParameter {
24    /// Parse a list of typed parameters from a token stream.
25    pub fn parse_typed_parameters(input: TokenStream) -> IResult<TokenStream, Vec<TypedParameter>, ParserError> {
26        log::debug!("BEGIN > parse_typed_parameters {:?}", input.span());
27        let (output, params) = many0(pair(many1(var), opt(preceded(Token::Dash, Type::parse_type))))(input)?;
28        let params = params
29            .into_iter()
30            .flat_map(|(names, type_)| {
31                names.into_iter().map(move |name| TypedParameter {
32                    name,
33                    type_: type_.clone().unwrap_or_default(),
34                })
35            })
36            .collect();
37        log::debug!("Parsed typed parameters: {params:?}");
38        log::debug!("END < parse_typed_parameters {:?}", output.span());
39        Ok((output, params))
40    }
41
42    /// Convert the typed parameter to PDDL.
43    pub fn to_pddl(&self) -> String {
44        format!("{} - {}", self.name, self.type_.to_pddl())
45    }
46}
pddl_parser/domain/typed_parameter.rs

pddl_parser/domain/
typed_parameter.rs
