Refactor AnnexB bitstream conversion for AVC/H.264

media/formats/mp4/avc.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/formats/mp4/avc.h"
 
 #include <algorithm>
 
 #include "base/logging.h"
 #include "media/base/decrypt_config.h"
@@ skipping 289 matching lines @@
         NOTREACHED() << "AdvanceToNextNALU() returned kUnsupportedStream!";
         return false;
 
       case H264Parser::kEOStream:
         done = true;
     }
   }
 
   return order_state >= kAfterFirstVCL;
 }
+
+AVCBitstreamConverter::AVCBitstreamConverter(
+    scoped_ptr<AVCDecoderConfigurationRecord> avc_config)
+  : avc_config_(avc_config.Pass()) {
+    DCHECK(avc_config_);
+}
+
+AVCBitstreamConverter::~AVCBitstreamConverter() {
+}
+
+bool AVCBitstreamConverter::ConvertFrame(
+    std::vector<uint8>* frame_buf,
+    bool is_keyframe,
+    std::vector<SubsampleEntry>* subsamples) const {
+  // Convert the AVC NALU length fields to Annex B headers, as expected by
+  // decoding libraries. Since this may enlarge the size of the buffer, we also
+  // update the clear byte count for each subsample if encryption is used to
+  // account for the difference in size between the length prefix and Annex B
+  // start code.
+  RCHECK(AVC::ConvertFrameToAnnexB(avc_config_->length_size, frame_buf,
+                                   subsamples));
+
+  if (is_keyframe) {
+    // If this is a keyframe, we (re-)inject SPS and PPS headers at the start of
+    // a frame. If subsample info is present, we also update the clear byte
+    // count for that first subsample.
+    RCHECK(AVC::InsertParamSetsAnnexB(*avc_config_, frame_buf, subsamples));
+  }
+
+  DCHECK(AVC::IsValidAnnexB(*frame_buf, *subsamples));
+  return true;
+}
+
 }  // namespace mp4
 }  // namespace media