Function msg

tinygrad/runtime/support/objc.py:30–34

(sel:str, restype=id_, argtypes=[], retain=False, clsmeth=False)

dispatch_data_create = returns_retained(dispatch_data_create)

def msg(sel:str, restype=id_, argtypes=[], retain=False, clsmeth=False):
  # Using attribute access returns a new reference so setting restype is safe
  (sender:=lib["objc_msgSend"]).restype, sender.argtypes = restype, [id_, id_]+list(argtypes) if argtypes else []
  def f(ptr, *args): return sender(ptr._objc_class_ if clsmeth else ptr, getsel(sel.encode()), *args)
  return returns_retained(f) if retain else f

if TYPE_CHECKING:
  import _ctypes