mistserver/util/MP4/box.cpp
Erik Zandvliet 75634c6512 Test5
2011-03-21 14:24:56 +01:00

167 lines
4.1 KiB
C++

#pragma once
#include <stdint.h>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>
struct BoxHeader {
uint32_t TotalSize;
uint32_t BoxType;
};//BoxHeader struct
class Box {
public:
Box();
Box(uint32_t BoxType);
Box(uint8_t * Content, uint32_t length);
~Box();
void SetBoxType(uint32_t BoxType);
uint32_t GetBoxType();
void SetPayload(uint32_t Size, uint8_t * Data, uint32_t Index = 0);
uint32_t GetPayloadSize();
uint8_t * GetPayload();
uint8_t * GetPayload(uint32_t Index, uint32_t & Size);
uint32_t GetBoxedDataSize();
uint8_t * GetBoxedData( );
static uint8_t * uint32_to_uint8( uint32_t data );
static uint8_t * uint16_to_uint8( uint16_t data );
static uint8_t * uint8_to_uint8( uint8_t data );
BoxHeader GetHeader( );
void ResetPayload( );
uint8_t * Payload;
BoxHeader header;
uint32_t PayloadSize;
};//Box Class
Box::Box() {
Payload = NULL;
PayloadSize = 0;
}
Box::Box(uint32_t BoxType) {
header.BoxType = BoxType;
Payload = NULL;
PayloadSize = 0;
}
Box::Box(uint8_t * Content, uint32_t length) {
header.TotalSize = (Content[0] << 24) + (Content[1] << 16) + (Content[2] << 8) + (Content[3]);
if(header.TotalSize != length) { std::cerr << "Warning: length sizes differ\n"; }
header.BoxType = (Content[4] << 24) + (Content[5] << 16) + (Content[6] << 8) + (Content[7]);
std::cerr << "Created new box with type \""
<< (char)(header.BoxType >> 24)
<< (char)((header.BoxType << 8) >> 24)
<< (char)((header.BoxType << 16) >> 24)
<< (char)((header.BoxType << 24) >> 24)
<< "\"\n";
PayloadSize = length-8;
Payload = new uint8_t[PayloadSize];
memcpy( Payload, &Content[8], PayloadSize );
}
Box::~Box() {
}
void Box::SetBoxType(uint32_t BoxType) {
header.BoxType = BoxType;
}
uint32_t Box::GetBoxType() {
return header.BoxType;
}
void Box::SetPayload(uint32_t Size, uint8_t * Data, uint32_t Index) {
uint8_t * tempchar = NULL;
if ( Index + Size > PayloadSize ) {
if ( Payload ) {
tempchar = new uint8_t[PayloadSize];
memcpy( tempchar, Payload, PayloadSize );
delete Payload;
}
PayloadSize = Index + Size;
Payload = new uint8_t[PayloadSize];
if( tempchar ) {
memcpy( Payload, tempchar, Index );
} else {
for(uint32_t i = 0; i < Index; i++) { Payload[i] = 0; }
}
memcpy( &Payload[Index], Data, Size );
header.TotalSize = PayloadSize + 8;
if( tempchar ) {
delete tempchar;
}
} else {
memcpy( &Payload[Index], Data, Size );
}
}
uint32_t Box::GetPayloadSize() {
return PayloadSize;
}
uint8_t * Box::GetPayload() {
uint8_t * temp = new uint8_t[PayloadSize];
memcpy( temp, Payload, PayloadSize );
return temp;
}
uint8_t * Box::GetPayload(uint32_t Index, uint32_t & Size) {
if(Index > PayloadSize) { return NULL; }
if(Index + Size > PayloadSize) { Size = PayloadSize - Index; }
uint8_t * temp = new uint8_t[Size - Index];
memcpy( temp, &Payload[Index], Size - Index );
return temp;
}
uint32_t Box::GetBoxedDataSize() {
return header.TotalSize;
}
uint8_t * Box::GetBoxedData( ) {
uint8_t * temp = new uint8_t[header.TotalSize];
memcpy( temp, uint32_to_uint8(header.TotalSize), 4 );
memcpy( &temp[4], uint32_to_uint8(header.BoxType), 4 );
memcpy( &temp[8], Payload, PayloadSize );
return temp;
}
uint8_t * Box::uint32_to_uint8( uint32_t data ) {
uint8_t * temp = new uint8_t[4];
temp[0] = (data >> 24) & 0x000000FF;
temp[1] = (data >> 16 ) & 0x000000FF;
temp[2] = (data >> 8 ) & 0x000000FF;
temp[3] = (data ) & 0x000000FF;
return temp;
}
uint8_t * Box::uint16_to_uint8( uint16_t data ) {
uint8_t * temp = new uint8_t[2];
temp[0] = (data >> 8) & 0x00FF;
temp[1] = (data ) & 0x00FF;
return temp;
}
uint8_t * Box::uint8_to_uint8( uint8_t data ) {
uint8_t * temp = new uint8_t[1];
temp[0] = data;
return temp;
}
BoxHeader Box::GetHeader( ) {
return header;
}
void Box::ResetPayload( ) {
header.TotalSize -= PayloadSize;
PayloadSize = 0;
if(Payload) {
delete Payload;
Payload = NULL;
}
}