// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef EASYMEDIA_MPP_ENCODER_H
#define EASYMEDIA_MPP_ENCODER_H

#include "encoder.h"
#include "mpp_inc.h"
#include "mpp_rc_api.h"
#include "rk_venc_cfg.h"

namespace easymedia
{

    // A encoder which call the mpp interface directly.
    // Mpp is always video process module.
    class MPPEncoder : public VideoEncoder
    {
      public:
        MPPEncoder();
        virtual ~MPPEncoder();

        virtual bool Init() override;

        // sync encode the raw input buffer to output buffer
        virtual int Process(const std::shared_ptr<MediaBuffer>& input, std::shared_ptr<MediaBuffer>& output,
                            std::shared_ptr<MediaBuffer> extra_output) override;

        virtual int SendInput(const std::shared_ptr<MediaBuffer>&) override;
        virtual std::shared_ptr<MediaBuffer> FetchOutput() override;

        // Encoder statistics enable switch. After enabling, calculate the
        // actual bps and actual fps every theoretical fps period.
        // Input parameters:
        //   value = true, enable encoder statistics.
        //   value = false, disable encoding statistics.
        void set_statistics_switch(bool value);
        int get_statistics_bps();
        int get_statistics_fps();

#ifdef MPP_SUPPORT_HW_OSD
        int OsdPaletteSet(uint32_t* ptl_data);
        int OsdRegionSet(OsdRegionData* region_data);
        int OsdRegionGet(OsdRegionData* region_data);
#endif

        // Set sei info by userdata.
        int SetUserData(const char* data, uint16_t len);
        void ClearUserData();
        void RestartUserData();
        void EnableUserDataAllFrame(bool value);

        // for updating roi regions config.
        int RoiUpdateRegions(EncROIRegion* regions, int region_cnt);

        virtual void QueryChange(uint32_t change, void* value, int32_t size);

      protected:
        MppCodingType coding_type;
        uint32_t output_mb_flags;
        std::string rc_api_brief_name;
        // call before Init()
        void SetMppCodeingType(MppCodingType type);
        virtual bool CheckConfigChange(std::pair<uint32_t, std::shared_ptr<ParameterBuffer>>)
        {
            return true;
        }
        // Control before encoding.
        int EncodeControl(int cmd, void* param);

        virtual int PrepareMppFrame(const std::shared_ptr<MediaBuffer>& input, std::shared_ptr<MediaBuffer>& mdinfo,
                                    MppFrame& frame);
        virtual int PrepareMppPacket(std::shared_ptr<MediaBuffer>& output, MppPacket& packet);
        virtual int PrepareMppExtraBuffer(std::shared_ptr<MediaBuffer> extra_output, MppBuffer& buffer);
        int Process(MppFrame frame, MppPacket& packet, MppBuffer& mv_buf);

      private:
        std::shared_ptr<MPPContext> mpp_ctx;

        // Statistics switch
        bool encoder_sta_en;
        // Statistics variable
        uint32_t encoded_bps;
        float encoded_fps;

        size_t stream_size_1s;
        size_t frame_cnt_1s;
        int64_t last_ts;
        int64_t cur_ts;

#ifdef MPP_SUPPORT_HW_OSD
        MppEncOSDData osd_data;
#endif // MPP_SUPPORT_HW_OSD

        // for roi regions config.
        MppEncROICfg roi_cfg;

#define MPP_ENCODER_USERDATA_MAX_SIZE 1024
        char userdata[MPP_ENCODER_USERDATA_MAX_SIZE];
        uint16_t userdata_len;
        uint16_t userdata_frame_id;
        uint8_t userdata_all_frame_en;
        MppEncUserData mpp_ud;

        friend class MPPMJPEGConfig;
        friend class MPPCommonConfig;
    };

} // namespace easymedia

#endif // EASYMEDIA_MPP_ENCODER_H