def has_solution(self, *query: Union[Atom, Not]): var_store = {} if len(query) == 1: query = query[0] predicate_name = query.get_predicate().get_name() query_args = swipy.swipy_new_term_refs(query.get_predicate().get_arity()) for ind, arg in enumerate(query.get_arguments()): _to_swipy_ref(arg, query_args + ind, var_store) pred = swipy.swipy_predicate(predicate_name, query.get_predicate().get_arity(), None) query = swipy.swipy_open_query(pred, query_args) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return True if r else False else: swipy_objs = [_lit_to_swipy(x, var_store) if isinstance(x, Atom) else _neg_to_swipy(x, var_store) for x in query] first = swipy_objs[0] rest = _conjoin_literals(swipy_objs[1:]) compound_arg = swipy.swipy_new_term_refs(2) swipy.swipy_put_term(compound_arg, first) swipy.swipy_put_term(compound_arg + 1, rest) predicate = swipy.swipy_predicate(",", 2, None) query = swipy.swipy_open_query(predicate, compound_arg) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return True if r else False
def query(self, *query, **kwargs): if 'max_solutions' in kwargs: max_solutions = kwargs['max_solutions'] else: max_solutions = -1 vars_of_interest = [[y for y in x.get_arguments() if isinstance(y, Variable)] for x in query] vars_of_interest = reduce(lambda x, y: x + y, vars_of_interest, []) vars_of_interest = reduce(lambda x, y: x + [y] if y not in x else x, vars_of_interest, []) var_store = {} for v in vars_of_interest: tmp_v = swipy.swipy_new_term_ref() swipy.swipy_put_variable(tmp_v) var_store[v] = tmp_v if len(query) == 1: query = query[0] predicate_name = query.get_predicate().get_name() query_args = swipy.swipy_new_term_refs(query.get_predicate().get_arity()) for ind, arg in enumerate(query.get_arguments()): _to_swipy_ref(arg, query_args + ind, var_store) pred = swipy.swipy_predicate(predicate_name, query.get_predicate().get_arity(), None) query = swipy.swipy_open_query(pred, query_args) else: swipy_objs = [_lit_to_swipy(x, var_store) if isinstance(x, Atom) else _neg_to_swipy(x, var_store) for x in query] first = swipy_objs[0] rest = _conjoin_literals(swipy_objs[1:]) compound_arg = swipy.swipy_new_term_refs(2) swipy.swipy_put_term(compound_arg, first) swipy.swipy_put_term(compound_arg + 1, rest) predicate = swipy.swipy_predicate(",", 2, None) query = swipy.swipy_open_query(predicate, compound_arg) r = swipy.swipy_next_solution(query) all_solutions = [] var_index_var = dict([(var_store[v], v) for v in var_store]) while r and max_solutions != 0: max_solutions -= 1 tmp_solution = {} for var in var_store: tmp_solution[var] = _read_swipy(var_store[var], swipy_term_to_var=var_index_var) all_solutions.append(tmp_solution) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return all_solutions
def use_module(self, module_name: str, **kwargs): if module_name.startswith('library'): # create library functor library_term = swipy.swipy_new_atom("library") library_funct = swipy.swipy_new_functor(library_term, 1) # create inner module name: library([name]) module_inner_name = module_name[:-1].replace('library(', '') inner_name = swipy.swipy_new_term_ref() swipy.swipy_put_atom_chars(inner_name, module_inner_name) # construct library(name) full_module = swipy.swipy_new_term_ref() swipy.swipy_cons_functor(full_module, library_funct, inner_name) else: full_module = swipy.swipy_new_term_ref() swipy.swipy_put_atom_chars(full_module, module_name) # load module use_module = swipy.swipy_predicate("use_module", 1, None) query = swipy.swipy_open_query(use_module, full_module) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) if r == 0: raise Exception(f"could not load module {module_name}") return r
def retract(self, clause: Union[Atom, Clause]): if isinstance(clause, Atom): lit = _lit_to_swipy(clause, {}) else: lit = _cl_to_swipy(clause, {}) retract = swipy.swipy_predicate("retract", 1, None) query = swipy.swipy_open_query(retract, lit) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return r
def asserta(self, clause): var_store = {} if isinstance(clause, Atom): swipl_object = _lit_to_swipy(clause, var_store) else: swipl_object = _cl_to_swipy(clause, var_store) asserta = swipy.swipy_predicate("asserta", 1, None) query = swipy.swipy_open_query(asserta, swipl_object) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return r
def consult(self, filename: str): string_term = swipy.swipy_new_term_ref() swipy.swipy_put_string_chars(string_term, filename) consult_pred = swipy.swipy_predicate("consult", 1, None) query = swipy.swipy_open_query(consult_pred, string_term) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) if r == 0: raise Exception(f"something wrong when consulting file {filename}") return r
def assertz(self, clause: Union[Atom, Clause]): var_store = {} if isinstance(clause, Atom): swipl_object = _lit_to_swipy(clause, var_store) elif isinstance(clause, Clause): swipl_object = _cl_to_swipy(clause, var_store) else: raise Exception(f"can only assertz atoms or clauses (got {clause})") asserta = swipy.swipy_predicate("assertz", 1, None) query = swipy.swipy_open_query(asserta, swipl_object) r = swipy.swipy_next_solution(query) swipy.swipy_close_query(query) return r
def _prepare_query(self, var_store, *query, max_time=None, max_depth=None, max_inference=None): """ Prepares the query Return swipl-transformed predicate and arguments needed to posed the query to the engine :param var_store: :param query: :param max_time: :param max_depth: :param max_inference: :return: """ if len(query) == 1: query = query[0] if not (max_time or max_depth or max_inference): predicate_name = query.get_predicate().get_name() query_args = swipy.swipy_new_term_refs(query.get_predicate().get_arity()) for ind, arg in enumerate(query.get_arguments()): _to_swipy_ref(arg, query_args + ind, var_store) pred = swipy.swipy_predicate(predicate_name, query.get_predicate().get_arity(), None) return pred, query_args else: query_term = _lit_to_swipy(query, var_store) if max_time: # if time limit should be imposed on the query pred = swipy.swipy_predicate("call_with_time_limit", 2, None) query_args = swipy.swipy_new_term_refs(2) swipy.swipy_put_integer(query_args, max_time) swipy.swipy_put_term(query_args+1, query_term) return pred, query_args elif max_depth or max_inference: pred = swipy.swipy_predicate("call_with_depth_limit", 3, None) if max_depth else swipy.swipy_predicate("call_with_inference_limit", 3, None) query_args = swipy.swipy_new_term_refs(3) swipy.swipy_put_term(query_args, query_term) if max_depth: _num_to_swipy_ref(max_depth, query_args + 1) else: _num_to_swipy_ref(max_inference, query_args + 1) swipy.swipy_put_variable(query_args + 2) return pred, query_args else: swipy_objs = [_lit_to_swipy(x, var_store) if isinstance(x, Atom) else _neg_to_swipy(x, var_store) for x in query] if not (max_time or max_depth or max_inference): first = swipy_objs[0] rest = _conjoin_literals(swipy_objs[1:]) compound_arg = swipy.swipy_new_term_refs(2) swipy.swipy_put_term(compound_arg, first) swipy.swipy_put_term(compound_arg + 1, rest) predicate = swipy.swipy_predicate(",", 2, None) return predicate, compound_arg else: query_term = _conjoin_literals(swipy_objs) if max_time: # if time limit should be imposed on the query wrapped_term = self._wrap_in_call_time(query_term, max_time) exception = swipy.swipy_new_atom('time_limit_exceeded') fail = swipy.swipy_new_atom('fail') pred = swipy.swipy_predicate("catch", 3, None) query_args = swipy.swipy_new_term_refs(3) swipy.swipy_put_term(query_args, wrapped_term) swipy.swipy_put_term(query_args + 1, exception) swipy.swipy_put_term(query_args + 2, fail) return pred, query_args elif max_depth or max_inference: pred = swipy.swipy_predicate("call_with_depth_limit", 3, None) if max_depth else swipy.swipy_predicate("call_with_inference_limit", 3, None) query_args = swipy.swipy_new_term_refs(3) swipy.swipy_put_term(query_args, query_term) if max_depth: _num_to_swipy_ref(max_depth, query_args + 1) else: _num_to_swipy_ref(max_inference, query_args + 1) swipy.swipy_put_variable(query_args + 2) return pred, query_args