Android中如何提取和生成mp4文件

隨著Android 4.4及以上版本的逐漸普及，Android 4.1引入的MediaExtractor類，以及Android 4.3引入的MediaMuxer類，終于可以開始正式地“發(fā)光發(fā)熱”了。

MediaMuxer類主要用于將音頻和視頻數(shù)據(jù)進(jìn)行混合生成多媒體文件（如：mp4文件），而MediaExtractor則剛好相反，主要用于多媒體文件的音視頻數(shù)據(jù)的分離。

本文將介紹如何利用Android SDK提供的MediaExtractor和MediaMuxer類來完成mp4文件的提取和生成，指出開發(fā)過程中會(huì)遇到的坑，并給出簡單的Demo示例代碼。

Demo的目標(biāo)：提取input.mp4文件中的視頻數(shù)據(jù)，生成除去音頻數(shù)據(jù)之后的純視頻output.mp4文件。代碼可以在本博文最后的附件中下載，也可以到我的Github中下載：

https://github.com/Jhuster/Android/tree/master/MediaDemo

由于Android SDK關(guān)于這兩個(gè)類的介紹真是少之又少，因此，在給出demo之前，我們先簡單地了解一下這兩個(gè)類吧。

1. MediaExtractor

該類主要用于音視頻混合數(shù)據(jù)的分離，接口比較簡單，首先要通過setDataSource(String path)函數(shù)設(shè)置數(shù)據(jù)源，數(shù)據(jù)源可以是本地文件地址，也可以使用HTTP協(xié)議的網(wǎng)絡(luò)碼流地址。

然后，可以通過下面的代碼塊，來獲取碼流的詳細(xì)信息，如：MimeType，分辨率、編碼格式、碼率、幀率等等。

int videoTrackIndex = -1;
int audioTrackIndex = -1;

for(int i = 0; i < mMediaExtractor.getTrackCount(); i++) {

    //獲取碼流的詳細(xì)格式/配置信息
    MediaFormat format = mMediaExtractor.getTrackFormat(i);

    String mime = format.getString(MediaFormat.KEY_MIME);
    if(mime.startsWith("video/")) {
        videoTrackIndex = i;
    }
    else if(mime.startsWith("audio/")) {
        audioTrackIndex = i;
    }

    ....
}

獲取到媒體文件的詳細(xì)信息之后，就可以選擇指定的通道，并分離和讀取數(shù)據(jù)了：

mMediaExtractor.selectTrack(videoTrackIndex); //選擇讀取視頻數(shù)據(jù)
while(true) {
    int sampleSize = mMediaExtractor.readSampleData(buffer, 0);  //讀取一幀數(shù)據(jù)
    if(sampleSize < 0) {
        break;
    }
    mMediaExtractor.advance(); //移動(dòng)到下一幀
}

mMediaExtractor.release(); //讀取結(jié)束后，要記得釋放資源

2. MediaMuxer

該類主要用于將音頻和視頻進(jìn)行混合生成多媒體文件，創(chuàng)建該類對(duì)象，需要傳入輸出的文件位置以及格式，構(gòu)造函數(shù)如下：

public MediaMuxer(String path, int format);

創(chuàng)建對(duì)象之后，一個(gè)比較重要的操作就是addTrack()，添加數(shù)據(jù)通道，該函數(shù)需要傳入MediaFormat對(duì)象，MediaFormat即媒體格式類，用于描述媒體的格式參數(shù)，如視頻幀率、音頻采樣率等。

在本示例中，可以直接使用MediaExtractor.getTrackFormat()解析得到的MediaFormat對(duì)象，如果你希望自己來創(chuàng)建這個(gè)MediaFormat對(duì)象的話，可以使用該類的如下靜態(tài)方法創(chuàng)建：

MediaFormat format = MediaFormat.createVideoFormat("video/avc",320,240);

注意，這里有一個(gè)比較大的坑，就是，如果手動(dòng)創(chuàng)建MediaFormat對(duì)象的話，一定要記得設(shè)置"csd-0"和"csd-1"這兩個(gè)參數(shù)：

byte[] csd0 = {x,x,x,x,x,x,x...}
byte[] csd1 = {x,x,x,x,x,x,x...}

format.setByteBuffer("csd-0",ByteBuffer.wrap(csd0));
format.setByteBuffer("csd-1",ByteBuffer.wrap(csd1));

至于"csd-0"和"csd-1"是什么，對(duì)于H264視頻的話，它對(duì)應(yīng)的是sps和pps，對(duì)于AAC音頻的話，對(duì)應(yīng)的是ADTS，做音視頻開發(fā)的人應(yīng)該都知道，它一般存在于編碼器生成的IDR幀之中。

通過 addTrack() 添加了數(shù)據(jù)通道之后，記錄下函數(shù)返回的 trackIndex，然后就可以調(diào)用 MediaMuxer.writeSampleData() 愉快地向mp4文件中寫入數(shù)據(jù)了。

這里會(huì)產(chǎn)生第二個(gè)坑，就是writeSampleData函數(shù)的最后一個(gè)參數(shù)是一個(gè)BufferInfo對(duì)象，你必須認(rèn)真地填入“正確”的值：

BufferInfo info = new BufferInfo();
info.offset = 0;
info.size = sampleSize;
info.flags = MediaCodec.BUFFER_FLAG_SYNC_FRAME;
info.presentationTimeUs = timestamp;

其中，

info.size 必須填入數(shù)據(jù)的大小

info.flags 需要給出是否為同步幀/關(guān)鍵幀

info.presentationTimeUs 必須給出正確的時(shí)間戳，注意單位是 us，例如，對(duì)于幀率為 x f/s 的視頻而言，時(shí)間戳的間隔就是 1000/x ms

跳過了這些坑，你就可以順利地完成mp4文件的寫入了，同樣，完成后記得關(guān)閉以及釋放資源：

mMediaMuxer.stop();
mMediaMuxer.release();

3. 小結(jié)

有了上面的簡單介紹和鋪墊，demo代碼就不難看懂了。運(yùn)行demo代碼的注意事項(xiàng)：

（1）Android 4.3以及以上系統(tǒng)的手機(jī)

（2）把 input.mp4文件拷貝到sdcard

代碼最核心的部分如下所示：

  protected boolean process() throws IOException {

      mMediaExtractor = new MediaExtractor();          
      mMediaExtractor.setDataSource(SDCARD_PATH+"/input.mp4");                
              
      int mVideoTrackIndex = -1;
      int framerate = 0;
      for(int i = 0; i < mMediaExtractor.getTrackCount(); i++) {
          MediaFormat format = mMediaExtractor.getTrackFormat(i);
          String mime = format.getString(MediaFormat.KEY_MIME);
          if(!mime.startsWith("video/")) {                
              continue;
          }
          framerate = format.getInteger(MediaFormat.KEY_FRAME_RATE);            
          mMediaExtractor.selectTrack(i);
          mMediaMuxer = new MediaMuxer(SDCARD_PATH+"/ouput.mp4", OutputFormat.MUXER_OUTPUT_MPEG_4);
          mVideoTrackIndex = mMediaMuxer.addTrack(format);  
          mMediaMuxer.start();
      }
      
      if(mMediaMuxer == null) {
          return false;
      }
      
      BufferInfo info = new BufferInfo();
      info.presentationTimeUs = 0;
      ByteBuffer buffer = ByteBuffer.allocate(500*1024);        
      while(true) {
          int sampleSize = mMediaExtractor.readSampleData(buffer, 0);
          if(sampleSize < 0) {
              break;
          }
          mMediaExtractor.advance();
          info.offset = 0;
          info.size = sampleSize;
          info.flags = MediaCodec.BUFFER_FLAG_SYNC_FRAME;        
          info.presentationTimeUs += 1000*1000/framerate;
          mMediaMuxer.writeSampleData(mVideoTrackIndex,buffer,info);
      }

      mMediaExtractor.release();
      
      mMediaMuxer.stop();
      mMediaMuxer.release();
      
      return true;
  }

4. 小結(jié)

關(guān)于Android中如何提取和生成mp4文件就總結(jié)到這里了，有任何疑問或者建議歡迎留言或者來信lujun.hust@gmail.com交流，或者關(guān)注我的新浪微博 @盧_俊 獲取最新的文章和資訊。