ffmpeg视频压缩代码

FFmpeg是一个开源的跨平台音视频处理框架，它提供了非常强大的音视频处理功能。其中，视频压缩是FFmpeg中的一个非常重要的功能之一。下面是一个使用FFmpeg进行视频压缩的示例代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>

int main(int argc, char *argv[]) {
    AVFormatContext *pFormatCtx = NULL;
    int i, videoStream;
    AVCodecContext *pCodecCtx = NULL;
    AVCodec *pCodec = NULL;
    AVFrame *pFrame = NULL;
    AVPacket packet;
    int frameFinished;
    struct SwsContext *sws_ctx = NULL;

    if (argc < 3) {
        printf("Usage: %s <input_file> <output_file>\n", argv);
        return -1;
    }

    av_register_all();
    if (avformat_open_input(&pFormatCtx, argv, NULL, NULL) != 0) {
        return -1;
    }
    if (avformat_find_stream_info(pFormatCtx, NULL) < 0) {
        return -1;
    }
    av_dump_format(pFormatCtx, 0, argv, 0);

    videoStream = -1;
    for (i = 0; i < pFormatCtx->nb_streams; i++) {
        if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
            videoStream = i;
            break;
        }
    }
    if (videoStream == -1) {
        return -1;
    }
    pCodecCtx = avcodec_alloc_context3(NULL);
    if (!pCodecCtx) {
        return -1;
    }
    avcodec_parameters_to_context(pCodecCtx, pFormatCtx->streams[videoStream]->codecpar);
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if (pCodec == NULL) {
        return -1;
    }
    if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0) {
        return -1;
    }
    pFrame = av_frame_alloc();
    sws_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,
        pCodecCtx->width, pCodecCtx->height, AV_PIX_FMT_YUV420P,
        SWS_BILINEAR, NULL, NULL, NULL);
    FILE *fp_out = fopen(argv, "wb");
    while (av_read_frame(pFormatCtx, &packet) >= 0) {
        if (packet.stream_index == videoStream) {
            avcodec_decode_video2(pCodecCtx, pFrame, &frameFinished, &packet);
            if (frameFinished) {
                uint8_t *out_buffer = (uint8_t *)av_malloc(av_image_get_buffer_size(AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height, 1));
                av_image_fill_arrays(pFrame->data, pFrame->linesize, out_buffer,
                    AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height, 1);
                sws_scale(sws_ctx, (const uint8_t* const*)pFrame->data,
                    pFrame->linesize, 0, pCodecCtx->height,
                    pFrame->data, pFrame->linesize);
                AVPacket pkt = { 0 };
                av_init_packet(&pkt);
                pkt.data = NULL;
                pkt.size = 0;
                int got_output = 0;
                AVCodecContext *enc_ctx = avcodec_alloc_context3(NULL);
                if (!enc_ctx) {
                    return -1;
                }
                enc_ctx->width = pCodecCtx->width / 2;
                enc_ctx->height = pCodecCtx->height / 2;
                enc_ctx->pix_fmt = AV_PIX_FMT_YUV420P;
                enc_ctx->bit_rate = pCodecCtx->bit_rate / 4;
                enc_ctx->time_base = (AVRational){ 1, 25 };
                enc_ctx->gop_size = 10;
                enc_ctx->max_b_frames = 1;
                enc_ctx->thread_count = 4;
                enc_ctx->qmin = 10;
                enc_ctx->qmax = 51;
                enc_ctx->qcompress = 0.6f;
                AVCodec *enc_codec = avcodec_find_encoder_by_name("libx264");
                if (!enc_codec) {
                    return -1;
                }
                if (avcodec_open2(enc_ctx, enc_codec, NULL) < 0) {
                    return -1;
                }
                if (avcodec_encode_video2(enc_ctx, &pkt, pFrame, &got_output) >= 0 && got_output) {
                    fwrite(pkt.data, pkt.size, 1, fp_out);
                    av_packet_unref(&pkt);
                }
                av_free(out_buffer);
                avcodec_free_context(&enc_ctx);
            }
        }
        av_packet_unref(&packet);
    }
    fclose(fp_out);
    av_frame_free(&pFrame);
    avcodec_close(pCodecCtx);
    avformat_close_input(&pFormatCtx);
    return 0;
}

这段代码实现了从输入文件中读取视频帧，将其压缩后写入到输出文件中。其中使用了libx264作为视频编码器，压缩后的视频格式为H.264。你可以根据自己的需求修改编码器的参数来得到不同的压缩效果。