在C++中使用ffmpeg从内存中的多个图像流式传输RTSP

#include <iostream>
#include <fstream>
extern "C" {
  #include <libavformat/avformat.h>
  #include <libavcodec/avcodec.h>
}

int main() {
  // 初始化FFmpeg
  av_register_all();
  avcodec_register_all();

  // 打开输入文件
  AVFormatContext* formatContext = avformat_alloc_context();
  if (avformat_open_input(&formatContext, "input.mp4", nullptr, nullptr) < 0) {
    std::cout << "无法打开输入文件" << std::endl;
    return -1;
  }

  // 查找视频流
  if (avformat_find_stream_info(formatContext, nullptr) < 0) {
    std::cout << "无法查找视频流" << std::endl;
    return -1;
  }

  // 打开输出RTSP流
  AVOutputFormat* outputFormat = av_guess_format("rtsp", "rtsp://localhost:8554/live", nullptr);
  if (!outputFormat) {
    std::cout << "无法确定输出格式" << std::endl;
    return -1;
  }
  AVFormatContext* outputFormatContext = avformat_alloc_context();
  outputFormatContext->oformat = outputFormat;

  // 创建视频流
  AVStream* outputStream = avformat_new_stream(outputFormatContext, nullptr);
  if (!outputStream) {
    std::cout << "无法创建视频流" << std::endl;
    return -1;
  }

  // 复制视频流参数
  AVCodecParameters* codecParameters = formatContext->streams[0]->codecpar;
  avcodec_parameters_copy(outputStream->codecpar, codecParameters);

  // 打开输出流
  if (avio_open(&outputFormatContext->pb, "rtsp://localhost:8554/live", AVIO_FLAG_WRITE) < 0) {
    std::cout << "无法打开输出流" << std::endl;
    return -1;
  }

  // 写入文件头
  if (avformat_write_header(outputFormatContext, nullptr) < 0) {
    std::cout << "无法写入文件头" << std::endl;
    return -1;
  }

  // 循环读取帧并写入输出流
  AVPacket packet;
  while (av_read_frame(formatContext, &packet) >= 0) {
    if (packet.stream_index == 0) {
      // 修改时间戳
      packet.pts = av_rescale_q(packet.pts, formatContext->streams[0]->time_base, outputStream->time_base);
      packet.dts = av_rescale_q(packet.dts, formatContext->streams[0]->time_base, outputStream->time_base);
      packet.duration = av_rescale_q(packet.duration, formatContext->streams[0]->time_base, outputStream->time_base);
      packet.pos = -1;

      // 写入帧
      av_interleaved_write_frame(outputFormatContext, &packet);
    }

    av_packet_unref(&packet);
  }

  // 写入文件尾
  av_write_trailer(outputFormatContext);

  // 清理资源
  avformat_close_input(&formatContext);
  avio_close(outputFormatContext->pb);
  avformat_free_context(outputFormatContext);

  return 0;
}