rtcllm/rtcllm/aav.py
2024-09-11 13:11:03 +08:00

47 lines
1.2 KiB
Python

import random
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
from aiortc import MediaStreamTrack, VideoStreamTrack, AudioStreamTrack
class MyMediaPlayer(MediaPlayer):
pass
class MyTrackBase(MediaStreamTrack):
def __init__(self, source=None):
super().__init__()
self.reloaded = False
self.source = source
self.set_track()
print(f'{self.kind=}, {self.__class__.__name__}')
def set_track(self):
if self.kind == 'audio':
self.track = self.source.audio
else:
self.track = self.source.video
self.track.on('ended', self.reload)
def reload(self):
print(f'reload the source----{self.source._file_path}')
self.set_source(MyMediaPlayer(self.source._file_path))
self.reloaded = True
def set_source(self, source):
self.source = source
self.set_track()
async def recv(self):
print(f'{self.__class__.__name__}, {self.source.duration=}, {self.source.time=}')
if self.source is None:
return None
if self.track.readyState != 'live':
self.reload()
f = await self.track.recv()
return f
class MyAudioStreamTrack(MyTrackBase):
kind = 'audio'
class MyVideoStreamTrack(VideoStreamTrack, MyTrackBase):
kind = 'video'