/*
 * Copyright 2020 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.
 */

#undef LOG_TAG
#define LOG_TAG "LayerHistory"
#define ATRACE_TAG ATRACE_TAG_GRAPHICS

#include "LayerHistory.h"

#ifdef MTK_SF_MSYNC_3
#include <android-base/properties.h>
#endif
#include <android-base/stringprintf.h>
#include <cutils/properties.h>
#include <gui/TraceUtils.h>
#include <utils/Log.h>
#include <utils/Timers.h>

#include <algorithm>
#include <cmath>
#include <string>
#include <utility>

#include "../Layer.h"
#include "EventThread.h"
#include "LayerInfo.h"

namespace android::scheduler {

namespace {

#ifdef MTK_SF_MSYNC_3
static bool sShowContentDetectionDetailLog = false;
static bool isSupportContentDetection() {
    static bool enable = false;
    static bool read = false;
    if (!read) {
        enable = android::base::GetBoolProperty("debug.sf.use_content_detection_for_refresh_rate", false);
        sShowContentDetectionDetailLog = android::base::GetBoolProperty("debug.sf.show_content_detection_detail_log", false);
        read = true;
    }
    return enable;
}
static bool isShowCDDetailLog() {
    return sShowContentDetectionDetailLog;
}
static int init __attribute((unused)) = (isSupportContentDetection(), 0);
#endif

#ifdef MTK_SF_HINT_LOW_POWER
static bool isSupportHintLowPower() {
    static bool enable = false;
    static bool read = false;
    if (!read) {
        enable = android::base::GetBoolProperty("debug.sf.hint_low_power", false);
        read = true;
    }
    return enable;
}
#endif

bool isLayerActive(const LayerInfo& info, nsecs_t threshold) {
#ifdef MTK_SF_MSYNC_3
    // Layers with an explicit vote are not always kept active
    if (!isSupportContentDetection() && info.getSetFrameRateVote().rate.isValid()) {
#else
    // Layers with an explicit vote are always kept active
    if (info.getSetFrameRateVote().rate.isValid()) {
#endif
        return true;
    }

    return info.isVisible() && info.getLastUpdatedTime() >= threshold;
}

bool traceEnabled() {
    return property_get_bool("debug.sf.layer_history_trace", false);
}

bool useFrameRatePriority() {
    char value[PROPERTY_VALUE_MAX];
    property_get("debug.sf.use_frame_rate_priority", value, "1");
    return atoi(value);
}

void trace(const LayerInfo& info, LayerHistory::LayerVoteType type, int fps) {
    const auto traceType = [&](LayerHistory::LayerVoteType checkedType, int value) {
        ATRACE_INT(info.getTraceTag(checkedType), type == checkedType ? value : 0);
    };

    traceType(LayerHistory::LayerVoteType::NoVote, 1);
    traceType(LayerHistory::LayerVoteType::Heuristic, fps);
    traceType(LayerHistory::LayerVoteType::ExplicitDefault, fps);
    traceType(LayerHistory::LayerVoteType::ExplicitExactOrMultiple, fps);
    traceType(LayerHistory::LayerVoteType::ExplicitExact, fps);
    traceType(LayerHistory::LayerVoteType::Min, 1);
    traceType(LayerHistory::LayerVoteType::Max, 1);

    ALOGD("%s: %s @ %d Hz", __FUNCTION__, info.getName().c_str(), fps);
}

LayerHistory::LayerVoteType getVoteType(LayerInfo::FrameRateCompatibility compatibility,
                                        bool contentDetectionEnabled) {
    LayerHistory::LayerVoteType voteType;
    if (!contentDetectionEnabled || compatibility == LayerInfo::FrameRateCompatibility::NoVote) {
        voteType = LayerHistory::LayerVoteType::NoVote;
    } else if (compatibility == LayerInfo::FrameRateCompatibility::Min) {
        voteType = LayerHistory::LayerVoteType::Min;
    } else {
        voteType = LayerHistory::LayerVoteType::Heuristic;
    }
    return voteType;
}

} // namespace

LayerHistory::LayerHistory()
      : mTraceEnabled(traceEnabled()), mUseFrameRatePriority(useFrameRatePriority()) {
    LayerInfo::setTraceEnabled(mTraceEnabled);
}

LayerHistory::~LayerHistory() = default;

void LayerHistory::registerLayer(Layer* layer, bool contentDetectionEnabled) {
    std::lock_guard lock(mLock);
    LOG_ALWAYS_FATAL_IF(findLayer(layer->getSequence()).first != LayerStatus::NotFound,
                        "%s already registered", layer->getName().c_str());
    LayerVoteType type =
            getVoteType(layer->getDefaultFrameRateCompatibility(), contentDetectionEnabled);
#ifdef MTK_SF_MSYNC_3
    if (isSupportContentDetection()) {
        if (layer->getName().find("NavigationBar") != std::string::npos
                || layer->getWindowType() == gui::WindowInfo::Type::STATUS_BAR
                || layer->getName().find("RefreshRateOverlay") != std::string::npos
                || layer->getName().find("StageCoordinatorSplitDivider") != std::string::npos) {
            type = scheduler::LayerHistory::LayerVoteType::NoVote;
        } else if (layer->getWindowType() == gui::WindowInfo::Type::WALLPAPER) {
            // For live wallpaper
            type = scheduler::LayerHistory::LayerVoteType::Heuristic;
        }
    }
#endif
#ifdef MTK_SF_HINT_LOW_POWER
    if (!contentDetectionEnabled && (isSupportHintLowPower() || mIsPELT32Enabled)) {
        //ALOGI("%s: naem=%s", __func__, layer->getDebugName());
        if (layer->getName().find("NavigationBar") != std::string::npos
                || layer->getWindowType() == gui::WindowInfo::Type::STATUS_BAR
                || layer->getName().find("RefreshRateOverlay") != std::string::npos
                || layer->getName().find("StageCoordinatorSplitDivider") != std::string::npos) {
            type = scheduler::LayerHistory::LayerVoteType::NoVote;
        } else {
            type = scheduler::LayerHistory::LayerVoteType::Heuristic;
        }
    }
#endif
    auto info = std::make_unique<LayerInfo>(layer->getName(), layer->getOwnerUid(), type);

    // The layer can be placed on either map, it is assumed that partitionLayers() will be called
    // to correct them.
    mInactiveLayerInfos.insert({layer->getSequence(), std::make_pair(layer, std::move(info))});
}

void LayerHistory::deregisterLayer(Layer* layer) {
    std::lock_guard lock(mLock);
    if (!mActiveLayerInfos.erase(layer->getSequence())) {
        if (!mInactiveLayerInfos.erase(layer->getSequence())) {
            LOG_ALWAYS_FATAL("%s: unknown layer %p", __FUNCTION__, layer);
        }
    }
}

void LayerHistory::record(int32_t id, const LayerProps& layerProps, nsecs_t presentTime,
                          nsecs_t now, LayerUpdateType updateType) {
    std::lock_guard lock(mLock);
    auto [found, layerPair] = findLayer(id);
    if (found == LayerStatus::NotFound) {
        // Offscreen layer
        ALOGV("%s: %d not registered", __func__, id);
        return;
    }

    const auto& info = layerPair->second;
    info->setLastPresentTime(presentTime, now, updateType, mModeChangePending, layerProps);

    // Set frame rate to attached choreographer.
    // TODO(b/260898223): Change to use layer hierarchy and handle frame rate vote.
    if (updateType == LayerUpdateType::SetFrameRate) {
        auto range = mAttachedChoreographers.equal_range(id);
        auto it = range.first;
        while (it != range.second) {
            sp<EventThreadConnection> choreographerConnection = it->second.promote();
            if (choreographerConnection) {
                choreographerConnection->frameRate = layerProps.setFrameRateVote.rate;
                it++;
            } else {
                it = mAttachedChoreographers.erase(it);
            }
        }
    }

    // Activate layer if inactive.
    if (found == LayerStatus::LayerInInactiveMap) {
        mActiveLayerInfos.insert(
                {id, std::make_pair(layerPair->first, std::move(layerPair->second))});
        mInactiveLayerInfos.erase(id);
    }
}

void LayerHistory::setDefaultFrameRateCompatibility(Layer* layer, bool contentDetectionEnabled) {
    std::lock_guard lock(mLock);
    auto id = layer->getSequence();

    auto [found, layerPair] = findLayer(id);
    if (found == LayerStatus::NotFound) {
        // Offscreen layer
        ALOGV("%s: %s not registered", __func__, layer->getName().c_str());
        return;
    }

    const auto& info = layerPair->second;
    info->setDefaultLayerVote(
            getVoteType(layer->getDefaultFrameRateCompatibility(), contentDetectionEnabled));
}

auto LayerHistory::summarize(const RefreshRateSelector& selector, nsecs_t now) -> Summary {
    ATRACE_CALL();
    Summary summary;

    std::lock_guard lock(mLock);

    partitionLayers(now);

    for (const auto& [key, value] : mActiveLayerInfos) {
        auto& info = value.second;
        const auto frameRateSelectionPriority = info->getFrameRateSelectionPriority();
        const auto layerFocused = Layer::isLayerFocusedBasedOnPriority(frameRateSelectionPriority);
#ifdef MTK_SF_MSYNC_3
        if (isShowCDDetailLog()) {
            ALOGI("%s: active layer, name=%s, priority=%d, focused=%d", __func__, info->getName().c_str(), frameRateSelectionPriority,
                layerFocused);
        }
#else
        ALOGV("%s has priority: %d %s focused", info->getName().c_str(), frameRateSelectionPriority,
              layerFocused ? "" : "not");
#endif

        ATRACE_FORMAT("%s", info->getName().c_str());
        const auto vote = info->getRefreshRateVote(selector, now);
        // Skip NoVote layer as those don't have any requirements
        if (vote.type == LayerVoteType::NoVote) {
            continue;
        }

        // Compute the layer's position on the screen
        const Rect bounds = Rect(info->getBounds());
        const ui::Transform transform = info->getTransform();
        constexpr bool roundOutwards = true;
        Rect transformed = transform.transform(bounds, roundOutwards);

        const float layerArea = transformed.getWidth() * transformed.getHeight();
        float weight = mDisplayArea ? layerArea / mDisplayArea : 0.0f;
        ATRACE_FORMAT_INSTANT("%s %s (%d%)", ftl::enum_string(vote.type).c_str(),
                              to_string(vote.fps).c_str(), weight * 100);
#ifdef MTK_SF_MSYNC_3
        auto& layer = value.first;
        if (isShowCDDetailLog()) {
            ALOGI("%s: has vote, name=%s, isProtected=%d, weight=%.2f, mDisplayArea=%d, layerArea=%.2f, fps=%d, vote.type=%d", __func__,
                info->getName().c_str(), layer->isProtected(), weight, mDisplayArea, layerArea, vote.fps.getIntValue(), vote.type);
        }
        Fps tempFps = vote.fps;
        if ((vote.type == LayerHistory::LayerVoteType::ExplicitDefault
                || vote.type == LayerHistory::LayerVoteType::ExplicitExactOrMultiple)
                && vote.fps.getIntValue() <= 60) {
            bool hasOtherUI = false;
            std::string fpsPkgName = getPkgName(info->getName());
            if (layerFocused && !fpsPkgName.empty()) {
                for (const auto& [key2, value2] : mActiveLayerInfos) {
                    auto& info2 = value2.second;
                    std::string otherPkgName = getPkgName(info2->getName());
                    if (!otherPkgName.empty() && otherPkgName.find(fpsPkgName) == std::string::npos) {
                        hasOtherUI = true;
                        break;
                    }
                }
            }
            if (!layerFocused || hasOtherUI) {
                tempFps = 120_Hz;
                if (isShowCDDetailLog()) {
                    ALOGI("%s: layerFocused=%d, hasOtherUI=%d, name=%s, promote %d as %d", __func__,
                        layerFocused, hasOtherUI, info->getName().c_str(), vote.fps.getIntValue(), tempFps.getIntValue());
                }
            }
        }
        summary.push_back({info->getName(), info->getOwnerUid(), vote.type, tempFps,
                           vote.seamlessness, weight, layerFocused, layer->isProtected()});
#else
        summary.push_back({info->getName(), info->getOwnerUid(), vote.type, vote.fps,
                           vote.seamlessness, weight, layerFocused});
#endif

        if (CC_UNLIKELY(mTraceEnabled)) {
            trace(*info, vote.type, vote.fps.getIntValue());
        }
    }

    return summary;
}

void LayerHistory::partitionLayers(nsecs_t now) {
    ATRACE_CALL();
    const nsecs_t threshold = getActiveLayerThreshold(now);

    // iterate over inactive map
    LayerInfos::iterator it = mInactiveLayerInfos.begin();
    while (it != mInactiveLayerInfos.end()) {
        auto& [layerUnsafe, info] = it->second;
        if (isLayerActive(*info, threshold)) {
            // move this to the active map

            mActiveLayerInfos.insert({it->first, std::move(it->second)});
            it = mInactiveLayerInfos.erase(it);
        } else {
            if (CC_UNLIKELY(mTraceEnabled)) {
                trace(*info, LayerVoteType::NoVote, 0);
            }
            info->onLayerInactive(now);
            it++;
        }
    }

    // iterate over active map
    it = mActiveLayerInfos.begin();
    while (it != mActiveLayerInfos.end()) {
        auto& [layerUnsafe, info] = it->second;
        if (isLayerActive(*info, threshold)) {
            // Set layer vote if set
            const auto frameRate = info->getSetFrameRateVote();
            const auto voteType = [&]() {
                switch (frameRate.type) {
                    case Layer::FrameRateCompatibility::Default:
                        return LayerVoteType::ExplicitDefault;
                    case Layer::FrameRateCompatibility::Min:
                        return LayerVoteType::Min;
                    case Layer::FrameRateCompatibility::ExactOrMultiple:
                        return LayerVoteType::ExplicitExactOrMultiple;
                    case Layer::FrameRateCompatibility::NoVote:
                        return LayerVoteType::NoVote;
                    case Layer::FrameRateCompatibility::Exact:
                        return LayerVoteType::ExplicitExact;
                }
            }();

            if (frameRate.rate.isValid() || voteType == LayerVoteType::NoVote) {
                const auto type = info->isVisible() ? voteType : LayerVoteType::NoVote;
                info->setLayerVote({type, frameRate.rate, frameRate.seamlessness});
            } else {
                info->resetLayerVote();
            }
#ifdef MTK_SF_MSYNC_3
            {
                if (frameRate.rate.getIntValue() == REFRESH_RATE_SET_DEFAULT
                        || frameRate.rate.getIntValue() == REFRESH_RATE_CONTROL_BY_AP
                        || frameRate.rate.getIntValue() > REFRESH_RATE_SET_DEFAULT_BY_AP) {
                    if (isShowCDDetailLog()) {
                        ALOGI("%s: %s has specific fps, setLayerVote as Heuristic with fps=%d",
                                __func__, info->getName().c_str(), frameRate.rate.getIntValue());
                    }
                    info->setLayerVote({LayerVoteType::Heuristic, frameRate.rate, frameRate.seamlessness});
                }
            }
#endif

            it++;
        } else {
            if (CC_UNLIKELY(mTraceEnabled)) {
                trace(*info, LayerVoteType::NoVote, 0);
            }
            info->onLayerInactive(now);
            // move this to the inactive map
            mInactiveLayerInfos.insert({it->first, std::move(it->second)});
            it = mActiveLayerInfos.erase(it);
        }
    }
}

void LayerHistory::clear() {
    std::lock_guard lock(mLock);
    for (const auto& [key, value] : mActiveLayerInfos) {
        value.second->clearHistory(systemTime());
    }
}

std::string LayerHistory::dump() const {
    std::lock_guard lock(mLock);
    return base::StringPrintf("{size=%zu, active=%zu}",
                              mActiveLayerInfos.size() + mInactiveLayerInfos.size(),
                              mActiveLayerInfos.size());
}

float LayerHistory::getLayerFramerate(nsecs_t now, int32_t id) const {
    std::lock_guard lock(mLock);
    auto [found, layerPair] = findLayer(id);
    if (found != LayerStatus::NotFound) {
        return layerPair->second->getFps(now).getValue();
    }
    return 0.f;
}

void LayerHistory::attachChoreographer(int32_t layerId,
                                       const sp<EventThreadConnection>& choreographerConnection) {
    std::lock_guard lock(mLock);
    mAttachedChoreographers.insert({layerId, wp<EventThreadConnection>(choreographerConnection)});
}

auto LayerHistory::findLayer(int32_t id) -> std::pair<LayerStatus, LayerPair*> {
    // the layer could be in either the active or inactive map, try both
    auto it = mActiveLayerInfos.find(id);
    if (it != mActiveLayerInfos.end()) {
        return {LayerStatus::LayerInActiveMap, &(it->second)};
    }
    it = mInactiveLayerInfos.find(id);
    if (it != mInactiveLayerInfos.end()) {
        return {LayerStatus::LayerInInactiveMap, &(it->second)};
    }
    return {LayerStatus::NotFound, nullptr};
}

#ifdef MTK_SF_HINT_LOW_POWER
void LayerHistory::setPELT32(bool enabled) {
    std::lock_guard lock(mLock);
    mIsPELT32Enabled = enabled;
}
#endif

#ifdef MTK_SF_MSYNC_3
std::string LayerHistory::getPkgName(const std::string& layerName) {
    std::size_t startPos = layerName.find('[');
    if (startPos != std::string::npos) {
        startPos++;
    } else {
        startPos = 0;
    }
    std::size_t len = layerName.find('/');
    if (len != std::string::npos) {
        len = len - startPos;
        return layerName.substr(startPos, len);
    }
    return "";
}
#endif

} // namespace android::scheduler