intelligine/intelligine/simulation/object/brain/part/transport/AntTakeBrainPart.py

from intelligine.simulation.object.brain.part.transport.TakeBrainPart import TakeBrainPart
from intelligine.synergy.object.ressource.Ressource import Resource
from intelligine.cst import MOVE_MODE_EXPLO, TYPE_RESOURCE_EXPLOITABLE, \
    MOVE_MODE_GOHOME, MOVE_MODE_NURSE, TYPE_NURSERY


class AntTakeBrainPart(TakeBrainPart):

    # TODO: methode __init_ pour la classe ? Pour surcharger ici.
    _mode_matches = {
        MOVE_MODE_EXPLO: [TYPE_RESOURCE_EXPLOITABLE],
        MOVE_MODE_NURSE: [TYPE_NURSERY],
        MOVE_MODE_GOHOME: []
    }

    @classmethod
    def can_take(cls, context, object_id, object_to_take_id):
        # Pour le moment si le type de l'objet fait partie des types admis pour le mode courant du porteur, c'est bon.
        return cls._match_with_mode(context, object_id, object_to_take_id)

    def done(self, obj, take_object, context):
        # TODO: Ranger ca ? Truc plus dynamique/configurable ?
        if isinstance(take_object, Resource):
            obj.get_brain().switch_to_mode(MOVE_MODE_GOHOME)
            obj.get_movement_molecule_gland().appose()