Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit e38e0586 authored by chelseahao's avatar chelseahao
Browse files

Modify channel selection in audio stream media service based on intent data. 

Flag: com.android.settingslib.flags.audio_stream_play_pause_by_modify_source
Test: atest
Bug: 384976631
Change-Id: Ice2e5094af3eb9ff593ad7aa228d37bef03e1197
parent 3c8e4222

src/com/android/settings/connecteddevice/audiosharing/audiostreams/AudioStreamMediaService.java

+103 −12
Original line number Diff line number Diff line
@@ -19,7 +19,9 @@ package com.android.settings.connecteddevice.audiosharing.audiostreams; 
import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcast.EXTRA_PRIVATE_BROADCAST_RECEIVE_DATA;

import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant.LocalBluetoothLeBroadcastSourceState.DECRYPTION_FAILED;

import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant.LocalBluetoothLeBroadcastSourceState.PAUSED;

import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant.LocalBluetoothLeBroadcastSourceState.PAUSED_BY_RECEIVER;

import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant.LocalBluetoothLeBroadcastSourceState.STREAMING;

import static com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant.UNKNOWN_CHANNEL;



import static java.util.Collections.emptyList;
@@ -69,6 +71,7 @@ import com.android.settingslib.utils.ThreadUtils; 
import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;



public class AudioStreamMediaService extends Service {

    static final String BROADCAST_ID = "audio_stream_media_service_broadcast_id";
@@ -96,7 +99,7 @@ public class AudioStreamMediaService extends Service { 
                            LEAVE_BROADCAST_ACTION,

                            LEAVE_BROADCAST_TEXT,

                            com.android.settings.R.drawable.ic_clear);

    private final PlaybackState.Builder mPlayStatePausingBuilder =

    private final PlaybackState.Builder mPlayStatePausedByReceiverBuilder =

            new PlaybackState.Builder()

                    .setActions(PlaybackState.ACTION_PLAY_PAUSE | PlaybackState.ACTION_SEEK_TO)

                    .setState(
@@ -107,7 +110,7 @@ public class AudioStreamMediaService extends Service { 
                            LEAVE_BROADCAST_ACTION,

                            LEAVE_BROADCAST_TEXT,

                            com.android.settings.R.drawable.ic_clear);

    private final PlaybackState.Builder mPlayStateHysteresisBuilder =

    private final PlaybackState.Builder mPlayStatePausedByHostBuilder =

            new PlaybackState.Builder()

                    .setState(

                            PlaybackState.STATE_PAUSED,
@@ -129,9 +132,12 @@ public class AudioStreamMediaService extends Service { 
    private int mLatestPositiveVolume = 25;

    private boolean mHysteresisModeFixAvailable;

    private int mBroadcastId;

    private int mSourceId;

    @VisibleForTesting

    @Nullable

    Map<BluetoothDevice, LocalBluetoothLeBroadcastSourceState> mStateByDevice;

    @VisibleForTesting

    Map<BluetoothDevice, Set<Integer>> mSelectedChannelCacheByDevice = new HashMap<>();

    @Nullable private LocalBluetoothManager mLocalBtManager;

    @Nullable private AudioStreamsHelper mAudioStreamsHelper;

    @Nullable private LocalBluetoothLeBroadcastAssistant mLeBroadcastAssistant;
@@ -286,11 +292,13 @@ public class AudioStreamMediaService extends Service { 
        int sourceId = data.getSourceId();

        var state = data.getState();

        String programInfo = data.getProgramInfo();

        Set<Integer> selectedChannelIndex = data.getSelectedChannelIndex();



        // Service not running yet.

        if (mBroadcastId == 0) {

            Log.d(TAG, "handleIntentData(): sending " + data + " to handleInitialSetup()");

            handleInitialSetup(broadcastId, device, state, sourceId, programInfo);

            handleInitialSetup(broadcastId, device, state, sourceId, programInfo,

                    selectedChannelIndex);

            return;

        }
@@ -298,7 +306,8 @@ public class AudioStreamMediaService extends Service { 
        // broadcast Id to handle.

        if (mBroadcastId != broadcastId) {

            Log.d(TAG, "handleIntentData(): sending " + data + " to handleNewBroadcastId()");

            handleNewBroadcastId(broadcastId, device, state, sourceId, programInfo);

            handleNewBroadcastId(broadcastId, device, state, sourceId, programInfo,

                    selectedChannelIndex);

            return;

        }
@@ -306,43 +315,60 @@ public class AudioStreamMediaService extends Service { 
        if (mStateByDevice != null && (!mStateByDevice.containsKey(device) || mStateByDevice.get(

                device) != state)) {

            Log.d(TAG, "handleIntentData(): sending " + data + " to handleNewDeviceOrState()");

            handleNewDeviceOrState(device, state, sourceId, programInfo);

            handleNewDeviceOrState(device, state, sourceId, programInfo, selectedChannelIndex);

        }



        Log.d(TAG, "handleIntentData(): nothing to update.");

    }



    private void handleInitialSetup(int broadcastId, BluetoothDevice device,

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo) {

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo,

            Set<Integer> selectedChannelIndex) {

        if (state == DECRYPTION_FAILED) {

            Log.d(TAG, "handleInitialSetup() : decryption failed. Service will not start.");

            stopSelf();

            return;

        }

        mBroadcastId = broadcastId;

        mSourceId = sourceId;

        mStateByDevice = new HashMap<>();

        mStateByDevice.put(device, state);

        if (!selectedChannelIndex.equals(UNKNOWN_CHANNEL)) {

            mSelectedChannelCacheByDevice.put(device, selectedChannelIndex);

            Log.d(TAG, "mSelectedChannelCacheByDevice:" + mSelectedChannelCacheByDevice);

        }

        MediaSession.Token token = getOrCreateLocalMediaSession(

                getBroadcastName(device, sourceId, programInfo));

        startForeground(NOTIFICATION_ID, buildNotification(token));

    }



    private void handleNewBroadcastId(int broadcastId, BluetoothDevice device,

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo) {

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo,

            Set<Integer> selectedChannelIndex) {

        if (state == DECRYPTION_FAILED) {

            Log.d(TAG, "handleNewBroadcastId() : decryption failed. Ignore.");

            return;

        }

        mBroadcastId = broadcastId;

        mSourceId = sourceId;

        mStateByDevice = new HashMap<>();

        mStateByDevice.put(device, state);

        if (!selectedChannelIndex.equals(UNKNOWN_CHANNEL)) {

            mSelectedChannelCacheByDevice.put(device, selectedChannelIndex);

            Log.d(TAG, "mSelectedChannelCacheByDevice:" + mSelectedChannelCacheByDevice);

        }

        updateMediaSessionAndNotify(device, sourceId, programInfo);

    }



    private void handleNewDeviceOrState(BluetoothDevice device,

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo) {

            LocalBluetoothLeBroadcastSourceState state, int sourceId, String programInfo,

            Set<Integer> selectedChannelIndex) {

        if (mStateByDevice != null) {

            mStateByDevice.put(device, state);

            if (!selectedChannelIndex.equals(UNKNOWN_CHANNEL)) {

                mSelectedChannelCacheByDevice.put(device, selectedChannelIndex);

                Log.d(TAG, "mSelectedChannelCacheByDevice:" + mSelectedChannelCacheByDevice);

            }

        }

        if (getDeviceInValidState().isEmpty()) {

            Log.d(TAG, "handleNewDeviceOrState() : no device is in valid state. Stop service.");
@@ -370,10 +396,34 @@ public class AudioStreamMediaService extends Service { 
    }



    private PlaybackState getPlaybackState() {

        if (Flags.audioStreamPlayPauseByModifySource()) {

            if (isAnyDeviceStreaming()) {

                return mPlayStatePlayingBuilder.build();

            }

            if (isAnyDeviceReceiverPaused()) {

                return mPlayStatePausedByReceiverBuilder.build();

            }

            if (isAllDeviceHysteresis()) {

                return mPlayStatePausedByHostBuilder.build();

            }

            Log.w(TAG, "getPlaybackState() : devices in unexpected state: " + mStateByDevice);

            return mPlayStatePausedByHostBuilder.build();

        }

        if (isAllDeviceHysteresis()) {

            return mPlayStateHysteresisBuilder.build();

            return mPlayStatePausedByHostBuilder.build();

        }

        return mIsMuted ? mPlayStatePausedByReceiverBuilder.build()

                : mPlayStatePlayingBuilder.build();

    }



    private boolean isAnyDeviceStreaming() {

        return mStateByDevice != null

                && mStateByDevice.values().stream().anyMatch(v -> v == STREAMING);

    }

        return mIsMuted ? mPlayStatePausingBuilder.build() : mPlayStatePlayingBuilder.build();



    private boolean isAnyDeviceReceiverPaused() {

        return mStateByDevice != null

                && mStateByDevice.values().stream().anyMatch(v -> v == PAUSED_BY_RECEIVER);

    }



    private boolean isAllDeviceHysteresis() {
@@ -408,8 +458,9 @@ public class AudioStreamMediaService extends Service { 
                        .setSmallIcon(com.android.settingslib.R.drawable.ic_bt_le_audio_sharing)

                        .setStyle(mediaStyle)

                        .setContentText(getString(

                                isAllDeviceHysteresis() ? BROADCAST_STREAM_PAUSED_TEXT :

                                        BROADCAST_LISTENING_NOW_TEXT))

                                (isAnyDeviceReceiverPaused() || isAllDeviceHysteresis())

                                        ? BROADCAST_STREAM_PAUSED_TEXT

                                        : BROADCAST_LISTENING_NOW_TEXT))

                        .setSilent(true);

        return notificationBuilder.build();

    }
@@ -521,6 +572,12 @@ public class AudioStreamMediaService extends Service { 
        @Override

        public void onDeviceVolumeChanged(

                @NonNull BluetoothDevice device, @IntRange(from = -255, to = 255) int volume) {

            if (Flags.audioStreamPlayPauseByModifySource()) {

                Log.d(TAG,

                        "onDeviceVolumeChanged() : skip as audioStreamPlayPauseByModifySource "

                                + "flag is on.");

                return;

            }

            if (!getDeviceInValidState().contains(device)) {

                Log.w(TAG, "onDeviceVolumeChanged() : device not in valid state list");

                return;
@@ -620,6 +677,23 @@ public class AudioStreamMediaService extends Service { 
    }



    private void handleOnPlay() {

        if (Flags.audioStreamPlayPauseByModifySource()) {

            getDeviceInValidState().forEach(device -> {

                if (mStateByDevice == null || mLocalBtManager == null) {

                    return;

                }

                var state = mStateByDevice.get(device);

                if (state != PAUSED_BY_RECEIVER

                        || !mSelectedChannelCacheByDevice.containsKey(device)) {

                    Log.d(TAG, "onPlay() skipped. Not paused or no channel cache.");

                    return;

                }

                BluetoothUtils.modifySelectedChannelIndex(

                        mLocalBtManager.getProfileManager(), device, mSourceId,

                        mSelectedChannelCacheByDevice.get(device), true);

            });

            return;

        }

        getDeviceInValidState().forEach(device -> {

            Log.d(TAG, "onPlay() setting volume for device : " + device + " volume: "

                    + mLatestPositiveVolume);
@@ -628,6 +702,23 @@ public class AudioStreamMediaService extends Service { 
    }



    private void handleOnPause() {

        if (Flags.audioStreamPlayPauseByModifySource()) {

            getDeviceInValidState().forEach(device -> {

                if (mStateByDevice == null || mLocalBtManager == null) {

                    return;

                }

                var state = mStateByDevice.get(device);

                if (state != STREAMING

                        || !mSelectedChannelCacheByDevice.containsKey(device)) {

                    Log.d(TAG, "onPause() skipped. Not streaming or no channel selected: " + state);

                    return;

                }

                BluetoothUtils.modifySelectedChannelIndex(

                        mLocalBtManager.getProfileManager(), device, mSourceId,

                        mSelectedChannelCacheByDevice.get(device), false);

            });

            return;

        }

        getDeviceInValidState().forEach(device -> {

            Log.d(TAG, "onPause() setting volume for device : " + device + " volume: " + 0);

            setDeviceVolume(device, /* volume= */ 0);

tests/robotests/src/com/android/settings/connecteddevice/audiosharing/audiostreams/AudioStreamMediaServiceTest.java

+122 −18

File changed.

Preview size limit exceeded, changes collapsed.

tests/robotests/src/com/android/settings/connecteddevice/audiosharing/audiostreams/PrivateBroadcastReceiverTest.java

+5 −2
Original line number Diff line number Diff line
@@ -54,6 +54,7 @@ import org.robolectric.annotation.Config; 
import org.robolectric.shadow.api.Shadow;

import org.robolectric.shadows.ShadowApplication;



import java.util.HashSet;

import java.util.List;



@RunWith(RobolectricTestRunner.class)
@@ -102,7 +103,8 @@ public class PrivateBroadcastReceiverTest { 
        Intent intent = new Intent(ACTION_LE_AUDIO_PRIVATE_BROADCAST_RECEIVED);

        intent.putExtra(EXTRA_PRIVATE_BROADCAST_RECEIVE_DATA,

                new PrivateBroadcastReceiveData(/* sink= */ null, /* sourceId= */

                        0, /* broadcastId= */ 0, /* programInfo= */ "", /* state= */ STREAMING));

                        0, /* broadcastId= */ 0, /* programInfo= */ "", /* state= */

                        STREAMING, /* selectedChannelIndex= */ new HashSet<>()));

        PrivateBroadcastReceiver receiver = getPrivateBroadcastReceiver(intent);

        receiver.onReceive(mContext, intent);
@@ -120,7 +122,8 @@ public class PrivateBroadcastReceiverTest { 
                        /* sourceId= */ 0,

                        /* broadcastId= */ 0,

                        /* programInfo= */ "",

                        /* state= */ STREAMING));

                        /* state= */ STREAMING,

                        /* selectedChannelIndex= */ new HashSet<>()));

        PrivateBroadcastReceiver receiver = getPrivateBroadcastReceiver(intent);

        receiver.onReceive(mContext, intent);