Mpeg2 TS stream parser for media source.

media/mpeg2/es_parser_h264.cc

Index: media/mpeg2/es_parser_h264.cc
diff --git a/media/mpeg2/es_parser_h264.cc b/media/mpeg2/es_parser_h264.cc
new file mode 100644
index 0000000000000000000000000000000000000000..9bf2903f0f2cc0fb49807cc2fa7db2c3c0427960
--- /dev/null
+++ b/media/mpeg2/es_parser_h264.cc
@@ -0,0 +1,572 @@
+// Copyright (c) 2013 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/mpeg2/es_parser_h264.h"
+
+#include "base/basictypes.h"
+#include "base/logging.h"
+#include "media/base/bit_reader.h"
+#include "media/base/stream_parser_buffer.h"
+#include "media/base/video_decoder_config.h"
+#include "media/base/video_frame.h"
+#include "media/mpeg2/mpeg2ts_common.h"
+#include "ui/gfx/rect.h"
+#include "ui/gfx/size.h"
+
+#define RCHECK(x) \
+    do { \
+      if (!(x)) { \
+        DLOG(ERROR) << "Failure while parsing H264: " << #x; \
+        return false; \
+      } \
+    } while (0)
+
+namespace {
+
+const int kExtendedSar = 255;
+
+const int kTableSarWidth[14] = {
+  1, 1, 12, 10, 16, 40, 24, 20, 32, 80, 18, 15, 64, 160
+};
+
+const int kTableSarHeight[14] = {
+  1, 1, 11, 11, 11, 33, 11, 11, 11, 33, 11, 11, 33, 99
+};
+
+class ByteReaderChainedBuffer {
+ public:
+  ByteReaderChainedBuffer(const uint8* buf0, int size0,
+                          const uint8* buf1, int size1)
+      : buf0_(buf0),
+        size0_(size0),
+        buf1_(buf1),
+        size1_(size1) { }
+
+  uint8 Get(int offset) const {
+    DCHECK_GE(offset, 0);
+    DCHECK_LT(offset, size0_ + size1_);
+    if (offset < size0_) {
+      return buf0_[offset];
+    }
+    return buf1_[offset - size0_];
+  }
+
+  int GetSize() const {
+    return (size0_ + size1_);
+  }
+
+ private:
+  const uint8* const buf0_;
+  const int size0_;
+  const uint8* const buf1_;
+  const int size1_;
+};
+
+}  // namespace
+
+namespace media {
+namespace mpeg2ts {
+
+EsParserH264::EsParserH264(
+    NewVideoConfigCB new_video_config_cb,
+    EmitBufferCB emit_buffer_cb)
+  : nal_es_pos_(0),
+    new_video_config_cb_(new_video_config_cb),
+    emit_buffer_cb_(emit_buffer_cb),
+    is_video_config_known_(false),
+    profile_idc_(0),
+    level_idc_(0),
+    pic_width_in_mbs_minus1_(0),
+    pic_height_in_map_units_minus1_(0) {
+}
+
+EsParserH264::~EsParserH264() {
+}
+
+void EsParserH264::Parse(const uint8* buf, int size,
+                         bool is_pts_valid, base::TimeDelta pts,
+                         bool is_dts_valid, base::TimeDelta dts) {
+  // Note: Parse is invoked each time a PES packet has been reassembled.
+  // Unfortunately, a PES packet does not necessarily map
+  // to an h264 access unit, although the HLS recommandation is to use one PES
+  // for each access unit (but this is just a recommandation and some streams
+  // do not comply with this recommandation).
+
+  // Link position |raw_es_.size()| in the ES stream with a timing descriptor.
+  // HLS recommandation: "In AVC video, you should have both a DTS and a
+  // PTS in each PES header".
+  // TODO(damienv): What if the stream is not compliant and both the PTS and the
+  // DTS are not valid ?
+  TimingDesc timing_desc;
+  timing_desc.pts = pts;
+  if (is_dts_valid) {
+    timing_desc.dts = dts;
+  } else {
+    timing_desc.dts = pts;
+  }
+  timing_desc_list_.push_back(
+      std::pair<int, TimingDesc>(raw_es_.size(), timing_desc));
+
+  // Add NALs from the incoming buffer.
+  FindNals(buf, size);
+
+  // Find access units based on AUD.
+  std::list<NalDescList::iterator> access_unit_list;
+  FindAccessUnits(&access_unit_list);
+  if (access_unit_list.empty()) {
+    int old_size = raw_es_.size();
+    raw_es_.resize(old_size + size);
+    memcpy(&raw_es_[old_size], buf, size);
+    DiscardEs(raw_es_.size() - 4);
+  }
+
+  // Make sure that all the frames to be emitted are in the ES buffer.
+  int last_position = (access_unit_list.back())->position;
+  int copy_size = last_position - raw_es_.size();
+  if (copy_size > 0) {
+    int copy_size = last_position - raw_es_.size();
+    int old_size = raw_es_.size();
+    raw_es_.resize(old_size + copy_size);
+    memcpy(&raw_es_[old_size], buf, copy_size);
+    buf += copy_size;
+    size -= copy_size;
+  }
+
+  // Emit all frames.
+  std::list<NalDescList::iterator>::iterator it0 = access_unit_list.begin();
+  std::list<NalDescList::iterator>::iterator it1 = it0;
+  ++it1;
+  LOG_IF(WARNING, (*it0)->position != 0)
+      << "Needs to discard some ES data before getting the 1st access unit: "
+      << (*it0)->position;
+  for (; it1 != access_unit_list.end(); ++it0, ++it1) {
+    int nxt_frame_position = (*it1)->position;
+    EmitFrame(*it0, *it1, nxt_frame_position);
+  }
+
+  // Discard emitted frames.
+  DiscardEs(last_position);
+
+  // Finally copy the incomplete access unit to the ES buffer.
+  int old_size = raw_es_.size();
+  raw_es_.resize(old_size + size);
+  memcpy(&raw_es_[old_size], buf, size);
+}
+
+void EsParserH264::Flush() {
+  // Find access units based on AUD.
+  std::list<NalDescList::iterator> access_unit_list;
+  FindAccessUnits(&access_unit_list);
+
+  // At this point, there can be at most one access unit in the buffer.
+  DCHECK_GE(access_unit_list.size(), 1u);
+  if (!access_unit_list.empty()) {
+    // Force emitting the last access unit (even it might be incomplete).
+    int nxt_frame_position = raw_es_.size();
+    NalDescList::iterator cur_frame = *(access_unit_list.begin());
+    NalDescList::iterator nxt_frame = nal_desc_list_.end();
+    EmitFrame(cur_frame, nxt_frame, nxt_frame_position);
+  }
+}
+
+void EsParserH264::FindNals(const uint8* buf, int size) {
+  ByteReaderChainedBuffer byte_reader(
+      &raw_es_[0], raw_es_.size(),
+      buf, size);
+
+  DCHECK_GE(nal_es_pos_, 0);
+  DCHECK_LT(nal_es_pos_, byte_reader.GetSize());
+
+  // Resume NAL segmentation where it was left.
+  for ( ; nal_es_pos_ < byte_reader.GetSize() - 4; nal_es_pos_++) {
+    // Make sure the syncword is either 00 00 00 01 or 00 00 01
+    if (byte_reader.Get(nal_es_pos_ + 0) != 0 ||
+        byte_reader.Get(nal_es_pos_ + 1) != 0) {
+      continue;
+    }
+    int syncword_length = 0;
+    if (byte_reader.Get(nal_es_pos_ + 2) == 0 &&
+        byte_reader.Get(nal_es_pos_ + 3) == 1) {
+      syncword_length = 4;
+    } else if (byte_reader.Get(nal_es_pos_ + 2) == 1) {
+      syncword_length = 3;
+    } else {
+      continue;
+    }
+
+    // Retrieve the NAL type.
+    int nal_header = byte_reader.Get(nal_es_pos_ + syncword_length);
+    int forbidden_zero_bit = (nal_header >> 7) & 0x1;
+    NalDesc nal_desc;
+    nal_desc.position = nal_es_pos_;
+    nal_desc.nal_unit_type = static_cast<NalUnitType>(nal_header & 0x1f);
+    if (forbidden_zero_bit != 0) {
+      nal_desc.nal_unit_type = kNalUnitTypeInvalid;
+    }
+    VLOG(LOG_LEVEL_ES) << "nal: offset=" << nal_desc.position
+                       << " type=" << nal_desc.nal_unit_type;
+    nal_desc_list_.push_back(nal_desc);
+    nal_es_pos_ += syncword_length;
+  }
+}
+
+void EsParserH264::FindAccessUnits(
+    std::list<NalDescList::iterator>* access_unit_list) {
+  // Get the H264 access units based on AUD.
+  // Mpeg2TS spec: "2.14 Carriage of Rec. ITU-T H.264 | ISO/IEC 14496-10 video"
+  // "Each AVC access unit shall contain an access unit delimiter NAL Unit;"
+  for (NalDescList::iterator it = nal_desc_list_.begin();
+       it != nal_desc_list_.end(); ++it) {
+    if (it->nal_unit_type == kNalUnitTypeAUD) {
+      VLOG(LOG_LEVEL_ES) << "aud found @ pos=" << it->position;
+      access_unit_list->push_back(it);
+    }
+  }
+}
+
+void EsParserH264::EmitFrame(
+    NalDescList::iterator cur_frame,
+    NalDescList::iterator nxt_frame,
+    int nxt_frame_position) {
+  // Current frame position = position of the 1st NAL of the frame.
+  int cur_frame_position = cur_frame->position;
+  int access_unit_size = nxt_frame_position - cur_frame_position;
+
+  // Get the access unit timing info.
+  TimingDesc current_timing_desc;
+  while (!timing_desc_list_.empty() &&
+         timing_desc_list_.front().first <= cur_frame_position) {
+    current_timing_desc = timing_desc_list_.front().second;
+    timing_desc_list_.pop_front();
+  }
+
+  // Check whether this is a key frame + light NAL parsing to get some
+  // relevant information (e.g. SPS/PPS).
+  // Note: it would have been nice to get the keyframe decision based
+  // on the Mpeg2TS random_access_indicator but encoders sometimes just don't
+  // bother setting this flag in the MPEG2 TS stream.
+  bool is_key_frame = true;
+  for (NalDescList::iterator it = cur_frame; it != nxt_frame; ++it) {
+    if (it->nal_unit_type == kNalUnitTypeNonIdrSlice) {
+      is_key_frame = false;
+    }
+    NalDescList::iterator next_nal_it = it;
+    ++next_nal_it;
+    int cur_nal_position = it->position;
+    int nxt_nal_position = (next_nal_it == nxt_frame)
+        ? nxt_frame_position : next_nal_it->position;
+    int nal_size = nxt_nal_position - cur_nal_position;
+    DCHECK_LE(cur_nal_position + nal_size, static_cast<int>(raw_es_.size()));
+    NalParser(&raw_es_[cur_nal_position], nal_size);
+  }
+
+  // Emit the current frame.
+  VLOG(LOG_LEVEL_ES) << "is_key_frame = " << is_key_frame;
+  scoped_refptr<StreamParserBuffer> stream_parser_buffer =
+      StreamParserBuffer::CopyFrom(
+          &raw_es_[cur_frame_position],
+          access_unit_size,
+          is_key_frame);
+  stream_parser_buffer->SetDecodeTimestamp(current_timing_desc.dts);
+  stream_parser_buffer->set_timestamp(current_timing_desc.pts);
+  emit_buffer_cb_.Run(stream_parser_buffer);
+}
+
+void EsParserH264::DiscardEs(int nbytes) {
+  if (nbytes <= 0) {
+    return;
+  }
+
+  // Update the NAL list accordingly.
+  while (!nal_desc_list_.empty() &&
+         nal_desc_list_.front().position < nbytes) {
+    nal_desc_list_.pop_front();
+  }
+  for (NalDescList::iterator it = nal_desc_list_.begin();
+       it != nal_desc_list_.end(); ++it) {
+    DCHECK(it->position >= nbytes);
+    it->position -= nbytes;
+  }
+  nal_es_pos_ -= nbytes;
+  if (nal_es_pos_ < 0) {
+    nal_es_pos_ = 0;
+  }
+
+  // Update the timing information accordingly.
+  std::list<std::pair<int, TimingDesc> >::iterator timing_it
+      = timing_desc_list_.begin();
+  for (; timing_it != timing_desc_list_.end(); ++timing_it) {
+    timing_it->first -= nbytes;
+  }
+
+  // Discard |nbytes| of ES.
+  int old_size = raw_es_.size();
+  int new_size = old_size - nbytes;
+  CHECK_LE(nbytes, old_size);
+  if (new_size > 0) {
+    memmove(&raw_es_[0], &raw_es_[nbytes], new_size);
+  }
+  raw_es_.resize(new_size);
+}
+
+void EsParserH264::NalParser(const uint8* buf, int size) {
+  // Discard the annexB syncword.
+  if (size < 3) {
+    LOG(WARNING) << "NalParser: incomplete NAL";
+    return;
+  }
+  DCHECK_EQ(buf[0], 0);
+  DCHECK_EQ(buf[1], 0);
+  if (buf[2] == 1) {
+    buf += 3;
+    size -= 3;
+  } else {
+    buf += 4;
+    size -= 4;
+  }
+
+  // Get the NAL header.
+  if (size < 1) {
+    LOG(WARNING) << "NalParser: incomplete NAL";
+    return;
+  }
+  int nal_header = buf[0];
+  buf += 1;
+  size -= 1;
+
+  int forbidden_zero_bit = (nal_header >> 7) & 0x1;
+  if (forbidden_zero_bit != 0) {
+    return;
+  }
+  int nal_ref_idc = (nal_header >> 5) & 0x3;
+  int nal_unit_type = nal_header & 0x1f;
+
+  // TODO(damienv):
+  // The nal start code emulation prevention should be un-done,
+  // before parsing the NAL content.
+
+  // Process the NAL content.
+  if (nal_unit_type == kNalUnitTypeSPS) {
+    VLOG(LOG_LEVEL_ES) << "NAL: SPS";
+    if (nal_ref_idc == 0) {
+      // Should not be 0 for a SPS.
+      return;
+    }
+    ProcessSPS(buf, size);
+  } else if (nal_unit_type == kNalUnitTypeIdrSlice) {
+    VLOG(LOG_LEVEL_ES) << "NAL: IDR slice";
+    ProcessSliceLayer(buf, size);
+  } else if (nal_unit_type == kNalUnitTypeNonIdrSlice) {
+    VLOG(LOG_LEVEL_ES) << "NAL: Non IDR slice";
+    ProcessSliceLayer(buf, size);
+  } else if (nal_unit_type == kNalUnitTypePPS) {
+    VLOG(LOG_LEVEL_ES) << "NAL: PPS";
+  } else if (nal_unit_type == kNalUnitTypeAUD) {
+    VLOG(LOG_LEVEL_ES) << "NAL: AUD";
+  } else {
+    VLOG(LOG_LEVEL_ES) << "NAL: " << nal_unit_type;
+  }
+}
+
+bool EsParserH264::ProcessSPS(const uint8* buf, int size) {
+  if (size <= 0) {
+    return false;
+  }
+  BitReader bit_reader(buf, size);
+
+  int profile_idc;
+  RCHECK(bit_reader.ReadBits(8, &profile_idc));
+  int constraint_setX_flag;
+  RCHECK(bit_reader.ReadBits(8, &constraint_setX_flag));
+  int level_idc;
+  RCHECK(bit_reader.ReadBits(8, &level_idc));
+  uint32 seq_parameter_set_id;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &seq_parameter_set_id));
+  uint32 log2_max_frame_num_minus4;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &log2_max_frame_num_minus4));
+  uint32 pic_order_cnt_type;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &pic_order_cnt_type));
+
+  if (pic_order_cnt_type > 2) {
+    // Bitstream error: pic_order_cnt_type shall be in the range of 0 to 2.
+    return false;
+  }
+  if (pic_order_cnt_type == 0) {
+    uint32 log2_max_pic_order_cnt_lsb_minus4;
+    RCHECK(ReadBitsExpGolomb(&bit_reader, &log2_max_pic_order_cnt_lsb_minus4));
+  } else if (pic_order_cnt_type == 1) {
+    NOTIMPLEMENTED();
+    LOG(FATAL) << "pic_order_cnt_type = 1 not supported yet";
+  }
+
+  uint32 num_ref_frames;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &num_ref_frames));
+  int gaps_in_frame_num_value_allowed_flag;
+  RCHECK(bit_reader.ReadBits(1, &gaps_in_frame_num_value_allowed_flag));
+  uint32 pic_width_in_mbs_minus1;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &pic_width_in_mbs_minus1));
+  uint32 pic_height_in_map_units_minus1;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &pic_height_in_map_units_minus1));
+
+  int frame_mbs_only_flag;
+  RCHECK(bit_reader.ReadBits(1, &frame_mbs_only_flag));
+  if (!frame_mbs_only_flag) {
+    int mb_adaptive_frame_field_flag;
+    RCHECK(bit_reader.ReadBits(1, &mb_adaptive_frame_field_flag));
+  }
+
+  int direct_8x8_inference_flag;
+  RCHECK(bit_reader.ReadBits(1, &direct_8x8_inference_flag));
+
+  bool frame_cropping_flag;
+  uint32 frame_crop_left_offset = 0;
+  uint32 frame_crop_right_offset = 0;
+  uint32 frame_crop_top_offset = 0;
+  uint32 frame_crop_bottom_offset = 0;
+  RCHECK(bit_reader.ReadBits(1, &frame_cropping_flag));
+  if (frame_cropping_flag) {
+    RCHECK(ReadBitsExpGolomb(&bit_reader, &frame_crop_left_offset));
+    RCHECK(ReadBitsExpGolomb(&bit_reader, &frame_crop_right_offset));
+    RCHECK(ReadBitsExpGolomb(&bit_reader, &frame_crop_top_offset));
+    RCHECK(ReadBitsExpGolomb(&bit_reader, &frame_crop_bottom_offset));
+  }
+
+  bool vui_parameters_present_flag;
+  RCHECK(bit_reader.ReadBits(1, &vui_parameters_present_flag));
+  int sar_width = 1;
+  int sar_height = 1;
+  if (vui_parameters_present_flag) {
+    // Read only the aspect ratio information from the VUI section.
+    // TODO(damienv): check whether other VUI info are useful.
+    bool aspect_ratio_info_present_flag = false;
+    RCHECK(bit_reader.ReadBits(1, &aspect_ratio_info_present_flag));
+    if (aspect_ratio_info_present_flag) {
+      int aspect_ratio_idc;
+      RCHECK(bit_reader.ReadBits(8, &aspect_ratio_idc));
+      if (aspect_ratio_idc == kExtendedSar) {
+        RCHECK(bit_reader.ReadBits(16, &sar_width));
+        RCHECK(bit_reader.ReadBits(16, &sar_height));
+      } else if (aspect_ratio_idc < 14) {
+        sar_width = kTableSarWidth[aspect_ratio_idc];
+        sar_height = kTableSarHeight[aspect_ratio_idc];
+      }
+    }
+  }
+
+  LOG_IF(WARNING, sar_width != sar_height)
+      << "Non square pixel not supported yet:"
+      << " sar_width=" << sar_width
+      << " sar_height=" << sar_height;
+
+  if (is_video_config_known_ &&
+      profile_idc == profile_idc_ &&
+      level_idc == level_idc_ &&
+      pic_width_in_mbs_minus1 == pic_width_in_mbs_minus1_ &&
+      pic_height_in_map_units_minus1 == pic_height_in_map_units_minus1_) {
+    // This is the same SPS as the previous one.
+    return true;
+  }
+  is_video_config_known_ = true;
+  profile_idc_ = profile_idc;
+  level_idc_ = level_idc;
+  pic_width_in_mbs_minus1_ = pic_width_in_mbs_minus1;
+  pic_height_in_map_units_minus1_ = pic_height_in_map_units_minus1;
+
+  // TODO(damienv):
+  // Assuming the SPS is used right away by the PPS
+  // and the slice headers is a strong assumption.
+  // In theory, we should process the SPS and PPS
+  // and only when one of the slice header is switching
+  // the PPS id, the video decoder config should be changed.
+  LOG(INFO) << "Profile IDC: " << profile_idc;
+  LOG(INFO) << "Level IDC: " << level_idc;
+  LOG(INFO) << "Pic width: " << (pic_width_in_mbs_minus1 + 1) * 16;
+  LOG(INFO) << "Pic height: " << (pic_height_in_map_units_minus1 + 1) * 16;
+  LOG(INFO) << "log2_max_frame_num_minus4: " << log2_max_frame_num_minus4;
+
+  // TODO(damienv): a MAP unit can be either 16 or 32 pixels.
+  // although it's 16 pixels for progressive non MBAFF frames.
+  gfx::Size coded_size((pic_width_in_mbs_minus1 + 1) * 16,
+                       (pic_height_in_map_units_minus1 + 1) * 16);
+  gfx::Rect visible_rect(
+      frame_crop_left_offset,
+      frame_crop_top_offset,
+      (coded_size.width() - frame_crop_right_offset) - frame_crop_left_offset,
+      (coded_size.height() - frame_crop_bottom_offset) - frame_crop_top_offset);
+
+  // TODO(damienv): calculate the natural size based
+  // on the possible aspect ratio coded in the VUI parameters.
+  gfx::Size natural_size(visible_rect.width(),
+                         visible_rect.height());
+
+  VideoDecoderConfig video_decoder_config(
+      kCodecH264,
+      VIDEO_CODEC_PROFILE_UNKNOWN,    // TODO(damienv)
+      VideoFrame::YV12,
+      coded_size,
+      visible_rect,
+      natural_size,
+      NULL, 0,
+      false);
+  new_video_config_cb_.Run(video_decoder_config);
+
+  return true;
+}
+
+bool EsParserH264::ProcessSliceLayer(const uint8* buf, int size) {
+  if (size <= 0) {
+    return false;
+  }
+  BitReader bit_reader(buf, size);
+
+  // Read only the slice header.
+  // TODO(damienv): frame_num
+  uint32 first_mb_in_slice;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &first_mb_in_slice));
+  uint32 slice_type;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &slice_type));
+  uint32 pic_parameter_set_id;
+  RCHECK(ReadBitsExpGolomb(&bit_reader, &pic_parameter_set_id));
+
+  VLOG(LOG_LEVEL_ES) << "first_mb_in_slice: " << first_mb_in_slice;
+  VLOG(LOG_LEVEL_ES) << "slice_type: " << slice_type;
+  return true;
+}
+
+bool EsParserH264::ReadBitsExpGolomb(
+    BitReader* bit_reader, uint32* exp_golomb_value) {
+  // TODO(damienv): this should be a member function of BitReader.
+
+  // Get the number of leading zeros.
+  int zero_count = 0;
+  for (zero_count = 0; ; zero_count++) {
+    int one_bit;
+    if (!bit_reader->ReadBits(1, &one_bit)) {
+      return false;
+    }
+    if (one_bit != 0) {
+      break;
+    }
+  }
+
+  // Read the actual value.
+  uint32 base_value = (1 << zero_count) - 1;
+  uint32 value = 0;
+  for (int bit_count = 0; bit_count < zero_count; bit_count++) {
+    int one_bit;
+    if (!bit_reader->ReadBits(1, &one_bit)) {
+      return false;
+    }
+    if (one_bit != 0) {
+      value += (1 << (zero_count-1 - bit_count));
+    }
+  }
+
+  *exp_golomb_value = base_value + value;
+  return true;
+}
+
+}  // namespace mpeg2ts
+}  // namespace media
+