阅读 97

FFmpeg-解封装

解封装步骤

  • 创建解封装上下文 avformat_open_input
  • 检索流信息 avformat_find_stream_info
  • 初始化音频信息
  • 初始化视频信息
  • 初始化frame av_frame_alloc
  • 初始化pkt av_packet_alloc
  • 从输入文件中读取数据 av_read_frame
    • 解码音频数据
    • 解码视频数据
  • 刷新缓冲区
  • 销毁资源

初始化解码器


// 初始化解码器
- (int)initDecoder:(AVCodecContext**)decodeCtx streamIdx:(int *)streamIdx mediaType:(AVMediaType)type {
    // 根据Type寻找最合适的流信息
    // 返回值是流索引
    int ret = av_find_best_stream(_fmtCtx, type, -1, -1, nullptr, 0);
    RET(av_find_best_stream);
    // 检验流
    *streamIdx = ret;
    AVStream *stream = _fmtCtx->streams[*streamIdx];
    if (!stream) {
        return -1;
    }
    // 为当前流找到合适的解码器
    AVCodec *decoder = avcodec_find_decoder(stream->codecpar->codec_id);
    if (!decoder) {
        return -1;
    }
    // 初始化解码器上下文
    *decodeCtx = avcodec_alloc_context3(decoder);
    if (!decodeCtx) {
        return -1;
    }
    // 从流中拷贝参数到解码器上下文中
    ret = avcodec_parameters_to_context(*decodeCtx, stream->codecpar);
    RET(avcodec_parameters_to_context);
    // 打开解码器
    ret = avcodec_open2(*decodeCtx, decoder, nullptr);
    RET(avcodec_open2);
    return 0;
}

初始化音频信息


// 初始化音频信息
- (int)initAudionInfo {
    int ret = [self initDecoder:&_aDecodeCtx streamIdx:&_aStreamIdx mediaType:AVMEDIA_TYPE_AUDIO];
    RET(initDecoder);
    [[NSFileManager defaultManager]createFileAtPath:self.aOut.filename contents:nil attributes:nil];
    aOutFile = [NSFileHandle fileHandleForWritingAtPath:self.aOut.filename];
    // 保存音频参数
    _aOut.sampleRate = _aDecodeCtx->sample_rate;
    _aOut.sampleFmt = _aDecodeCtx->sample_fmt;
    _aOut.chLayout = (int)_aDecodeCtx->channel_layout;
    // 音频样本帧的大小
    _sampleSize = av_get_bytes_per_sample((AVSampleFormat)_aOut.sampleFmt);
    _sampleFrameSize = _sampleSize * _aDecodeCtx->channels;
    return 0;
}

初始化视频信息

// 初始化视频信息
- (int)initVideoInfo {
    int ret = [self initDecoder:&_vDecodeCtx streamIdx:&_vStreamIdx mediaType:AVMEDIA_TYPE_VIDEO];
    RET(initDecoder);
    [[NSFileManager defaultManager]createFileAtPath:self.vOut.filename contents:nil attributes:nil];
    vOutFile = [NSFileHandle fileHandleForWritingAtPath:self.vOut.filename];
    if (!vOutFile) {
        return -1;
    }
    // 保存视频参数
    _vOut.width = _vDecodeCtx->width;
    _vOut.height = _vDecodeCtx->height;
    _vOut.pixFmt = _vDecodeCtx->pix_fmt;
    // 帧率
    AVRational frameRate = av_guess_frame_rate(_fmtCtx, _fmtCtx->streams[_vStreamIdx], nullptr);
    _vOut.fps = frameRate.num / frameRate.den;
    // 创建用于存放一帧解码图片的缓冲区
    ret = av_image_alloc(_imgBuf, _imageLinesizes, _vOut.width , _vOut.height, (AVPixelFormat)_vOut.pixFmt, 1);
    RET(av_image_alloc);
    _imgSize = ret;
    return 0;
}

解码


// 解码
- (int)decode: (AVCodecContext *)decodeCtx packet:(AVPacket*)pkt func: (void(^)())handler {
    // 发送压缩数据到解码器
    int ret = avcodec_send_packet(decodeCtx, pkt);
    RET(avcodec_send_packet);
    while (true) {
        // 获取解码后的数据
        ret = avcodec_receive_frame(decodeCtx, _frame);
        if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
            return 0;
        }
        RET(avcodec_receive_frame);
        // 执行写入文件的代码
        handler();
    }
    return 1;
}

写入音频信息到文件


- (void)writeAudioFrame {
    // libfdk_aac解码器,解码出来的PCM格式:s16
    // aac解码器,解码出来的PCM格式:ftlp
    /**
     - planar的内存布局(左右声道隔离的)
        - LLLLLLLLLLL RRRRRRRRRRRR
     - 非planar的内存布局
        - LR LR LR
     */
    if (av_sample_fmt_is_planar((AVSampleFormat)_aOut.sampleFmt)) {
        // planar格式写入文件时,以非planar写入,这样ffplay才能正确播放
        //  LLLL RRRR DDDD FFFF
        // 外层循环:每一个声道的样本数
        for (int si = 0; si < _frame->nb_samples; si++) {
            // 内层循环:有多少个声道
            for (int ci = 0; ci < _aDecodeCtx->channels; ci++) {
                char *begin = (char*)(_frame->data[ci] + si * _sampleSize);
                [aOutFile writeData:[NSData dataWithBytes:begin length:_sampleSize]];
            }
        }
    } else {
        [aOutFile writeData:[NSData dataWithBytes:_frame->data[0] length:_frame->nb_samples * _sampleFrameSize]];
        
        // _aOutFile.write((char *) _frame->data[0], _frame->linesize[0]);
        // 不用这句话的原因是:linesize[0]的值是始终固定的,但是在最后几帧的时候,样本数不一定能填满整个frame
    }
}

写入视频信息到文件


- (void)writeVideoFrame {
    av_image_copy(_imgBuf,
                  _imageLinesizes,
                  (const uint8_t**)_frame->data,
                  _frame->linesize,
                  (AVPixelFormat)_vOut.pixFmt,
                  _vOut.width,
                  _vOut.height);
    // 将缓冲区的数据写入文件
    [vOutFile writeData:[NSData dataWithBytes:_imgBuf[0] length:_imgSize]];
    
}

解封装所用到的API总结

  • 创建解封装上下文
avformat_open_input(&_fmtCtx, infileName.UTF8String, nullptr, nullptr);
  • 检索流信息
avformat_find_stream_info(_fmtCtx, nullptr);
  • 打印流信息
av_dump_format(_fmtCtx, 0, infileName.UTF8String, 0);
  • 初始化Packet
packet = av_packet_alloc();
packet->data = nullptr;
packet->size = 0;
  • 初始化frame
_frame = av_frame_alloc();
  • 从frameContext中读取流数据到Packet
av_read_frame(_fmtCtx, packet)
  • 释放Packet
av_packet_unref(packet);
  • 根据Type寻找最合适的流信息
av_find_best_stream(_fmtCtx, type, -1, -1, nullptr, 0);
  • 为当前流找到合适的解码器
AVCodec *decoder = avcodec_find_decoder(stream->codecpar->codec_id);
  • 初始化解码器上下文
*decodeCtx = avcodec_alloc_context3(decoder);
  • 从流中拷贝参数到解码器上下文中
 avcodec_parameters_to_context(*decodeCtx, stream->codecpar);
  • 打开解码器
avcodec_open2(*decodeCtx, decoder, nullptr);
  • 发送压缩数据到解码器
avcodec_send_packet(decodeCtx, pkt);
  • 获取解码后的数据
avcodec_receive_frame(decodeCtx, _frame);

作者:lieon

原文链接:https://www.jianshu.com/p/d047a277b08f

文章分类
后端
文章标签
版权声明:本站是系统测试站点,无实际运营。本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 XXXXXXo@163.com 举报,一经查实,本站将立刻删除。
相关推荐