// 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/base/audio_stream_info.h"

#include <inttypes.h>

#include "packager/base/logging.h"
#include "packager/base/strings/string_number_conversions.h"
#include "packager/base/strings/stringprintf.h"
#include "packager/media/base/limits.h"

namespace shaka {
namespace media {

namespace {
std::string AudioCodecToString(Codec codec) {
  switch (codec) {
    case kCodecAAC:
      return "AAC";
    case kCodecAC3:
      return "AC3";
    case kCodecDTSC:
      return "DTSC";
    case kCodecDTSE:
      return "DTSE";
    case kCodecDTSH:
      return "DTSH";
    case kCodecDTSL:
      return "DTSL";
    case kCodecDTSM:
      return "DTS-";
    case kCodecDTSP:
      return "DTS+";
    case kCodecEAC3:
      return "EAC3";
    case kCodecAC4:
      return "AC4";
    case kCodecFlac:
      return "FLAC";
    case kCodecOpus:
      return "Opus";
    case kCodecVorbis:
      return "Vorbis";
    case kCodecMP3:
      return "MP3";
    default:
      NOTIMPLEMENTED() << "Unknown Audio Codec: " << codec;
      return "UnknownCodec";
  }
}

FourCC CodecToFourCC(Codec codec) {
  switch (codec) {
    case kCodecMha1:
      return FOURCC_mha1;
    case kCodecMhm1:
      return FOURCC_mhm1;
    default:
      return FOURCC_NULL;
  }
}
}  // namespace

AudioStreamInfo::AudioStreamInfo(
    int track_id, uint32_t time_scale, uint64_t duration, Codec codec,
    const std::string& codec_string, const uint8_t* codec_config,
    size_t codec_config_size, uint8_t sample_bits, uint8_t num_channels,
    uint32_t sampling_frequency, uint64_t seek_preroll_ns,
    uint64_t codec_delay_ns, uint32_t max_bitrate, uint32_t avg_bitrate,
    const std::string& language, bool is_encrypted)
    : StreamInfo(kStreamAudio, track_id, time_scale, duration, codec,
                 codec_string, codec_config, codec_config_size, language,
                 is_encrypted),
      sample_bits_(sample_bits),
      num_channels_(num_channels),
      sampling_frequency_(sampling_frequency),
      seek_preroll_ns_(seek_preroll_ns),
      codec_delay_ns_(codec_delay_ns),
      max_bitrate_(max_bitrate),
      avg_bitrate_(avg_bitrate) {}

AudioStreamInfo::~AudioStreamInfo() {}

bool AudioStreamInfo::IsValidConfig() const {
  return codec() != kUnknownCodec && num_channels_ != 0 &&
         num_channels_ <= limits::kMaxChannels && sample_bits_ > 0 &&
         sample_bits_ <= limits::kMaxBitsPerSample && sampling_frequency_ > 0 &&
         sampling_frequency_ <= limits::kMaxSampleRate;
}

std::string AudioStreamInfo::ToString() const {
  std::string str = base::StringPrintf(
      "%s codec: %s\n sample_bits: %d\n num_channels: %d\n "
      "sampling_frequency: %d\n language: %s\n",
      StreamInfo::ToString().c_str(), AudioCodecToString(codec()).c_str(),
      sample_bits_, num_channels_, sampling_frequency_, language().c_str());
  if (seek_preroll_ns_ != 0) {
    base::StringAppendF(&str, " seek_preroll_ns: %" PRIu64 "\n",
                        seek_preroll_ns_);
  }
  if (codec_delay_ns_ != 0) {
    base::StringAppendF(&str, " codec_delay_ns: %" PRIu64 "\n",
                        codec_delay_ns_);
  }
  return str;
}

std::unique_ptr<StreamInfo> AudioStreamInfo::Clone() const {
  return std::unique_ptr<StreamInfo>(new AudioStreamInfo(*this));
}

std::string AudioStreamInfo::GetCodecString(Codec codec,
                                            uint8_t audio_object_type) {
  switch (codec) {
    case kCodecAAC:
      return "mp4a.40." + base::UintToString(audio_object_type);
    case kCodecAC3:
      return "ac-3";
    case kCodecDTSC:
      return "dtsc";
    case kCodecDTSE:
      return "dtse";
    case kCodecDTSH:
      return "dtsh";
    case kCodecDTSL:
      return "dtsl";
    case kCodecDTSM:
      return "dts-";
    case kCodecDTSP:
      return "dts+";
    case kCodecEAC3:
      return "ec-3";
    case kCodecAC4:
      // ETSI TS 103 190-2 Digital Audio Compression (AC-4) Standard; Part 2:
      // Immersive and personalized audio E.13. audio_object_type is composed of
      // bitstream_version (3bits), presentation_version (2bits) and
      // mdcompat (3bits).
      return base::StringPrintf(
        "ac-4.%02d.%02d.%02d", (audio_object_type & 0xE0) >> 5,
        (audio_object_type & 0x18) >> 3, audio_object_type & 0x7);
    case kCodecFlac:
      return "flac";
    case kCodecOpus:
      return "opus";
    case kCodecMP3:
      return "mp3";
    case kCodecVorbis:
      return "vorbis";
    case kCodecMha1:
    case kCodecMhm1:
      // The signalling of the codecs parameters is according to RFC6381 [11]
      //   and ISO/IEC 23008-3 clause 21 [7].
      // The value consists of the following two parts separated by a dot:
      //  - the sample entry 4CC code ('mha1', 'mha2', 'mhm1', 'mhm2')
      //  - ‘0x’ followed by the hex value of the profile-levelid, as defined 
      //      in in ISO/IEC 23008-3 [7]
      return base::StringPrintf("%s.0x%02x", 
                                FourCCToString(CodecToFourCC(codec)).c_str(), 
                                audio_object_type);
    default:
      NOTIMPLEMENTED() << "Codec: " << codec;
      return "unknown";
  }
}

}  // namespace media
}  // namespace shaka