yumoqing/rtcllm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2024-09-11 13:20:58 +08:00
parent 8f00665eb6
commit 35de168787
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
rtcllm/aav.py
View File

@ -8,33 +8,30 @@ class MyMediaPlayer(MediaPlayer):
class MyTrackBase(MediaStreamTrack): class MyTrackBase(MediaStreamTrack):
def __init__(self, source=None): def __init__(self, source=None):
super().__init__() super().__init__()
self.reloaded = False
self.source = source self.source = source
self.set_track() self.set_source_track()
print(f'{self.kind=}, {self.__class__.__name__}') print(f'{self.kind=}, {self.__class__.__name__}')
def set_track(self): def set_source_track(self):
if self.kind == 'audio': if self.kind == 'audio':
self.track = self.source.audio self.track = self.source.audio
else: else:
self.track = self.source.video self.track = self.source.video
self.track.on('ended', self.reload)
def reload(self): def source_reload(self):
print(f'reload the source----{self.source._file_path}') print(f'reload the source----{self.source._file_path}')
self.set_source(MyMediaPlayer(self.source._file_path)) self.set_source(MyMediaPlayer(self.source._file_path))
self.reloaded = True
def set_source(self, source): def set_source(self, source):
self.source = source self.source = source
self.set_track() self.set_source_track()
async def recv(self): async def recv(self):
print(f'{self.__class__.__name__}, {self.source.duration=}, {self.source.time=}') print(f'{self.__class__.__name__}, {self.source.duration=}, {self.source.time=}')
if self.source is None: if self.source is None:
return None return None
if self.track.readyState != 'live': if self.track.readyState != 'live':
self.reload() self.source_reload()
f = await self.track.recv() f = await self.track.recv()
return f return f