Jerry/voice-bridge
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix audio encoder

Browse source 
Signed-off-by: Aron Heinecke <aron.heinecke@t-online.de>
This commit is contained in:
Aron Heinecke 2021-05-11 00:17:26 +02:00
parent 8a8c21bdca
commit b4bbf5e777
2 changed files with 12 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/discord.rs
View file

@ -394,7 +394,7 @@ impl VoiceEventHandler for Receiver {
let time = std::time::Instant::now();
let mut lock = self.sink.lock().await;
let dur = time.elapsed();
if dur > Duration::from_nanos(500) {
if dur.as_millis() > 1 {
eprintln!("Akquiring lock took {}ms",dur.as_millis());
}
if let Some(buffer) = lock.get_mut(&packet.ssrc) {

21
src/main.rs
View file

@ -153,7 +153,12 @@ async fn main() -> Result<()> {
// a2t.set_volume(config.volume);
// a2t.set_playing(true);
// }
let encoder = audiopus::coder::Encoder::new(
audiopus::SampleRate::Hz48000,
audiopus::Channels::Stereo,
audiopus::Application::Voip)
.expect("Can't construct encoder!");
let encoder = Arc::new(Mutex::new(encoder));
let mut interval = tokio::time::interval(Duration::from_millis(TICK_TIME));
// tokio::spawn(async {
@ -207,14 +212,13 @@ async fn main() -> Result<()> {
//eprintln!("Tick took {}ms",dur.as_millis());
}
let start = std::time::Instant::now();
if let Some(processed) = process_audio(&voice_buffer).await {
if let Some(processed) = process_audio(&voice_buffer,&encoder).await {
con.send_audio(processed)?;
let dur = start.elapsed();
if dur >= Duration::from_millis(1) {
eprintln!("Audio pipeline took {}ms",dur.as_millis());
}
}
}
_ = tokio::signal::ctrl_c() => { break; }
r = events => {
@ -233,7 +237,7 @@ async fn main() -> Result<()> {
async fn process_audio(voice_buffer: &AudioBufferDiscord) -> Option<OutPacket> {
async fn process_audio(voice_buffer: &AudioBufferDiscord, encoder: &Arc<Mutex<Encoder>>) -> Option<OutPacket> {
let mut buffer_map;
{
let mut lock = voice_buffer.lock().await;
@ -243,6 +247,7 @@ async fn process_audio(voice_buffer: &AudioBufferDiscord) -> Option<OutPacket> {
return None;
}
let mut encoded = [0; 1024];
let encoder_c = encoder.clone();
let res = task::spawn_blocking(move || {
let start = std::time::Instant::now();
let mut data: Vec<i16> = Vec::with_capacity(STEREO_20MS);
@ -258,13 +263,9 @@ async fn process_audio(voice_buffer: &AudioBufferDiscord) -> Option<OutPacket> {
}
}
let encoder = audiopus::coder::Encoder::new(
audiopus::SampleRate::Hz48000,
audiopus::Channels::Stereo,
audiopus::Application::Voip)
.expect("Can't construct encoder!");
let length = match encoder.encode(&data, &mut encoded) {
let lock = encoder_c.try_lock().expect("Can't reach encoder!");
let length = match lock.encode(&data, &mut encoded) {
Err(e) => {eprintln!("Failed to encode voice: {}",e); return None;},
Ok(size) => size,
};