selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #447 from Monadical-SAS/fix-duration

Browse Source 
Fix audio duration calculation
This commit is contained in:
Sergey Mankovsky
2025-01-24 15:04:16 +01:00
committed by GitHub
parent d0d019f1b6 43562391b7
commit 8aa3f049ef
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
server/reflector/processors/audio_file_writer.py
View File

@@ -22,6 +22,7 @@ class AudioFileWriterProcessor(Processor):
self.out_container = None
self.out_stream = None
self.last_packet = None
self.duration = 0
async def _push(self, data: av.AudioFrame):
if not self.out_container:
@@ -39,9 +40,11 @@ class AudioFileWriterProcessor(Processor):
)
else:
raise ValueError("Only mp3 and wav files are supported")
for packet in self.out_stream.encode(data):
self.out_container.mux(packet)
self.last_packet = packet
self.duration += packet.duration
await self.emit(data)
async def _flush(self):
@@ -49,13 +52,11 @@ class AudioFileWriterProcessor(Processor):
for packet in self.out_stream.encode():
self.out_container.mux(packet)
self.last_packet = packet
self.duration += packet.duration
try:
if self.last_packet is not None:
duration = round(
float(
(self.last_packet.pts * self.last_packet.duration)
* self.last_packet.time_base
),
float(self.duration * self.last_packet.time_base * 1000),
2,
)
except Exception: