OpenHarmony-v6.0-Release/s

/*
 * Copyright (c) 2024 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 "feature/capture/rs_ui_capture_task_parallel.h"

#include <memory>
#include <sys/mman.h>

#include "draw/surface.h"
#include "draw/color.h"
#include "rs_trace.h"

#include "common/rs_background_thread.h"
#include "common/rs_obj_abs_geometry.h"
#include "feature/capture/rs_surface_capture_task_parallel.h"
#include "feature/uifirst/rs_uifirst_manager.h"
#include "feature_cfg/graphic_feature_param_manager.h"
#include "memory/rs_tag_tracker.h"
#include "params/rs_surface_render_params.h"
#include "pipeline/render_thread/rs_uni_render_util.h"
#include "pipeline/rs_base_render_node.h"
#include "pipeline/rs_screen_render_node.h"
#include "pipeline/main_thread/rs_main_thread.h"
#include "pipeline/rs_paint_filter_canvas.h"
#include "pipeline/main_thread/rs_render_service_connection.h"
#include "pipeline/rs_surface_render_node.h"
#include "pipeline/rs_uni_render_judgement.h"
#include "platform/common/rs_log.h"
#include "platform/drawing/rs_surface.h"
#include "render/rs_drawing_filter.h"
#include "render/rs_skia_filter.h"
#include "screen_manager/rs_screen_manager.h"
#include "screen_manager/rs_screen_mode_info.h"
#include "drawable/rs_canvas_render_node_drawable.h"
#include "pipeline/rs_canvas_render_node.h"

#ifdef RS_ENABLE_VK
#include "platform/ohos/backend/native_buffer_utils.h"
#endif

#ifdef RS_PROFILER_ENABLED
#include "rs_profiler_capture_recorder.h"
#endif

namespace OHOS {
namespace Rosen {
constexpr uint32_t ALPHA_MASK = 0xFF000000;
namespace {
    const std::string UICAPTURE_TASK_PREFIX = "uicapture_task_";
};

#ifdef RS_ENABLE_GPU
static inline void DrawCapturedImg(Drawing::Image& image,
    Drawing::Surface& surface, const Drawing::BackendTexture& backendTexture,
    Drawing::TextureOrigin& textureOrigin, Drawing::BitmapFormat& bitmapFormat)
{
    RSPaintFilterCanvas canvas(&surface);
    auto gpuContext = canvas.GetGPUContext();
    if (gpuContext == nullptr) {
        RS_LOGE("DrawCapturedImg failed: gpuContext is nullptr");
        return;
    }
    image.BuildFromTexture(*gpuContext, backendTexture.GetTextureInfo(),
        textureOrigin, bitmapFormat, nullptr);
    canvas.DrawImage(image, 0.f, 0.f, Drawing::SamplingOptions());
    surface.FlushAndSubmit(true);
}
#endif

uint32_t PixelMapSamplingDump(std::unique_ptr<Media::PixelMap>& pixelmap, int32_t x, int32_t y)
{
    if (pixelmap == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::PixelMapSamplingDump fail: pixelmap is nullptr");
        return 0;
    }
    if (x < 0 || y < 0 || x >= pixelmap->GetWidth() || y >= pixelmap->GetHeight()) {
        RS_LOGE("RSUiCaptureTaskParallel::PixelMapSamplingDump fail: x or y invalid");
        return 0;
    }
    uint32_t pixel = 0;
    pixelmap->ReadPixel({x, y}, pixel);
    return pixel;
}

bool RSUiCaptureTaskParallel::IsRectValid(NodeId nodeId, const Drawing::Rect& specifiedAreaRect)
{
    RS_LOGD("RSUiCaptureTaskParallel::IsRectValid: NodeId:[%{public}" PRIu64 "],"
        " Rect Left is [%{public}f], Top is [%{public}f],"
        " Right is [%{public}f], Bottom is [%{public}f],",
        nodeId, specifiedAreaRect.GetLeft(), specifiedAreaRect.GetTop(),
        specifiedAreaRect.GetRight(), specifiedAreaRect.GetBottom());
    auto node = RSMainThread::Instance()->GetContext().GetNodeMap().GetRenderNode(nodeId);
    if (node == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::IsRectValid: Invalid nodeId:[%{public}" PRIu64 "]", nodeId);
        return false;
    }
    if (!specifiedAreaRect.IsValid()) {
        RS_LOGD("RSUiCaptureTaskParallel::IsRectValid: specifiedAreaRect is an invalid rect");
        return false;
    }
    if ((specifiedAreaRect.GetWidth() > node->GetRenderProperties().GetBoundsWidth()) ||
        (specifiedAreaRect.GetHeight() > node->GetRenderProperties().GetBoundsHeight()) ||
        (specifiedAreaRect.GetLeft() < 0) || (specifiedAreaRect.GetTop() < 0) ||
        (specifiedAreaRect.GetRight() > node->GetRenderProperties().GetBoundsWidth()) ||
        (specifiedAreaRect.GetBottom() > node->GetRenderProperties().GetBoundsHeight())) {
        RS_LOGE("RSUiCaptureTaskParallel::IsRectValid: specifiedAreaRect is out of bounds");
        return false;
    }
    return true;
}

void RSUiCaptureTaskParallel::Capture(NodeId id, sptr<RSISurfaceCaptureCallback> callback,
    const RSSurfaceCaptureConfig& captureConfig, const Drawing::Rect& specifiedAreaRect)
{
    if (callback == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::Capture nodeId:[%{public}" PRIu64 "], callback is nullptr", id);
        return;
    }

    RS_LOGI("RSUiCaptureTaskParallel::Capture nodeId:[%{public}" PRIu64 "]", id);
    captureCount_++;
    std::shared_ptr<RSUiCaptureTaskParallel> captureHandle =
        std::make_shared<RSUiCaptureTaskParallel>(id, captureConfig);
    if (captureHandle == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::Capture captureHandle is nullptr!");
        ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
        return;
    }
    if (!captureConfig.uiCaptureInRangeParam.useBeginNodeSize && !captureHandle->UpdateStartAndEndNodeRect()) {
        RS_LOGE("RSUiCaptureTaskParallel::Capture UpdateStartAndEndNodeRect error!");
        ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
        return;
    }
    if (!captureHandle->CreateResources(specifiedAreaRect)) {
        RS_LOGE("RSUiCaptureTaskParallel::Capture CreateResources failed");
        ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
        return;
    }
    Drawing::Rect chosenRect;
    Drawing::Rect invalidRect(0.f, 0.f, 0.f, 0.f);
    if (!IsRectValid(id, specifiedAreaRect)) {
        chosenRect = invalidRect;
        RS_LOGD("RSUiCaptureTaskParallel::Capture invalid rect");
    } else {
        chosenRect = specifiedAreaRect;
    }
    captureHandle->IsStartEndSameNode();
    auto taskName = UICAPTURE_TASK_PREFIX + std::to_string(id);
    std::function<void()> captureTask = [captureHandle, id, captureConfig, callback, chosenRect]() -> void {
        RSSystemProperties::SetForceHpsBlurDisabled(true);
        if (!captureHandle->Run(callback, chosenRect)) {
            ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
        }
        RSSystemProperties::SetForceHpsBlurDisabled(false);
    };
    // Make the priority of uicapture task is lower than render task
    RSUniRenderThread::Instance().PostTask(captureTask, taskName, 0, AppExecFwk::EventQueue::Priority::HIGH);
}

bool RSUiCaptureTaskParallel::CreateResources(const Drawing::Rect& specifiedAreaRect)
{
    if (ROSEN_EQ(captureConfig_.scaleX, 0.f) || ROSEN_EQ(captureConfig_.scaleY, 0.f) ||
        captureConfig_.scaleX < 0.f || captureConfig_.scaleY < 0.f) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: SurfaceCapture scale is invalid.");
        return false;
    }
    auto node = RSMainThread::Instance()->GetContext().GetNodeMap().GetRenderNode(nodeId_);
    if (node == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: Invalid nodeId:[%{public}" PRIu64 "]",
            nodeId_);
        return false;
    }

    if (node->GetType() != RSRenderNodeType::ROOT_NODE &&
        node->GetType() != RSRenderNodeType::CANVAS_NODE &&
        node->GetType() != RSRenderNodeType::CANVAS_DRAWING_NODE &&
        node->GetType() != RSRenderNodeType::SURFACE_NODE) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: Invalid RSRenderNodeType!");
        return false;
    }
    Drawing::RectF targetRect = specifiedAreaRect;
    if (HasEndNodeRect() && !isStartEndNodeSame_) {
        auto endNode = RSMainThread::Instance()->GetContext().GetNodeMap().GetRenderNode(endNodeId_);
        if (endNode == nullptr) {
            RS_LOGE("RSUiCaptureTaskParallel::CreateResources: Invalid endNodeId:[%{public}" PRIu64 "]", endNodeId_);
            return false;
        }
        if (auto surfaceNode = endNode->ReinterpretCastTo<RSSurfaceRenderNode>()) {
            endNodeDrawable_ = std::static_pointer_cast<DrawableV2::RSRenderNodeDrawable>(
                DrawableV2::RSRenderNodeDrawableAdapter::OnGenerate(surfaceNode));
        } else if (auto canvasNode = endNode->ReinterpretCastTo<RSCanvasRenderNode>()) {
            endNodeDrawable_ = std::static_pointer_cast<DrawableV2::RSRenderNodeDrawable>(
                DrawableV2::RSRenderNodeDrawableAdapter::OnGenerate(canvasNode));
        } else {
            RS_LOGE("RSUiCaptureTaskParallel::CreateResources: Invalid RSRenderEndNode");
            return false;
        }
        const auto& endNodeParams = endNodeDrawable_->GetRenderParams();
        targetRect = endNodeParams->GetBounds();
    }
#ifdef RS_ENABLE_VK
    float nodeBoundsWidth = node->GetRenderProperties().GetBoundsWidth();
    float nodeBoundsHeight = node->GetRenderProperties().GetBoundsHeight();
    int32_t width = ceil(nodeBoundsWidth * captureConfig_.scaleX);
    int32_t height = ceil(nodeBoundsHeight * captureConfig_.scaleY);
    if (width > 0 && static_cast<int32_t>(OHOS::Rosen::NativeBufferUtils::VKIMAGE_LIMIT_SIZE) / width < height) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: image is too large, width:%{public}d, height::%{public}d",
            width, height);
        return false;
    }
    RS_LOGD("RSUiCaptureTaskParallel::CreateResources: Origin nodeBoundsWidth is [%{public}f,]"
        " Origin nodeBoundsHeight is [%{public}f]", nodeBoundsWidth, nodeBoundsHeight);
#endif
    if (auto surfaceNode = node->ReinterpretCastTo<RSSurfaceRenderNode>()) {
        // Determine whether cache can be used
        auto curNode = surfaceNode;
        auto parentNode = RSBaseRenderNode::ReinterpretCast<RSSurfaceRenderNode>(surfaceNode->GetParent().lock());
        if (parentNode && parentNode->IsLeashWindow() && parentNode->ShouldPaint() &&
            (parentNode->GetLastFrameUifirstFlag() == MultiThreadCacheType::LEASH_WINDOW ||
            parentNode->GetLastFrameUifirstFlag() == MultiThreadCacheType::NONFOCUS_WINDOW)) {
            curNode = parentNode;
        }

        nodeDrawable_ = std::static_pointer_cast<DrawableV2::RSRenderNodeDrawable>(
            DrawableV2::RSRenderNodeDrawableAdapter::OnGenerate(curNode));
        if (IsRectValid(nodeId_, targetRect)) {
            pixelMap_ = CreatePixelMapByRect(targetRect);
        } else {
            pixelMap_ = CreatePixelMapByNode(curNode);
        }
    } else if (auto canvasNode = node->ReinterpretCastTo<RSCanvasRenderNode>()) {
        nodeDrawable_ = std::static_pointer_cast<DrawableV2::RSRenderNodeDrawable>(
            DrawableV2::RSRenderNodeDrawableAdapter::OnGenerate(canvasNode));
        if (IsRectValid(nodeId_, targetRect)) {
            pixelMap_ = CreatePixelMapByRect(targetRect);
        } else {
            pixelMap_ = CreatePixelMapByNode(canvasNode);
        }
    } else {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: Invalid RSRenderNode!");
        return false;
    }

    if (pixelMap_ == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateResources: pixelMap_ is nullptr!");
        return false;
    }
    return true;
}

bool RSUiCaptureTaskParallel::Run(sptr<RSISurfaceCaptureCallback> callback, const Drawing::Rect& specifiedAreaRect)
{
    RS_TRACE_NAME_FMT("RSUiCaptureTaskParallel::TakeSurfaceCapture. NodeId: [%" PRIu64 "], from pid: [%d]",
        nodeId_, ExtractPid(nodeId_));

#if defined(RS_ENABLE_GL) || defined(RS_ENABLE_VK)
    auto renderContext = RSUniRenderThread::Instance().GetRenderEngine()->GetRenderContext();
    auto grContext = renderContext != nullptr ? renderContext->GetDrGPUContext() : nullptr;
    std::string nodeName("RSUiCaptureTaskParallel");
    RSTagTracker tagTracker(renderContext != nullptr ? renderContext->GetSharedDrGPUContext() : nullptr,
        nodeId_, RSTagTracker::TAGTYPE::TAG_CAPTURE, nodeName);
#endif
    auto surface = CreateSurface(pixelMap_);
    if (surface == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::Run: surface is nullptr!");
        return false;
    }
    if (!nodeDrawable_) {
        RS_LOGE("RSUiCaptureTaskParallel::Run: Invalid RSRenderNodeDrawable!");
        return false;
    }

    RSPaintFilterCanvas canvas(surface.get());
    canvas.Scale(captureConfig_.scaleX, captureConfig_.scaleY);
    canvas.SetDisableFilterCache(true);
    canvas.SetUICapture(true);
    const auto& nodeParams = nodeDrawable_->GetRenderParams();
    if (UNLIKELY(!nodeParams)) {
        RS_LOGE("RSUiCaptureTaskParallel::Run: RenderParams is nullptr!");
        return false;
    }
#ifdef RS_PROFILER_ENABLED
    // check if capturing was triggered, if so - add the recording canvas
    if (auto canvasRec = RSCaptureRecorder::GetInstance().TryComponentScreenshotCapture(
        static_cast<float>(canvas.GetWidth()), static_cast<float>(canvas.GetHeight()))) {
        canvas.AddCanvas(canvasRec);
    }
#endif
    Drawing::Matrix relativeMatrix = Drawing::Matrix();
    relativeMatrix.Set(Drawing::Matrix::Index::SCALE_X, captureConfig_.scaleX);
    relativeMatrix.Set(Drawing::Matrix::Index::SCALE_Y, captureConfig_.scaleY);
    int32_t rectLeft = specifiedAreaRect.GetLeft();
    int32_t rectTop = specifiedAreaRect.GetTop();
    Drawing::scalar xOffset = static_cast<Drawing::scalar>(-1 * rectLeft);
    Drawing::scalar yOffset = static_cast<Drawing::scalar>(-1 * rectTop);
    relativeMatrix.Set(Drawing::Matrix::Index::TRANS_X, xOffset);
    relativeMatrix.Set(Drawing::Matrix::Index::TRANS_Y, yOffset);
    RS_LOGD("RSUiCaptureTaskParallel::Run: specifiedAreaRect offsetX is [%{public}f], offsetY is [%{public}f]",
        xOffset, yOffset);
    Drawing::Matrix invertMatrix;
    if (HasEndNodeRect()) {
        if (endMatrix_.Invert(invertMatrix)) {
            relativeMatrix.PreConcat(invertMatrix);
        }
        relativeMatrix.PreConcat(startMatrix_);
    } else {
        if (nodeParams->GetMatrix().Invert(invertMatrix)) {
            relativeMatrix.PreConcat(invertMatrix);
        }
    }
    canvas.SetMatrix(relativeMatrix);

    RSUiFirstProcessStateCheckerHelper stateCheckerHepler(
        nodeParams->GetFirstLevelNodeId(), nodeParams->GetUifirstRootNodeId());
    RSUniRenderThread::SetCaptureParam(CaptureParam(true, true, false, false, false, false, false, false,
        captureConfig_.uiCaptureInRangeParam.endNodeId));
    DrawableV2::RSRenderNodeDrawable::ClearSnapshotProcessedNodeCount();
    if (HasEndNodeRect() && !isStartEndNodeSame_) {
        auto offScreenWidth = nodeParams->GetBounds().GetWidth();
        auto offScreenHeight = nodeParams->GetBounds().GetHeight();
        auto originSurface = canvas.GetSurface();
        if (originSurface == nullptr) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: originSurface is nullptr");
            return false;
        }
        auto offScreenSurface = originSurface->MakeSurface(offScreenWidth, offScreenHeight);
        if (offScreenSurface == nullptr) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: offScreenSurface is nullptr");
            return false;
        }
        RSPaintFilterCanvas offScreenCanvas(offScreenSurface.get());
        offScreenCanvas.SetDisableFilterCache(true);
        offScreenCanvas.SetUICapture(true);
        offScreenCanvas.CopyHDRConfiguration(canvas);
        offScreenCanvas.CopyConfigurationToOffscreenCanvas(canvas);
        offScreenCanvas.ResetMatrix();
        nodeDrawable_->OnCapture(offScreenCanvas);
        auto image = offScreenSurface->GetImageSnapshot();
        if (!image) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: image is nullptr");
            return false;
        }
        RS_LOGD("RSUiCaptureTaskParallel::Run: offScreenSurface width: %{public}d, height: %{public}d",
            offScreenSurface->Width(), offScreenSurface->Height());
        canvas.DrawImage(*image, 0, 0, Drawing::SamplingOptions());
    } else {
        nodeDrawable_->OnCapture(canvas);
    }
    RS_LOGI("RSUiCaptureTaskParallel::Run: NodeId: [%{public}" PRIu64 "], "
        "the number of total processedNodes: [%{public}d]",
        nodeId_, DrawableV2::RSRenderNodeDrawable::GetSnapshotProcessedNodeCount());
    DrawableV2::RSRenderNodeDrawable::ClearSnapshotProcessedNodeCount();
    RSUniRenderThread::ResetCaptureParam();
#ifdef RS_PROFILER_ENABLED
    // finish capturing if started
    RSCaptureRecorder::GetInstance().EndComponentScreenshotCapture();
#endif
#if (defined (RS_ENABLE_GL) || defined (RS_ENABLE_VK)) && (defined RS_ENABLE_EGLIMAGE)
#ifdef RS_ENABLE_UNI_RENDER
    bool snapshotDmaEnabled = system::GetBoolParameter("rosen.snapshotDma.enabled", true);
    bool isEnableFeature = GetFeatureParamValue("CaptureConfig",
        &CaptureBaseParam::IsSnapshotWithDMAEnabled).value_or(false);
    if (snapshotDmaEnabled && isEnableFeature) {
        RSUniRenderUtil::OptimizedFlushAndSubmit(surface, grContext, GetFeatureParamValue("UICaptureConfig",
            &UICaptureParam::IsUseOptimizedFlushAndSubmitEnabled).value_or(false));
        auto copytask =
            RSUiCaptureTaskParallel::CreateSurfaceSyncCopyTask(
                surface, std::move(pixelMap_), nodeId_, captureConfig_, callback);
        if (!copytask) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: create capture task failed!");
            return false;
        }
        RSBackgroundThread::Instance().PostTask(copytask);
        return true;
    } else {
        std::shared_ptr<Drawing::Image> img(surface.get()->GetImageSnapshot());
        if (!img) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: img is nullptr");
            return false;
        }
        if (!CopyDataToPixelMap(img, pixelMap_)) {
            RS_LOGE("RSUiCaptureTaskParallel::Run: CopyDataToPixelMap failed");
            return false;
        }
    }
#endif
#endif
    // To get dump image
    // execute "param set rosen.dumpsurfacetype.enabled 3 && setenforce 0"
    RSBaseRenderUtil::WritePixelMapToPng(*pixelMap_);
    RS_LOGI("RSUiCaptureTaskParallel::Capture DMADisable capture success nodeId:[%{public}" PRIu64
            "], pixelMap width: %{public}d, height: %{public}d",
        nodeId_, pixelMap_->GetWidth(), pixelMap_->GetHeight());
    ProcessUiCaptureCallback(callback, nodeId_, captureConfig_, pixelMap_.get());
    return true;
}

std::unique_ptr<Media::PixelMap> RSUiCaptureTaskParallel::CreatePixelMapByRect(
    const Drawing::Rect& specifiedAreaRect) const
{
    float pixmapWidth = specifiedAreaRect.GetWidth();
    float pixmapHeight = specifiedAreaRect.GetHeight();
    Media::InitializationOptions opts;
    opts.size.width = ceil(pixmapWidth * captureConfig_.scaleX);
    opts.size.height = ceil(pixmapHeight * captureConfig_.scaleY);
    RS_LOGD("RSUiCaptureTaskParallel::CreatePixelMapByRect:"
        " origin pixelmap width is [%{public}f], height is [%{public}f],"
        " created pixelmap width is [%{public}u], height is [%{public}u],"
        " the scale is scaleX:[%{public}f], scaleY:[%{public}f]",
        pixmapWidth, pixmapHeight, opts.size.width, opts.size.height,
        captureConfig_.scaleX, captureConfig_.scaleY);
    return Media::PixelMap::Create(opts);
}

std::unique_ptr<Media::PixelMap> RSUiCaptureTaskParallel::CreatePixelMapByNode(
    std::shared_ptr<RSRenderNode> node) const
{
    float pixmapWidth = node->GetRenderProperties().GetBoundsWidth();
    float pixmapHeight = node->GetRenderProperties().GetBoundsHeight();
    Media::InitializationOptions opts;
    opts.size.width = ceil(pixmapWidth * captureConfig_.scaleX);
    opts.size.height = ceil(pixmapHeight * captureConfig_.scaleY);
    RS_LOGD("RSUiCaptureTaskParallel::CreatePixelMapByNode: NodeId:[%{public}" PRIu64 "],"
        " origin pixelmap width is [%{public}f], height is [%{public}f],"
        " created pixelmap width is [%{public}u], height is [%{public}u],"
        " the scale is scaleX:[%{public}f], scaleY:[%{public}f]",
        node->GetId(), pixmapWidth, pixmapHeight, opts.size.width, opts.size.height,
        captureConfig_.scaleX, captureConfig_.scaleY);
    return Media::PixelMap::Create(opts);
}

std::shared_ptr<Drawing::Surface> RSUiCaptureTaskParallel::CreateSurface(
    const std::unique_ptr<Media::PixelMap>& pixelmap) const
{
    if (pixelmap == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateSurface: pixelmap == nullptr");
        return nullptr;
    }
    auto address = const_cast<uint32_t*>(pixelmap->GetPixel32(0, 0));
    if (address == nullptr) {
        RS_LOGE("RSUiCaptureTaskParallel::CreateSurface: address == nullptr");
        return nullptr;
    }
    Drawing::ImageInfo info = Drawing::ImageInfo{pixelmap->GetWidth(), pixelmap->GetHeight(),
        Drawing::ColorType::COLORTYPE_RGBA_8888, Drawing::AlphaType::ALPHATYPE_PREMUL};

#if (defined RS_ENABLE_GL) && (defined RS_ENABLE_EGLIMAGE)
    if (RSSystemProperties::GetGpuApiType() == GpuApiType::OPENGL) {
        auto renderContext = RSUniRenderThread::Instance().GetRenderEngine()->GetRenderContext();
        if (renderContext == nullptr) {
            RS_LOGE("RSUiCaptureTaskParallel::CreateSurface: renderContext is nullptr");
            return nullptr;
        }
        renderContext->SetUpGpuContext(nullptr);
        return Drawing::Surface::MakeRenderTarget(renderContext->GetDrGPUContext(), false, info);
    }
#endif
#ifdef RS_ENABLE_VK
    if (RSSystemProperties::GetGpuApiType() == GpuApiType::VULKAN ||
        RSSystemProperties::GetGpuApiType() == GpuApiType::DDGR) {
        return Drawing::Surface::MakeRenderTarget(
            RSUniRenderThread::Instance().GetRenderEngine()->GetSkContext().get(), false, info);
    }
#endif

    return Drawing::Surface::MakeRasterDirect(info, address, pixelmap->GetRowBytes());
}

#ifdef RS_ENABLE_UNI_RENDER
std::function<void()> RSUiCaptureTaskParallel::CreateSurfaceSyncCopyTask(
    std::shared_ptr<Drawing::Surface> surface, std::unique_ptr<Media::PixelMap> pixelMap,
    NodeId id, const RSSurfaceCaptureConfig& captureConfig, sptr<RSISurfaceCaptureCallback> callback,
    int32_t rotation)
{
    Drawing::BackendTexture backendTexture = surface->GetBackendTexture();
    if (!backendTexture.IsValid()) {
        RS_LOGE("RSUiCaptureTaskParallel: SkiaSurface bind Image failed: BackendTexture is invalid");
        return {};
    }
    auto wrapper = std::make_shared<std::tuple<std::unique_ptr<Media::PixelMap>>>();
    std::get<0>(*wrapper) = std::move(pixelMap);
    auto wrapperSf = std::make_shared<std::tuple<std::shared_ptr<Drawing::Surface>>>();
    std::get<0>(*wrapperSf) = std::move(surface);
    std::function<void()> copytask = [
        wrapper, captureConfig, callback, backendTexture, wrapperSf, id, rotation]() -> void {
        RS_TRACE_NAME_FMT("copy and send capture useDma:%d", captureConfig.useDma);
        if (!backendTexture.IsValid()) {
            RS_LOGE("RSUiCaptureTaskParallel: Surface bind Image failed: BackendTexture is invalid");
            ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
            RSUniRenderUtil::ClearNodeCacheSurface(
                std::move(std::get<0>(*wrapperSf)), nullptr, UNI_MAIN_THREAD_INDEX, 0);
            return;
        }
        auto pixelmap = std::move(std::get<0>(*wrapper));
        if (pixelmap == nullptr) {
            RS_LOGE("RSUiCaptureTaskParallel: pixelmap == nullptr");
            ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
            RSUniRenderUtil::ClearNodeCacheSurface(
                std::move(std::get<0>(*wrapperSf)), nullptr, UNI_MAIN_THREAD_INDEX, 0);
            return;
        }

        Drawing::ImageInfo info = Drawing::ImageInfo{ pixelmap->GetWidth(), pixelmap->GetHeight(),
            Drawing::COLORTYPE_RGBA_8888, Drawing::ALPHATYPE_PREMUL };
        Drawing::TextureOrigin textureOrigin = Drawing::TextureOrigin::BOTTOM_LEFT;
        Drawing::BitmapFormat bitmapFormat =
            Drawing::BitmapFormat{ Drawing::COLORTYPE_RGBA_8888, Drawing::ALPHATYPE_PREMUL };
        std::shared_ptr<Drawing::Surface> surface;
        auto grContext = RSBackgroundThread::Instance().GetShareGPUContext();
        if (!grContext) {
            RS_LOGE("RSUiCaptureTaskParallel: SharedGPUContext get failed");
            ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
            return;
        }
#if defined(ROSEN_OHOS) && defined(RS_ENABLE_VK)
        DmaMem dmaMem;
        if (captureConfig.useDma && GetFeatureParamValue("UICaptureConfig",
            &UICaptureParam::IsUseDMAProcessEnabled).value_or(false) &&
            (RSSystemProperties::GetGpuApiType() == GpuApiType::VULKAN ||
            RSSystemProperties::GetGpuApiType() == GpuApiType::DDGR)) {
            sptr<SurfaceBuffer> surfaceBuffer = dmaMem.DmaMemAlloc(info, pixelmap);
            surface = dmaMem.GetSurfaceFromSurfaceBuffer(surfaceBuffer, grContext);
            if (surface == nullptr) {
                RS_LOGE("RSUiCaptureTaskParallel: GetSurfaceFromSurfaceBuffer fail.");
                ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
                RSUniRenderUtil::ClearNodeCacheSurface(
                    std::move(std::get<0>(*wrapperSf)), nullptr, UNI_MAIN_THREAD_INDEX, 0);
                return;
            }
            auto tmpImg = std::make_shared<Drawing::Image>();
            DrawCapturedImg(*tmpImg, *surface, backendTexture, textureOrigin, bitmapFormat);
            RS_LOGI("RSUiCaptureTaskParallel::Capture DMA success nodeId:[%{public}" PRIu64
                "], pixelMap width: %{public}d, height: %{public}d",
                id, pixelmap->GetWidth(), pixelmap->GetHeight());
        } else {
#else
        {
#endif
            auto tmpImg = std::make_shared<Drawing::Image>();
            tmpImg->BuildFromTexture(*grContext, backendTexture.GetTextureInfo(),
                textureOrigin, bitmapFormat, nullptr);
            if (!CopyDataToPixelMap(tmpImg, pixelmap)) {
                RS_LOGE("RSUiCaptureTaskParallel: CopyDataToPixelMap failed");
                ProcessUiCaptureCallback(callback, id, captureConfig, nullptr);
                RSUniRenderUtil::ClearNodeCacheSurface(
                    std::move(std::get<0>(*wrapperSf)), nullptr, UNI_MAIN_THREAD_INDEX, 0);
                return;
            }
        }
        if (rotation) {
            pixelmap->rotate(rotation);
        }
        // To get dump image
        // execute "param set rosen.dumpsurfacetype.enabled 3 && setenforce 0"
        RSBaseRenderUtil::WritePixelMapToPng(*pixelmap);
        auto pixelDump = PixelMapSamplingDump(pixelmap, pixelmap->GetWidth() / 2, 0) |
                         PixelMapSamplingDump(pixelmap, 0, pixelmap->GetHeight() / 2) |
                         PixelMapSamplingDump(pixelmap, pixelmap->GetWidth() / 2, pixelmap->GetHeight() / 2) |
                         PixelMapSamplingDump(pixelmap, pixelmap->GetWidth() - 1, pixelmap->GetHeight() / 2) |
                         PixelMapSamplingDump(pixelmap, pixelmap->GetWidth() / 2, pixelmap->GetHeight() - 1);
        if ((pixelDump & ALPHA_MASK) == 0) {
            RS_LOGW("RSUiCaptureTaskParallel::CreateSurfaceSyncCopyTask pixelmap is transparent");
        }
        RS_LOGI("RSUiCaptureTaskParallel::Capture capture success nodeId:[%{public}" PRIu64
                "], pixelMap width: %{public}d, height: %{public}d",
            id, pixelmap->GetWidth(), pixelmap->GetHeight());
        ProcessUiCaptureCallback(callback, id, captureConfig, pixelmap.get());
        RSBackgroundThread::Instance().CleanGrResource();
        RSUniRenderUtil::ClearNodeCacheSurface(
            std::move(std::get<0>(*wrapperSf)), nullptr, UNI_MAIN_THREAD_INDEX, 0);
    };
    return copytask;
}
#endif

void RSUiCaptureTaskParallel::ProcessUiCaptureCallback(sptr<RSISurfaceCaptureCallback> callback, NodeId id,
    const RSSurfaceCaptureConfig& captureConfig, Media::PixelMap* pixelmap)
{
    callback->OnSurfaceCapture(id, captureConfig, pixelmap);
    RSUiCaptureTaskParallel::captureCount_--;
    RSMainThread::Instance()->RequestNextVSync();
}

bool RSUiCaptureTaskParallel::HasEndNodeRect() const
{
    return captureConfig_.uiCaptureInRangeParam.endNodeId != INVALID_NODEID &&
        !captureConfig_.uiCaptureInRangeParam.useBeginNodeSize;
}

bool RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect()
{
    auto startNode = RSMainThread::Instance()->GetContext().GetNodeMap().GetRenderNode(nodeId_);
    if (!startNode) {
        RS_LOGE("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect start node nullptr %{public}" PRIu64, nodeId_);
        return false;
    }
    startRect_ = startNode->GetRenderProperties().GetBoundsGeometry()->GetAbsRect();
    RS_LOGI("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect startRect %{public}s", startRect_.ToString().c_str());

    NodeId endNodeId = captureConfig_.uiCaptureInRangeParam.endNodeId;
    endNodeId_ = endNodeId;
    if (!HasEndNodeRect()) {
        RS_LOGE("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect end node invalid %{public}" PRIu64, endNodeId);
        return false;
    }

    auto endNode = RSMainThread::Instance()->GetContext().GetNodeMap().GetRenderNode(endNodeId);
    if (!endNode) {
        RS_LOGE("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect end node nullptr %{public}" PRIu64, endNodeId);
        return false;
    }
    std::vector<NodeId> nodeIdVec;
    startNode->CollectAllChildren(startNode, nodeIdVec);
    if (std::find(nodeIdVec.begin(), nodeIdVec.end(), endNodeId) == nodeIdVec.end()) {
        RS_LOGE("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect endNode not on tree %{public}" PRIu64, endNodeId);
        return false;
    }

    endRect_ = endNode->GetRenderProperties().GetBoundsGeometry()->GetAbsRect();
    RS_LOGI("RSUiCaptureTaskParallel::UpdateStartAndEndNodeRect endRect %{public}s", endRect_.ToString().c_str());
    startMatrix_ = startNode->GetRenderProperties().GetBoundsGeometry()->GetAbsMatrix();
    endMatrix_ = endNode->GetRenderProperties().GetBoundsGeometry()->GetAbsMatrix();

    return true;
}

bool RSUiCaptureTaskParallel::IsStartEndSameNode()
{
    isStartEndNodeSame_ = (nodeId_ == captureConfig_.uiCaptureInRangeParam.endNodeId);
    RS_LOGD("RSUiCaptureTaskParallel::IsStartEndSameNode isStartEndNodeSame_: %{public}d", isStartEndNodeSame_);
    return isStartEndNodeSame_;
}
} // namespace Rosen
} // namespace OHOS