mp4-latm-depack.cpp
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:9k
源码类别:

Symbian

开发平台:

C/C++

  1. /* ***** BEGIN LICENSE BLOCK ***** 
  2.  * Version: RCSL 1.0/RPSL 1.0 
  3.  *  
  4.  * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved. 
  5.  *      
  6.  * The contents of this file, and the files included with this file, are 
  7.  * subject to the current version of the RealNetworks Public Source License 
  8.  * Version 1.0 (the "RPSL") available at 
  9.  * http://www.helixcommunity.org/content/rpsl unless you have licensed 
  10.  * the file under the RealNetworks Community Source License Version 1.0 
  11.  * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl, 
  12.  * in which case the RCSL will apply. You may also obtain the license terms 
  13.  * directly from RealNetworks.  You may not use this file except in 
  14.  * compliance with the RPSL or, if you have a valid RCSL with RealNetworks 
  15.  * applicable to this file, the RCSL.  Please see the applicable RPSL or 
  16.  * RCSL for the rights, obligations and limitations governing use of the 
  17.  * contents of the file.  
  18.  *  
  19.  * This file is part of the Helix DNA Technology. RealNetworks is the 
  20.  * developer of the Original Code and owns the copyrights in the portions 
  21.  * it created. 
  22.  *  
  23.  * This file, and the files included with this file, is distributed and made 
  24.  * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 
  25.  * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES, 
  26.  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS 
  27.  * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 
  28.  * 
  29.  * Technology Compatibility Kit Test Suite(s) Location: 
  30.  *    http://www.helixcommunity.org/content/tck 
  31.  * 
  32.  * Contributor(s): 
  33.  *  
  34.  * ***** END LICENSE BLOCK ***** */ 
  35. #include "hlxclib/string.h"
  36. #include "mp4-latm-depack.h"
  37. #define DEFAULT_PAYLOAD_BUF_SIZE 1024
  38. #define DEFAULT_STAGING_BUF_SIZE 1024
  39. MP4LATMDepack::MP4LATMDepack() :
  40.     m_ulSampleRate(0),
  41.     m_ulFrameDuration(0),
  42.     m_pCallback(0),
  43.     m_pUserData(0),
  44.     m_bConfigPresent(FALSE),
  45.     m_bNeedTime(TRUE),
  46.     m_ulCurrentTime(0),
  47.     m_pStagingBuf(new UINT8[DEFAULT_STAGING_BUF_SIZE]),
  48.     m_ulStagingBufSize(DEFAULT_PAYLOAD_BUF_SIZE),
  49.     m_ulBytesStaged(0),
  50.     m_pSlotLengths(0),
  51.     m_pPayloadBuf(new UINT8[DEFAULT_PAYLOAD_BUF_SIZE]),
  52.     m_ulPayloadBufSize(DEFAULT_PAYLOAD_BUF_SIZE),
  53.     m_bHadLoss(FALSE),
  54.     m_bNeedMarker(FALSE)
  55. {}
  56. MP4LATMDepack::~MP4LATMDepack()
  57. {
  58.     delete [] m_pStagingBuf;
  59.     m_pStagingBuf = 0;
  60.     delete [] m_pSlotLengths;
  61.     m_pSlotLengths = 0;
  62.     delete [] m_pPayloadBuf;
  63.     m_pPayloadBuf = 0;
  64. }
  65. BOOL MP4LATMDepack::Init(ULONG32 ulProfileID,
  66.  ULONG32 ulObject,
  67.  ULONG32 ulBitrate,
  68.  BOOL bConfigPresent,
  69.  const UINT8* pStreamMuxConfig,
  70.  ULONG32 ulMuxConfigSize,
  71.  OnFrameCB pFrameCB,
  72.  void* pUserData)
  73. {
  74.     BOOL ret = FALSE;
  75.     if (pStreamMuxConfig)
  76.     {
  77. Bitstream bs;
  78. bs.SetBuffer(pStreamMuxConfig);
  79. // A config was specified so try to unpack it
  80. ret = HandleMuxConfig(bs);
  81.     }
  82.     else if (bConfigPresent)
  83.     {
  84. // A config will be specified in the stream so signal success
  85. // since we know we are going to get a config soon
  86. ret = TRUE;
  87.     }
  88.     if (ret)
  89.     {
  90. m_bConfigPresent = bConfigPresent;
  91. m_pCallback = pFrameCB;
  92. m_pUserData = pUserData;
  93.     }
  94.     return ret;
  95. }
  96. BOOL MP4LATMDepack::GetCodecConfig(const UINT8*& pConfig, 
  97.    ULONG32& ulConfigSize)
  98. {
  99.     BOOL ret = FALSE;
  100.     if (m_muxConfig.NumStreams() > 0)
  101.     {
  102. // Currently we just hand out the config info for stream 0
  103. const MP4AAudioSpec& audioSpec = 
  104.     m_muxConfig.GetStream(0).GetAudioSpec();
  105. pConfig = audioSpec.Config();
  106. ulConfigSize = audioSpec.ConfigSize();
  107. ret = TRUE;
  108.     }
  109.     return ret;
  110. }
  111. BOOL MP4LATMDepack::SetTimeBase(ULONG32 ulSampleRate)
  112. {
  113.     m_ulSampleRate = ulSampleRate;
  114.     return TRUE;
  115. }
  116. BOOL MP4LATMDepack::SetFrameDuration(ULONG32 ulFrameDuration)
  117. {
  118.     /* MBO: While setting of m_ulFrameDuration is conceptually correct,
  119.     it requires m_ulFrameDuration to be correct at all times.
  120.     Thus, it introduces a burden on external components 
  121.     to provide this information in timely manner and adjust 
  122.     when AUDuration changes mid-stream.
  123.     Since rendering component will handle consecutive, equally
  124.     time-stamped audio packets as contiguous audio (at least
  125.     for a significant time-span), the setting of m_ulFrameDuration
  126.     to 0 will work reliably in all cases.
  127.     m_ulFrameDuration = ulFrameDuration; */
  128.     return TRUE;
  129. }
  130. BOOL MP4LATMDepack::Reset() // Completely reset the depacketizer state
  131. {
  132.     m_bNeedTime = TRUE;
  133.     m_ulBytesStaged = 0;
  134.     m_bHadLoss = FALSE;
  135.     m_bNeedMarker = FALSE;
  136.     return FALSE;
  137. }
  138. BOOL MP4LATMDepack::Flush() // Indicates end of stream
  139. {
  140.     return FALSE;
  141. }
  142. BOOL MP4LATMDepack::OnPacket(ULONG32 ulTime, const UINT8* pData, 
  143.      ULONG32 ulSize, BOOL bMarker)
  144. {
  145.     BOOL failed = FALSE;
  146.     if (!m_bNeedMarker)
  147.     {
  148. const UINT8* pBuffer = pData;
  149. if (m_bNeedTime)
  150. {
  151.     m_ulCurrentTime = ulTime;
  152.     
  153.     m_bNeedTime = FALSE;
  154. }
  155. if ((!bMarker) || (m_ulBytesStaged != 0))
  156. {
  157.     if ((m_ulBytesStaged + ulSize) > m_ulStagingBufSize)
  158.     {
  159. // We need to make the staging buffer larger
  160. UINT8* pNewBuf = new UINT8[m_ulBytesStaged + ulSize];
  161. memcpy(pNewBuf, m_pStagingBuf, m_ulBytesStaged); /* Flawfinder: ignore */
  162. delete [] m_pStagingBuf;
  163. m_pStagingBuf = pNewBuf;
  164.     }
  165.     memcpy(&m_pStagingBuf[m_ulBytesStaged], pData, ulSize); /* Flawfinder: ignore */
  166.     m_ulBytesStaged += ulSize;
  167.     pBuffer = m_pStagingBuf;
  168. }
  169. if (bMarker)
  170. {
  171.     ProcessStagingBuffer(pBuffer);
  172. }
  173.     }
  174.     else if (bMarker)
  175.     {
  176. // We just got the marker we needed. We should be
  177. // synced up again now
  178. m_bNeedMarker = FALSE;
  179.     }
  180.     return !failed;
  181. }
  182. BOOL MP4LATMDepack::OnLoss(ULONG32 ulNumPackets) // called to indicate lost 
  183.                                                  // packets
  184. {
  185.     m_bHadLoss = TRUE;
  186.     // Signal that we need a marked packet before we are
  187.     // synced up again
  188.     m_bNeedMarker = TRUE;
  189.     // Toss the bytes we currently have staged since we
  190.     // have no way of determining how much data was lost
  191.     m_ulBytesStaged = 0;
  192.     // Signal that we need to update m_ulCurrentTime
  193.     // when we lock back onto the bitstream.
  194.     m_bNeedTime = TRUE;
  195.     return TRUE;
  196. }
  197. BOOL MP4LATMDepack::ProcessStagingBuffer(const UINT8* pBuffer)
  198. {
  199.     BOOL failed = FALSE;
  200.     Bitstream bs;
  201.     
  202.     bs.SetBuffer(pBuffer);
  203.     if (m_bConfigPresent)
  204.     {
  205. if (bs.GetBits(1) == 0)
  206. {
  207.     // Not sure what we are supposed to do if this fails
  208.     if (!HandleMuxConfig(bs))
  209. failed = TRUE;
  210. }
  211.     }
  212.     if (!failed)
  213.     {
  214. for (ULONG32 i = 0; i < m_muxConfig.NumSubFrames(); i++)
  215. {
  216.     ULONG32 ulFrameTime = m_ulCurrentTime;
  217.     ulFrameTime += (i * m_ulFrameDuration); 
  218.     GetPayloadLengths(bs);
  219.     GetPayloads(bs, ulFrameTime);
  220. }
  221. m_bHadLoss = FALSE;
  222. m_ulBytesStaged = 0;
  223. m_bNeedTime = TRUE;
  224.     }
  225.     return !failed;
  226. }
  227. BOOL MP4LATMDepack::HandleMuxConfig(Bitstream& bs)
  228. {
  229.     BOOL ret = m_muxConfig.Unpack(bs);
  230.     
  231.     if (ret)
  232.     {
  233. delete [] m_pSlotLengths;
  234. m_pSlotLengths = new ULONG32[m_muxConfig.NumStreams()];
  235.     }
  236.     return ret;
  237. }
  238. void MP4LATMDepack::GetPayloadLengths(Bitstream& bs)
  239. {
  240.     if (m_muxConfig.AllSameTiming())
  241.     {
  242. for (ULONG32 prog = 0; prog < m_muxConfig.NumPrograms(); prog++)
  243. {
  244.     for (ULONG32 layer = 0; layer < m_muxConfig.NumLayers(prog); layer++)
  245.     {
  246. ULONG32 streamID = m_muxConfig.GetStreamID(prog, layer);
  247. if (m_muxConfig.GetStream(streamID).GetLengthType() == 0)
  248. {
  249.     ULONG32 tmp;
  250.     m_pSlotLengths[streamID] = 0;
  251.     
  252.     do
  253.     { 
  254. tmp = bs.GetBits(8);
  255. m_pSlotLengths[streamID] += tmp;
  256.     } while (tmp == 0xff);
  257.     // Resize the payload buffer if we encounter a payload
  258.     // size that is larger than our buffer
  259.     if (m_pSlotLengths[streamID] > m_ulPayloadBufSize)
  260.     {
  261. m_ulPayloadBufSize = m_pSlotLengths[streamID];
  262. delete [] m_pPayloadBuf;
  263. m_pPayloadBuf = new UINT8 [m_ulPayloadBufSize];
  264.     }
  265. }
  266.     }
  267. }
  268.     }
  269. }
  270. void MP4LATMDepack::GetPayloads(Bitstream& bs, ULONG32 ulTime)
  271. {
  272.     if (m_muxConfig.AllSameTiming())
  273.     {
  274. for (ULONG32 prog = 0; prog < m_muxConfig.NumPrograms(); prog++)
  275. {
  276.     for (ULONG32 layer = 0; layer < m_muxConfig.NumLayers(prog); layer++)
  277.     {
  278. ULONG32 streamID = m_muxConfig.GetStreamID(prog, layer);
  279. bs.GetBits(m_pSlotLengths[streamID] << 3, m_pPayloadBuf);
  280. // Currently we just hand out packets for stream 0
  281. if ((streamID == 0) && m_pCallback)
  282. {
  283.     m_pCallback( m_pUserData, ulTime, 
  284.  m_pPayloadBuf,
  285.  m_pSlotLengths[streamID],
  286.  m_bHadLoss);
  287. }
  288.     }
  289. }
  290.     }
  291. }