/***************************************************************************************
 *
 *  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
 *
 *  By downloading, copying, installing or using the software you agree to this license.
 *  If you do not agree to this license, do not download, install, 
 *  copy or use the software.
 *
 *  Copyright (C) 2014-2022, Happytimesoft Corporation, all rights reserved.
 *
 *  Redistribution and use in binary forms, with or without modification, are permitted.
 *
 *  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 _VIDEO_ENCODER_H
#define _VIDEO_ENCODER_H

#include "linked_list.h"

extern "C" 
{
#include <libavutil/opt.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
#include <libswresample/swresample.h>
#include <libavutil/intreadwrite.h>
#include <libavutil/avstring.h>
#include <libavutil/base64.h>
#include <libavutil/imgutils.h>
}

typedef void (*VideoDataCallback)(uint8 *data, int size, int waitnext, void *pUserdata);

typedef struct
{
	VideoDataCallback pCallback;
	void *            pUserdata;
	BOOL              bFirst;
} VideoEncoderCB;

typedef struct
{
    int             SrcWidth;       // source video width
    int             SrcHeight;      // source video height
    AVPixelFormat   SrcPixFmt;      // source video pixel format
    int             Updown;         // Image needs to be upside down
    
    int             DstCodec;       // output video codec, see media_format.h
    int             DstWidth;       // output video width
    int             DstHeight;      // output video height
    double          DstFramerate;   // output video frame rate
    int             DstBitrate;     // output bitrate
} VideoEncoderParam;

class CVideoEncoder
{
public:
    CVideoEncoder();
    ~CVideoEncoder();

    BOOL                init(VideoEncoderParam * params);
    void                uninit();
    BOOL                encode(uint8 * data, int size);
    BOOL                encode(AVFrame * pFrame);
    void                addCallback(VideoDataCallback pCallback, void *pUserdata);
    void                delCallback(VideoDataCallback pCallback, void *pUserdata);
    char *              getAuxSDPLine(int rtp_pt);
    BOOL                getExtraData(uint8 ** extradata, int * extralen);
    void                procData(uint8 * data, int size);
    
private:    
    void                flush();    
    BOOL                isCallbackExist(VideoDataCallback pCallback, void *pUserdata);
    char *              getH264AuxSDPLine(int rtp_pt);
    char *              getH265AuxSDPLine(int rtp_pt);
    char *              getMP4AuxSDPLine(int rtp_pt);
    int                 computeBitrate(AVCodecID codec, int width, int height, double framerate, int quality);

private:
    AVCodecID           m_nCodecId;    
    AVPixelFormat       m_nDstPixFmt;

    VideoEncoderParam   m_EncoderParams;

    AVCodecContext *    m_pCodecCtx;
    AVFrame *           m_pFrameSrc;
    AVFrame *           m_pFrameYuv;
    int64_t             m_pts;
    
    int                 m_nFrameIndex;
    BOOL                m_bInited;

    void *              m_pCallbackMutex;
	LINKED_LIST *       m_pCallbackList;
};

#endif // _VIDEO_ENCODER_H