98 lines
2.5 KiB
C++
98 lines
2.5 KiB
C++
// Copyright 2014 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include <packager/media/formats/mp4/box.h>
|
|
|
|
#include <glog/logging.h>
|
|
#include <packager/media/formats/mp4/box_buffer.h>
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
namespace mp4 {
|
|
|
|
Box::Box() : box_size_(0) {}
|
|
Box::~Box() {}
|
|
|
|
bool Box::Parse(BoxReader* reader) {
|
|
DCHECK(reader);
|
|
BoxBuffer buffer(reader);
|
|
return ReadWriteInternal(&buffer);
|
|
}
|
|
|
|
void Box::Write(BufferWriter* writer) {
|
|
DCHECK(writer);
|
|
// Compute and update box size.
|
|
uint32_t size = ComputeSize();
|
|
DCHECK_EQ(size, box_size_);
|
|
|
|
size_t buffer_size_before_write = writer->Size();
|
|
BoxBuffer buffer(writer);
|
|
CHECK(ReadWriteInternal(&buffer));
|
|
DCHECK_EQ(box_size_, writer->Size() - buffer_size_before_write)
|
|
<< FourCCToString(BoxType());
|
|
}
|
|
|
|
void Box::WriteHeader(BufferWriter* writer) {
|
|
DCHECK(writer);
|
|
// Compute and update box size.
|
|
uint32_t size = ComputeSize();
|
|
DCHECK_EQ(size, box_size_);
|
|
|
|
size_t buffer_size_before_write = writer->Size();
|
|
BoxBuffer buffer(writer);
|
|
CHECK(ReadWriteHeaderInternal(&buffer));
|
|
DCHECK_EQ(HeaderSize(), writer->Size() - buffer_size_before_write);
|
|
}
|
|
|
|
uint32_t Box::ComputeSize() {
|
|
box_size_ = static_cast<uint32_t>(ComputeSizeInternal());
|
|
return box_size_;
|
|
}
|
|
|
|
uint32_t Box::HeaderSize() const {
|
|
const uint32_t kFourCCSize = 4;
|
|
// We don't support 64-bit size.
|
|
return kFourCCSize + sizeof(uint32_t);
|
|
}
|
|
|
|
bool Box::ReadWriteHeaderInternal(BoxBuffer* buffer) {
|
|
if (buffer->Reading()) {
|
|
// Skip for read mode, which is handled already in BoxReader.
|
|
} else {
|
|
CHECK(buffer->ReadWriteUInt32(&box_size_));
|
|
FourCC fourcc = BoxType();
|
|
CHECK(buffer->ReadWriteFourCC(&fourcc));
|
|
}
|
|
return true;
|
|
}
|
|
|
|
FullBox::FullBox() = default;
|
|
FullBox::~FullBox() = default;
|
|
|
|
uint32_t FullBox::HeaderSize() const {
|
|
// Additional 1-byte version and 3-byte flags.
|
|
return Box::HeaderSize() + 1 + 3;
|
|
}
|
|
|
|
bool FullBox::ReadWriteHeaderInternal(BoxBuffer* buffer) {
|
|
RCHECK(Box::ReadWriteHeaderInternal(buffer));
|
|
|
|
uint32_t vflags;
|
|
if (buffer->Reading()) {
|
|
RCHECK(buffer->ReadWriteUInt32(&vflags));
|
|
this->version = vflags >> 24;
|
|
this->flags = vflags & 0x00FFFFFF;
|
|
} else {
|
|
vflags = (this->version << 24) | this->flags;
|
|
RCHECK(buffer->ReadWriteUInt32(&vflags));
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // namespace mp4
|
|
} // namespace media
|
|
} // namespace shaka
|