1 /* 2 * Copyright (c) 2021-2023 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef RS_SURFACE_CAPTURE_TASK 17 #define RS_SURFACE_CAPTURE_TASK 18 #define EGL_EGLEXT_PROTOTYPES 19 #define GL_GLEXT_PROTOTYPES 20 21 #include "common/rs_common_def.h" 22 #include "system/rs_system_parameters.h" 23 #include "draw/canvas.h" 24 #include "draw/surface.h" 25 #include "utils/matrix.h" 26 #include "ipc_callbacks/surface_capture_callback.h" 27 #include "pipeline/render_thread/rs_base_render_engine.h" 28 #include "pipeline/rs_screen_render_node.h" 29 #include "pipeline/rs_effect_render_node.h" 30 #include "pipeline/rs_surface_render_node.h" 31 #include "pixel_map.h" 32 #include "visitor/rs_node_visitor.h" 33 namespace OHOS { 34 namespace Rosen { 35 bool CopyDataToPixelMap(std::shared_ptr<Drawing::Image> img, const std::unique_ptr<Media::PixelMap>& pixelmap); 36 class RSSurfaceCaptureVisitor : public RSNodeVisitor { 37 public: 38 RSSurfaceCaptureVisitor(const RSSurfaceCaptureConfig& captureConfig, bool isUniRender); 39 ~RSSurfaceCaptureVisitor() noexcept override = default; PrepareChildren(RSRenderNode & node)40 void PrepareChildren(RSRenderNode& node) override {} PrepareCanvasRenderNode(RSCanvasRenderNode & node)41 void PrepareCanvasRenderNode(RSCanvasRenderNode& node) override {} PrepareLogicalDisplayRenderNode(RSLogicalDisplayRenderNode & node)42 void PrepareLogicalDisplayRenderNode(RSLogicalDisplayRenderNode& node) override {} PrepareProxyRenderNode(RSProxyRenderNode & node)43 void PrepareProxyRenderNode(RSProxyRenderNode& node) override {} PrepareRootRenderNode(RSRootRenderNode & node)44 void PrepareRootRenderNode(RSRootRenderNode& node) override {} PrepareScreenRenderNode(RSScreenRenderNode & node)45 void PrepareScreenRenderNode(RSScreenRenderNode& node) override {} PrepareSurfaceRenderNode(RSSurfaceRenderNode & node)46 void PrepareSurfaceRenderNode(RSSurfaceRenderNode& node) override {} PrepareEffectRenderNode(RSEffectRenderNode & node)47 void PrepareEffectRenderNode(RSEffectRenderNode& node) override {} 48 49 void ProcessChildren(RSRenderNode& node) override; ProcessCanvasRenderNode(RSCanvasRenderNode & node)50 void ProcessCanvasRenderNode(RSCanvasRenderNode& node) override {} 51 void ProcessLogicalDisplayRenderNode(RSLogicalDisplayRenderNode& node) override; ProcessProxyRenderNode(RSProxyRenderNode & node)52 void ProcessProxyRenderNode(RSProxyRenderNode& node) override {} ProcessRootRenderNode(RSRootRenderNode & node)53 void ProcessRootRenderNode(RSRootRenderNode& node) override {} 54 void ProcessScreenRenderNode(RSScreenRenderNode& node) override; 55 void ProcessSurfaceRenderNode(RSSurfaceRenderNode& node) override; ProcessEffectRenderNode(RSEffectRenderNode & node)56 void ProcessEffectRenderNode(RSEffectRenderNode& node) override {} 57 58 void SetSurface(Drawing::Surface* surface); IsDisplayNode(bool isDisplayNode)59 void IsDisplayNode(bool isDisplayNode) 60 { 61 isDisplayNode_ = isDisplayNode; 62 } 63 IsUniRender()64 bool IsUniRender() const 65 { 66 return isUniRender_; 67 } 68 69 private: 70 void ProcessSurfaceRenderNodeWithoutUni(RSSurfaceRenderNode& node); 71 void CaptureSingleSurfaceNodeWithoutUni(RSSurfaceRenderNode& node); 72 void CaptureSurfaceInDisplayWithoutUni(RSSurfaceRenderNode& node); 73 74 std::unique_ptr<RSPaintFilterCanvas> canvas_ = nullptr; 75 bool isDisplayNode_ = false; 76 RSSurfaceCaptureConfig captureConfig_ = {}; 77 bool isUniRender_ = false; 78 std::shared_ptr<RSBaseRenderEngine> renderEngine_ = nullptr; 79 }; 80 81 class RSSurfaceCaptureTask { 82 public: RSSurfaceCaptureTask(NodeId nodeId,const RSSurfaceCaptureConfig & captureConfig)83 explicit RSSurfaceCaptureTask(NodeId nodeId, 84 const RSSurfaceCaptureConfig& captureConfig) 85 : nodeId_(nodeId), captureConfig_(captureConfig) {} 86 ~RSSurfaceCaptureTask() = default; 87 88 bool Run(sptr<RSISurfaceCaptureCallback> callback); 89 90 private: 91 std::shared_ptr<RSSurfaceCaptureVisitor> visitor_ = nullptr; 92 93 std::shared_ptr<Drawing::Surface> CreateSurface(const std::unique_ptr<Media::PixelMap>& pixelmap, 94 std::shared_ptr<Drawing::ColorSpace> colorSpace = nullptr); 95 96 std::unique_ptr<Media::PixelMap> CreatePixelMapBySurfaceNode(std::shared_ptr<RSSurfaceRenderNode> node, 97 bool isUniRender = false); 98 99 std::unique_ptr<Media::PixelMap> CreatePixelMapByDisplayNode(std::shared_ptr<RSLogicalDisplayRenderNode> node, 100 bool isUniRender = false); 101 102 NodeId nodeId_ = INVALID_NODEID; 103 104 RSSurfaceCaptureConfig captureConfig_ = {}; 105 106 ScreenRotation screenCorrection_ = ScreenRotation::ROTATION_0; 107 }; 108 } // namespace Rosen 109 } // namespace OHOS 110 111 #endif // RS_SURFACE_CAPTURE_TASK 112