OpenHarmony-v5.0.1-Release/s

/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * 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.
 */

#include "scene_session_dirty_manager.h"

#include <parameters.h>
#include "screen_session_manager/include/screen_session_manager_client.h"
#include "session_manager/include/scene_session_manager.h"
#include "window_helper.h"
#include "fold_screen_state_internel.h"

namespace OHOS::Rosen {
namespace {
constexpr float DIRECTION0 = 0 ;
constexpr float DIRECTION90 = 90 ;
constexpr float DIRECTION180 = 180 ;
constexpr float DIRECTION270 = 270 ;
constexpr HiviewDFX::HiLogLabel LABEL = {LOG_CORE, HILOG_DOMAIN_WINDOW, "SceneSessionDirtyManager"};
constexpr unsigned int POINTER_CHANGE_AREA_COUNT = 8;
constexpr int POINTER_CHANGE_AREA_SIXTEEN = 16;
constexpr int POINTER_CHANGE_AREA_DEFAULT = 0;
constexpr int POINTER_CHANGE_AREA_FIVE = 5;
constexpr unsigned int TRANSFORM_DATA_LEN = 9;
constexpr int UPDATE_TASK_DURATION = 10;
const std::string UPDATE_WINDOW_INFO_TASK = "UpdateWindowInfoTask";
static int32_t g_screenRotationOffset = system::GetIntParameter<int32_t>("const.fold.screen_rotation.offset", 0);
constexpr float ZORDER_UIEXTENSION_INDEX = 0.1;
} //namespace

static bool operator==(const MMI::Rect left, const MMI::Rect right)
{
    return ((left.x == right.x) && (left.y == right.y) && (left.width == right.width) && (left.height == right.height));
}

MMI::Direction ConvertDegreeToMMIRotation(float degree, MMI::DisplayMode displayMode)
{
    MMI::Direction rotation = MMI::DIRECTION0;
    if (NearEqual(degree, DIRECTION0)) {
        rotation = MMI::DIRECTION0;
    }
    if (NearEqual(degree, DIRECTION90)) {
        rotation = MMI::DIRECTION90;
    }
    if (NearEqual(degree, DIRECTION180)) {
        rotation = MMI::DIRECTION180;
    }
    if (NearEqual(degree, DIRECTION270)) {
        rotation = MMI::DIRECTION270;
    }
    if ((displayMode == MMI::DisplayMode::FULL && g_screenRotationOffset != 0) ||
        (displayMode == MMI::DisplayMode::MAIN && FoldScreenStateInternel::IsSingleDisplayPocketFoldDevice())) {
        switch (rotation) {
            case MMI::DIRECTION0:
                rotation = MMI::DIRECTION90;
                break;
            case MMI::DIRECTION90:
                rotation = MMI::DIRECTION180;
                break;
            case MMI::DIRECTION180:
                rotation = MMI::DIRECTION270;
                break;
            case MMI::DIRECTION270:
                rotation = MMI::DIRECTION0;
                break;
            default:
                rotation = MMI::DIRECTION0;
                break;
        }
    }
    return rotation;
}

bool CmpMMIWindowInfo(const MMI::WindowInfo& a, const MMI::WindowInfo& b)
{
    return a.defaultHotAreas.size() > b.defaultHotAreas.size();
}

Vector2f CalRotationToTranslate(const MMI::Direction& displayRotation, float width, float height,
    const Vector2f& offset, float& rotate)
{
    Vector2f translate = offset;
    switch (displayRotation) {
        case MMI::DIRECTION0: {
            break;
        }
        case MMI::DIRECTION270: {
            translate.x_ = offset.y_;
            translate.y_ = height - offset.x_;
            rotate = -M_PI_2;
            break;
        }
        case MMI::DIRECTION180:
            translate.x_ = width - offset.x_;
            translate.y_ = height - offset.y_;
            rotate = M_PI;
            break;
        case MMI::DIRECTION90: {
            translate.x_ = width - offset.y_;
            translate.y_ = offset.x_;
            rotate = M_PI_2;
            break;
        }
        default:
            break;
    }
    return translate;
}

void SceneSessionDirtyManager::CalNotRotateTransform(const sptr<SceneSession>& sceneSession, Matrix3f& transform,
    bool useUIExtension) const
{
    if (sceneSession == nullptr) {
        WLOGFE("sceneSession is nullptr");
        return;
    }
    auto sessionProperty = sceneSession->GetSessionProperty();
    if (sessionProperty == nullptr) {
        WLOGFE("sessionProperty is nullptr");
        return;
    }
    auto displayId = sessionProperty->GetDisplayId();
    auto displayMode = Rosen::ScreenSessionManagerClient::GetInstance().GetFoldDisplayMode();
    std::map<ScreenId, ScreenProperty> screensProperties =
        Rosen::ScreenSessionManagerClient::GetInstance().GetAllScreensProperties();
    if (screensProperties.find(displayId) == screensProperties.end()) {
        return;
    }
    auto screenProperty = screensProperties[displayId];
    MMI::Direction displayRotation = ConvertDegreeToMMIRotation(screenProperty.GetRotation(),
        static_cast<MMI::DisplayMode>(displayMode));
    float width = screenProperty.GetBounds().rect_.GetWidth();
    float height = screenProperty.GetBounds().rect_.GetHeight();
    Vector2f scale(sceneSession->GetScaleX(), sceneSession->GetScaleY());
    Vector2f offset = sceneSession->GetSessionGlobalPosition(useUIExtension);
    float rotate = 0.0f;
    Vector2f translate = CalRotationToTranslate(displayRotation, width, height, offset, rotate);
    transform = transform.Translate(translate).Rotate(rotate).Scale(scale, sceneSession->GetPivotX(),
        sceneSession->GetPivotY());
    transform = transform.Inverse();
}

void SceneSessionDirtyManager::CalTransform(const sptr<SceneSession>& sceneSession, Matrix3f& transform,
    bool useUIExtension) const
{
    if (sceneSession == nullptr) {
        WLOGFE("sceneSession is nullptr");
        return;
    }
    transform = Matrix3f::IDENTITY;
    bool isRotate = sceneSession->GetSessionInfo().isRotable_;
    auto displayMode = Rosen::ScreenSessionManagerClient::GetInstance().GetFoldDisplayMode();
    if (isRotate || !sceneSession->GetSessionInfo().isSystem_ ||
        static_cast<MMI::DisplayMode>(displayMode) == MMI::DisplayMode::FULL ||
        (static_cast<MMI::DisplayMode>(displayMode) == MMI::DisplayMode::MAIN &&
         FoldScreenStateInternel::IsSingleDisplayPocketFoldDevice())) {
        Vector2f scale(sceneSession->GetScaleX(), sceneSession->GetScaleY());
        Vector2f translate = sceneSession->GetSessionGlobalPosition(useUIExtension);
        transform = transform.Translate(translate);
        transform = transform.Scale(scale, sceneSession->GetPivotX(), sceneSession->GetPivotY());
        transform = transform.Inverse();
        return;
    }
    CalNotRotateTransform(sceneSession, transform, useUIExtension);
}


void SceneSessionDirtyManager::UpdateDefaultHotAreas(sptr<SceneSession> sceneSession,
    std::vector<MMI::Rect>& touchHotAreas,
    std::vector<MMI::Rect>& pointerHotAreas) const
{
    if (sceneSession == nullptr) {
        WLOGFE("sceneSession is nullptr");
        return;
    }
    WSRect windowRect = sceneSession->GetSessionGlobalRect();
    uint32_t touchOffset = 0;
    uint32_t pointerOffset = 0;
    bool isMidScene = sceneSession->GetIsMidScene();
    bool isAppMainWindowOrPip = sceneSession->GetWindowType() == WindowType::WINDOW_TYPE_APP_MAIN_WINDOW ||
                                sceneSession->GetWindowType() == WindowType::WINDOW_TYPE_PIP;
    if (isAppMainWindowOrPip && !isMidScene) {
        float vpr = 1.5f; // 1.5: default vp
        auto sessionProperty = sceneSession->GetSessionProperty();
        if (sessionProperty != nullptr) {
            auto displayId = sessionProperty->GetDisplayId();
            auto screenSession = ScreenSessionManagerClient::GetInstance().GetScreenSession(displayId);
            if (screenSession != nullptr) {
                vpr = screenSession->GetScreenProperty().GetDensity();
            }
        }
        touchOffset = static_cast<uint32_t>(HOTZONE_TOUCH * vpr);
        pointerOffset = static_cast<uint32_t>(HOTZONE_POINTER * vpr);
    }

    MMI::Rect touchRect = {
        .x = -touchOffset,
        .y = -touchOffset,
        .width = windowRect.width_ + static_cast<int32_t>(touchOffset * 2),  // 2 : double touchOffset
        .height = windowRect.height_ + static_cast<int32_t>(touchOffset * 2) // 2 : double touchOffset
    };

    MMI::Rect pointerRect = {
        .x = -pointerOffset,
        .y = -pointerOffset,
        .width = windowRect.width_ + static_cast<int32_t>(pointerOffset * 2),  // 2 : double pointerOffset
        .height = windowRect.height_ + static_cast<int32_t>(pointerOffset * 2) // 2 : double pointerOffset
    };

    touchHotAreas.emplace_back(touchRect);
    pointerHotAreas.emplace_back(pointerRect);
}

void SceneSessionDirtyManager::UpdateHotAreas(sptr<SceneSession> sceneSession, std::vector<MMI::Rect>& touchHotAreas,
    std::vector<MMI::Rect>& pointerHotAreas) const
{
    if (sceneSession == nullptr) {
        WLOGFE("sceneSession is nullptr");
        return;
    }
    WSRect windowRect = sceneSession->GetSessionGlobalRect();
    const std::vector<Rect>& hotAreas = sceneSession->GetTouchHotAreas();
    for (auto area : hotAreas) {
        MMI::Rect rect;
        rect.x = area.posX_;
        rect.y = area.posY_;
        rect.width = static_cast<int32_t>(area.width_);
        rect.height = static_cast<int32_t>(area.height_);
        auto iter = std::find_if(touchHotAreas.begin(), touchHotAreas.end(),
            [&rect](const MMI::Rect& var) { return rect == var; });
        if (iter != touchHotAreas.end()) {
            continue;
        }
        touchHotAreas.emplace_back(rect);
        pointerHotAreas.emplace_back(rect);
        if (touchHotAreas.size() == static_cast<uint32_t>(MMI::WindowInfo::MAX_HOTAREA_COUNT)) {
            auto sessionid = sceneSession->GetWindowId();
            WLOGFE("id = %{public}d hotAreas size > %{public}d", sessionid, static_cast<int>(hotAreas.size()));
            break;
        }
    }
    if (touchHotAreas.empty()) {
        return UpdateDefaultHotAreas(sceneSession, touchHotAreas, pointerHotAreas);
    }
}

static void AddDialogSessionMapItem(const sptr<SceneSession>& session,
    std::map<int32_t, sptr<SceneSession>>& dialogMap)
{
    const auto& mainSession = session->GetMainSession();
    if (mainSession == nullptr) {
        return;
    }
    bool isTopmostModalSubWindow = false;
    const auto& property = session->GetSessionProperty();
    if (property != nullptr && property->IsTopmost()) {
        isTopmostModalSubWindow = true;
    }
    if (auto iter = dialogMap.find(mainSession->GetPersistentId());
        iter != dialogMap.end() && iter->second != nullptr) {
        auto& targetSession = iter->second;
        if (targetSession->GetSessionProperty() &&
            targetSession->GetSessionProperty()->IsTopmost() &&
            !isTopmostModalSubWindow) {
            return;
        }
        if (targetSession->GetZOrder() > session->GetZOrder()) {
            return;
        }
    }
    dialogMap[mainSession->GetPersistentId()] = session;
    TLOGD(WmsLogTag::WMS_DIALOG, "Add dialog session, id: %{public}d, mainSessionId: %{public}d",
        session->GetPersistentId(), mainSession->GetPersistentId());
}

std::map<int32_t, sptr<SceneSession>> SceneSessionDirtyManager::GetDialogSessionMap(
    const std::map<int32_t, sptr<SceneSession>>& sessionMap) const
{
    std::map<int32_t, sptr<SceneSession>> dialogMap;
    for (const auto& elem: sessionMap) {
        const auto& session = elem.second;
        if (session == nullptr || session->GetForceHideState() != ForceHideState::NOT_HIDDEN) {
            continue;
        }
        bool isModalSubWindow = false;
        const auto& property = session->GetSessionProperty();
        if (property != nullptr) {
            isModalSubWindow = WindowHelper::IsModalSubWindow(property->GetWindowType(), property->GetWindowFlags());
        }
        if (isModalSubWindow || session->GetWindowType() == WindowType::WINDOW_TYPE_DIALOG) {
            AddDialogSessionMapItem(session, dialogMap);
        }
    }
    return dialogMap;
}

bool SceneSessionDirtyManager::IsFilterSession(const sptr<SceneSession>& sceneSession) const
{
    if (sceneSession == nullptr) {
        return true;
    }

    if (sceneSession->IsSystemInput()) {
        return false;
    } else if (sceneSession->IsSystemSession() && sceneSession->IsVisible() && sceneSession->IsSystemActive()) {
        return false;
    }
    if (!Rosen::SceneSessionManager::GetInstance().IsSessionVisible(sceneSession)) {
        return true;
    }
    return false;
}

void SceneSessionDirtyManager::NotifyWindowInfoChange(const sptr<SceneSession>& sceneSession,
    const WindowUpdateType& type, const bool startMoving)
{
    if (sceneSession == nullptr) {
        WLOGFW("sceneSession is null");
        return;
    }

    if (type == WindowUpdateType::WINDOW_UPDATE_ADDED || type == WindowUpdateType::WINDOW_UPDATE_REMOVED||
        type == WindowUpdateType::WINDOW_UPDATE_ACTIVE) {
            WLOGFD("[EventDispatch] wid = %{public}d, winType = %{public}d",
                sceneSession->GetWindowId(), static_cast<int>(type));
    }
    ResetFlushWindowInfoTask();
}

void SceneSessionDirtyManager::ResetFlushWindowInfoTask()
{
    sessionDirty_.store(true);
    bool hasPostTask = false;
    if (hasPostTask_.compare_exchange_strong(hasPostTask, true)) {
        auto task = [this]() {
            hasPostTask_.store(false);
            if (!sessionDirty_.load() || flushWindowInfoCallback_ == nullptr) {
                return;
            }
            flushWindowInfoCallback_();
        };
        SceneSessionManager::GetInstance().PostFlushWindowInfoTask(task,
            UPDATE_WINDOW_INFO_TASK, UPDATE_TASK_DURATION);
    }
}

void SceneSessionDirtyManager::AddModalExtensionWindowInfo(std::vector<MMI::WindowInfo>& windowInfoList,
    MMI::WindowInfo windowInfo, const sptr<SceneSession>& sceneSession)
{
    if (sceneSession == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "sceneSession is nullptr");
        return;
    }

    auto extensionInfo = sceneSession->GetLastModalUIExtensionEventInfo();
    windowInfo.id = extensionInfo.persistentId;
    if (extensionInfo.windowRect.width_ != 0 || extensionInfo.windowRect.height_ != 0) {
        MMI::Rect windowRect = {
            .x = extensionInfo.windowRect.posX_,
            .y = extensionInfo.windowRect.posY_,
            .width = extensionInfo.windowRect.width_,
            .height = extensionInfo.windowRect.height_
        };
        windowInfo.area =  windowRect;
        std::vector<MMI::Rect> touchHotAreas;
        MMI::Rect touchRect = {
            .x = 0,
            .y = 0,
            .width = extensionInfo.windowRect.width_,
            .height = extensionInfo.windowRect.height_
        };
        touchHotAreas.emplace_back(touchRect);
        windowInfo.defaultHotAreas = touchHotAreas;
        windowInfo.pointerHotAreas = touchHotAreas;
        Matrix3f transform;
        CalTransform(sceneSession, transform, true);
        std::vector<float> transformData(transform.GetData(), transform.GetData() + TRANSFORM_DATA_LEN);
        windowInfo.transform = transformData;
    }

    windowInfo.agentWindowId = extensionInfo.persistentId;
    windowInfo.pid = extensionInfo.pid;
    std::vector<int32_t> pointerChangeAreas(POINTER_CHANGE_AREA_COUNT, 0);
    windowInfo.pointerChangeAreas = pointerChangeAreas;
    windowInfo.zOrder = windowInfo.zOrder + ZORDER_UIEXTENSION_INDEX;

    windowInfoList.emplace_back(windowInfo);
}

auto SceneSessionDirtyManager::GetFullWindowInfoList() ->
std::pair<std::vector<MMI::WindowInfo>, std::vector<std::shared_ptr<Media::PixelMap>>>
{
    std::vector<MMI::WindowInfo> windowInfoList;
    std::vector<std::shared_ptr<Media::PixelMap>> pixelMapList;
    const auto sceneSessionMap = Rosen::SceneSessionManager::GetInstance().GetSceneSessionMap();
    // all input event should trans to dialog window if dialog exists
    const auto dialogMap = GetDialogSessionMap(sceneSessionMap);
    uint32_t maxHotAreasNum = 0;
    for (const auto& sceneSessionValuePair : sceneSessionMap) {
        const auto& sceneSessionValue = sceneSessionValuePair.second;
        if (sceneSessionValue == nullptr) {
            continue;
        }
        WLOGFD("[EventDispatch] FullSceneSessionInfoUpdate windowName = %{public}s bundleName = %{public}s"
            " windowId = %{public}d activeStatus = %{public}d", sceneSessionValue->GetWindowName().c_str(),
            sceneSessionValue->GetSessionInfo().bundleName_.c_str(), sceneSessionValue->GetWindowId(),
            sceneSessionValue->GetForegroundInteractiveStatus());
        auto [windowInfo, pixelMap] = GetWindowInfo(sceneSessionValue, WindowAction::WINDOW_ADD);
        auto iter = (sceneSessionValue->GetMainSessionId() == INVALID_SESSION_ID) ?
            dialogMap.find(sceneSessionValue->GetPersistentId()) :
            dialogMap.find(sceneSessionValue->GetMainSessionId());
        if (iter != dialogMap.end() && iter->second != nullptr &&
            sceneSessionValue->GetPersistentId() != iter->second->GetPersistentId() &&
            iter->second->GetZOrder() > sceneSessionValue->GetZOrder()) {
            windowInfo.agentWindowId = static_cast<int32_t>(iter->second->GetPersistentId());
            windowInfo.pid = static_cast<int32_t>(iter->second->GetCallingPid());
        } else if (sceneSessionValue->HasModalUIExtension()) {
            AddModalExtensionWindowInfo(windowInfoList, windowInfo, sceneSessionValue);
        }
        TLOGD(WmsLogTag::WMS_EVENT, "windowId = %{public}d, agentWindowId = %{public}d, zOrder = %{public}f",
            windowInfo.id, windowInfo.agentWindowId, windowInfo.zOrder);
        windowInfoList.emplace_back(windowInfo);
        pixelMapList.emplace_back(pixelMap);
        // set the number of hot areas to the maximum number of hot areas when it exceeds the maximum number
        // to avoid exceeding socket buff limits
        if (windowInfo.defaultHotAreas.size() > maxHotAreasNum) {
            maxHotAreasNum = windowInfo.defaultHotAreas.size();
        }
    }
    if (maxHotAreasNum > MMI::WindowInfo::DEFAULT_HOTAREA_COUNT) {
        std::sort(windowInfoList.begin(), windowInfoList.end(), CmpMMIWindowInfo);
    }
    return {windowInfoList, pixelMapList};
}

void SceneSessionDirtyManager::UpdatePointerAreas(sptr<SceneSession> sceneSession,
    std::vector<int32_t>& pointerChangeAreas) const
{
    if (!sceneSession) {
        TLOGE(WmsLogTag::WMS_EVENT, "sceneSession is null");
        return;
    }
    auto sessionProperty = sceneSession->GetSessionProperty();
    if (!sessionProperty) {
        TLOGE(WmsLogTag::WMS_EVENT, "sessionProperty is null");
        return;
    }
    bool dragEnabled = sessionProperty->GetDragEnabled();
    if (dragEnabled) {
        float vpr = 1.5f; // 1.5: default vp
        auto displayId = sessionProperty->GetDisplayId();
        auto screenSession = ScreenSessionManagerClient::GetInstance().GetScreenSession(displayId);
        if (screenSession != nullptr) {
            vpr = screenSession->GetScreenProperty().GetDensity();
        }
        int32_t pointerAreaFivePx = static_cast<int32_t>(POINTER_CHANGE_AREA_FIVE * vpr);
        int32_t pointerAreaSixteenPx = static_cast<int32_t>(POINTER_CHANGE_AREA_SIXTEEN * vpr);
        if (sceneSession->GetSessionInfo().isSetPointerAreas_) {
            pointerChangeAreas = {POINTER_CHANGE_AREA_DEFAULT, POINTER_CHANGE_AREA_DEFAULT,
                POINTER_CHANGE_AREA_DEFAULT, pointerAreaFivePx, pointerAreaSixteenPx,
                pointerAreaFivePx, pointerAreaSixteenPx, pointerAreaFivePx};
            return;
        }
        auto limits = sessionProperty->GetWindowLimits();
        if (limits.minWidth_ == limits.maxWidth_ && limits.minHeight_ != limits.maxHeight_) {
            pointerChangeAreas = {POINTER_CHANGE_AREA_DEFAULT, pointerAreaFivePx,
                POINTER_CHANGE_AREA_DEFAULT, POINTER_CHANGE_AREA_DEFAULT, POINTER_CHANGE_AREA_DEFAULT,
                pointerAreaFivePx, POINTER_CHANGE_AREA_DEFAULT,  POINTER_CHANGE_AREA_DEFAULT};
        } else if (limits.minWidth_ != limits.maxWidth_ && limits.minHeight_ == limits.maxHeight_) {
            pointerChangeAreas = {POINTER_CHANGE_AREA_DEFAULT, POINTER_CHANGE_AREA_DEFAULT,
                POINTER_CHANGE_AREA_DEFAULT, pointerAreaFivePx, POINTER_CHANGE_AREA_DEFAULT,
                POINTER_CHANGE_AREA_DEFAULT, POINTER_CHANGE_AREA_DEFAULT, pointerAreaFivePx};
        } else if (limits.minWidth_ != limits.maxWidth_ && limits.minHeight_ != limits.maxHeight_) {
            pointerChangeAreas = {pointerAreaSixteenPx, pointerAreaFivePx,
                pointerAreaSixteenPx, pointerAreaFivePx, pointerAreaSixteenPx,
                pointerAreaFivePx, pointerAreaSixteenPx, pointerAreaFivePx};
        }
    } else {
        WLOGFD("UpdatePointerAreas sceneSession is: %{public}d dragEnabled is false", sceneSession->GetPersistentId());
    }
}

void SceneSessionDirtyManager::UpdatePrivacyMode(const sptr<SceneSession>& sceneSession,
    MMI::WindowInfo& windowInfo) const
{
    if (sceneSession == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "sceneSession is nullptr");
        return;
    }
    windowInfo.privacyMode = MMI::SecureFlag::DEFAULT_MODE;
    sptr<WindowSessionProperty> windowSessionProperty = sceneSession->GetSessionProperty();
    if (windowSessionProperty == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "windowSessionProperty is nullptr");
        return;
    }
    if (windowSessionProperty->GetPrivacyMode() || windowSessionProperty->GetSystemPrivacyMode() ||
        sceneSession->GetCombinedExtWindowFlags().privacyModeFlag) {
        windowInfo.privacyMode = MMI::SecureFlag::PRIVACY_MODE;
    }
}

void SceneSessionDirtyManager::UpdateWindowFlags(DisplayId displayId, const sptr<SceneSession>& sceneSession,
    MMI::WindowInfo& windowInfo) const
{
    windowInfo.flags = 0;
    auto screenSession = ScreenSessionManagerClient::GetInstance().GetScreenSession(displayId);
    if (screenSession != nullptr) {
        if (!screenSession->IsTouchEnabled() || !sceneSession->GetSystemTouchable() ||
            !sceneSession->GetForegroundInteractiveStatus()) {
            windowInfo.flags |= MMI::WindowInfo::FLAG_BIT_UNTOUCHABLE;
        }
    }
}

std::pair<MMI::WindowInfo, std::shared_ptr<Media::PixelMap>> SceneSessionDirtyManager::GetWindowInfo(
    const sptr<SceneSession>& sceneSession, const SceneSessionDirtyManager::WindowAction& action) const
{
    if (sceneSession == nullptr) {
        WLOGFE("sceneSession is nullptr");
        return {};
    }
    sptr<WindowSessionProperty> windowSessionProperty = sceneSession->GetSessionProperty();
    if (windowSessionProperty == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "GetSessionProperty is nullptr");
        return {};
    }
    Matrix3f transform;
    WSRect windowRect = sceneSession->GetSessionGlobalRect();
    auto pid = sceneSession->GetCallingPid();
    auto uid = sceneSession->GetCallingUid();
    auto windowId = sceneSession->GetWindowId();
    auto displayId = windowSessionProperty->GetDisplayId();
    CalTransform(sceneSession, transform);
    std::vector<float> transformData(transform.GetData(), transform.GetData() + TRANSFORM_DATA_LEN);

    auto agentWindowId = sceneSession->GetWindowId();
    auto zOrder = sceneSession->GetZOrder();
    std::vector<int32_t> pointerChangeAreas(POINTER_CHANGE_AREA_COUNT, 0);
    auto windowMode = windowSessionProperty->GetWindowMode();
    auto maxMode = windowSessionProperty->GetMaximizeMode();
    WindowType windowType = windowSessionProperty->GetWindowType();
    bool isMainWindow = Rosen::WindowHelper::IsMainWindow(windowType);
    bool isDecorDialog = Rosen::WindowHelper::IsDialogWindow(windowType) && windowSessionProperty->IsDecorEnable();
    bool isDecorSubWindow = WindowHelper::IsSubWindow(windowType) && windowSessionProperty->IsDecorEnable();
    if ((windowMode == Rosen::WindowMode::WINDOW_MODE_FLOATING &&
        (isMainWindow || isDecorDialog || isDecorSubWindow) &&
        maxMode != Rosen::MaximizeMode::MODE_AVOID_SYSTEM_BAR) || (sceneSession->GetSessionInfo().isSetPointerAreas_)) {
            UpdatePointerAreas(sceneSession, pointerChangeAreas);
    }
    std::vector<MMI::Rect> touchHotAreas;
    std::vector<MMI::Rect> pointerHotAreas;
    UpdateHotAreas(sceneSession, touchHotAreas, pointerHotAreas);
    auto pixelMap = windowSessionProperty->GetWindowMask();
    MMI::WindowInfo windowInfo = {
        .id = windowId,
        .pid = sceneSession->IsStartMoving() ? static_cast<int32_t>(getpid()) : pid,
        .uid = uid,
        .area = { windowRect.posX_, windowRect.posY_, windowRect.width_, windowRect.height_ },
        .defaultHotAreas = touchHotAreas,
        .pointerHotAreas = pointerHotAreas,
        .agentWindowId = agentWindowId,
        .action = static_cast<MMI::WINDOW_UPDATE_ACTION>(action),
        .displayId = displayId,
        .zOrder = zOrder,
        .pointerChangeAreas = pointerChangeAreas,
        .transform = transformData,
        .pixelMap = pixelMap.get(),
        .windowInputType = static_cast<MMI::WindowInputType>(sceneSession->GetSessionInfo().windowInputType_),
        .windowType = static_cast<int32_t>(windowType),
    };
    UpdateWindowFlags(displayId, sceneSession, windowInfo);
    if (windowSessionProperty->GetWindowFlags() & static_cast<uint32_t>(WindowFlag::WINDOW_FLAG_HANDWRITING)) {
        windowInfo.flags |= MMI::WindowInfo::FLAG_BIT_HANDWRITING;
    }
    UpdatePrivacyMode(sceneSession, windowInfo);
    windowInfo.uiExtentionWindowInfo = GetSecSurfaceWindowinfoList(sceneSession, windowInfo, transform);
    return {windowInfo, pixelMap};
}

void SceneSessionDirtyManager::RegisterFlushWindowInfoCallback(const FlushWindowInfoCallback &&callback)
{
    flushWindowInfoCallback_ = std::move(callback);
}

void SceneSessionDirtyManager::ResetSessionDirty()
{
    sessionDirty_.store(false);
}

std::string DumpRect(const std::vector<MMI::Rect>& rects)
{
    std::string rectStr = "";
    for (const auto& rect : rects) {
        rectStr = rectStr + " hot : [ " + std::to_string(rect.x) +" , " + std::to_string(rect.y) +
        " , " + std::to_string(rect.width) + " , " + std::to_string(rect.height) + "]";
    }
    return rectStr;
}

std::string DumpWindowInfo(const MMI::WindowInfo& info)
{
    std::string infoStr = "windowInfo:";
    infoStr = infoStr + "windowId: " + std::to_string(info.id) + " pid : " + std::to_string(info.pid) +
        " uid: " + std::to_string(info.uid) + " area: [ " + std::to_string(info.area.x) + " , " +
        std::to_string(info.area.y) +  " , " + std::to_string(info.area.width) + " , " +
        std::to_string(info.area.height) + "] agentWindowId:" + std::to_string(info.agentWindowId) + " flags:" +
        std::to_string(info.flags)  +" displayId: " + std::to_string(info.displayId) +
        " action: " + std::to_string(static_cast<int>(info.action)) + " zOrder: " + std::to_string(info.zOrder);
    return infoStr + DumpRect(info.defaultHotAreas);
}

std::string DumpSecRectInfo(const SecRectInfo & secRectInfo)
{
    std::string infoStr = " area: [ " + std::to_string(secRectInfo.relativeCoords.GetLeft()) + " , " +
        std::to_string(secRectInfo.relativeCoords.GetTop()) +  " , " +
        std::to_string(secRectInfo.relativeCoords.GetWidth()) + " , " +
        std::to_string(secRectInfo.relativeCoords.GetHeight()) + "]" +
        " scaleX:" + std::to_string(secRectInfo.scale[0]) + " scaleY:" + std::to_string(secRectInfo.scale[1]) +
        " anchorX:" + std::to_string(secRectInfo.anchor[0]) + " anchorY:" + std::to_string(secRectInfo.anchor[1]);
    return infoStr;
}

std::string DumpSecSurfaceInfo(const SecSurfaceInfo& secSurfaceInfo)
{
    std::string infoStr = "hostPid:" + std::to_string(secSurfaceInfo.hostPid) +
        " uiExtensionPid:" + std::to_string(secSurfaceInfo.uiExtensionPid) +
        " hostNodeId:" + std::to_string(secSurfaceInfo.hostNodeId) +
        " uiExtensionNodeId:" + std::to_string(secSurfaceInfo.uiExtensionNodeId);
    return infoStr;
}

MMI::WindowInfo SceneSessionDirtyManager::MakeWindowInfoFormHostWindow(const SecRectInfo& secRectInfo,
    const MMI::WindowInfo& hostWindowinfo) const
{
    MMI::WindowInfo windowinfo;
    windowinfo.id = hostWindowinfo.id;
    windowinfo.pid = hostWindowinfo.pid;
    windowinfo.uid = hostWindowinfo.uid;
    windowinfo.area = hostWindowinfo.area;
    windowinfo.agentWindowId = hostWindowinfo.agentWindowId;
    windowinfo.action = hostWindowinfo.action;
    windowinfo.displayId = hostWindowinfo.displayId;
    windowinfo.flags = hostWindowinfo.flags;
    windowinfo.privacyMode = hostWindowinfo.privacyMode;
    windowinfo.transform = hostWindowinfo.transform;
    return windowinfo;
}

Matrix3f CoordinateSystemHostWindowToScreen(const Matrix3f hostTransform, const SecRectInfo& secRectInfo)
{
    Matrix3f transform = Matrix3f::IDENTITY;
    Vector2f translate(secRectInfo.relativeCoords.GetLeft(), secRectInfo.relativeCoords.GetTop());
    transform = transform.Translate(translate);
    Vector2f scale(secRectInfo.scale[0], secRectInfo.scale[1]);
    transform = transform.Scale(scale, secRectInfo.anchor[0], secRectInfo.anchor[1]);
    transform = hostTransform.Inverse() * transform;
    return transform;
}

MMI::Rect CalRectInScreen(const Matrix3f& transform, const SecRectInfo& secRectInfo)
{
    auto topLeft = transform * Vector3f(0, 0, 1.0);
    auto bottomRight = transform * Vector3f(secRectInfo.relativeCoords.GetWidth(),
        secRectInfo.relativeCoords.GetHeight(), 1.0);
    auto left = std::min(topLeft[0], bottomRight[0]);
    auto top = std::min(topLeft[1], bottomRight[1]);
    auto topLeftX = static_cast<int32_t>(topLeft[0]);
    auto topLeftY = static_cast<int32_t>(topLeft[1]);
    auto bottomRightX = static_cast<int32_t>(bottomRight[0]);
    auto bottomRightY = static_cast<int32_t>(bottomRight[1]);
    if ((topLeftX > 0 && bottomRightX < INT32_MIN + topLeftX) ||
        (topLeftX < 0 && bottomRightX > INT32_MAX + topLeftX)) {
        TLOGE(WmsLogTag::WMS_EVENT, "data overflows topLeftX:%{public}d bottomRightX:%{public}d",
            topLeftX, bottomRightX);
    }
    if ((topLeftY > 0 && bottomRightY < INT32_MIN + topLeftY) ||
        (topLeftY < 0 && bottomRightY > INT32_MAX + topLeftY)) {
        TLOGE(WmsLogTag::WMS_EVENT, "data overflows topLeftY:%{public}d bottomRightY:%{public}d",
            topLeftY, bottomRightY);
    }
    auto width = std::abs(topLeftX - bottomRightX);
    auto height = std::abs(topLeftY - bottomRightY);
    return MMI::Rect{ left, top, width, height};
}


MMI::WindowInfo SceneSessionDirtyManager::GetHostComponentWindowInfo(const SecSurfaceInfo& secSurfaceInfo,
    const MMI::WindowInfo& hostWindowinfo, const Matrix3f hostTransform) const
{
    MMI::WindowInfo windowinfo;
    const auto& secRectInfoList = secSurfaceInfo.upperNodes;
    if (secRectInfoList.size() > 0) {
        windowinfo = MakeWindowInfoFormHostWindow(secRectInfoList[0], hostWindowinfo);
    }
    for (const auto& secRectInfo : secRectInfoList) {
        windowinfo.pid = secSurfaceInfo.hostPid;
        MMI::Rect hotArea = { secRectInfo.relativeCoords.GetLeft(), secRectInfo.relativeCoords.GetTop(),
            secRectInfo.relativeCoords.GetWidth(), secRectInfo.relativeCoords.GetHeight() };
        windowinfo.defaultHotAreas.emplace_back(hotArea);
        windowinfo.pointerHotAreas.emplace_back(hotArea);
    }
    return windowinfo;
}

MMI::WindowInfo SceneSessionDirtyManager::GetSecComponentWindowInfo(const SecSurfaceInfo& secSurfaceInfo,
    const MMI::WindowInfo& hostWindowinfo, const sptr<SceneSession>& sceneSession, const Matrix3f hostTransform) const
{
    if (sceneSession == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "sceneSession is nullptr");
        return {};
    }
    MMI::WindowInfo windowinfo;
    const auto& secRectInfo = secSurfaceInfo.uiExtensionRectInfo;
    windowinfo = MakeWindowInfoFormHostWindow(secRectInfo, hostWindowinfo);
    windowinfo.id = sceneSession->GetUIExtPersistentIdBySurfaceNodeId(secSurfaceInfo.uiExtensionNodeId);
    if (windowinfo.id == 0) {
        TLOGE(WmsLogTag::WMS_EVENT, "GetUIExtPersistentId ERROR");
        return {};
    }
    windowinfo.agentWindowId = windowinfo.id;
    windowinfo.pid = secSurfaceInfo.uiExtensionPid;
    windowinfo.privacyUIFlag = true;
    auto transform = CoordinateSystemHostWindowToScreen(hostTransform, secRectInfo);
    windowinfo.area = CalRectInScreen(transform, secRectInfo);
    MMI::Rect hotArea = { 0, 0, secRectInfo.relativeCoords.GetWidth(), secRectInfo.relativeCoords.GetHeight() };
    windowinfo.defaultHotAreas.emplace_back(hotArea);
    windowinfo.pointerHotAreas.emplace_back(hotArea);
    // 屏幕坐标系到控件坐标系转换
    transform = transform.Inverse();
    std::vector<float> transformData(transform.GetData(), transform.GetData() + TRANSFORM_DATA_LEN);
    windowinfo.transform = transformData;
    return windowinfo;
}

bool operator==(const SecRectInfo& a, const SecRectInfo& b)
{
    return (a.relativeCoords == b.relativeCoords && a.scale == b.scale && a.anchor == b.anchor);
}

bool operator!=(const SecRectInfo& a, const SecRectInfo& b)
{
    return !(a == b);
}

bool operator==(const SecSurfaceInfo& a, const SecSurfaceInfo& b)
{
    return (a.uiExtensionRectInfo == b.uiExtensionRectInfo && a.hostPid == b.hostPid &&
        a.uiExtensionNodeId == b.uiExtensionNodeId && a.uiExtensionPid == b.uiExtensionPid &&
        a.hostNodeId == b.hostNodeId && a.upperNodes == b.upperNodes);
}

void DumpSecSurfaceInfoMap(const std::map<uint64_t, std::vector<SecSurfaceInfo>>& secSurfaceInfoMap)
{
    TLOGI(WmsLogTag::WMS_EVENT, "secSurfaceInfoMap size:%{public}d", static_cast<int>(secSurfaceInfoMap.size()));
    for (auto& e : secSurfaceInfoMap) {
        auto hostNodeId = e.first;
        TLOGI(WmsLogTag::WMS_EVENT, "hostNodeId:%{public}" PRIu64 " secSurfaceInfoList size:%{public}d",
            hostNodeId, static_cast<int>(e.second.size()));
        for (const auto& secSurfaceInfo : e.second) {
            auto surfaceInfoStr = DumpSecSurfaceInfo(secSurfaceInfo);
            auto rectInfoStr = DumpSecRectInfo(secSurfaceInfo.uiExtensionRectInfo);
            TLOGI(WmsLogTag::WMS_EVENT, "secSurfaceInfo:%{public}s secRectInfo:%{public}s", surfaceInfoStr.c_str(),
                rectInfoStr.c_str());
            for (const auto& secRectInfo : secSurfaceInfo.upperNodes) {
                auto infoStr = DumpSecRectInfo(secRectInfo);
                TLOGI(WmsLogTag::WMS_EVENT, "hostRectInfo:%{public}s", infoStr.c_str());
            }
        }
    }
}

void SceneSessionDirtyManager::UpdateSecSurfaceInfo(const std::map<uint64_t,
    std::vector<SecSurfaceInfo>>& secSurfaceInfoMap)
{
    std::unique_lock<std::shared_mutex> lock(secSurfaceInfoMutex_);
    if (secSurfaceInfoMap.size() != secSurfaceInfoMap_.size() || secSurfaceInfoMap_ != secSurfaceInfoMap) {
        secSurfaceInfoMap_ = secSurfaceInfoMap;
        ResetFlushWindowInfoTask();
        DumpSecSurfaceInfoMap(secSurfaceInfoMap_);
    }
}

std::vector<MMI::WindowInfo> SceneSessionDirtyManager::GetSecSurfaceWindowinfoList(
    const sptr<SceneSession>& sceneSession, const MMI::WindowInfo& hostWindowinfo, const Matrix3f& hostTransform) const
{
    if (secSurfaceInfoMap_.size() == 0) {
        return {};
    }
    if (sceneSession == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "sceneSession is nullptr");
        return {};
    }
    auto surfaceNode = sceneSession->GetSurfaceNode();
    if (surfaceNode == nullptr) {
        TLOGE(WmsLogTag::WMS_EVENT, "surfaceNode is nullptr");
        return {};
    }
    std::vector<SecSurfaceInfo> secSurfaceInfoList;
    auto surfaceNodeId = surfaceNode->GetId();
    {
        std::shared_lock<std::shared_mutex> lock(secSurfaceInfoMutex_);
        auto iter = secSurfaceInfoMap_.find(surfaceNodeId);
        if (iter == secSurfaceInfoMap_.end()) {
            return {};
        }
        secSurfaceInfoList = iter->second;
    }
    std::vector<MMI::WindowInfo> windowinfoList;
    int seczOrder = 0;
    MMI::WindowInfo windowinfo;
    for (const auto& secSurfaceInfo : secSurfaceInfoList) {
        windowinfo = GetSecComponentWindowInfo(secSurfaceInfo, hostWindowinfo, sceneSession, hostTransform);
        windowinfo.zOrder = seczOrder++;
        windowinfoList.emplace_back(windowinfo);
        windowinfo = GetHostComponentWindowInfo(secSurfaceInfo, hostWindowinfo, hostTransform);
        windowinfo.zOrder = seczOrder++;
        windowinfoList.emplace_back(windowinfo);
    }
    return windowinfoList;
}
} //namespace OHOS::Rosen