CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-02 19:01:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(rtc): 2-way REMB with in-place encoder bitrate update (#1131)

Browse Source 
* implement remb other direction

* update test
This commit is contained in:
Harisreedhar
2026-05-30 04:41:10 +05:30
committed by GitHub
parent 3ff327e670
commit 9f6a19c1d2
5 changed files with 63 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_rtc.py
+38 -8
View File
@@ -79,7 +79,8 @@ def test_send_video() -> None:
'receiver_track': video_track,
'codec': 'vp8'
},
'bitrate': ctypes.c_uint(0)
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
send_video(rtc_peer, bytes(1024), 0)
@@ -106,7 +107,8 @@ def test_send_audio() -> None:
'receiver_track': audio_track,
'codec': 'opus'
},
'bitrate': ctypes.c_uint(0)
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
send_audio(rtc_peer, bytes(960), 0)
@@ -127,7 +129,8 @@ def test_delete_peers() -> None:
'receiver_track': 0,
'codec': 'vp8'
},
'bitrate': ctypes.c_uint(0)
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
]
@@ -150,16 +153,43 @@ def test_wire_remb(video_codec : VideoCodec, payload_type : int) -> None:
'receiver_track': video_sender_track,
'codec': video_codec
},
'bitrate': ctypes.c_uint(0)
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
wire_remb(video_sender_track, rtc_peer.get('bitrate'))
wire_remb(video_sender_track, rtc_peer.get('sender_bitrate'))
assert rtc_peer.get('bitrate').value == 0
assert rtc_peer.get('sender_bitrate').value == 0
handle_remb(0, 6000000, ctypes.addressof(rtc_peer.get('bitrate')))
handle_remb(0, 6000000, ctypes.addressof(rtc_peer.get('sender_bitrate')))
assert rtc_peer.get('bitrate').value == 6000
assert rtc_peer.get('sender_bitrate').value == 6000
datachannel_library.rtcDeletePeerConnection(peer_connection)
@pytest.mark.parametrize('video_codec, payload_type', [ ('av1', 35), ('vp8', 96) ])
def test_wire_remb_receiver(video_codec : VideoCodec, payload_type : int) -> None:
datachannel_library = datachannel_module.create_static_library()
peer_connection = create_peer_connection()
video_receiver_track = add_video_track(peer_connection, 'recvonly', video_codec, payload_type)
rtc_peer : RtcPeer =\
{
'peer_connection': peer_connection,
'video':
{
'sender_track': video_receiver_track,
'receiver_track': video_receiver_track,
'codec': video_codec
},
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
wire_remb(video_receiver_track, rtc_peer.get('receiver_bitrate'))
handle_remb(0, 6000000, ctypes.addressof(rtc_peer.get('receiver_bitrate')))
assert rtc_peer.get('receiver_bitrate').value == 6000
datachannel_library.rtcDeletePeerConnection(peer_connection)
tests/test_stream_helper.py
+15 -1
View File
@@ -178,7 +178,8 @@ def test_run_peer_loop() -> None:
'receiver_track': video_receiver_track,
'codec': 'vp8'
},
'bitrate': ctypes.c_uint(0)
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
session_id = 'test-run-peer-loop'
@@ -297,3 +298,16 @@ def test_process_video(video_codec : VideoCodec, session_id : str) -> None:
assert 'm=video' in sdp_answer
assert 'a=recvonly' in sdp_answer
assert 'a=sendonly' in sdp_answer
for peer in rtc_store.get_peers(session_id):
sender_bitrate = peer.get('sender_bitrate')
receiver_bitrate = peer.get('receiver_bitrate')
assert sender_bitrate.value == 0
assert receiver_bitrate.value == 0
rtc.handle_remb(0, 6000000, ctypes.addressof(sender_bitrate))
assert sender_bitrate.value == 6000
rtc.handle_remb(0, 4000000, ctypes.addressof(receiver_bitrate))
assert receiver_bitrate.value == 4000