#include "XPushStream.h"

XPushStream::XPushStream(QObject *parent)
    : QObject{parent}
{

}


bool XPushStream::Init(QString _qstrOutStreamUrl, QString _qstrPushType, AVFormatContext* _InCtx)
{
    if(_qstrPushType.isEmpty())
    {
        return false;
    }

    if(_qstrOutStreamUrl.isEmpty())
    {
        return false;
    }

    if (_InCtx == nullptr)
    {
        return false;
    }

    m_InFmtCtx = _InCtx;

    int ret = 0;
    //如果是rtmp 填写flv  如果是rtsp，则填写rtsp
    //创建context

    if (m_OutFmtCtx == nullptr)
    {
        ret = avformat_alloc_output_context2(&m_OutFmtCtx, NULL, _qstrPushType.toStdString().c_str(), _qstrOutStreamUrl.toStdString().c_str());
    }
    if (!m_OutFmtCtx)
    {
        avformat_free_context(m_OutFmtCtx);
        m_OutFmtCtx = nullptr;
        return false;
    }
    //输出流的相关参数
    for (size_t i = 0; i < m_InFmtCtx->nb_streams; i++)
    {
        AVStream *inStream = m_InFmtCtx->streams[i];
        AVStream *outStream = avformat_new_stream(m_OutFmtCtx, NULL);
        if (!outStream)
        {
            return false;
        }
        ret = avcodec_parameters_from_context(outStream->codecpar, inStream->codec);
        if (ret < 0)
        {
            return false;
        }
        outStream->codecpar->codec_tag = 0;
    }

    AVDictionary * opts = NULL;
    av_dict_set(&opts, "rw_timeout", "1000000", 0);//设置超时时间 单位mcs
    ret = avio_open2(&m_OutFmtCtx->pb, _qstrOutStreamUrl.toStdString().c_str(), AVIO_FLAG_WRITE,NULL,&opts);
    if (ret < 0)
    {
        avformat_free_context(m_OutFmtCtx);
        m_OutFmtCtx = nullptr;
        return false;
    }
    //写入文件头 会改变time_base 所以需要重新计算时间
    avformat_write_header(m_OutFmtCtx, 0);
    return true;
}

int XPushStream::PushStream(AVPacket *_pPushAVPacket)
{
    int ret;
    AVPacket TempPacket;
    m_qmutPushStream.lock();
    av_init_packet(&TempPacket);
    ret = av_packet_ref(&TempPacket, _pPushAVPacket);

    //需要重新计算pts dts
    if (_pPushAVPacket->pts != AV_NOPTS_VALUE)
    {
        AVStream *inStream, *outStream;
        inStream = m_InFmtCtx->streams[_pPushAVPacket->stream_index];
        outStream = m_OutFmtCtx->streams[_pPushAVPacket->stream_index];
        av_packet_rescale_ts(&TempPacket, inStream->time_base, outStream->time_base);
        TempPacket.pos = -1;

        if (TempPacket.stream_index > 2)
        {
            m_qmutPushStream.unlock();
            return -3;
        }
        ret = av_interleaved_write_frame(m_OutFmtCtx, &TempPacket);
    }
    m_qmutPushStream.unlock();
    return ret;
}

void XPushStream::Clear()
{
    if (m_OutFmtCtx)
    {
        //写文件尾
        av_write_trailer(m_OutFmtCtx);
        /* close output */
        if (m_OutFmtCtx && !(m_OutFmtCtx->oformat->flags & AVFMT_NOFILE))
        {
            //关闭输出流
            avio_closep(&m_OutFmtCtx->pb);
        }
        //释放context
        avformat_free_context(m_OutFmtCtx);
        m_OutFmtCtx = nullptr;
    }
}