cyphesis/data/rulesets/basic/scripts/mind/Memory.py

# This file is distributed under the terms of the GNU General Public license.
# Copyright (C) 1999 Aloril (See the file COPYING for details).

from common import const


class Memory:
    def __init__(self, a_map):
        self.events = []
        self.map = a_map

    def destroy(self):
        self.events = None
        self.map = None

    def recall_place(self, location, radius, a_filter):
        try:
            if isinstance(a_filter, list):
                for i in a_filter:
                    result = self.map.find_by_location_query(location, radius, i)
                    if len(result) != 0:
                        return result
            else:
                return self.map.find_by_location_query(location, radius, a_filter)
        except RuntimeError:
            # Expect the location to be incomplete and handle it
            return None

    def remember_event(self, event):
        """add new memory with age"""
        self.events.append([event, 1.0])

    def recall_event(self, event, cmp):
        """return list of memories with same command"""
        found = []
        for (e, age) in self.events:
            if cmp(*(event, e)):
                found.append(e)
        return found

    def forget(self):
        """age memories and forgot too old ones"""
        for m in self.events:
            m[1] = m[1] - 0.1
        # remove forgotten things
        self.events = [m for m in self.events if m[1] > const.fzero]