shaka-packager/packager/media/formats/mp2t/adts_header.cc

// Copyright 2014 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include "packager/media/formats/mp2t/adts_header.h"

#include "packager/media/base/bit_reader.h"
#include "packager/media/formats/mp2t/mp2t_common.h"
#include "packager/media/formats/mpeg/adts_constants.h"

namespace shaka {
namespace media {
namespace mp2t {

AdtsHeader::AdtsHeader()
    : valid_config_(false),
      profile_(0),
      sampling_frequency_index_(0),
      channel_configuration_(0) {}

size_t AdtsHeader::GetAdtsFrameSize(const uint8_t* data, size_t num_bytes) {
  if (num_bytes < 6)
    return 0;
  return ((static_cast<int>(data[5]) >> 5) |
          (static_cast<int>(data[4]) << 3) |
          ((static_cast<int>(data[3]) & 0x3) << 11));
}

size_t AdtsHeader::GetAdtsHeaderSize(const uint8_t* data, size_t num_bytes) {
  if (num_bytes < 2)
    return 0;
  if (data[1] & 0x01)
    return kAdtsHeaderMinSize;
  return kAdtsHeaderMinSize + sizeof(uint16_t);  // Header + CRC.
}

bool AdtsHeader::Parse(const uint8_t* adts_frame, size_t adts_frame_size) {
  CHECK(adts_frame);

  valid_config_ = false;

  BitReader frame(adts_frame, adts_frame_size);
  // Verify frame starts with sync bits (0xfff).
  uint32_t sync;
  RCHECK(frame.ReadBits(12, &sync));
  RCHECK(sync == 0xfff);
  // Skip MPEG version and layer.
  RCHECK(frame.SkipBits(3));
  // Get "protection absent" flag.
  uint8_t protection_absent;
  RCHECK(frame.ReadBits(1, &protection_absent));
  // Get profile.
  RCHECK(frame.ReadBits(2, &profile_));
  // Get sampling frequency.
  RCHECK(frame.ReadBits(4, &sampling_frequency_index_));
  RCHECK(sampling_frequency_index_ < kAdtsFrequencyTableSize);
  // Skip private stream bit.
  RCHECK(frame.SkipBits(1));
  // Get number of audio channels.
  RCHECK(frame.ReadBits(3, &channel_configuration_));
  RCHECK((channel_configuration_ > 0) &&
         (channel_configuration_ < kAdtsNumChannelsTableSize));
  // Skip originality, home and copyright info.
  RCHECK(frame.SkipBits(4));
  // Verify that the frame size matches input parameters.
  uint16_t frame_size;
  RCHECK(frame.ReadBits(13, &frame_size));
  RCHECK(frame_size == adts_frame_size);
  // Skip buffer fullness indicator.
  RCHECK(frame.SkipBits(11));
  uint8_t num_blocks_minus_1;
  RCHECK(frame.ReadBits(2, &num_blocks_minus_1));
  if (num_blocks_minus_1) {
    NOTIMPLEMENTED() << "ADTS frames with more than one data block "
                        "not supported.";
    return false;
  }

  valid_config_ = true;
  return true;
}

bool AdtsHeader::GetAudioSpecificConfig(std::vector<uint8_t>* buffer) const {
  DCHECK(buffer);
  if (!valid_config_)
    return false;

  buffer->resize(2);
  (*buffer)[0] = ((profile_ + 1) << 3) | (sampling_frequency_index_ >> 1);
  (*buffer)[1] = ((sampling_frequency_index_ & 1) << 7) |
      (channel_configuration_ << 3);
  return true;
}

uint8_t AdtsHeader::GetObjectType() const {
  return profile_ + 1;
}

uint32_t AdtsHeader::GetSamplingFrequency() const {
  DCHECK_LT(sampling_frequency_index_, kAdtsFrequencyTableSize);
  return kAdtsFrequencyTable[sampling_frequency_index_];
}

uint8_t AdtsHeader::GetNumChannels() const {
  DCHECK_GT(channel_configuration_, 0);
  DCHECK_LT(channel_configuration_, kAdtsNumChannelsTableSize);
  return kAdtsNumChannelsTable[channel_configuration_];
}

}  // namespace mp2t
}  // namespace media
}  // namespace shaka