mirror of
https://github.com/facefusion/facefusion.git
synced 2026-08-09 02:26:04 +02:00
Henry RuhsandGitHub
1ca152b934
* fix import * introduce collect vision frame from asset, add is_vision_frame everywhere, rename to vision_buffer * fix lint * fix lint * fix lint
29 lines
1.3 KiB
Python
29 lines
1.3 KiB
Python
import ctypes
|
|
import threading
|
|
from functools import partial
|
|
|
|
from facefusion.libraries import datachannel as datachannel_module
|
|
from facefusion.types import FrameHandler
|
|
|
|
|
|
def create_receive_event(track : int, frame_handler : FrameHandler) -> threading.Event:
|
|
datachannel_library = datachannel_module.create_static_library()
|
|
receive_event = threading.Event()
|
|
|
|
frame_callback = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_void_p, ctypes.c_int, ctypes.c_void_p, ctypes.c_void_p)(partial(dispatch_frame, frame_handler))
|
|
close_callback = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_void_p)(partial(dispatch_event, receive_event))
|
|
datachannel_library.rtcSetFrameCallback(track, frame_callback)
|
|
datachannel_library.rtcSetClosedCallback(track, close_callback)
|
|
receive_event.frame_callback = frame_callback # type: ignore[attr-defined]
|
|
receive_event.close_callback = close_callback # type: ignore[attr-defined]
|
|
|
|
return receive_event
|
|
|
|
|
|
def dispatch_frame(frame_handler : FrameHandler, track : int, data : ctypes.c_void_p, size : int, info : ctypes.c_void_p, pointer : ctypes.c_void_p) -> None:
|
|
frame_handler(ctypes.string_at(data, size), ctypes.cast(info, ctypes.POINTER(ctypes.c_uint32)).contents.value)
|
|
|
|
|
|
def dispatch_event(event : threading.Event, track : int, pointer : ctypes.c_void_p) -> None:
|
|
event.set()
|