# Source code for pymunk._collision_handler

    __version__ = "$Id$"
    __docformat__ = "reStructuredText"

    import sys
    import warnings

    from . import _chipmunk_cffi
    cp = _chipmunk_cffi.lib
    ffi = _chipmunk_cffi.ffi

    from ._arbiter import Arbiter

    class CollisionHandler(object):
        """A collision handler is a set of 4 function callbacks for the different
        collision events that pymunk recognizes.

        Collision callbacks are closely associated with Arbiter objects. You
        should familiarize yourself with those as well.

        Note #1: Shapes tagged as sensors (Shape.sensor == true) never generate
        collisions that get processed, so collisions between sensors shapes and
        other shapes will never call the postSolve() callback. They still
        generate begin(), and separate() callbacks, and the preSolve() callback
        is also called every frame even though there is no collision response.
        Note #2: preSolve() callbacks are called before the sleeping algorithm
        runs. If an object falls asleep, its postSolve() callback won't be
        called until it's reawoken.
        """
        def __init__(self, _handler, space, *args, **kwargs):
            """Initialize a CollisionHandler object from the Chipmunk equivalent
            struct and the Space.

            .. note::
                You should never need to create an instance of this class directly.
            """
            self._handler = _handler
            self._space = space
            self._begin = None
            self._pre_solve = None
            self._post_solve = None
            self._separate = None

            self._data = {}

        def _get_data(self):
            return self._data
        data = property(_get_data, doc="""Data property that get passed on into the
        callbacks.

        Usefull if the callback needs some extra data to perform its function.
        """)

        def _set_begin(self, func):

            @ffi.callback("typedef cpBool (*cpCollisionBeginFunc)"
                "(cpArbiter *arb, cpSpace *space, cpDataPointer userData)")
            def cf(_arb, _space, _):
                x = func(Arbiter(_arb, self._space), self._space, self._data)
                if isinstance(x,int):
                    return x

                if sys.version_info[0] >= 3:
                    func_name = func.__code__.co_name
                    filename = func.__code__.co_filename
                    lineno = func.__code__.co_firstlineno
                else:
                    func_name = func.func_name
                    filename = func.func_code.co_filename
                    lineno = func.func_code.co_firstlineno

                warnings.warn_explicit(
                    "Function '" + func_name + "' should return a bool to"
                    " indicate if the collision should be processed or not when"
                    " used as 'begin' or 'pre_solve' collision callback.",
                    UserWarning, filename, lineno, func.__module__)
                return True

            self._begin = cf
            self._handler.beginFunc = cf

        def _get_begin(self):
            return self._begin

        begin = property(_get_begin, _set_begin,
            doc="""Two shapes just started touching for the first time this step.

            ``func(arbiter, space) -> bool``

            Return true from the callback to process the collision normally or
            false to cause pymunk to ignore the collision entirely. If you return
            false, the `pre_solve` and `post_solve` callbacks will never be run,
            but you will still recieve a separate event when the shapes stop
            overlapping.
            """)

        def _set_pre_solve(self, func):

            @ffi.callback("typedef cpBool (*cpCollisionPreSolveFunc)"
                "(cpArbiter *arb, cpSpace *space, cpDataPointer userData)")
            def cf(_arb, _space, _data):
                x = func(Arbiter(_arb, self._space), self._space, self._data)
                if isinstance(x,int):
                    return x

                if sys.version_info[0] >= 3:
                    func_name = func.__code__.co_name
                    filename = func.__code__.co_filename
                    lineno = func.__code__.co_firstlineno
                else:
                    func_name = func.func_name
                    filename = func.func_code.co_filename
                    lineno = func.func_code.co_firstlineno

                warnings.warn_explicit(
                    "Function '" + func_name + "' should return a bool to"
                    " indicate if the collision should be processed or not when"
                    " used as 'begin' or 'pre_solve' collision callback.",
                    UserWarning, filename, lineno, func.__module__)
                return True

            self._pre_solve = cf
            self._handler.preSolveFunc = cf

        def _get_pre_solve(self):
            return self._pre_solve

        pre_solve = property(_get_pre_solve, _set_pre_solve,
            doc="""Two shapes are touching during this step.

            ``func(arbiter, space) -> bool``

            Return false from the callback to make pymunk ignore the collision
            this step or true to process it normally. Additionally, you may
            override collision values using Arbiter.friction, Arbiter.elasticity
            or Arbiter.surfaceVelocity to provide custom friction, elasticity,
            or surface velocity values. See Arbiter for more info.
            """)

        def _set_post_solve(self, func):

            @ffi.callback("typedef void (*cpCollisionPostSolveFunc)"
                "(cpArbiter *arb, cpSpace *space, cpDataPointer userData)")
            def cf(_arb, _space, _data):
                func(Arbiter(_arb, self._space), self._space, self._data)

            self._post_solve = cf
            self._handler.postSolveFunc = cf

        def _get_post_solve(self):
            return self._post_solve

        post_solve = property(_get_post_solve, _set_post_solve,
            doc="""Two shapes are touching and their collision response has been
            processed.

            ``func(arbiter, space)``

            You can retrieve the collision impulse or kinetic energy at this
            time if you want to use it to calculate sound volumes or damage
            amounts. See Arbiter for more info.
            """)

        def _set_separate(self, func):

            @ffi.callback("typedef void (*cpCollisionSeparateFunc)"
                "(cpArbiter *arb, cpSpace *space, cpDataPointer userData)")
            def cf(_arb, _space, _data):
                func(Arbiter(_arb, self._space), self._space, self._data)

            self._separate = cf
            self._handler.separateFunc = cf

        def _get_separate(self):
            return self._separate

        separate = property(_get_separate, _set_separate,
            doc="""Two shapes have just stopped touching for the first time this
            step.

            ``func(arbiter, space)``

            To ensure that begin()/separate() are always called in balanced
            pairs, it will also be called when removing a shape while its in
            contact with something or when deallocating the space.
            """)