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

流媒体/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. MP4RootAtom::MP4RootAtom() 
  23. : MP4Atom(NULL)
  24. {
  25. ExpectChildAtom("ftyp", Required, OnlyOne);
  26. ExpectChildAtom("moov", Required, OnlyOne);
  27. ExpectChildAtom("mdat", Optional, Many);
  28. ExpectChildAtom("free", Optional, Many);
  29. ExpectChildAtom("skip", Optional, Many);
  30. ExpectChildAtom("udta", Optional, Many);
  31. ExpectChildAtom("moof", Optional, Many);
  32. }
  33. void MP4RootAtom::BeginWrite(bool use64) 
  34. {
  35. // only call under MP4Create() control
  36. WriteAtomType("ftyp", OnlyOne);
  37. m_pChildAtoms[GetLastMdatIndex()]->BeginWrite(m_pFile->Use64Bits());
  38. }
  39. void MP4RootAtom::Write()
  40. {
  41. // no-op
  42. }
  43. void MP4RootAtom::FinishWrite(bool use64)
  44. {
  45. // finish writing last mdat atom
  46. u_int32_t mdatIndex = GetLastMdatIndex();
  47. m_pChildAtoms[mdatIndex]->FinishWrite(m_pFile->Use64Bits());
  48. // write all atoms after last mdat
  49. u_int32_t size = m_pChildAtoms.Size();
  50. for (u_int32_t i = mdatIndex + 1; i < size; i++) {
  51. m_pChildAtoms[i]->Write();
  52. }
  53. }
  54. void MP4RootAtom::BeginOptimalWrite() 
  55. {
  56. WriteAtomType("ftyp", OnlyOne);
  57. WriteAtomType("moov", OnlyOne);
  58. WriteAtomType("udta", Many);
  59. m_pChildAtoms[GetLastMdatIndex()]->BeginWrite(m_pFile->Use64Bits());
  60. }
  61. void MP4RootAtom::FinishOptimalWrite() 
  62. {
  63. // finish writing mdat
  64. m_pChildAtoms[GetLastMdatIndex()]->FinishWrite(m_pFile->Use64Bits());
  65. // find moov atom
  66. u_int32_t size = m_pChildAtoms.Size();
  67. MP4Atom* pMoovAtom = NULL;
  68. u_int32_t i;
  69. for (i = 0; i < size; i++) {
  70. if (!strcmp("moov", m_pChildAtoms[i]->GetType())) {
  71. pMoovAtom = m_pChildAtoms[i];
  72. break;
  73. }
  74. }
  75. ASSERT(i < size);
  76. // rewrite moov so that updated chunkOffsets are written to disk
  77. m_pFile->SetPosition(pMoovAtom->GetStart());
  78. u_int64_t oldSize = pMoovAtom->GetSize();
  79. pMoovAtom->Write();
  80. // sanity check
  81. u_int64_t newSize = pMoovAtom->GetSize();
  82. ASSERT(oldSize == newSize);
  83. }
  84. u_int32_t MP4RootAtom::GetLastMdatIndex()
  85. {
  86. for (int32_t i = m_pChildAtoms.Size() - 1; i >= 0; i--) {
  87. if (!strcmp("mdat", m_pChildAtoms[i]->GetType())) {
  88. return i;
  89. }
  90. }
  91. ASSERT(false);
  92. return (u_int32_t)-1;
  93. }
  94. void MP4RootAtom::WriteAtomType(const char* type, bool onlyOne)
  95. {
  96. u_int32_t size = m_pChildAtoms.Size();
  97. for (u_int32_t i = 0; i < size; i++) {
  98. if (!strcmp(type, m_pChildAtoms[i]->GetType())) {
  99. m_pChildAtoms[i]->Write();
  100. if (onlyOne) {
  101. break;
  102. }
  103. }
  104. }
  105. }