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/rs_display_render_node.h" 28 #include "pipeline/rs_effect_render_node.h" 29 #include "pipeline/rs_surface_render_node.h" 30 #include "pixel_map.h" 31 #include "rs_base_render_engine.h" 32 #include "visitor/rs_node_visitor.h" 33 34 namespace OHOS { 35 namespace Rosen { 36 bool CopyDataToPixelMap(std::shared_ptr<Drawing::Image> img, const std::unique_ptr<Media::PixelMap>& pixelmap, 37 std::shared_ptr<Drawing::ColorSpace> colorSpace = nullptr); 38 class RSSurfaceCaptureVisitor : public RSNodeVisitor { 39 public: 40 RSSurfaceCaptureVisitor(const RSSurfaceCaptureConfig& captureConfig, bool isUniRender); 41 ~RSSurfaceCaptureVisitor() noexcept override = default; PrepareChildren(RSRenderNode & node)42 void PrepareChildren(RSRenderNode& node) override {} PrepareCanvasRenderNode(RSCanvasRenderNode & node)43 void PrepareCanvasRenderNode(RSCanvasRenderNode& node) override {} PrepareDisplayRenderNode(RSDisplayRenderNode & node)44 void PrepareDisplayRenderNode(RSDisplayRenderNode& node) override {} PrepareProxyRenderNode(RSProxyRenderNode & node)45 void PrepareProxyRenderNode(RSProxyRenderNode& node) override {} PrepareRootRenderNode(RSRootRenderNode & node)46 void PrepareRootRenderNode(RSRootRenderNode& node) override {} PrepareSurfaceRenderNode(RSSurfaceRenderNode & node)47 void PrepareSurfaceRenderNode(RSSurfaceRenderNode& node) override {} PrepareEffectRenderNode(RSEffectRenderNode & node)48 void PrepareEffectRenderNode(RSEffectRenderNode& node) override {} 49 50 void ProcessChildren(RSRenderNode& node) override; ProcessCanvasRenderNode(RSCanvasRenderNode & node)51 void ProcessCanvasRenderNode(RSCanvasRenderNode& node) override {} 52 void ProcessDisplayRenderNode(RSDisplayRenderNode& node) override; ProcessProxyRenderNode(RSProxyRenderNode & node)53 void ProcessProxyRenderNode(RSProxyRenderNode& node) override {} ProcessRootRenderNode(RSRootRenderNode & node)54 void ProcessRootRenderNode(RSRootRenderNode& 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 bool Run(sptr<RSISurfaceCaptureCallback> callback); 88 private: 89 std::shared_ptr<RSSurfaceCaptureVisitor> visitor_ = nullptr; 90 91 std::shared_ptr<Drawing::Surface> CreateSurface(const std::unique_ptr<Media::PixelMap>& pixelmap, 92 std::shared_ptr<Drawing::ColorSpace> colorSpace = nullptr); 93 94 std::unique_ptr<Media::PixelMap> CreatePixelMapBySurfaceNode(std::shared_ptr<RSSurfaceRenderNode> node, 95 bool isUniRender = false); 96 97 std::unique_ptr<Media::PixelMap> CreatePixelMapByDisplayNode(std::shared_ptr<RSDisplayRenderNode> node, 98 bool isUniRender = false); 99 NodeId nodeId_ = INVALID_NODEID; 100 101 RSSurfaceCaptureConfig captureConfig_ = {}; 102 103 ScreenRotation screenCorrection_ = ScreenRotation::ROTATION_0; 104 }; 105 } // namespace Rosen 106 } // namespace OHOS 107 108 #endif // RS_SURFACE_CAPTURE_TASK 109