1 /*
2  * Copyright (c) 2022 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 SURFACE_CAPTURE_FUTURE_H
17 #define SURFACE_CAPTURE_FUTURE_H
18 
19 #include "future.h"
20 #include "pixel_map.h"
21 #include "transaction/rs_render_service_client.h"
22 
23 namespace OHOS {
24 namespace Rosen {
25 class SurfaceCaptureFuture : public SurfaceCaptureCallback, public Future<std::shared_ptr<Media::PixelMap>> {
26 public:
27     SurfaceCaptureFuture() = default;
~SurfaceCaptureFuture()28     ~SurfaceCaptureFuture() {};
OnSurfaceCapture(std::shared_ptr<Media::PixelMap> pixelmap)29     void OnSurfaceCapture(std::shared_ptr<Media::PixelMap> pixelmap) override
30     {
31         FutureCall(pixelmap);
32     }
33 protected:
Call(std::shared_ptr<Media::PixelMap> pixelmap)34     void Call(std::shared_ptr<Media::PixelMap> pixelmap) override
35     {
36         if (!flag_) {
37             pixelMap_ = pixelmap;
38             flag_ = true;
39         }
40     }
IsReady()41     bool IsReady() override
42     {
43         return flag_;
44     }
FetchResult()45     std::shared_ptr<Media::PixelMap> FetchResult() override
46     {
47         return pixelMap_;
48     }
49 private:
50     bool flag_ = false;
51     std::shared_ptr<Media::PixelMap> pixelMap_ = nullptr;
52 };
53 } // Rosen
54 } // OHOS
55 #endif  // SURFACE_CAPTURE_FUTURE_H