tv/mediaquality/aidl/vts/functional/VtsHalMediaQualityTest.cpp

/*
 * Copyright (C) 2024 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.
 */
#define LOG_TAG "ambient_backlight_aidl_hal_test"

#include <aidl/Gtest.h>
#include <aidl/Vintf.h>
#include <aidl/android/hardware/tv/mediaquality/AmbientBacklightEvent.h>
#include <aidl/android/hardware/tv/mediaquality/AmbientBacklightSettings.h>
#include <aidl/android/hardware/tv/mediaquality/BnMediaQualityCallback.h>
#include <aidl/android/hardware/tv/mediaquality/BnPictureProfileAdjustmentListener.h>
#include <aidl/android/hardware/tv/mediaquality/BnSoundProfileAdjustmentListener.h>
#include <aidl/android/hardware/tv/mediaquality/IMediaQuality.h>
#include <aidl/android/hardware/tv/mediaquality/PictureParameters.h>
#include <aidl/android/hardware/tv/mediaquality/PictureProfile.h>
#include <aidl/android/hardware/tv/mediaquality/SoundParameters.h>
#include <aidl/android/hardware/tv/mediaquality/SoundProfile.h>

#include <android/binder_auto_utils.h>
#include <android/binder_manager.h>
#include <binder/IServiceManager.h>
#include <binder/ProcessState.h>
#include <future>

using aidl::android::hardware::graphics::common::PixelFormat;
using aidl::android::hardware::tv::mediaquality::AmbientBacklightEvent;
using aidl::android::hardware::tv::mediaquality::AmbientBacklightSettings;
using aidl::android::hardware::tv::mediaquality::AmbientBacklightSource;
using aidl::android::hardware::tv::mediaquality::BnMediaQualityCallback;
using aidl::android::hardware::tv::mediaquality::BnPictureProfileAdjustmentListener;
using aidl::android::hardware::tv::mediaquality::BnSoundProfileAdjustmentListener;
using aidl::android::hardware::tv::mediaquality::IMediaQuality;
using aidl::android::hardware::tv::mediaquality::PictureParameters;
using aidl::android::hardware::tv::mediaquality::PictureProfile;
using aidl::android::hardware::tv::mediaquality::SoundParameters;
using aidl::android::hardware::tv::mediaquality::SoundProfile;
using android::ProcessState;
using android::String16;
using ndk::ScopedAStatus;
using ndk::SpAIBinder;

#define ASSERT_OK(ret) ASSERT_TRUE(ret.isOk())
#define EXPECT_OK(ret) EXPECT_TRUE(ret.isOk())

class MediaQualityCallback : public BnMediaQualityCallback {
  public:
    explicit MediaQualityCallback(
            const std::function<void(const AmbientBacklightEvent& event)>& on_hal_event_cb)
        : on_hal_event_cb_(on_hal_event_cb) {}
    ScopedAStatus notifyAmbientBacklightEvent(const AmbientBacklightEvent& event) override {
        on_hal_event_cb_(event);
        return ScopedAStatus::ok();
    }

  private:
    std::function<void(const AmbientBacklightEvent& event)> on_hal_event_cb_;
};

class PictureProfileAdjustmentListener : public BnPictureProfileAdjustmentListener {
  public:
    explicit PictureProfileAdjustmentListener(
            const std::function<void(const PictureProfile& pictureProfile)>&
                    on_hal_picture_profile_adjust)
        : on_hal_picture_profile_adjust_(on_hal_picture_profile_adjust) {}
    ScopedAStatus onPictureProfileAdjusted(const PictureProfile& pictureProfile) override {
        on_hal_picture_profile_adjust_(pictureProfile);
        return ScopedAStatus::ok();
    }

  private:
    std::function<void(const PictureProfile& pictureProfile)> on_hal_picture_profile_adjust_;
};

class SoundProfileAdjustmentListener : public BnSoundProfileAdjustmentListener {
  public:
    explicit SoundProfileAdjustmentListener(
            const std::function<void(const SoundProfile& soundProfile)>&
                    on_hal_sound_profile_adjust)
        : on_hal_sound_profile_adjust_(on_hal_sound_profile_adjust) {}
    ScopedAStatus onSoundProfileAdjusted(const SoundProfile& soundProfile) override {
        on_hal_sound_profile_adjust_(soundProfile);
        return ScopedAStatus::ok();
    }

  private:
    std::function<void(const SoundProfile& soundProfile)> on_hal_sound_profile_adjust_;
};

class MediaQualityAidl : public testing::TestWithParam<std::string> {
  public:
    virtual void SetUp() override {
        mediaquality = IMediaQuality::fromBinder(
                SpAIBinder(AServiceManager_waitForService(GetParam().c_str())));
        ASSERT_NE(mediaquality, nullptr);
    }
    std::shared_ptr<IMediaQuality> mediaquality;
};

TEST_P(MediaQualityAidl, TestSetAmbientBacklightDetectionEnabled) {
    std::promise<void> open_cb_promise;
    std::future<void> open_cb_future{open_cb_promise.get_future()};
    std::shared_ptr<MediaQualityCallback> callback =
            ndk::SharedRefBase::make<MediaQualityCallback>([&open_cb_promise](auto event) {
                EXPECT_EQ(event.getTag(), AmbientBacklightEvent::Tag::enabled);
                EXPECT_EQ(event.template get<AmbientBacklightEvent::Tag::enabled>(), true);
                open_cb_promise.set_value();
                return ScopedAStatus::ok();
            });
    ASSERT_OK(mediaquality->setCallback(callback));
    ASSERT_OK(mediaquality->setAmbientBacklightDetectionEnabled(true));
    std::chrono::milliseconds timeout{10000};
    EXPECT_EQ(open_cb_future.wait_for(timeout), std::future_status::ready);
}

TEST_P(MediaQualityAidl, TestGetAmbientBacklightDetectionEnabled) {
    bool enabled;
    ASSERT_OK(mediaquality->getAmbientBacklightDetectionEnabled(&enabled));
}

TEST_P(MediaQualityAidl, TestSetMediaQualityCallback) {
    std::shared_ptr<MediaQualityCallback> callback = ndk::SharedRefBase::make<MediaQualityCallback>(
            [](auto /* event */) { return ScopedAStatus::ok(); });
    ASSERT_OK(mediaquality->setCallback(callback));
}

TEST_P(MediaQualityAidl, TestSetPictureProfileAdjustmentListener) {
    std::shared_ptr<PictureProfileAdjustmentListener> listener =
            ndk::SharedRefBase::make<PictureProfileAdjustmentListener>(
                    [](auto /*picture profile*/) { return ScopedAStatus::ok(); });
    ASSERT_OK(mediaquality->setPictureProfileAdjustmentListener(listener));
}

TEST_P(MediaQualityAidl, TestGetPictureParameters) {
    PictureParameters pictureParams;
    auto result = mediaquality->getPictureParameters(1, &pictureParams);
    ASSERT_TRUE(result.isOk());
    ASSERT_EQ(pictureParams.pictureParameters.size(), 2);
}

TEST_P(MediaQualityAidl, TestSetSoundProfileAdjustmentListener) {
    std::shared_ptr<SoundProfileAdjustmentListener> listener =
            ndk::SharedRefBase::make<SoundProfileAdjustmentListener>(
                    [](auto /*sound profile*/) { return ScopedAStatus::ok(); });
    ASSERT_OK(mediaquality->setSoundProfileAdjustmentListener(listener));
}

TEST_P(MediaQualityAidl, TestGetSoundParameters) {
    SoundParameters soundParams;
    auto result = mediaquality->getSoundParameters(1, &soundParams);
    ASSERT_TRUE(result.isOk());
    ASSERT_EQ(soundParams.soundParameters.size(), 2);
}

TEST_P(MediaQualityAidl, TestSetAmbientBacklightDetector) {
    AmbientBacklightSettings in_settings = {
            .packageName = "com.android.mediaquality",
            .source = AmbientBacklightSource::VIDEO,
            .colorFormat = PixelFormat::RGB_888,
            .hZonesNumber = 32,
            .vZonesNumber = 20,
            .hasLetterbox = true,
            .threshold = 0,
    };
    ASSERT_OK(mediaquality->setAmbientBacklightDetector(in_settings));
}

GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(MediaQualityAidl);

INSTANTIATE_TEST_SUITE_P(
        PerInstance, MediaQualityAidl,
        testing::ValuesIn(android::getAidlHalInstanceNames(IMediaQuality::descriptor)),
        android::PrintInstanceNameToString);

int main(int argc, char** argv) {
    ::testing::InitGoogleTest(&argc, argv);
    ProcessState::self()->setThreadPoolMaxThreadCount(1);
    ProcessState::self()->startThreadPool();
    return RUN_ALL_TESTS();
}