pddl_parser/domain/
constant.rs

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

pddl_parser/domain/
constant.rs
