ffmpeg使用png编码器把rgb24编码为png图像

version

#define LIBAVUTIL_VERSION_MAJOR 58

#define LIBAVUTIL_VERSION_MINOR 12

#define LIBAVUTIL_VERSION_MICRO 100

note

不使用AVOutputFormat

code

void CFfmpegOps::EncodeRGB24ToPNG(const char *infile, const char *width_str, const char *height_str, const char *outfile)
{
    if ((!infile) || (!width_str) || (!height_str) || (!outfile))
    {
        return;
    }

    int width = 0;
    int height = 0;
    AVFrame *rgb24frame = nullptr;
    int rgb24frame_bytes = 0;
    AVPixelFormat rgb24_fmt = AV_PIX_FMT_RGB24;
    int ret = -1;
    FILE *in_fp = nullptr;
    size_t n = 0;
    AVPacket *pkt = nullptr;
    const AVCodec *png_encoder = nullptr;
    AVCodecContext *png_encoder_ctx = nullptr;
    FILE *out_fp = nullptr;

    try
    {
        width = std::stoi(width_str);
        height = std::stoi(height_str);
    }
    catch (const std::exception &e)
    {
        return;
    }

    rgb24frame = av_frame_alloc();
    if (!rgb24frame)
    {
        printf("av_frame_alloc error\n");
        goto end;
    }
    rgb24frame->format = rgb24_fmt;
    rgb24frame->width = width;
    rgb24frame->height = height;

    rgb24frame_bytes = av_image_get_buffer_size(rgb24_fmt, width, height, 1);

    ret = av_frame_get_buffer(rgb24frame, 0);
    if (ret < 0)
    {
        printf("av_frame_get_buffer error(%s)\n", GetFfmpegERR(ret));
        goto end;
    }

    in_fp = fopen(infile, "rb");
    if (!in_fp)
    {
        printf("fopen error\n");
        goto end;
    }

    n = fread(rgb24frame->data[0], sizeof(uint8_t), width * height * 3, in_fp);
    if ((int)n != (width * height * 3))
    {
        printf("n != (width * height * 3)\n");
        goto end;
    }

    pkt = av_packet_alloc();
    if (!pkt)
    {
        printf("av_packet_alloc error\n");
        goto end;
    }

    png_encoder = avcodec_find_encoder(AV_CODEC_ID_PNG);
    if (!png_encoder)
    {
        printf("avcodec_find_encoder error\n");
        goto end;
    }

    png_encoder_ctx = avcodec_alloc_context3(png_encoder);
    if (!png_encoder_ctx)
    {
        printf("avcodec_alloc_context3 error\n");
        goto end;
    }
    png_encoder_ctx->pix_fmt = rgb24_fmt;
    png_encoder_ctx->width = width;
    png_encoder_ctx->height = height;
    png_encoder_ctx->time_base.num = 1;
    png_encoder_ctx->time_base.den = 25;
    png_encoder_ctx->framerate.num = 25;
    png_encoder_ctx->framerate.den = 1;
    png_encoder_ctx->bit_rate = rgb24frame_bytes * png_encoder_ctx->framerate.num * 8;

    ret = avcodec_open2(png_encoder_ctx, png_encoder, nullptr);
    if (ret < 0)
    {
        printf("avcodec_open2 error(%s)\n", GetFfmpegERR(ret));
        goto end;
    }

    out_fp = fopen(outfile, "wb");
    if (!out_fp)
    {
        printf("fopen error\n");
        goto end;
    }

    ret = avcodec_send_frame(png_encoder_ctx, rgb24frame);
    if (ret != 0)
    {
        printf("avcodec_send_frame error(%s)\n", GetFfmpegERR(ret));
        goto end;
    }

    while (1)
    {
        ret = avcodec_receive_packet(png_encoder_ctx, pkt);
        if (ret != 0)
        {
            if (ret == AVERROR(EAGAIN))
            {
                continue;
            }

            printf("avcodec_receive_packet error(%s)\n", GetFfmpegERR(ret));
            break;
        }

        n = fwrite(pkt->data, pkt->size, sizeof(uint8_t), out_fp);

        av_packet_unref(pkt);

        break;
    }

end:
    if (out_fp)
    {
        fclose(out_fp);
        out_fp = nullptr;
    }

    if (png_encoder_ctx)
    {
        avcodec_free_context(&png_encoder_ctx);
        png_encoder_ctx = nullptr;
    }

    if (pkt)
    {
        av_packet_free(&pkt);
        pkt = nullptr;
    }

    if (in_fp)
    {
        fclose(in_fp);
        in_fp = nullptr;
    }

    if (rgb24frame)
    {
        av_frame_free(&rgb24frame);
        rgb24frame = nullptr;
    }
}

performance

相关推荐
xinghuitunan5 分钟前
蓝桥杯顺子日期(填空题)
c语言·蓝桥杯
van叶~7 分钟前
算法妙妙屋-------1.递归的深邃回响:二叉树的奇妙剪枝
c++·算法
Half-up7 分钟前
C语言心型代码解析
c语言·开发语言
knighthood200117 分钟前
解决:ros进行gazebo仿真,rviz没有显示传感器数据
c++·ubuntu·ros
懒大王就是我43 分钟前
C语言网络编程 -- TCP/iP协议
c语言·网络·tcp/ip
半盏茶香1 小时前
【C语言】分支和循环详解(下)猜数字游戏
c语言·开发语言·c++·算法·游戏
小堇不是码农1 小时前
在VScode中配置C_C++环境
c语言·c++·vscode
Jack黄从零学c++1 小时前
C++ 的异常处理详解
c++·经验分享
小肥象不是小飞象1 小时前
(六千字心得笔记)零基础C语言入门第八课——函数(上)
c语言·开发语言·笔记·1024程序员节
励志成为嵌入式工程师6 小时前
c语言简单编程练习9
c语言·开发语言·算法·vim