/*
 * 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/PictureParameter.h>
#include <aidl/android/hardware/tv/mediaquality/PictureParameters.h>
#include <aidl/android/hardware/tv/mediaquality/PictureProfile.h>
#include <aidl/android/hardware/tv/mediaquality/SoundParameter.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::ParamCapability;
using aidl::android::hardware::tv::mediaquality::PictureParameter;
using aidl::android::hardware::tv::mediaquality::PictureParameters;
using aidl::android::hardware::tv::mediaquality::PictureProfile;
using aidl::android::hardware::tv::mediaquality::SoundParameter;
using aidl::android::hardware::tv::mediaquality::SoundParameters;
using aidl::android::hardware::tv::mediaquality::SoundProfile;
using aidl::android::hardware::tv::mediaquality::VendorParamCapability;
using aidl::android::hardware::tv::mediaquality::VendorParameterIdentifier;
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();
    }

    ScopedAStatus onParamCapabilityChanged(int64_t, const std::vector<ParamCapability>&) override {
        return ScopedAStatus::ok();
    }

    ScopedAStatus onVendorParamCapabilityChanged(int64_t,
                                                 const std::vector<VendorParamCapability>&) {
        return ScopedAStatus::ok();
    }

    ScopedAStatus onRequestPictureParameters(int64_t) { 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();
    }

    ScopedAStatus onParamCapabilityChanged(int64_t, const std::vector<ParamCapability>&) override {
        return ScopedAStatus::ok();
    }

    ScopedAStatus onVendorParamCapabilityChanged(int64_t,
                                                 const std::vector<VendorParamCapability>&) {
        return ScopedAStatus::ok();
    }

    ScopedAStatus onRequestSoundParameters(int64_t) { 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, TestSendDefaultPictureParameters) {
    PictureParameters pictureParameters;
    std::vector<PictureParameter> picParams;

    PictureParameter brightnessParam;
    brightnessParam.set<PictureParameter::Tag::brightness>(0.5f);
    picParams.push_back(brightnessParam);

    PictureParameter contrastParam;
    contrastParam.set<PictureParameter::Tag::contrast>(50);
    picParams.push_back(contrastParam);

    pictureParameters.pictureParameters = picParams;
    ASSERT_OK(mediaquality->sendDefaultPictureParameters(pictureParameters));
}

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, TestSendDefaultSoundParameters) {
    SoundParameters soundParameters;
    std::vector<SoundParameter> soundParams;

    SoundParameter balanceParam;
    balanceParam.set<SoundParameter::Tag::balance>(50);
    soundParams.push_back(balanceParam);

    SoundParameter bassParam;
    bassParam.set<SoundParameter::Tag::bass>(50);
    soundParams.push_back(bassParam);

    soundParameters.soundParameters = soundParams;
    ASSERT_OK(mediaquality->sendDefaultSoundParameters(soundParameters));
}

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));
}

TEST_P(MediaQualityAidl, TestIsAutoPqSupported) {
    bool supported;
    ASSERT_OK(mediaquality->isAutoPqSupported(&supported));
}

TEST_P(MediaQualityAidl, TestGetAutoPqEnabled) {
    bool enabled;
    ASSERT_OK(mediaquality->getAutoPqEnabled(&enabled));
}

TEST_P(MediaQualityAidl, TestSetAutoPqEnabled) {
    ASSERT_OK(mediaquality->setAutoPqEnabled(true));
}

TEST_P(MediaQualityAidl, TestIsAutoSrSupported) {
    bool supported;
    ASSERT_OK(mediaquality->isAutoSrSupported(&supported));
}

TEST_P(MediaQualityAidl, TestGetAutoSrEnabled) {
    bool enabled;
    ASSERT_OK(mediaquality->getAutoSrEnabled(&enabled));
}

TEST_P(MediaQualityAidl, TestSetAutoSrEnabled) {
    ASSERT_OK(mediaquality->setAutoSrEnabled(true));
}

TEST_P(MediaQualityAidl, TestIsAutoAqSupported) {
    bool supported;
    ASSERT_OK(mediaquality->isAutoAqSupported(&supported));
}

TEST_P(MediaQualityAidl, TestGetAutoAqEnabled) {
    bool enabled;
    ASSERT_OK(mediaquality->getAutoAqEnabled(&enabled));
}

TEST_P(MediaQualityAidl, TestSetAutoAqEnabled) {
    ASSERT_OK(mediaquality->setAutoAqEnabled(true));
}

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();
}