Examples with AVCodecParameters org.bytedeco.ffmpeg.avcodec.AVCodecParameters used on opensource projects

Example 1 with AVCodecParameters

use of org.bytedeco.ffmpeg.avcodec.AVCodecParameters in project blog_demos by zq2599.

the class PushMp4 method grabAndPush.

/**
 * 读取指定的mp4文件，推送到SRS服务器
 * @param sourceFilePath 视频文件的绝对路径
 * @param PUSH_ADDRESS 推流地址
 * @throws Exception
 */
private static void grabAndPush(String sourceFilePath, String PUSH_ADDRESS) throws Exception {
    // ffmepg日志级别
    avutil.av_log_set_level(avutil.AV_LOG_INFO);
    FFmpegLogCallback.set();
    // 实例化帧抓取器对象，将文件路径传入
    FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(MP4_FILE_PATH);
    long startTime = System.currentTimeMillis();
    log.info("开始初始化帧抓取器");
    // 初始化帧抓取器，例如数据结构（时间戳、编码器上下文、帧对象等），
    // 如果入参等于true，还会调用avformat_find_stream_info方法获取流的信息，放入AVFormatContext类型的成员变量oc中
    grabber.start(true);
    log.info("帧抓取器初始化完成，耗时[{}]毫秒", System.currentTimeMillis() - startTime);
    // grabber.start方法中，初始化的解码器信息存在放在grabber的成员变量oc中
    AVFormatContext avFormatContext = grabber.getFormatContext();
    // 文件内有几个媒体流（一般是视频流+音频流）
    int streamNum = avFormatContext.nb_streams();
    // 没有媒体流就不用继续了
    if (streamNum < 1) {
        log.error("文件内不存在媒体流");
        return;
    }
    // 取得视频的帧率
    int frameRate = (int) grabber.getVideoFrameRate();
    log.info("视频帧率[{}]，视频时长[{}]秒，媒体流数量[{}]", frameRate, avFormatContext.duration() / 1000000, avFormatContext.nb_streams());
    // 遍历每一个流，检查其类型
    for (int i = 0; i < streamNum; i++) {
        AVStream avStream = avFormatContext.streams(i);
        AVCodecParameters avCodecParameters = avStream.codecpar();
        log.info("流的索引[{}]，编码器类型[{}]，编码器ID[{}]", i, avCodecParameters.codec_type(), avCodecParameters.codec_id());
    }
    // 视频宽度
    int frameWidth = grabber.getImageWidth();
    // 视频高度
    int frameHeight = grabber.getImageHeight();
    // 音频通道数量
    int audioChannels = grabber.getAudioChannels();
    log.info("视频宽度[{}]，视频高度[{}]，音频通道数[{}]", frameWidth, frameHeight, audioChannels);
    // 实例化FFmpegFrameRecorder，将SRS的推送地址传入
    FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(SRS_PUSH_ADDRESS, frameWidth, frameHeight, audioChannels);
    // 设置编码格式
    recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);
    // 设置封装格式
    recorder.setFormat("flv");
    // 一秒内的帧数
    recorder.setFrameRate(frameRate);
    // 两个关键帧之间的帧数
    recorder.setGopSize(frameRate);
    // 设置音频通道数，与视频源的通道数相等
    recorder.setAudioChannels(grabber.getAudioChannels());
    startTime = System.currentTimeMillis();
    log.info("开始初始化帧抓取器");
    // 初始化帧录制器，例如数据结构（音频流、视频流指针，编码器），
    // 调用av_guess_format方法，确定视频输出时的封装方式，
    // 媒体上下文对象的内存分配，
    // 编码器的各项参数设置
    recorder.start();
    log.info("帧录制初始化完成，耗时[{}]毫秒", System.currentTimeMillis() - startTime);
    Frame frame;
    startTime = System.currentTimeMillis();
    log.info("开始推流");
    long videoTS = 0;
    int videoFrameNum = 0;
    int audioFrameNum = 0;
    int dataFrameNum = 0;
    // 假设一秒钟15帧，那么两帧间隔就是(1000/15)毫秒
    int interVal = 1000 / frameRate;
    // 发送完一帧后sleep的时间，不能完全等于(1000/frameRate)，不然会卡顿，
    // 要更小一些，这里取八分之一
    interVal /= 8;
    // 持续从视频源取帧
    while (null != (frame = grabber.grab())) {
        videoTS = 1000 * (System.currentTimeMillis() - startTime);
        // 时间戳
        recorder.setTimestamp(videoTS);
        // 有图像，就把视频帧加一
        if (null != frame.image) {
            videoFrameNum++;
        }
        // 有声音，就把音频帧加一
        if (null != frame.samples) {
            audioFrameNum++;
        }
        // 有数据，就把数据帧加一
        if (null != frame.data) {
            dataFrameNum++;
        }
        // 取出的每一帧，都推送到SRS
        recorder.record(frame);
        // 停顿一下再推送
        Thread.sleep(interVal);
    }
    log.info("推送完成，视频帧[{}]，音频帧[{}]，数据帧[{}]，耗时[{}]秒", videoFrameNum, audioFrameNum, dataFrameNum, (System.currentTimeMillis() - startTime) / 1000);
    // 关闭帧录制器
    recorder.close();
    // 关闭帧抓取器
    grabber.close();
}