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