/*
 * Copyright (c) 2023-2024 Shenzhen Kaihong Digital Industry Development 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 OHOS_SHARING_SCREEN_CAPTURE_CONSUMER_H
#define OHOS_SHARING_SCREEN_CAPTURE_CONSUMER_H

#include <mutex>
#include "audio_aac_codec.h"
#include "audio_source_capturer.h"
#include "codec_factory.h"
#include "magic_enum.hpp"
#include "mediachannel/base_consumer.h"
#include "video_source_encoder.h"
#include "video_source_screen.h"

namespace OHOS {
namespace Sharing {
class ScreenCaptureConsumer : public BaseConsumer,
                              public VideoSourceEncoderListener,
                              public std::enable_shared_from_this<ScreenCaptureConsumer> {
public:
    class AudioEncoderReceiver : public FrameDestination {
    public:
        explicit AudioEncoderReceiver(std::weak_ptr<ScreenCaptureConsumer> parent) : parent_(parent){};

        void OnFrame(const Frame::Ptr &frame) override;

    private:
        std::weak_ptr<ScreenCaptureConsumer> parent_;
    };

    ScreenCaptureConsumer();
    ~ScreenCaptureConsumer();

public:
    int32_t Release() override;
    int32_t ReleaseScreenBuffer();
    int32_t HandleEvent(SharingEvent &event) override;

    void OnInitVideoCaptureError();
    void OnFrameBufferUsed() override;
    void UpdateOperation(ProsumerStatusMsg::Ptr &statusMsg) override;
    void OnFrame(const Frame::Ptr &frame, FRAME_TYPE frameType, bool keyFrame) override;

private:
    bool IsPaused();
    bool StopCapture();
    bool StartCapture();
    bool StopAudioCapture();
    bool StopVideoCapture();
    bool StartAudioCapture();
    bool StartVideoCapture();

    bool InitAudioCapture();
    bool InitAudioEncoder();
    bool Init(uint64_t screenId);
    bool InitVideoCapture(uint64_t screenId);

    void HandleProsumerInitState(SharingEvent &event);
    void HandleSpsFrame(BufferDispatcher::Ptr dispatcher, const Frame::Ptr &frame);
    void HandlePpsFrame(BufferDispatcher::Ptr dispatcher, const Frame::Ptr &frame);

private:
    bool paused_ = false;

    std::mutex mutex_;

    std::shared_ptr<VideoSourceScreen> videoSourceScreen_ = nullptr;
    std::shared_ptr<VideoSourceEncoder> videoSourceEncoder_ = nullptr;

    std::shared_ptr<AudioEncoder> audioEncoder_ = nullptr;
    std::shared_ptr<AudioSourceCapturer> audioSourceCapturer_ = nullptr;
    std::shared_ptr<AudioEncoderReceiver> audioEncoderReceiver_ = nullptr;
};

} // namespace Sharing
} // namespace OHOS
#endif