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 query(self, *query, **kwargs): if 'max_solutions' in kwargs: max_solutions = kwargs['max_solutions'] else: max_solutions = -1 if 'time_limit' in kwargs: time_limit = kwargs['time_limit'] else: time_limit = None if 'depth_limit' in kwargs: depth_limit = kwargs['depth_limit'] else: depth_limit = None if 'inference_limit' in kwargs: inference_limit = kwargs['inference_limit'] else: inference_limit = None 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) pred, compound_arg = self._prepare_query(var_store, *query, max_time=time_limit, max_depth=inference_limit, max_inference=inference_limit) query = swipy.swipy_open_query(pred, 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