windowmanager/utils/include/surface_reader.h

/*
 * Copyright (c) 2021-2022 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.
 */

#ifndef SURFACE_READER_H
#define SURFACE_READER_H

#include "refbase.h"

#include <surface.h>

#include "surface_reader_handler.h"

namespace OHOS {
namespace Rosen {
class SurfaceReader {
public:
    SurfaceReader();
    virtual ~SurfaceReader();

    bool Init();
    void DeInit();

    sptr<Surface> GetSurface() const;
    void SetHandler(sptr<SurfaceReaderHandler> handler);
private:
    class BufferListener : public IBufferConsumerListener {
    public:
        explicit BufferListener(SurfaceReader &surfaceReader): surfaceReader_(surfaceReader)
        {
        }
        ~BufferListener() noexcept override = default;
        void OnBufferAvailable() override
        {
            surfaceReader_.OnVsync();
        }

    private:
        SurfaceReader &surfaceReader_;
    };
    friend class BufferListener;

    void OnVsync();
    bool ProcessBuffer(const sptr<SurfaceBuffer> &buf);

    sptr<IBufferConsumerListener> listener_ = nullptr;
    sptr<Surface> csurface_ = nullptr; // cosumer surface
    sptr<Surface> psurface_ = nullptr; // producer surface
    sptr<SurfaceBuffer> prevBuffer_ = nullptr;
    sptr<SurfaceReaderHandler> handler_ = nullptr;
};
}
}

#endif // SURFACE_READER_H