mp4descriptor.cpp
上传用户:sun1608
上传日期:2007-02-02
资源大小:6116k
文件大小:5k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*
  2.  * The contents of this file are subject to the Mozilla Public
  3.  * License Version 1.1 (the "License"); you may not use this file
  4.  * except in compliance with the License. You may obtain a copy of
  5.  * the License at http://www.mozilla.org/MPL/
  6.  * 
  7.  * Software distributed under the License is distributed on an "AS
  8.  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  9.  * implied. See the License for the specific language governing
  10.  * rights and limitations under the License.
  11.  * 
  12.  * The Original Code is MPEG4IP.
  13.  * 
  14.  * The Initial Developer of the Original Code is Cisco Systems Inc.
  15.  * Portions created by Cisco Systems Inc. are
  16.  * Copyright (C) Cisco Systems Inc. 2001.  All Rights Reserved.
  17.  * 
  18.  * Contributor(s): 
  19.  * Dave Mackie dmackie@cisco.com
  20.  */
  21. #include "mp4common.h"
  22. MP4Descriptor::MP4Descriptor(u_int8_t tag) {
  23. m_tag = tag;
  24. m_pParentAtom = NULL;
  25. m_start = 0;
  26. m_size = 0;
  27. m_readMutatePoint = 0;
  28. }
  29. MP4Descriptor::~MP4Descriptor() 
  30. {
  31. for (u_int32_t i = 0; i < m_pProperties.Size(); i++) {
  32. delete m_pProperties[i];
  33. }
  34. }
  35. void MP4Descriptor::AddProperty(MP4Property* pProperty) 
  36. {
  37. ASSERT(pProperty);
  38. m_pProperties.Add(pProperty);
  39. pProperty->SetParentAtom(m_pParentAtom);
  40. }
  41. bool MP4Descriptor::FindContainedProperty(const char *name,
  42. MP4Property** ppProperty, u_int32_t* pIndex)
  43. {
  44. u_int32_t numProperties = m_pProperties.Size();
  45. for (u_int32_t i = 0; i < numProperties; i++) {
  46. if (m_pProperties[i]->FindProperty(name, ppProperty, pIndex)) { 
  47. return true;
  48. }
  49. }
  50. return false;
  51. }
  52. void MP4Descriptor::Generate()
  53. {
  54. // generate properties
  55. for (u_int32_t i = 0; i < m_pProperties.Size(); i++) {
  56. m_pProperties[i]->Generate();
  57. }
  58. }
  59. void MP4Descriptor::Read(MP4File* pFile)
  60. {
  61. ReadHeader(pFile);
  62. ReadProperties(pFile, 0, m_readMutatePoint);
  63. Mutate();
  64. ReadProperties(pFile, m_readMutatePoint);
  65. // flush any leftover read bits
  66. pFile->FlushReadBits();
  67. }
  68. void MP4Descriptor::ReadHeader(MP4File* pFile)
  69. {
  70. VERBOSE_READ(pFile->GetVerbosity(),
  71. printf("ReadDescriptor: pos = 0x%llxn", 
  72. pFile->GetPosition()));
  73. // read tag and length
  74. u_int8_t tag = pFile->ReadUInt8();
  75. if (m_tag) {
  76. ASSERT(tag == m_tag);
  77. } else {
  78. m_tag = tag;
  79. }
  80. m_size = pFile->ReadMpegLength();
  81. m_start = pFile->GetPosition();
  82. VERBOSE_READ(pFile->GetVerbosity(),
  83. printf("ReadDescriptor: tag 0x%02x data size %u (0x%x)n", 
  84. m_tag, m_size, m_size));
  85. }
  86. void MP4Descriptor::ReadProperties(MP4File* pFile, 
  87. u_int32_t propStartIndex, u_int32_t propCount)
  88. {
  89. u_int32_t numProperties = MIN(propCount, 
  90. m_pProperties.Size() - propStartIndex);
  91. for (u_int32_t i = propStartIndex; 
  92.   i < propStartIndex + numProperties; i++) {
  93. MP4Property* pProperty = m_pProperties[i];
  94. int32_t remaining = m_size - (pFile->GetPosition() - m_start);
  95. if (pProperty->GetType() == DescriptorProperty) {
  96.     if (remaining > 0) {
  97. // place a limit on how far this sub-descriptor looks
  98. ((MP4DescriptorProperty*)pProperty)->SetSizeLimit(remaining);
  99. pProperty->Read(pFile);
  100. } // else do nothing, empty descriptor
  101. } else {
  102. // non-descriptor property
  103. if (remaining >= 0) {
  104. pProperty->Read(pFile);
  105. if (pProperty->GetType() == TableProperty) {
  106. VERBOSE_READ_TABLE(pFile->GetVerbosity(), 
  107. printf("Read: "); pProperty->Dump(stdout, 0, true));
  108. } else {
  109. VERBOSE_READ(pFile->GetVerbosity(), 
  110. printf("Read: "); pProperty->Dump(stdout, 0, true));
  111. }
  112. } else {
  113. VERBOSE_ERROR(pFile->GetVerbosity(),
  114. printf("Overran descriptor, tag %u data size %u property %un",
  115. m_tag, m_size, i));
  116. throw new MP4Error("overran descriptor",
  117.  "MP4Descriptor::ReadProperties");
  118. }
  119. }
  120. }
  121. void MP4Descriptor::Write(MP4File* pFile)
  122. {
  123. // call virtual function to adapt properties before writing
  124. Mutate();
  125. u_int32_t numProperties = m_pProperties.Size();
  126. if (numProperties == 0) {
  127. WARNING(numProperties == 0);
  128. return;
  129. }
  130. // write tag and length placeholder
  131. pFile->WriteUInt8(m_tag);
  132. u_int64_t lengthPos = pFile->GetPosition();
  133. pFile->WriteMpegLength(0);
  134. u_int64_t startPos = pFile->GetPosition();
  135. for (u_int32_t i = 0; i < numProperties; i++) {
  136. m_pProperties[i]->Write(pFile);
  137. }
  138. // align with byte boundary (rarely necessary)
  139. pFile->PadWriteBits();
  140. // go back and write correct length
  141. u_int64_t endPos = pFile->GetPosition();
  142. pFile->SetPosition(lengthPos);
  143. pFile->WriteMpegLength(endPos - startPos);
  144. pFile->SetPosition(endPos);
  145. }
  146. void MP4Descriptor::WriteToMemory(MP4File* pFile,
  147. u_int8_t** ppBytes, u_int64_t* pNumBytes)
  148. {
  149. // use memory buffer to save descriptor in memory
  150. // instead of going directly to disk
  151. pFile->EnableMemoryBuffer();
  152. Write(pFile);
  153. pFile->DisableMemoryBuffer(ppBytes, pNumBytes);
  154. }
  155. void MP4Descriptor::Dump(FILE* pFile, u_int8_t indent, bool dumpImplicits)
  156. {
  157. // call virtual function to adapt properties before dumping
  158. Mutate();
  159. u_int32_t numProperties = m_pProperties.Size();
  160. if (numProperties == 0) {
  161. WARNING(numProperties == 0);
  162. return;
  163. }
  164. for (u_int32_t i = 0; i < numProperties; i++) {
  165. m_pProperties[i]->Dump(pFile, indent, dumpImplicits);
  166. }
  167. }
  168. u_int8_t MP4Descriptor::GetDepth() 
  169. {
  170. if (m_pParentAtom) {
  171. return m_pParentAtom->GetDepth();
  172. }
  173. return 0;
  174. }