xref: /ohos5.0/foundation/arkui/ace_engine/frameworks/core/components_ng/pattern/node_container/node_container_pattern.cpp

/*
 * 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 "core/components_ng/pattern/node_container/node_container_pattern.h"

#include "base/utils/utils.h"
#include "core/components_v2/inspector/inspector_constants.h"
#include "core/pipeline_ng/pipeline_context.h"

namespace OHOS::Ace::NG {
namespace {
bool CheckBeforeAddNode(const RefPtr<UINode>& hostNode, const RefPtr<UINode>& newNode)
{
    CHECK_NULL_RETURN(hostNode, false);
    CHECK_NULL_RETURN(newNode, true);
    if (!(newNode->IsArkTsFrameNode()) && !newNode->GetIsRootBuilderNode()) {
        TAG_LOGW(AceLogTag::ACE_NODE_CONTAINER, "Cannot return node created by declarative UI function");
        return false;
    }
    auto parent = newNode->GetParent();
    if (parent && parent != hostNode) {
        TAG_LOGF(AceLogTag::ACE_NODE_CONTAINER,
            "Add [id:%{public}d][tag:%{public}s] to [id:%{public}d][tag:%{public}s] with previous parent "
            "[id:%{public}d][tag:%{public}s]",
            newNode->GetId(), newNode->GetTag().c_str(), hostNode->GetId(), hostNode->GetTag().c_str(), parent->GetId(),
            parent->GetTag().c_str());
    }
    return true;
}
} // namespace

void NodeContainerPattern::RemakeNode()
{
    auto host = GetHost();
    CHECK_NULL_VOID(host);
    auto newNode = FireMakeFunction();
    auto oldChild = host->GetChildAtIndex(0);
    if ((!oldChild && !newNode) || (oldChild && oldChild == newNode)) {
        return;
    }
    host->RemoveChildAtIndex(0);
    if (newNode) {
        CHECK_NULL_VOID(CheckBeforeAddNode(host, newNode));
        host->AddChild(newNode, 0);
        newNode->UpdateGeometryTransition();
    }
    OnAddBaseNode();
    host->MarkNeedFrameFlushDirty(NG::PROPERTY_UPDATE_MEASURE);
}

void NodeContainerPattern::CleanChild()
{
    auto host = GetHost();
    CHECK_NULL_VOID(host);
    host->RemoveChildAtIndex(0);
    host->MarkNeedFrameFlushDirty(NG::PROPERTY_UPDATE_MEASURE);
}

bool NodeContainerPattern::OnDirtyLayoutWrapperSwap(const RefPtr<LayoutWrapper>& dirty, const DirtySwapConfig& config)
{
    if (config.skipMeasure && config.skipLayout) {
        return false;
    }
    auto host = GetHost();
    CHECK_NULL_RETURN(host, false);
    auto context = host->GetContext();
    CHECK_NULL_RETURN(context, false);
    if (config.frameSizeChange) {
        auto geometryNode = dirty->GetGeometryNode();
        auto size = geometryNode->GetFrameSize();
        context->AddAfterLayoutTask([weak = WeakClaim(this), size]() {
            auto pattern = weak.Upgrade();
            CHECK_NULL_VOID(pattern);
            pattern->FireOnResize(size);
        });
    }
    if (surfaceId_ != 0U && !exportTextureNode_.Invalid()) {
        context->AddAfterLayoutTask([weak = WeakClaim(this)]() {
            auto pattern = weak.Upgrade();
            CHECK_NULL_VOID(pattern);
            auto ret = pattern->HandleTextureExport(false);
            if (!ret) {
                TAG_LOGW(AceLogTag::ACE_NODE_CONTAINER, "DoTextureExport fail");
            }
        });
    }
    return false;
}

bool NodeContainerPattern::HandleTextureExport(bool isStop)
{
    auto exportTextureNode = GetExportTextureNode();
    CHECK_NULL_RETURN(exportTextureNode, false);
    auto exportTextureRenderContext = exportTextureNode->GetRenderContext();
    CHECK_NULL_RETURN(exportTextureRenderContext, false);
    auto host = GetHost();
    if (host) {
        auto renderContext = host->GetRenderContext();
        CHECK_NULL_RETURN(renderContext, false);
        renderContext->SetIsNeedRebuildRSTree(isStop);
    }
    if (isStop) {
        return exportTextureRenderContext->StopTextureExport();
    }
    return exportTextureRenderContext->DoTextureExport(surfaceId_);
}

void NodeContainerPattern::OnDetachFromFrameNode(FrameNode* /* frameNode */)
{
    HandleTextureExport(true);
}

RefPtr<FrameNode> NodeContainerPattern::GetExportTextureNode() const
{
    auto exportTextureNode = exportTextureNode_.Upgrade();
    CHECK_NULL_RETURN(exportTextureNode, nullptr);
    auto exportTextureFrameNode = DynamicCast<FrameNode>(exportTextureNode);
    while (exportTextureNode && !exportTextureFrameNode) {
        exportTextureNode = exportTextureNode->GetFirstChild();
        exportTextureFrameNode = DynamicCast<FrameNode>(exportTextureNode);
    }
    return exportTextureFrameNode;
}

void NodeContainerPattern::ResetExportTextureInfo()
{
    surfaceId_ = 0U;
    exportTextureNode_.Reset();
}

void NodeContainerPattern::SetExportTextureInfoIfNeeded()
{
    ResetExportTextureInfo();
    auto host = GetHost();
    CHECK_NULL_VOID(host);
    auto viewNode = host->GetChildAtIndex(0);
    CHECK_NULL_VOID(viewNode);
    if (!viewNode->IsNeedExportTexture()) {
        return;
    }
    auto parent = host->GetAncestorNodeOfFrame();
    if (parent) {
        auto nodeContainer = parent->GetNodeContainer();
        if (nodeContainer) {
            auto parentViewNode = nodeContainer->GetChildAtIndex(0);
            CHECK_NULL_VOID(parentViewNode);
            auto parnetExportTextureInfo = parentViewNode->GetExportTextureInfo();
            if (parnetExportTextureInfo &&
                parnetExportTextureInfo->GetCurrentRenderType() == NodeRenderType::RENDER_TYPE_TEXTURE) {
                return;
            }
        }
    }
    auto exportTextureInfo = viewNode->GetExportTextureInfo();
    exportTextureNode_ = viewNode;
    surfaceId_ = StringUtils::StringToLongUint(exportTextureInfo->GetSurfaceId());
}

void NodeContainerPattern::OnAddBaseNode()
{
    HandleTextureExport(true);
    SetExportTextureInfoIfNeeded();
}

void NodeContainerPattern::OnMountToParentDone()
{
    SetExportTextureInfoIfNeeded();
}
} // namespace OHOS::Ace::NG