xbmc/cores/AudioEngine/Utils/AEBitstreamPacker.cpp

   1 /*
   2  *  Copyright (C) 2010-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "AEBitstreamPacker.h"
  10
  11 #include "AEPackIEC61937.h"
  12 #include "AEStreamInfo.h"
  13 #include "utils/log.h"
  14
  15 #include <array>
  16 #include <stddef.h>
  17 #include <stdint.h>
  18 #include <string.h>
  19
  20 namespace
  21 {
  22 constexpr auto BURST_HEADER_SIZE = 8;
  23 constexpr auto EAC3_MAX_BURST_PAYLOAD_SIZE = 24576 - BURST_HEADER_SIZE;
  24 } // namespace
  25
  26 CAEBitstreamPacker::CAEBitstreamPacker()
  27 {
  28   Reset();
  29 }
  30
  31 CAEBitstreamPacker::~CAEBitstreamPacker()
  32 {
  33 }
  34
  35 void CAEBitstreamPacker::Pack(CAEStreamInfo &info, uint8_t* data, int size)
  36 {
  37   m_pauseDuration = 0;
  38   switch (info.m_type)
  39   {
  40     case CAEStreamInfo::STREAM_TYPE_TRUEHD:
  41       m_dataSize = CAEPackIEC61937::PackTrueHD(data + IEC61937_DATA_OFFSET,
  42                                                size - IEC61937_DATA_OFFSET, m_packedBuffer);
  43       break;
  44
  45     case CAEStreamInfo::STREAM_TYPE_DTSHD:
  46     case CAEStreamInfo::STREAM_TYPE_DTSHD_MA:
  47       PackDTSHD (info, data, size);
  48       break;
  49
  50     case CAEStreamInfo::STREAM_TYPE_AC3:
  51       m_dataSize = CAEPackIEC61937::PackAC3(data, size, m_packedBuffer);
  52       break;
  53
  54     case CAEStreamInfo::STREAM_TYPE_EAC3:
  55       PackEAC3 (info, data, size);
  56       break;
  57
  58     case CAEStreamInfo::STREAM_TYPE_DTSHD_CORE:
  59     case CAEStreamInfo::STREAM_TYPE_DTS_512:
  60       m_dataSize = CAEPackIEC61937::PackDTS_512(data, size, m_packedBuffer, info.m_dataIsLE);
  61       break;
  62
  63     case CAEStreamInfo::STREAM_TYPE_DTS_1024:
  64       m_dataSize = CAEPackIEC61937::PackDTS_1024(data, size, m_packedBuffer, info.m_dataIsLE);
  65       break;
  66
  67     case CAEStreamInfo::STREAM_TYPE_DTS_2048:
  68       m_dataSize = CAEPackIEC61937::PackDTS_2048(data, size, m_packedBuffer, info.m_dataIsLE);
  69       break;
  70
  71     default:
  72       CLog::Log(LOGERROR, "CAEBitstreamPacker::Pack - no pack function");
  73   }
  74 }
  75
  76 bool CAEBitstreamPacker::PackPause(CAEStreamInfo &info, unsigned int millis, bool iecBursts)
  77 {
  78   // re-use last buffer
  79   if (m_pauseDuration == millis)
  80     return false;
  81
  82   switch (info.m_type)
  83   {
  84     case CAEStreamInfo::STREAM_TYPE_TRUEHD:
  85     case CAEStreamInfo::STREAM_TYPE_EAC3:
  86       m_dataSize = CAEPackIEC61937::PackPause(m_packedBuffer, millis, GetOutputChannelMap(info).Count() * 2, GetOutputRate(info), 4, info.m_sampleRate);
  87       m_pauseDuration = millis;
  88       break;
  89
  90     case CAEStreamInfo::STREAM_TYPE_AC3:
  91     case CAEStreamInfo::STREAM_TYPE_DTSHD:
  92     case CAEStreamInfo::STREAM_TYPE_DTSHD_MA:
  93     case CAEStreamInfo::STREAM_TYPE_DTSHD_CORE:
  94     case CAEStreamInfo::STREAM_TYPE_DTS_512:
  95     case CAEStreamInfo::STREAM_TYPE_DTS_1024:
  96     case CAEStreamInfo::STREAM_TYPE_DTS_2048:
  97       m_dataSize = CAEPackIEC61937::PackPause(m_packedBuffer, millis, GetOutputChannelMap(info).Count() * 2, GetOutputRate(info), 3, info.m_sampleRate);
  98       m_pauseDuration = millis;
  99       break;
 100
 101     default:
 102       CLog::Log(LOGERROR, "CAEBitstreamPacker::Pack - no pack function");
 103   }
 104
 105   if (!iecBursts)
 106   {
 107     memset(m_packedBuffer, 0, m_dataSize);
 108   }
 109
 110   return true;
 111 }
 112
 113 unsigned int CAEBitstreamPacker::GetSize() const
 114 {
 115   return m_dataSize;
 116 }
 117
 118 uint8_t* CAEBitstreamPacker::GetBuffer()
 119 {
 120   return m_packedBuffer;
 121 }
 122
 123 void CAEBitstreamPacker::Reset()
 124 {
 125   m_dataSize = 0;
 126   m_pauseDuration = 0;
 127   m_packedBuffer[0] = 0;
 128 }
 129
 130 void CAEBitstreamPacker::PackDTSHD(CAEStreamInfo &info, uint8_t* data, int size)
 131 {
 132   static const uint8_t dtshd_start_code[10] = { 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xfe, 0xfe };
 133   unsigned int dataSize = sizeof(dtshd_start_code) + 2 + size;
 134
 135   if (dataSize > m_dtsHDSize)
 136   {
 137     m_dtsHDSize = dataSize;
 138     m_dtsHD.resize(dataSize);
 139     memcpy(m_dtsHD.data(), dtshd_start_code, sizeof(dtshd_start_code));
 140   }
 141
 142   m_dtsHD[sizeof(dtshd_start_code) + 0] = ((uint16_t)size & 0xFF00) >> 8;
 143   m_dtsHD[sizeof(dtshd_start_code) + 1] = ((uint16_t)size & 0x00FF);
 144   memcpy(m_dtsHD.data() + sizeof(dtshd_start_code) + 2, data, size);
 145
 146   m_dataSize =
 147       CAEPackIEC61937::PackDTSHD(m_dtsHD.data(), dataSize, m_packedBuffer, info.m_dtsPeriod);
 148 }
 149
 150 void CAEBitstreamPacker::PackEAC3(CAEStreamInfo &info, uint8_t* data, int size)
 151 {
 152   unsigned int framesPerBurst = info.m_repeat;
 153
 154   if (m_eac3FramesPerBurst != framesPerBurst)
 155   {
 156     /* switched streams, discard partial burst */
 157     m_eac3Size = 0;
 158     m_eac3FramesPerBurst = framesPerBurst;
 159   }
 160
 161   if (m_eac3FramesPerBurst == 1)
 162   {
 163     /* simple case, just pass through */
 164     m_dataSize = CAEPackIEC61937::PackEAC3(data, size, m_packedBuffer);
 165   }
 166   else
 167   {
 168     /* multiple frames needed to achieve 6 blocks as required by IEC 61937-3:2007 */
 169
 170     if (m_eac3.size() == 0)
 171       m_eac3.resize(EAC3_MAX_BURST_PAYLOAD_SIZE);
 172
 173     unsigned int newsize = m_eac3Size + size;
 174     bool overrun = newsize > EAC3_MAX_BURST_PAYLOAD_SIZE;
 175
 176     if (!overrun)
 177     {
 178       memcpy(m_eac3.data() + m_eac3Size, data, size);
 179       m_eac3Size = newsize;
 180       m_eac3FramesCount++;
 181     }
 182
 183     if (m_eac3FramesCount >= m_eac3FramesPerBurst || overrun)
 184     {
 185       m_dataSize = CAEPackIEC61937::PackEAC3(m_eac3.data(), m_eac3Size, m_packedBuffer);
 186       m_eac3Size = 0;
 187       m_eac3FramesCount = 0;
 188     }
 189   }
 190 }
 191
 192 unsigned int CAEBitstreamPacker::GetOutputRate(const CAEStreamInfo& info)
 193 {
 194   unsigned int rate;
 195   switch (info.m_type)
 196   {
 197     case CAEStreamInfo::STREAM_TYPE_AC3:
 198       rate = info.m_sampleRate;
 199       break;
 200     case CAEStreamInfo::STREAM_TYPE_EAC3:
 201       rate = info.m_sampleRate * 4;
 202       break;
 203     case CAEStreamInfo::STREAM_TYPE_TRUEHD:
 204       if (info.m_sampleRate == 48000 ||
 205           info.m_sampleRate == 96000 ||
 206           info.m_sampleRate == 192000)
 207         rate = 192000;
 208       else
 209         rate = 176400;
 210       break;
 211     case CAEStreamInfo::STREAM_TYPE_DTS_512:
 212     case CAEStreamInfo::STREAM_TYPE_DTS_1024:
 213     case CAEStreamInfo::STREAM_TYPE_DTS_2048:
 214     case CAEStreamInfo::STREAM_TYPE_DTSHD_CORE:
 215       rate = info.m_sampleRate;
 216       break;
 217     case CAEStreamInfo::STREAM_TYPE_DTSHD:
 218     case CAEStreamInfo::STREAM_TYPE_DTSHD_MA:
 219       rate = 192000;
 220       break;
 221     default:
 222       rate = 48000;
 223       break;
 224   }
 225   return rate;
 226 }
 227
 228 CAEChannelInfo CAEBitstreamPacker::GetOutputChannelMap(const CAEStreamInfo& info)
 229 {
 230   int channels = 2;
 231   switch (info.m_type)
 232   {
 233     case CAEStreamInfo::STREAM_TYPE_AC3:
 234     case CAEStreamInfo::STREAM_TYPE_EAC3:
 235     case CAEStreamInfo::STREAM_TYPE_DTS_512:
 236     case CAEStreamInfo::STREAM_TYPE_DTS_1024:
 237     case CAEStreamInfo::STREAM_TYPE_DTS_2048:
 238     case CAEStreamInfo::STREAM_TYPE_DTSHD_CORE:
 239     case CAEStreamInfo::STREAM_TYPE_DTSHD:
 240       channels = 2;
 241       break;
 242
 243     case CAEStreamInfo::STREAM_TYPE_TRUEHD:
 244     case CAEStreamInfo::STREAM_TYPE_DTSHD_MA:
 245       channels = 8;
 246       break;
 247
 248     default:
 249       break;
 250   }
 251
 252   CAEChannelInfo channelMap;
 253   for (int i=0; i<channels; i++)
 254   {
 255     channelMap += AE_CH_RAW;
 256   }
 257
 258   return channelMap;
 259 }