2020-02-01 17:20:43 +01:00
|
|
|
|
mod head_type;
|
2020-01-31 17:24:13 +01:00
|
|
|
|
mod translate_body;
|
2020-01-31 17:19:44 +01:00
|
|
|
|
|
2020-02-01 17:20:43 +01:00
|
|
|
|
use head_type::*;
|
|
|
|
|
use translate_body::*;
|
2020-02-04 16:53:52 +01:00
|
|
|
|
use crate::traits::AssignVariableDeclarationDomain as _;
|
2020-01-25 12:55:23 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
struct PredicateDefinitions
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
pub parameters: std::rc::Rc<foliage::VariableDeclarations>,
|
2020-05-11 03:58:30 +02:00
|
|
|
|
pub definitions: Vec<foliage::OpenFormula>,
|
2020-02-02 02:53:04 +01:00
|
|
|
|
}
|
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
type Definitions =
|
|
|
|
|
std::collections::BTreeMap::<std::rc::Rc<foliage::PredicateDeclaration>, PredicateDefinitions>;
|
|
|
|
|
|
|
|
|
|
pub(crate) struct Translator<'p>
|
|
|
|
|
{
|
|
|
|
|
problem: &'p mut crate::Problem,// TODO: refactor
|
|
|
|
|
definitions: Definitions,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct Logger;
|
|
|
|
|
|
|
|
|
|
impl clingo::Logger for Logger
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
fn log(&mut self, code: clingo::Warning, message: &str)
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
log::warn!("clingo warning ({:?}): {}", code, message);
|
2020-02-02 02:53:04 +01:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
impl<'p> clingo::StatementHandler for Translator<'p>
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
|
|
|
|
fn on_statement(&mut self, statement: &clingo::ast::Statement) -> bool
|
|
|
|
|
{
|
|
|
|
|
match statement.statement_type()
|
|
|
|
|
{
|
|
|
|
|
clingo::ast::StatementType::Rule(ref rule) =>
|
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
if let Err(error) = self.read_rule(rule)
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
|
|
|
|
log::error!("could not translate input program: {}", error);
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
},
|
|
|
|
|
_ => log::debug!("read statement (other kind)"),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
true
|
|
|
|
|
}
|
|
|
|
|
}
|
2020-02-01 21:59:36 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
impl<'p> Translator<'p>
|
2020-02-02 02:53:04 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
pub fn new(problem: &'p mut crate::Problem) -> Self
|
2020-02-05 19:40:21 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
Self
|
|
|
|
|
{
|
|
|
|
|
problem,
|
|
|
|
|
definitions: Definitions::new(),
|
|
|
|
|
}
|
2020-02-05 19:40:21 +01:00
|
|
|
|
}
|
2020-02-05 01:10:33 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
pub fn translate<P>(&mut self, program_path: P) -> Result<(), crate::Error>
|
|
|
|
|
where
|
|
|
|
|
P: AsRef<std::path::Path>
|
2020-02-02 19:20:16 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// Read input program
|
2020-02-02 19:20:16 +01:00
|
|
|
|
let program = std::fs::read_to_string(program_path.as_ref())
|
2020-05-05 19:40:57 +02:00
|
|
|
|
.map_err(
|
|
|
|
|
|error| crate::Error::new_read_file(program_path.as_ref().to_path_buf(), error))?;
|
2020-02-02 19:20:16 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
clingo::parse_program_with_logger(&program, self, &mut Logger, std::u32::MAX)
|
2020-02-02 19:20:16 +01:00
|
|
|
|
.map_err(|error| crate::Error::new_translate(error))?;
|
2020-02-05 01:10:33 +01:00
|
|
|
|
|
|
|
|
|
log::info!("read input program “{}”", program_path.as_ref().display());
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let completed_definition = |predicate_declaration, definitions: &mut Definitions,
|
|
|
|
|
problem: &crate::Problem|
|
2020-02-02 17:57:27 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
match definitions.remove(predicate_declaration)
|
2020-02-02 17:57:27 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// This predicate symbol has at least one definition, so build the disjunction of those
|
|
|
|
|
Some(predicate_definitions) =>
|
2020-02-02 17:57:27 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let or_arguments = predicate_definitions.definitions.into_iter()
|
|
|
|
|
.map(|x| crate::existential_closure(x))
|
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
|
let or = foliage::Formula::or(or_arguments);
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let head_arguments = predicate_definitions.parameters.iter()
|
|
|
|
|
.map(|x| foliage::Term::variable(std::rc::Rc::clone(x)))
|
|
|
|
|
.collect::<Vec<_>>();
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let head_predicate = foliage::Formula::predicate(
|
|
|
|
|
std::rc::Rc::clone(predicate_declaration), head_arguments);
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let completed_definition =
|
|
|
|
|
foliage::Formula::if_and_only_if(vec![head_predicate, or]);
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
let open_formula = foliage::OpenFormula
|
2020-05-05 19:40:57 +02:00
|
|
|
|
{
|
|
|
|
|
free_variable_declarations: predicate_definitions.parameters,
|
|
|
|
|
formula: completed_definition,
|
|
|
|
|
};
|
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
crate::universal_closure(open_formula)
|
2020-05-05 19:40:57 +02:00
|
|
|
|
},
|
|
|
|
|
// This predicate has no definitions, so universally falsify it
|
|
|
|
|
None =>
|
2020-02-02 17:57:27 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let parameters = std::rc::Rc::new((0..predicate_declaration.arity)
|
|
|
|
|
.map(|_|
|
|
|
|
|
{
|
|
|
|
|
let variable_declaration = std::rc::Rc::new(
|
|
|
|
|
foliage::VariableDeclaration::new("<anonymous>".to_string()));
|
|
|
|
|
problem.assign_variable_declaration_domain(&variable_declaration,
|
|
|
|
|
crate::Domain::Program);
|
|
|
|
|
variable_declaration
|
|
|
|
|
})
|
|
|
|
|
.collect::<Vec<_>>());
|
|
|
|
|
|
|
|
|
|
let head_arguments = parameters.iter()
|
|
|
|
|
.map(|x| foliage::Term::variable(std::rc::Rc::clone(x)))
|
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
|
|
let head_predicate = foliage::Formula::predicate(
|
|
|
|
|
std::rc::Rc::clone(predicate_declaration), head_arguments);
|
|
|
|
|
|
|
|
|
|
let not = foliage::Formula::not(Box::new(head_predicate));
|
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
let open_formula = foliage::OpenFormula
|
2020-05-05 19:40:57 +02:00
|
|
|
|
{
|
|
|
|
|
free_variable_declarations: parameters,
|
|
|
|
|
formula: not,
|
|
|
|
|
};
|
2020-02-05 20:19:22 +01:00
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
crate::universal_closure(open_formula)
|
2020-05-05 19:40:57 +02:00
|
|
|
|
},
|
2020-02-05 20:19:22 +01:00
|
|
|
|
}
|
2020-05-05 19:40:57 +02:00
|
|
|
|
};
|
2020-02-05 20:19:22 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
for predicate_declaration in self.problem.predicate_declarations.borrow().iter()
|
2020-02-05 01:10:33 +01:00
|
|
|
|
{
|
2020-05-07 17:19:42 +02:00
|
|
|
|
// Don’t perform completion for input predicates
|
|
|
|
|
if self.problem.input_predicate_declarations.borrow().contains(predicate_declaration)
|
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let completed_definition =
|
|
|
|
|
completed_definition(predicate_declaration, &mut self.definitions, self.problem);
|
|
|
|
|
|
|
|
|
|
let statement = crate::problem::Statement::new(
|
2020-05-07 02:53:48 +02:00
|
|
|
|
crate::problem::StatementKind::Program, completed_definition)
|
2020-05-05 19:40:57 +02:00
|
|
|
|
.with_name(format!("completed_definition_{}_{}", predicate_declaration.name,
|
|
|
|
|
predicate_declaration.arity))
|
|
|
|
|
.with_description(format!("completed definition of {}/{}",
|
|
|
|
|
predicate_declaration.name, predicate_declaration.arity));
|
|
|
|
|
|
|
|
|
|
self.problem.add_statement(crate::problem::SectionKind::CompletedDefinitions,
|
|
|
|
|
statement);
|
2020-02-05 01:10:33 +01:00
|
|
|
|
}
|
2020-02-04 00:27:04 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
Ok(())
|
2020-02-05 01:10:33 +01:00
|
|
|
|
}
|
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
fn read_rule(&mut self, rule: &clingo::ast::Rule)
|
|
|
|
|
-> Result<(), crate::Error>
|
2020-02-05 01:10:33 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let head_type = determine_head_type(rule.head(), self.problem)?;
|
2020-02-05 01:10:33 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
match &head_type
|
2020-02-05 01:10:33 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
HeadType::SingleAtom(head_atom)
|
|
|
|
|
| HeadType::ChoiceWithSingleAtom(head_atom) =>
|
|
|
|
|
{
|
|
|
|
|
if !self.definitions.contains_key(&head_atom.predicate_declaration)
|
|
|
|
|
{
|
|
|
|
|
let parameters = std::rc::Rc::new((0..head_atom.predicate_declaration.arity)
|
|
|
|
|
.map(|_|
|
|
|
|
|
{
|
|
|
|
|
let variable_declaration = std::rc::Rc::new(
|
|
|
|
|
foliage::VariableDeclaration::new("<anonymous>".to_string()));
|
|
|
|
|
self.problem.assign_variable_declaration_domain(&variable_declaration,
|
|
|
|
|
crate::Domain::Program);
|
|
|
|
|
variable_declaration
|
|
|
|
|
})
|
|
|
|
|
.collect());
|
|
|
|
|
|
|
|
|
|
self.definitions.insert(
|
|
|
|
|
std::rc::Rc::clone(&head_atom.predicate_declaration),
|
|
|
|
|
PredicateDefinitions
|
|
|
|
|
{
|
|
|
|
|
parameters,
|
|
|
|
|
definitions: vec![],
|
|
|
|
|
});
|
|
|
|
|
}
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// TODO: refactor
|
|
|
|
|
let predicate_definitions =
|
|
|
|
|
self.definitions.get_mut(&head_atom.predicate_declaration).unwrap();
|
2020-02-01 19:20:46 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let parameters = std::rc::Rc::clone(&predicate_definitions.parameters);
|
|
|
|
|
let free_variable_declarations = std::cell::RefCell::new(vec![]);
|
|
|
|
|
let free_layer =
|
|
|
|
|
foliage::VariableDeclarationStackLayer::Free(free_variable_declarations);
|
|
|
|
|
let parameters_layer =
|
|
|
|
|
foliage::VariableDeclarationStackLayer::bound(&free_layer, parameters);
|
2020-02-01 19:20:46 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let mut definition_arguments =
|
|
|
|
|
translate_body(rule.body(), self.problem, ¶meters_layer)?;
|
2020-02-01 21:59:36 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// TODO: refactor
|
|
|
|
|
assert_eq!(predicate_definitions.parameters.len(), head_atom.arguments.len());
|
2020-02-01 21:59:36 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
if let HeadType::ChoiceWithSingleAtom(_) = head_type
|
|
|
|
|
{
|
|
|
|
|
let head_arguments = predicate_definitions.parameters.iter()
|
|
|
|
|
.map(|x| foliage::Term::variable(std::rc::Rc::clone(x)))
|
|
|
|
|
.collect::<Vec<_>>();
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let head_predicate = foliage::Formula::predicate(
|
|
|
|
|
std::rc::Rc::clone(&head_atom.predicate_declaration), head_arguments);
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
definition_arguments.push(head_predicate);
|
|
|
|
|
}
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let mut head_atom_arguments_iterator = head_atom.arguments.iter();
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
for parameter in predicate_definitions.parameters.iter()
|
|
|
|
|
{
|
|
|
|
|
let head_atom_argument = head_atom_arguments_iterator.next().unwrap();
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let translated_head_term = crate::translate::common::choose_value_in_term(
|
|
|
|
|
head_atom_argument, std::rc::Rc::clone(parameter), self.problem,
|
|
|
|
|
¶meters_layer)?;
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
definition_arguments.push(translated_head_term);
|
|
|
|
|
}
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// TODO: refactor
|
|
|
|
|
let free_variable_declarations = match free_layer
|
|
|
|
|
{
|
|
|
|
|
foliage::VariableDeclarationStackLayer::Free(free_variable_declarations)
|
|
|
|
|
=> free_variable_declarations.into_inner(),
|
|
|
|
|
_ => unreachable!(),
|
|
|
|
|
};
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let definition = match definition_arguments.len()
|
|
|
|
|
{
|
|
|
|
|
1 => definition_arguments.pop().unwrap(),
|
|
|
|
|
0 => foliage::Formula::true_(),
|
|
|
|
|
_ => foliage::Formula::and(definition_arguments),
|
|
|
|
|
};
|
2020-02-01 21:59:36 +01:00
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
let definition = foliage::OpenFormula
|
2020-05-05 19:40:57 +02:00
|
|
|
|
{
|
|
|
|
|
free_variable_declarations: std::rc::Rc::new(free_variable_declarations),
|
|
|
|
|
formula: definition,
|
|
|
|
|
};
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
predicate_definitions.definitions.push(definition);
|
|
|
|
|
},
|
|
|
|
|
HeadType::IntegrityConstraint =>
|
2020-02-02 02:32:32 +01:00
|
|
|
|
{
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let free_variable_declarations = std::cell::RefCell::new(vec![]);
|
|
|
|
|
let free_layer =
|
|
|
|
|
foliage::VariableDeclarationStackLayer::Free(free_variable_declarations);
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let mut arguments = translate_body(rule.body(), self.problem, &free_layer)?;
|
2020-02-02 02:32:32 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
// TODO: refactor
|
|
|
|
|
let free_variable_declarations = match free_layer
|
|
|
|
|
{
|
|
|
|
|
foliage::VariableDeclarationStackLayer::Free(free_variable_declarations)
|
|
|
|
|
=> free_variable_declarations.into_inner(),
|
|
|
|
|
_ => unreachable!(),
|
|
|
|
|
};
|
2020-04-17 03:28:18 +02:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let formula = match arguments.len()
|
|
|
|
|
{
|
|
|
|
|
1 => foliage::Formula::not(Box::new(arguments.pop().unwrap())),
|
|
|
|
|
0 => foliage::Formula::false_(),
|
|
|
|
|
_ => foliage::Formula::not(Box::new(foliage::Formula::and(arguments))),
|
|
|
|
|
};
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
let open_formula = foliage::OpenFormula
|
2020-05-05 19:40:57 +02:00
|
|
|
|
{
|
|
|
|
|
free_variable_declarations: std::rc::Rc::new(free_variable_declarations),
|
|
|
|
|
formula,
|
|
|
|
|
};
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-11 03:58:30 +02:00
|
|
|
|
let integrity_constraint = crate::universal_closure(open_formula);
|
2020-02-02 17:57:27 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
let statement = crate::problem::Statement::new(
|
2020-05-07 02:53:48 +02:00
|
|
|
|
crate::problem::StatementKind::Program, integrity_constraint)
|
2020-05-05 19:40:57 +02:00
|
|
|
|
.with_name("integrity_constraint".to_string())
|
|
|
|
|
.with_description("integrity constraint".to_string());
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
self.problem.add_statement(crate::problem::SectionKind::IntegrityConstraints,
|
|
|
|
|
statement);
|
|
|
|
|
},
|
|
|
|
|
HeadType::Trivial => log::info!("skipping trivial rule"),
|
|
|
|
|
}
|
2020-02-02 03:56:34 +01:00
|
|
|
|
|
2020-05-05 19:40:57 +02:00
|
|
|
|
Ok(())
|
2020-02-01 15:32:41 +01:00
|
|
|
|
}
|
2020-01-24 13:32:43 +01:00
|
|
|
|
}
|