在Android中,可以使用MediaMuxer来封装编码后的视频流和音频流到mp4容器中:
MediaMuxer facilitates muxing elementary streams. Currently supports mp4 or webm file as the output and at most one audio and/or one video elementary stream. MediaMuxer does not support muxing B-frames.
在官方文档的介绍中,MediaMuxer最多仅支持一个视频track和一个音频track,所以如果有多个音频track可以先把它们混合成为一个音频track然后再使用MediaMuxer封装到mp4容器中。
通常视频编码使用H.264(AVC)编码,音频编码使用AAC编码,在MediaFormat中我们可以看到各种编码格式:
public static final String MIMETYPE_VIDEO_AVC = "video/avc";
public static final String MIMETYPE_AUDIO_AAC = "audio/mp4a-latm";
public static final String MIMETYPE_TEXT_CEA_608 = "text/cea-608";
上面只各自摘取了视频、音频、字幕的一种编码格式,更多的编码格式可自行查看MediaFormat源码。
MediaMuxer的使用也比较简单,首先通过new MediaMuxer(String path, int format)指定视频文件输出路径和文件格式:
MediaMuxer mMediaMuxer = new MediaMuxer(mOutputVideoPath,
MediaMuxer.OutputFormat.MUXER_OUTPUT_MPEG_4);
创建MediaMuxer对象之后,一个比较重要的操作就是addTrack(MediaFormat format),添加媒体通道,该函数需要传入MediaFormat对象,通常从MediaExtractor或者MediaCodec中获取,如果希望自己创建的话可以直接new或者通过该类如下静态方法创建:
MediaFormat.createAudioFormat(...);
MediaFormat.createVideoFormat(...);
MediaFormat.createSubtitleFormat(...);
如果是自己创建MediaFormat要根据媒体类型配置好相应的keys,这在MediaFormat官方文档有详细说明。这里注意一定要设置csd参数,否则添加进MediaMuxer的MediaFormat会导致MediaMuxer调用stop()时抛出异常。
csd参数在官方文档中叫Codec-specific Data,详细介绍可以看MediaCodec官方文档 - Codec-specific Data部分。对于H.264来说,"csd-0"和"csd-1"分别对应sps和pps;对于AAC来说,"csd-0"对应ADTS。下面是自己创建视频MediaFormat的例子:
MediaFormat videoFormat = MediaFormat.createVideoFormat(MediaFormat.MIMETYPE_VIDEO_AVC, 1280, 720);
byte[] header_sps = {0, 0, 0, 1, 103, 100, 0, 31, -84, -76, 2, -128, 45, -56};
byte[] header_pps = {0, 0, 0, 1, 104, -18, 60, 97, 15, -1, -16, -121, -1, -8, 67, -1, -4, 33, -1, -2, 16, -1, -1, 8, 127, -1, -64};
videoFormat.setByteBuffer("csd-0", ByteBuffer.wrap(header_sps));
videoFormat.setByteBuffer("csd-1", ByteBuffer.wrap(header_pps));
videoFormat.setInteger(MediaFormat.KEY_COLOR_FORMAT, MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Planar);
videoFormat.setInteger(MediaFormat.KEY_MAX_INPUT_SIZE, 1920 * 1080);
videoFormat.setInteger(MediaFormat.KEY_CAPTURE_RATE, 25);
把MediaFormat添加到MediaMuxer后记录返回的track index,添加完所有track后调用start方法:
videoTrackIndex = mMediaMuxer.addTrack(format);
audioTrackIndex = mMediaMuxer.addTrack(format);
mMediaMuxer.start();
然后就可以调用MediaMuxer.writeSampleData()向mp4文件中写入数据了。这里要注意每次只能添加一帧视频数据或者单个Sample的音频数据,并且BufferInfo对象的值一定要设置正确:
info.offset = 0;
info.size = sampleSize;
info.flags = MediaCodec.BUFFER_FLAG_SYNC_FRAME;
info.presentationTimeUs = mVideoExtractor.getSampleTime();
mMediaMuxer.writeSampleData(videoTrackIndex, buffer, info);
- info.size 必须填入数据的大小
- info.flags 需要给出是否为同步帧/关键帧
- info.presentationTimeUs 必须给出正确的时间戳,注意单位是 us
结束写入后关闭以及释放资源:
mMediaMuxer.stop();
mMediaMuxer.release();