Bijna klaar!

util/MP4/box.cpp

@@ -0,0 +1,150 @@
+#pragma once
+
+#include <stdint.h>
+#include <cstdlib>
+#include <cstdio>
+#include <cstring>
+#include <cmath>
+#include <algorithm>
+
+struct BoxHeader {
+  uint32_t TotalSize;
+  uint32_t BoxType;
+};//BoxHeader struct
+
+class Box {
+  public:
+    Box();
+    Box(uint32_t BoxType);
+    ~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( );
+  private:
+    BoxHeader header;
+    uint8_t * Payload;
+    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() {
+}
+
+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;
+  }
+}