use super::Config;
use crate::catalog::hang::CatalogExt;
use crate::container::Frame;
pub struct Import<E: CatalogExt = ()> {
track: crate::container::Producer<crate::catalog::hang::Container>,
rendition: crate::catalog::AudioTrack<E>,
}
impl<E: CatalogExt> Import<E> {
pub fn new(
track: moq_net::TrackProducer,
catalog: crate::catalog::Producer<E>,
config: Config,
) -> crate::Result<Self> {
let mut audio = hang::catalog::AudioConfig::new(
hang::catalog::AudioCodec::Opus,
config.sample_rate,
config.channel_count,
);
audio.container = hang::catalog::Container::Legacy;
tracing::debug!(name = ?track.name(), config = ?audio, "starting track");
let mut rendition = catalog.audio_track(track.name());
rendition.set(audio);
Ok(Self {
track: crate::container::Producer::new(track, crate::catalog::hang::Container::Legacy),
rendition,
})
}
pub fn name(&self) -> &str {
self.track.name()
}
pub fn demand(&self) -> moq_net::TrackDemand {
self.track.track().demand()
}
pub fn finish(&mut self) -> crate::Result<()> {
self.track.finish()?;
Ok(())
}
pub fn seek(&mut self, sequence: u64) -> crate::Result<()> {
self.track.seek(sequence)?;
Ok(())
}
pub fn decode(&mut self, frame: &[u8], pts: Option<crate::container::Timestamp>) -> crate::Result<()> {
let timestamp = self.rendition.timestamp(pts)?;
self.track.write(Frame {
timestamp,
payload: bytes::Bytes::copy_from_slice(frame),
keyframe: true,
duration: None,
})?;
self.track.finish_group()?;
Ok(())
}
}