#!/usr/bin/env python3.4
#
#   Copyright 2018 - The Android Open Source Project
#
#   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.

import time
import acts.test_utils.bt.bt_test_utils as btutils
import acts.test_utils.power.PowerBTBaseTest as PBtBT
from acts import asserts
from acts.test_utils.bt import BtEnum

EXTRA_PLAY_TIME = 10


class PowerBTa2dpTest(PBtBT.PowerBTBaseTest):
    def __init__(self, configs):
        super().__init__(configs)
        req_params = ['codecs', 'tx_power_levels', 'atten_pl_settings']
        self.unpack_userparams(req_params)
        # Loop all codecs and tx power levels
        for codec_config in self.codecs:
            for tpl in self.tx_power_levels:
                self.generate_test_case(codec_config, tpl)

    def setup_test(self):
        super().setup_test()
        btutils.connect_phone_to_headset(self.dut, self.bt_device, 60)
        vol = self.dut.droid.getMaxMediaVolume() * self.volume
        self.dut.droid.setMediaVolume(0)
        time.sleep(1)
        self.dut.droid.setMediaVolume(int(vol))

    def generate_test_case(self, codec_config, tpl):
        def test_case_fn():
            self.measure_a2dp_power(codec_config, tpl)

        test_case_name = ('test_BTa2dp_{}_codec_at_PL{}'.format(
            codec_config['codec_type'], tpl))
        setattr(self, test_case_name, test_case_fn)

    def measure_a2dp_power(self, codec_config, tpl):

        current_codec = self.dut.droid.bluetoothA2dpGetCurrentCodecConfig()
        current_codec_type = BtEnum.BluetoothA2dpCodecType(
            current_codec['codecType']).name
        if current_codec_type != codec_config['codec_type']:
            codec_set = btutils.set_bluetooth_codec(self.dut, **codec_config)
            asserts.assert_true(codec_set, 'Codec configuration failed.')
        else:
            self.log.info('Current Codec is {}, no need to change'.format(
                current_codec_type))

        # Set attenuation so BT tx at desired power level
        self.log.info('Current Attenuation {} dB'.format(
            self.attenuator.get_atten()))
        tpl = 'PL' + str(tpl)
        PBtBT.ramp_attenuation(self.attenuator, self.atten_pl_settings[tpl][0])
        self.log.info('Setting Attenuator to {} dB'.format(
            self.atten_pl_settings[tpl][0]))

        self.media.play()
        self.log.info('Running A2DP with codec {} at {}'.format(
            codec_config['codec_type'], tpl))
        self.dut.droid.goToSleepNow()
        time.sleep(EXTRA_PLAY_TIME)
        self.measure_power_and_validate()