FFmpeg4.0笔记:rtsp2rtmp

Github

https://github.com/gongluck/FFmpeg4.0-study.git

#include <iostream>
using namespace std;

extern "C"
{
#include "libavformat/avformat.h"
#include "libavutil/time.h"
}
#pragma comment(lib, "avformat.lib")
#pragma comment(lib, "avutil.lib")
#pragma comment(lib, "avcodec.lib")

char av_error[AV_ERROR_MAX_STRING_SIZE] = { 0 };
#define av_err2str(errnum) \
    av_make_error_string(av_error, AV_ERROR_MAX_STRING_SIZE, errnum)

#define INFILE  "in.flv"
#define RTMP    "rtmp://192.168.140.128/live/test"
#define RTSP    "rtsp://184.72.239.149/vod/mp4://BigBuckBunny_175k.mov"

int file2rtmp()
{
    int ret = 0;
    //封装上下文
    AVFormatContext* ictx = nullptr;
    AVFormatContext* octx = nullptr;
    const char* iurl = INFILE;
    const char* ourl = RTMP;
    int64_t starttime;

    ret = avformat_network_init();
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //打开文件,解封文件头
    ret = avformat_open_input(&ictx, iurl, nullptr, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cerr << "open file " << iurl << " success." << endl;

    //获取音视频流信息,h264 flv
    ret = avformat_find_stream_info(ictx, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //打印媒体信息
    av_dump_format(ictx, 0, iurl, 0);

    //////////////////////////////

    //输出流
    ret = avformat_alloc_output_context2(&octx, av_guess_format(nullptr, INFILE, nullptr), nullptr, ourl);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cout << "octx create success." << endl;

    //配置输出流
    for (int i = 0; i < ictx->nb_streams; ++i)
    {
        //创建流
        AVStream* ostream = avformat_new_stream(octx, avcodec_find_encoder(ictx->streams[i]->codecpar->codec_id));
        if (ostream == nullptr)
            goto END;
        //复制配置信息
        ret = avcodec_parameters_copy(ostream->codecpar, ictx->streams[i]->codecpar);
        if (ret != 0)
        {
            cout << av_err2str(ret) << endl;
            goto END;
        }
        ostream->codecpar->codec_tag = 0;//标记不需要重新编解码
    }
    av_dump_format(octx, 0, ourl, 1);

    //////////////////////////////

    //推流
    ret = avio_open2(&octx->pb, ourl, AVIO_FLAG_WRITE, nullptr, nullptr);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //写入头信息
    ret = avformat_write_header(octx, nullptr);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //推流每一帧数据
    AVPacket pkt;
    starttime = av_gettime();
    while (av_read_frame(ictx, &pkt) == 0)
    {
        //计算转换pts dts
        AVRational itime = ictx->streams[pkt.stream_index]->time_base;
        AVRational otime = octx->streams[pkt.stream_index]->time_base;
        pkt.pts = av_rescale_q_rnd(pkt.pts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.dts = av_rescale_q_rnd(pkt.dts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.duration = av_rescale_q_rnd(pkt.duration, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.pos = -1;

        if (ictx->streams[pkt.stream_index]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
        {
            int64_t nowtime = av_gettime() - starttime;
            int64_t dts = pkt.dts * av_q2d(octx->streams[pkt.stream_index]->time_base) * 1000 * 1000;
            if(dts > nowtime)
                /*av_usleep(dts- nowtime)*/;
        }
        
        ret = av_interleaved_write_frame(octx, &pkt);
        av_packet_unref(&pkt);
        if (ret < 0)
        {
            cout << av_err2str(ret) << endl;
            goto END;
        }
    }
    ret = av_write_trailer(octx);//写文件尾
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

END:
    if (ictx != nullptr)
        avformat_close_input(&ictx);
    if (octx != nullptr)
    {
        avio_close(octx->pb);
        avformat_free_context(octx);
    }
    ret = avformat_network_deinit();
    if (ret != 0)
        cout << av_err2str(ret) << endl;
    return 0;
}

int rtsp2rtmp()
{
    int ret = 0;
    //封装上下文
    AVFormatContext* ictx = nullptr;
    AVFormatContext* octx = nullptr;
    const char* iurl = RTSP;
    const char* ourl = RTMP;
    int64_t starttime;

    ret = avformat_network_init();
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //打开文件,解封文件头
    ret = avformat_open_input(&ictx, iurl, nullptr, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cerr << "open file " << iurl << " success." << endl;

    //获取音视频流信息,h264 flv
    ret = avformat_find_stream_info(ictx, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //打印媒体信息
    av_dump_format(ictx, 0, iurl, 0);

    //////////////////////////////

    //输出流
    ret = avformat_alloc_output_context2(&octx, av_guess_format(nullptr, INFILE, nullptr), nullptr, ourl);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cerr << "octx create success." << endl;

    //配置输出流
    for (int i = 0; i < ictx->nb_streams; ++i)
    {
        //创建流
        AVStream* ostream = avformat_new_stream(octx, avcodec_find_encoder(ictx->streams[i]->codecpar->codec_id));
        if (ostream == nullptr)
            return -1;
        //复制配置信息
        ret = avcodec_parameters_copy(ostream->codecpar, ictx->streams[i]->codecpar);
        if (ret != 0)
        {
            cout << av_err2str(ret) << endl;
            goto END;
        }
        ostream->codecpar->codec_tag = 0;//标记不需要重新编解码
    }
    av_dump_format(octx, 0, ourl, 1);

    //////////////////////////////

    //推流
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    ret = avio_open(&octx->pb, ourl, AVIO_FLAG_WRITE);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //写入头信息
    ret = avformat_write_header(octx, nullptr);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }

    //推流每一帧数据
    AVPacket pkt;
    starttime = av_gettime();
    while (av_read_frame(ictx, &pkt) == 0)
    {
        //计算转换pts dts
        AVRational itime = ictx->streams[pkt.stream_index]->time_base;
        AVRational otime = octx->streams[pkt.stream_index]->time_base;
        pkt.pts = av_rescale_q_rnd(pkt.pts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.dts = av_rescale_q_rnd(pkt.dts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.duration = av_rescale_q_rnd(pkt.duration, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.pos = -1;

        ret = av_interleaved_write_frame(octx, &pkt);
        av_packet_unref(&pkt);
        if (ret < 0)
            cout << av_err2str(ret) << endl;//不用退出
    }

END:
    if (ictx != nullptr)
        avformat_close_input(&ictx);
    if (octx != nullptr)
    {
        avio_close(octx->pb);
        avformat_free_context(octx);
    }
    ret = avformat_network_deinit();
    if (ret != 0)
        cout << av_err2str(ret) << endl;
    return 0;
}

int main()
{
    //file2rtmp();
    rtsp2rtmp();
    system("pause");
    return 0;
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏转载gongluck的CSDN博客

利用FFmpeg对火眼一体摄像机的回调数据进行处理:YUV转H264,H264封装flv,所有输入都是在内存中。

整个工程代码下载地址 http://download.csdn.net/download/gongluck93/10175326 Code //#define ...

5136
来自专栏Android源码框架分析

Android内容服务ContentService原理浅析ContentService启动跟实质注册观察者流程通知流程总结

ContentService可以看做Android中一个系统级别的消息中心,可以说搭建了一个系统级的观察者模型,APP可以向消息中心注册观察者,选择订阅自己关心...

1003
来自专栏乐沙弥的世界

ORA-39126 KUPW$WORKER.PUT_DDLS [TABLE_STATISTICS]错误

--======================================================= -- ORA-39126 KUPW$WOR...

1024
来自专栏曾大稳的博客

使用SoudTouch实现变速变调

1082
来自专栏函数式编程语言及工具

Akka(7): FSM:通过状态变化来转换运算行为

  在上篇讨论里我们提到了become/unbecome。由于它们本质上是堆栈操作,所以只能在较少的状态切换下才能保证堆栈操作的协调及维持程序的清晰逻辑。对于比...

5628
来自专栏程序员互动联盟

【Windows编程】创建基本控件

前一篇文章我们一起学习了Windows编程基本框架,几乎所有的Windows编程都是以这个模式开始,剩下的就是如何怎么框架的基础上如何添加枝叶实现不同的功能了。...

2756
来自专栏AhDung

【C#】Excel舍入函数Round、RoundUp、RoundDown的C#版

本人在C#中进行小数舍入的时候常常会怀念Excel中的Round、RoundUp、RoundDown这几个函数,原因就是后者“接地气”,比较符合俺小老百姓的舍入...

722
来自专栏杨建荣的学习笔记

数据迁移部分问题总结(r2第3天)

按照计划在周二开始了数据迁移,本来之前也做了不少的准备工作。但是还是在迁移的过程中出现了一些问题。简单做一个总结。 1.constraint导致的数据rej...

2757
来自专栏杨建荣的学习笔记

海量数据切分抽取的实践场景(r11笔记第43天)

如果一个大表要抽取数据导出成csv文件,我们有什么策略,如何改进。 一、问题背景 今天开发的同学找到我,他们需要做一个数据统计分析,需要我提供一些支持,把一...

3316
来自专栏后端程序员的自我修养

CPython源码阅读笔记(1)

目前 CPython 的开发已经迁移到了 Github 上,可以直接去 Github clone 对应的分支。 我们将基于 Python 2.7.13 版本, ...

731

扫码关注云+社区