propositional_kb.py

from itertools import chain

from sympy import *
from sympy.logic.boolalg import conjuncts, to_cnf

from utils import *


class PropositionalKB:

    def __init__(self, sentence=None):
        self.game_map = None
        self.map_objects = None

        self.clauses = []
        if sentence:
            self.tell(sentence)

    def tell(self, sentence):
        self.clauses.extend(conjuncts(to_cnf(sentence)))

    def tell_percepts(self, game_map, map_objects):
        self.game_map = game_map
        self.map_objects = map_objects

    def entails(self, clauses, query, all_models):
        if len(clauses) > 0:
            first = clauses[0]
            rest = clauses[1:]
            models = chain(satisfiable(first & query, all_models=True))
            return chain(models, self.entails(rest, query, all_models))

    def ask(self, query):
        models = self.entails(self.clauses, query, all_models=True)
        return models

    def is_monster(self, state):
        return self.is_something(state, StaticMonster) or self.is_something(state, DynamicMonster)

    def is_boss(self, state):
        return self.is_something(state, Boss)

    def is_agent(self, state):
        return self.is_something(state, AgentPlaceholder)

    def is_something(self, state, clazz):
        if (state.x, state.y) in self.map_objects:
            if isinstance(self.map_objects[(state.x, state.y)], clazz):
                return True
        return False

    def is_power_up(self, state):
        return isinstance(self.map_objects[state], PowerUp)

    def is_safe(self, state):
        return not (self.is_monster(state) or self.is_agent(state))

    def has_enough_strength_for_boss(self, strength):
        return strength >= 90

    def has_enough_strength_for_monster(self, strength):
        return strength > 30

    def has_not_enough_strength(self, strength):
        return strength < 100

    def get_KB(self):
        return self.clauses

    # def ask(self, query):
    #     checkClause = KBentailsAlpha(self.clauses, query)
    #     if (checkClause == {}): return True
    #     else: return False

# propkb = PropsitionalKB()
# a,b = symbols('a,b')
# propkb.tell(Implies(a,b))
# print('here is my KB')
# print(propkb.getKB())

# propkb.tell(symbols('p'))
# print('now my kb is: ')
# print(propkb.getKB())