mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-12 18:32:18 +02:00
Harisreedhar
949d9cd276
* converted from type() factory to proper ctypes.Structure subclasses * change class to method * bring back rtcSetLocalDescription * rename rtc_bindings.py to datachannel.py * rename rtc_library to datachannel_library and rearrange methods * cleanup
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import ctypes
|
|
import threading
|
|
import time
|
|
from typing import Optional
|
|
|
|
from starlette.testclient import TestClient
|
|
|
|
from facefusion import rtc
|
|
from facefusion.types import SdpOffer
|
|
|
|
|
|
def create_sdp_offer() -> Optional[SdpOffer]:
|
|
datachannel_library = rtc.create_static_datachannel_library()
|
|
peer_connection = rtc.create_peer_connection(disable_auto_negotiation = True)
|
|
|
|
datachannel_library.rtcAddTrack(peer_connection, rtc.build_media_description('video', 96, 'VP8/90000', 'recvonly', 0))
|
|
datachannel_library.rtcAddTrack(peer_connection, rtc.build_media_description('audio', 111, 'opus/48000/2', 'recvonly', 1))
|
|
datachannel_library.rtcSetLocalDescription(peer_connection, b'offer')
|
|
|
|
buffer_size = 16384
|
|
buffer_string = ctypes.create_string_buffer(buffer_size)
|
|
wait_limit = time.monotonic() + 5
|
|
|
|
while time.monotonic() < wait_limit:
|
|
if datachannel_library.rtcGetLocalDescription(peer_connection, buffer_string, buffer_size) > 0:
|
|
sdp = buffer_string.value.decode()
|
|
datachannel_library.rtcDeletePeerConnection(peer_connection)
|
|
return sdp
|
|
|
|
time.sleep(0.05)
|
|
|
|
datachannel_library.rtcDeletePeerConnection(peer_connection)
|
|
return None
|
|
|
|
|
|
def open_websocket_stream(test_client : TestClient, subprotocols : list[str], source_content : bytes, ready_event : threading.Event, stop_event : threading.Event) -> None:
|
|
with test_client.websocket_connect('/stream', subprotocols = subprotocols) as websocket:
|
|
websocket.send_bytes(source_content)
|
|
websocket.receive_text()
|
|
ready_event.set()
|
|
stop_event.wait()
|