from rpython.rlib import _rffi_stacklet as _c
from rpython.rlib import debug
from rpython.rlib.objectmodel import we_are_translated, specialize
from rpython.rtyper.annlowlevel import llhelper


class StackletGcRootFinder(object):
    @staticmethod
    @specialize.arg(1)
    def new(thrd, callback, arg):
        h = _c.new(thrd._thrd, llhelper(_c.run_fn, callback), arg)
        if not h:
            raise MemoryError
        return h

    @staticmethod
    def switch(h):
        h = _c.switch(h)
        if not h:
            raise MemoryError
        return h

    @staticmethod
    def destroy(thrd, h):
        _c.destroy(thrd._thrd, h)
        if we_are_translated():
            debug.debug_print("not using a framework GC: "
                              "stacklet_destroy() may leak")

    is_empty_handle = staticmethod(_c.is_empty_handle)

    @staticmethod
    def get_null_handle():
        return _c.null_handle


gcrootfinder = StackletGcRootFinder    # class object