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

improve encoder collect

Browse Source
This commit is contained in:
harisreedhar
2026-05-26 22:25:55 +05:30
parent e813d7df95
commit d77be89177
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/codecs/aom_encoder.py
+3 -3
View File
@@ -49,7 +49,7 @@ def encode(aom_encoder : AomEncoder, input_buffer : bytes, frame_resolution : Re
def collect(aom_encoder : AomEncoder) -> bytes:
aom_library = aom_module.create_static_library()
output_buffer = bytes()
output_parts = []
packet_cursor = ctypes.c_void_p(0)
packet = aom_library.aom_codec_get_cx_data(aom_encoder, ctypes.byref(packet_cursor))
@@ -58,11 +58,11 @@ def collect(aom_encoder : AomEncoder) -> bytes:
if ctypes.c_int.from_address(packet).value == 0:
buffer_pointer = ctypes.c_void_p.from_address(packet + 8).value
buffer_size = ctypes.c_size_t.from_address(packet + 16).value
output_buffer += ctypes.string_at(buffer_pointer, buffer_size)
output_parts.append(ctypes.string_at(buffer_pointer, buffer_size))
packet = aom_library.aom_codec_get_cx_data(aom_encoder, ctypes.byref(packet_cursor))
return output_buffer
return b''.join(output_parts)
def destroy(aom_encoder : AomEncoder) -> None:
facefusion/codecs/vpx_encoder.py
+3 -3
View File
@@ -53,7 +53,7 @@ def encode(vpx_encoder : VpxEncoder, input_buffer : bytes, frame_resolution : Re
def collect(vpx_encoder : VpxEncoder) -> bytes:
vpx_library = vpx_module.create_static_library()
output_buffer = bytes()
output_parts = []
packet_cursor = ctypes.c_void_p(0)
packet = vpx_library.vpx_codec_get_cx_data(vpx_encoder, ctypes.byref(packet_cursor))
@@ -62,11 +62,11 @@ def collect(vpx_encoder : VpxEncoder) -> bytes:
if ctypes.c_int.from_address(packet).value == 0:
buffer_pointer = ctypes.c_void_p.from_address(packet + 8).value
buffer_size = ctypes.c_size_t.from_address(packet + 16).value
output_buffer += ctypes.string_at(buffer_pointer, buffer_size)
output_parts.append(ctypes.string_at(buffer_pointer, buffer_size))
packet = vpx_library.vpx_codec_get_cx_data(vpx_encoder, ctypes.byref(packet_cursor))
return output_buffer
return b''.join(output_parts)
def destroy(vpx_encoder : VpxEncoder) -> None: