Skip to content

Commit

Permalink
Communication workaround for Android 11+ (livekit#363) (livekit#364)
Browse files Browse the repository at this point in the history
* Communication mode workaround

* clean up implementation

* tests and spotless

* Inject dispatcher

* Tie communication workaround to playout instead of record
  • Loading branch information
davidliu authored Jan 25, 2024
1 parent 4a41b5a commit d8e44ee
Show file tree
Hide file tree
Showing 10 changed files with 289 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -104,6 +104,22 @@ class AudioOptions(
* Not used if [audioDeviceModule] is provided.
*/
val javaAudioDeviceModuleCustomizer: ((builder: JavaAudioDeviceModule.Builder) -> Unit)? = null,

/**
* On Android 11+, the audio mode will reset itself from [AudioManager.MODE_IN_COMMUNICATION] if
* there is no audio playback or capture for 6 seconds (for example when joining a room with
* no speakers and the local mic is muted.) This mode reset will cause unexpected
* behavior when trying to change the volume, causing it to not properly change the volume.
*
* We use a workaround by playing a silent audio track to keep the communication mode from
* resetting.
*
* Setting this flag to true will disable the workaround.
*
* This flag is a no-op when the audio mode is set to anything other than
* [AudioManager.MODE_IN_COMMUNICATION].
*/
val disableCommunicationModeWorkaround: Boolean = false,
)

/**
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,186 @@
/*
* Copyright 2024 LiveKit, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package io.livekit.android.audio

import android.annotation.SuppressLint
import android.media.AudioAttributes
import android.media.AudioFormat
import android.media.AudioManager
import android.media.AudioTrack
import android.os.Build
import androidx.annotation.RequiresApi
import io.livekit.android.dagger.InjectionNames
import io.livekit.android.util.CloseableCoroutineScope
import kotlinx.coroutines.MainCoroutineDispatcher
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.combine
import kotlinx.coroutines.flow.distinctUntilChanged
import kotlinx.coroutines.launch
import java.nio.ByteBuffer
import javax.inject.Inject
import javax.inject.Named
import javax.inject.Singleton

/**
* @see CommunicationWorkaroundImpl
*/
interface CommunicationWorkaround {

fun start()
fun stop()
fun onStartPlayout()
fun onStopPlayout()

fun dispose()
}

class NoopCommunicationWorkaround
@Inject
constructor() : CommunicationWorkaround {
override fun start() {
}

override fun stop() {
}

override fun onStartPlayout() {
}

override fun onStopPlayout() {
}

override fun dispose() {
}
}

/**
* Work around for communication mode resetting after 6 seconds if no audio playback or capture.
* Issue only happens on 11+ (version code R).
* https://issuetracker.google.com/issues/209493718
*/
@Singleton
@RequiresApi(Build.VERSION_CODES.R)
class CommunicationWorkaroundImpl
@Inject
constructor(
@Named(InjectionNames.DISPATCHER_MAIN)
dispatcher: MainCoroutineDispatcher,
) : CommunicationWorkaround {

private val coroutineScope = CloseableCoroutineScope(dispatcher)
private val started = MutableStateFlow(false)
private val playoutStopped = MutableStateFlow(true)

private var audioTrack: AudioTrack? = null

init {
coroutineScope.launch {
started.combine(playoutStopped) { a, b -> a to b }
.distinctUntilChanged()
.collect { (started, playoutStopped) ->
onStateChanged(started, playoutStopped)
}
}
}

override fun start() {
started.value = true
}

override fun stop() {
started.value = false
}

override fun onStartPlayout() {
playoutStopped.value = false
}

override fun onStopPlayout() {
playoutStopped.value = true
}

@SuppressLint("NewApi")
private fun onStateChanged(started: Boolean, playoutStopped: Boolean) {
if (started && playoutStopped) {
startAudioTrackIfNeeded()
} else {
stopAudioTrackIfNeeded()
}
}

@SuppressLint("Range")
private fun startAudioTrackIfNeeded() {
if (audioTrack != null) {
return
}

val sampleRate = 16000
val audioFormat = AudioFormat.ENCODING_PCM_16BIT
val bytesPerFrame = 1 * getBytesPerSample(audioFormat)
val framesPerBuffer: Int = sampleRate / 100 // 10 ms
val byteBuffer = ByteBuffer.allocateDirect(bytesPerFrame * framesPerBuffer)

audioTrack = AudioTrack.Builder()
.setAudioFormat(
AudioFormat.Builder()
.setEncoding(audioFormat)
.setSampleRate(sampleRate)
.setChannelMask(AudioFormat.CHANNEL_OUT_MONO)
.build(),
)
.setAudioAttributes(
AudioAttributes.Builder()
.setUsage(AudioAttributes.USAGE_VOICE_COMMUNICATION)
.setContentType(AudioAttributes.CONTENT_TYPE_SPEECH)
.build(),
)
.setBufferSizeInBytes(byteBuffer.capacity())
.setTransferMode(AudioTrack.MODE_STATIC)
.setSessionId(AudioManager.AUDIO_SESSION_ID_GENERATE)
.build()

audioTrack?.write(byteBuffer, byteBuffer.remaining(), AudioTrack.WRITE_BLOCKING)
audioTrack?.setLoopPoints(0, framesPerBuffer - 1, -1)

audioTrack?.play()
}

// Reference from Android code, AudioFormat.getBytesPerSample. BitPerSample / 8
// Default audio data format is PCM 16 bits per sample.
// Guaranteed to be supported by all devices
private fun getBytesPerSample(audioFormat: Int): Int {
return when (audioFormat) {
AudioFormat.ENCODING_PCM_8BIT -> 1
AudioFormat.ENCODING_PCM_16BIT, AudioFormat.ENCODING_IEC61937, AudioFormat.ENCODING_DEFAULT -> 2
AudioFormat.ENCODING_PCM_FLOAT -> 4
AudioFormat.ENCODING_INVALID -> throw IllegalArgumentException("Bad audio format $audioFormat")
else -> throw IllegalArgumentException("Bad audio format $audioFormat")
}
}

private fun stopAudioTrackIfNeeded() {
audioTrack?.stop()
audioTrack?.release()
audioTrack = null
}

override fun dispose() {
coroutineScope.close()
stop()
stopAudioTrackIfNeeded()
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,17 @@
package io.livekit.android.dagger

import android.media.AudioAttributes
import android.media.AudioManager
import android.os.Build
import dagger.Module
import dagger.Provides
import io.livekit.android.AudioType
import io.livekit.android.audio.AudioHandler
import io.livekit.android.audio.AudioSwitchHandler
import io.livekit.android.audio.CommunicationWorkaround
import io.livekit.android.audio.CommunicationWorkaroundImpl
import io.livekit.android.audio.NoopCommunicationWorkaround
import io.livekit.android.memory.CloseableManager
import javax.inject.Named
import javax.inject.Provider
import javax.inject.Singleton
Expand Down Expand Up @@ -62,4 +68,27 @@ internal object AudioHandlerModule {
audioStreamType = audioOutputType.audioStreamType
}
}

@Provides
@Singleton
@JvmSuppressWildcards
fun communicationWorkaround(
@Named(InjectionNames.OVERRIDE_DISABLE_COMMUNICATION_WORKAROUND)
disableCommunicationWorkaround: Boolean,
audioType: AudioType,
closeableManager: CloseableManager,
commWorkaroundImplProvider: Provider<CommunicationWorkaroundImpl>,
): CommunicationWorkaround {
return if (
!disableCommunicationWorkaround &&
Build.VERSION.SDK_INT >= Build.VERSION_CODES.R &&
audioType.audioMode == AudioManager.MODE_IN_COMMUNICATION
) {
commWorkaroundImplProvider.get().apply {
closeableManager.registerClosable { this.dispose() }
}
} else {
NoopCommunicationWorkaround()
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -52,5 +52,6 @@ internal object InjectionNames {
internal const val OVERRIDE_VIDEO_DECODER_FACTORY = "override_video_decoder_factory"
internal const val OVERRIDE_AUDIO_HANDLER = "override_audio_handler"
internal const val OVERRIDE_AUDIO_OUTPUT_TYPE = "override_audio_output_type"
internal const val OVERRIDE_DISABLE_COMMUNICATION_WORKAROUND = "override_disable_communication_workaround"
internal const val OVERRIDE_EGL_BASE = "override_egl_base"
}
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,10 @@ internal class OverridesModule(private val overrides: LiveKitOverrides) {
@Named(InjectionNames.OVERRIDE_AUDIO_OUTPUT_TYPE)
fun audioOutputType() = overrides.audioOptions?.audioOutputType

@Provides
@Named(InjectionNames.OVERRIDE_DISABLE_COMMUNICATION_WORKAROUND)
fun disableCommunicationWorkAround() = overrides.audioOptions?.disableCommunicationModeWorkaround ?: false

@Provides
@Named(InjectionNames.OVERRIDE_EGL_BASE)
@Nullable
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ import androidx.annotation.Nullable
import dagger.Module
import dagger.Provides
import io.livekit.android.LiveKit
import io.livekit.android.audio.CommunicationWorkaround
import io.livekit.android.memory.CloseableManager
import io.livekit.android.util.LKLog
import io.livekit.android.util.LoggingLevel
Expand Down Expand Up @@ -91,6 +92,7 @@ internal object RTCModule {
audioOutputAttributes: AudioAttributes,
appContext: Context,
closeableManager: CloseableManager,
communicationWorkaround: CommunicationWorkaround,
): AudioDeviceModule {
if (audioDeviceModuleOverride != null) {
return audioDeviceModuleOverride
Expand Down Expand Up @@ -130,6 +132,7 @@ internal object RTCModule {
LKLog.e { "onWebRtcAudioTrackError: $errorMessage" }
}
}

val audioRecordStateCallback: JavaAudioDeviceModule.AudioRecordStateCallback = object :
JavaAudioDeviceModule.AudioRecordStateCallback {
override fun onWebRtcAudioRecordStart() {
Expand All @@ -146,10 +149,12 @@ internal object RTCModule {
JavaAudioDeviceModule.AudioTrackStateCallback {
override fun onWebRtcAudioTrackStart() {
LKLog.v { "Audio playout starts" }
communicationWorkaround.onStartPlayout()
}

override fun onWebRtcAudioTrackStop() {
LKLog.v { "Audio playout stops" }
communicationWorkaround.onStopPlayout()
}
}

Expand Down
14 changes: 12 additions & 2 deletions livekit-android-sdk/src/main/java/io/livekit/android/room/Room.kt
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ import io.livekit.android.ConnectOptions
import io.livekit.android.RoomOptions
import io.livekit.android.Version
import io.livekit.android.audio.AudioHandler
import io.livekit.android.audio.CommunicationWorkaround
import io.livekit.android.dagger.InjectionNames
import io.livekit.android.e2ee.E2EEManager
import io.livekit.android.e2ee.E2EEOptions
Expand Down Expand Up @@ -69,6 +70,7 @@ constructor(
val audioHandler: AudioHandler,
private val closeableManager: CloseableManager,
private val e2EEManagerFactory: E2EEManager.Factory,
private val communicationWorkaround: CommunicationWorkaround,
) : RTCEngine.Listener, ParticipantListener {

private lateinit var coroutineScope: CoroutineScope
Expand Down Expand Up @@ -133,8 +135,16 @@ constructor(
var state: State by flowDelegate(State.DISCONNECTED) { new, old ->
if (new != old) {
when (new) {
State.CONNECTING -> audioHandler.start()
State.DISCONNECTED -> audioHandler.stop()
State.CONNECTING -> {
audioHandler.start()
communicationWorkaround.start()
}

State.DISCONNECTED -> {
audioHandler.stop()
communicationWorkaround.stop()
}

else -> {}
}
}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
/*
* Copyright 2024 LiveKit, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package io.livekit.android.util

import android.os.Handler
import android.os.Looper

fun Handler.runOrPost(r: Runnable) {
if (Looper.myLooper() == this.looper) {
r.run()
} else {
post(r)
}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright 2023 LiveKit, Inc.
* Copyright 2023-2024 LiveKit, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand All @@ -19,10 +19,15 @@ package io.livekit.android.mock.dagger
import dagger.Binds
import dagger.Module
import io.livekit.android.audio.AudioHandler
import io.livekit.android.audio.CommunicationWorkaround
import io.livekit.android.audio.NoAudioHandler
import io.livekit.android.audio.NoopCommunicationWorkaround

@Module
interface TestAudioHandlerModule {
@Binds
fun audioHandler(audioHandler: NoAudioHandler): AudioHandler

@Binds
fun communicationWorkaround(communicationWorkaround: NoopCommunicationWorkaround): CommunicationWorkaround
}
Loading

0 comments on commit d8e44ee

Please sign in to comment.