michael@0: /* michael@0: * Copyright (c) 2010 The WebM project authors. All Rights Reserved. michael@0: * michael@0: * Use of this source code is governed by a BSD-style license michael@0: * that can be found in the LICENSE file in the root of the source michael@0: * tree. An additional intellectual property rights grant can be found michael@0: * in the file PATENTS. All contributing project authors may michael@0: * be found in the AUTHORS file in the root of the source tree. michael@0: */ michael@0: #include "EbmlWriter.h" michael@0: #include michael@0: #include michael@0: #include michael@0: #include michael@0: #include "EbmlBufferWriter.h" michael@0: #if defined(_MSC_VER) michael@0: #define LITERALU64(n) n michael@0: #else michael@0: #define LITERALU64(n) n##LLU michael@0: #endif michael@0: michael@0: void Ebml_WriteLen(EbmlGlobal *glob, int64_t val) { michael@0: /* TODO check and make sure we are not > than 0x0100000000000000LLU */ michael@0: unsigned char size = 8; /* size in bytes to output */ michael@0: michael@0: /* mask to compare for byte size */ michael@0: int64_t minVal = 0xff; michael@0: michael@0: for (size = 1; size < 8; size ++) { michael@0: if (val < minVal) michael@0: break; michael@0: michael@0: minVal = (minVal << 7); michael@0: } michael@0: michael@0: val |= (((uint64_t)0x80) << ((size - 1) * 7)); michael@0: michael@0: Ebml_Serialize(glob, (void *) &val, sizeof(val), size); michael@0: } michael@0: michael@0: void Ebml_WriteString(EbmlGlobal *glob, const char *str) { michael@0: const size_t size_ = strlen(str); michael@0: const uint64_t size = size_; michael@0: Ebml_WriteLen(glob, size); michael@0: /* TODO: it's not clear from the spec whether the nul terminator michael@0: * should be serialized too. For now we omit the null terminator. michael@0: */ michael@0: Ebml_Write(glob, str, (unsigned long)size); michael@0: } michael@0: michael@0: void Ebml_WriteUTF8(EbmlGlobal *glob, const wchar_t *wstr) { michael@0: const size_t strlen = wcslen(wstr); michael@0: michael@0: /* TODO: it's not clear from the spec whether the nul terminator michael@0: * should be serialized too. For now we include it. michael@0: */ michael@0: const uint64_t size = strlen; michael@0: michael@0: Ebml_WriteLen(glob, size); michael@0: Ebml_Write(glob, wstr, (unsigned long)size); michael@0: } michael@0: michael@0: void Ebml_WriteID(EbmlGlobal *glob, unsigned long class_id) { michael@0: int len; michael@0: michael@0: if (class_id >= 0x01000000) michael@0: len = 4; michael@0: else if (class_id >= 0x00010000) michael@0: len = 3; michael@0: else if (class_id >= 0x00000100) michael@0: len = 2; michael@0: else michael@0: len = 1; michael@0: michael@0: Ebml_Serialize(glob, (void *)&class_id, sizeof(class_id), len); michael@0: } michael@0: michael@0: void Ebml_SerializeUnsigned32(EbmlGlobal *glob, unsigned long class_id, uint32_t ui) { michael@0: unsigned char sizeSerialized = 8 | 0x80; michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); michael@0: Ebml_Serialize(glob, &ui, sizeof(ui), 4); michael@0: } michael@0: michael@0: void Ebml_SerializeUnsigned64(EbmlGlobal *glob, unsigned long class_id, uint64_t ui) { michael@0: unsigned char sizeSerialized = 8 | 0x80; michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); michael@0: Ebml_Serialize(glob, &ui, sizeof(ui), 8); michael@0: } michael@0: michael@0: void Ebml_SerializeUnsigned(EbmlGlobal *glob, unsigned long class_id, unsigned long ui) { michael@0: unsigned char size = 8; /* size in bytes to output */ michael@0: unsigned char sizeSerialized = 0; michael@0: unsigned long minVal; michael@0: michael@0: Ebml_WriteID(glob, class_id); michael@0: minVal = 0x7fLU; /* mask to compare for byte size */ michael@0: michael@0: for (size = 1; size < 4; size ++) { michael@0: if (ui < minVal) { michael@0: break; michael@0: } michael@0: michael@0: minVal <<= 7; michael@0: } michael@0: michael@0: sizeSerialized = 0x80 | size; michael@0: Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); michael@0: Ebml_Serialize(glob, &ui, sizeof(ui), size); michael@0: } michael@0: /* TODO: perhaps this is a poor name for this id serializer helper function */ michael@0: void Ebml_SerializeBinary(EbmlGlobal *glob, unsigned long class_id, unsigned long bin) { michael@0: int size; michael@0: for (size = 4; size > 1; size--) { michael@0: if (bin & (unsigned int)0x000000ff << ((size - 1) * 8)) michael@0: break; michael@0: } michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_WriteLen(glob, size); michael@0: Ebml_WriteID(glob, bin); michael@0: } michael@0: michael@0: void Ebml_SerializeFloat(EbmlGlobal *glob, unsigned long class_id, double d) { michael@0: unsigned char len = 0x88; michael@0: michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_Serialize(glob, &len, sizeof(len), 1); michael@0: Ebml_Serialize(glob, &d, sizeof(d), 8); michael@0: } michael@0: michael@0: void Ebml_WriteSigned16(EbmlGlobal *glob, short val) { michael@0: signed long out = ((val & 0x003FFFFF) | 0x00200000) << 8; michael@0: Ebml_Serialize(glob, &out, sizeof(out), 3); michael@0: } michael@0: michael@0: void Ebml_SerializeString(EbmlGlobal *glob, unsigned long class_id, const char *s) { michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_WriteString(glob, s); michael@0: } michael@0: michael@0: void Ebml_SerializeUTF8(EbmlGlobal *glob, unsigned long class_id, wchar_t *s) { michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_WriteUTF8(glob, s); michael@0: } michael@0: michael@0: void Ebml_SerializeData(EbmlGlobal *glob, unsigned long class_id, unsigned char *data, unsigned long data_length) { michael@0: Ebml_WriteID(glob, class_id); michael@0: Ebml_WriteLen(glob, data_length); michael@0: Ebml_Write(glob, data, data_length); michael@0: } michael@0: michael@0: void Ebml_WriteVoid(EbmlGlobal *glob, unsigned long vSize) { michael@0: unsigned char tmp = 0; michael@0: unsigned long i = 0; michael@0: michael@0: Ebml_WriteID(glob, 0xEC); michael@0: Ebml_WriteLen(glob, vSize); michael@0: michael@0: for (i = 0; i < vSize; i++) { michael@0: Ebml_Write(glob, &tmp, 1); michael@0: } michael@0: } michael@0: michael@0: /* TODO Serialize Date */