winamp/Src/external_dependencies/libmp4v2/atom_root.cpp
2024-09-24 14:54:57 +02:00

124 lines
3.1 KiB
C++

/*
* The contents of this file are subject to the Mozilla Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is MPEG4IP.
*
* The Initial Developer of the Original Code is Cisco Systems Inc.
* Portions created by Cisco Systems Inc. are
* Copyright (C) Cisco Systems Inc. 2001. All Rights Reserved.
*
* Contributor(s):
* Dave Mackie dmackie@cisco.com
*/
#include "mp4common.h"
MP4RootAtom::MP4RootAtom()
: MP4Atom(NULL)
{
ExpectChildAtom("moov", Required, OnlyOne);
ExpectChildAtom("ftyp", Optional, OnlyOne);
ExpectChildAtom("mdat", Optional, Many);
ExpectChildAtom("free", Optional, Many);
ExpectChildAtom("skip", Optional, Many);
ExpectChildAtom("udta", Optional, Many);
ExpectChildAtom("moof", Optional, Many);
}
void MP4RootAtom::BeginWrite(bool use64)
{
// only call under MP4Create() control
WriteAtomType("ftyp", OnlyOne);
m_pChildAtoms[GetLastMdatIndex()]->BeginWrite(m_pFile->Use64Bits("mdat"));
}
void MP4RootAtom::Write()
{
// no-op
}
void MP4RootAtom::FinishWrite(bool use64)
{
// finish writing last mdat atom
u_int32_t mdatIndex = GetLastMdatIndex();
m_pChildAtoms[mdatIndex]->FinishWrite(m_pFile->Use64Bits("mdat"));
// write all atoms after last mdat
u_int32_t size = m_pChildAtoms.Size();
for (u_int32_t i = mdatIndex + 1; i < size; i++) {
m_pChildAtoms[i]->Write();
}
}
void MP4RootAtom::BeginOptimalWrite()
{
WriteAtomType("ftyp", OnlyOne);
WriteAtomType("moov", OnlyOne);
WriteAtomType("udta", Many);
m_pChildAtoms[GetLastMdatIndex()]->BeginWrite(m_pFile->Use64Bits("mdat"));
}
void MP4RootAtom::FinishOptimalWrite()
{
// finish writing mdat
m_pChildAtoms[GetLastMdatIndex()]->FinishWrite(m_pFile->Use64Bits("mdat"));
// find moov atom
u_int32_t size = m_pChildAtoms.Size();
MP4Atom* pMoovAtom = NULL;
u_int32_t i;
for (i = 0; i < size; i++) {
if (!strcmp("moov", m_pChildAtoms[i]->GetType())) {
pMoovAtom = m_pChildAtoms[i];
break;
}
}
ASSERT(i < size);
ASSERT(pMoovAtom != NULL);
// rewrite moov so that updated chunkOffsets are written to disk
m_pFile->SetPosition(pMoovAtom->GetStart());
u_int64_t oldSize = pMoovAtom->GetSize();
pMoovAtom->Write();
// sanity check
u_int64_t newSize = pMoovAtom->GetSize();
ASSERT(oldSize == newSize);
}
u_int32_t MP4RootAtom::GetLastMdatIndex()
{
for (int32_t i = m_pChildAtoms.Size() - 1; i >= 0; i--) {
if (!strcmp("mdat", m_pChildAtoms[i]->GetType())) {
return i;
}
}
ASSERT(false);
return (u_int32_t)-1;
}
void MP4RootAtom::WriteAtomType(const char* type, bool onlyOne)
{
u_int32_t size = m_pChildAtoms.Size();
for (u_int32_t i = 0; i < size; i++) {
if (!strcmp(type, m_pChildAtoms[i]->GetType())) {
m_pChildAtoms[i]->Write();
if (onlyOne) {
break;
}
}
}
}