libmkv/EbmlWriter.c

   1 // Copyright (c) 2010 The WebM project authors. All Rights Reserved.
   2 //
   3 // Use of this source code is governed by a BSD-style license
   4 // that can be found in the LICENSE file in the root of the source
   5 // tree. An additional intellectual property rights grant can be found
   6 // in the file PATENTS.  All contributing project authors may
   7 // be found in the AUTHORS file in the root of the source tree.
   8
   9
  10 #include "EbmlWriter.h"
  11 #include <stdlib.h>
  12 #include <wchar.h>
  13 #include <string.h>
  14 #if defined(_MSC_VER)
  15 #define LITERALU64(n) n
  16 #else
  17 #define LITERALU64(n) n##LLU
  18 #endif
  19
  20 void Ebml_WriteLen(EbmlGlobal *glob, long long val)
  21 {
  22     //TODO check and make sure we are not > than 0x0100000000000000LLU
  23     unsigned char size = 8; //size in bytes to output
  24     unsigned long long minVal = LITERALU64(0x00000000000000ff); //mask to compare for byte size
  25
  26     for (size = 1; size < 8; size ++)
  27     {
  28         if (val < minVal)
  29             break;
  30
  31         minVal = (minVal << 7);
  32     }
  33
  34     val |= (LITERALU64(0x000000000000080) << ((size - 1) * 7));
  35
  36     Ebml_Serialize(glob, (void *) &val, size);
  37 }
  38
  39 void Ebml_WriteString(EbmlGlobal *glob, const char *str)
  40 {
  41     const size_t size_ = strlen(str);
  42     const unsigned long long  size = size_;
  43     Ebml_WriteLen(glob, size);
  44     //TODO: it's not clear from the spec whether the nul terminator
  45     //should be serialized too.  For now we omit the null terminator.
  46     Ebml_Write(glob, str, size);
  47 }
  48
  49 void Ebml_WriteUTF8(EbmlGlobal *glob, const wchar_t *wstr)
  50 {
  51     const size_t strlen = wcslen(wstr);
  52
  53     //TODO: it's not clear from the spec whether the nul terminator
  54     //should be serialized too.  For now we include it.
  55     const unsigned long long  size = strlen;
  56
  57     Ebml_WriteLen(glob, size);
  58     Ebml_Write(glob, wstr, size);
  59 }
  60
  61 void Ebml_WriteID(EbmlGlobal *glob, unsigned long class_id)
  62 {
  63     if (class_id >= 0x01000000)
  64         Ebml_Serialize(glob, (void *)&class_id, 4);
  65     else if (class_id >= 0x00010000)
  66         Ebml_Serialize(glob, (void *)&class_id, 3);
  67     else if (class_id >= 0x00000100)
  68         Ebml_Serialize(glob, (void *)&class_id, 2);
  69     else
  70         Ebml_Serialize(glob, (void *)&class_id, 1);
  71 }
  72 void Ebml_SerializeUnsigned64(EbmlGlobal *glob, unsigned long class_id, uint64_t ui)
  73 {
  74     unsigned char sizeSerialized = 8 | 0x80;
  75     Ebml_WriteID(glob, class_id);
  76     Ebml_Serialize(glob, &sizeSerialized, 1);
  77     Ebml_Serialize(glob, &ui, 8);
  78 }
  79
  80 void Ebml_SerializeUnsigned(EbmlGlobal *glob, unsigned long class_id, unsigned long ui)
  81 {
  82     unsigned char size = 8; //size in bytes to output
  83     unsigned char sizeSerialized = 0;
  84     unsigned long minVal;
  85
  86     Ebml_WriteID(glob, class_id);
  87     minVal = 0x7fLU; //mask to compare for byte size
  88
  89     for (size = 1; size < 4; size ++)
  90     {
  91         if (ui < minVal)
  92         {
  93             break;
  94         }
  95
  96         minVal <<= 7;
  97     }
  98
  99     sizeSerialized = 0x80 | size;
 100     Ebml_Serialize(glob, &sizeSerialized, 1);
 101     Ebml_Serialize(glob, &ui, size);
 102 }
 103 //TODO: perhaps this is a poor name for this id serializer helper function
 104 void Ebml_SerializeBinary(EbmlGlobal *glob, unsigned long class_id, unsigned long bin)
 105 {
 106     int size;
 107     for (size=4; size > 1; size--)
 108     {
 109         if (bin & 0x000000ff << ((size-1) * 8))
 110             break;
 111     }
 112     Ebml_WriteID(glob, class_id);
 113     Ebml_WriteLen(glob, size);
 114     Ebml_WriteID(glob, bin);
 115 }
 116
 117 void Ebml_SerializeFloat(EbmlGlobal *glob, unsigned long class_id, double d)
 118 {
 119     unsigned char len = 0x88;
 120
 121     Ebml_WriteID(glob, class_id);
 122     Ebml_Serialize(glob, &len, 1);
 123     Ebml_Serialize(glob,  &d, 8);
 124 }
 125
 126 void Ebml_WriteSigned16(EbmlGlobal *glob, short val)
 127 {
 128     signed long out = ((val & 0x003FFFFF) | 0x00200000) << 8;
 129     Ebml_Serialize(glob, &out, 3);
 130 }
 131
 132 void Ebml_SerializeString(EbmlGlobal *glob, unsigned long class_id, const char *s)
 133 {
 134     Ebml_WriteID(glob, class_id);
 135     Ebml_WriteString(glob, s);
 136 }
 137
 138 void Ebml_SerializeUTF8(EbmlGlobal *glob, unsigned long class_id, wchar_t *s)
 139 {
 140     Ebml_WriteID(glob,  class_id);
 141     Ebml_WriteUTF8(glob,  s);
 142 }
 143
 144 void Ebml_SerializeData(EbmlGlobal *glob, unsigned long class_id, unsigned char *data, unsigned long data_length)
 145 {
 146     unsigned char size = 4;
 147     Ebml_WriteID(glob, class_id);
 148     Ebml_WriteLen(glob, data_length);
 149     Ebml_Write(glob,  data, data_length);
 150 }
 151
 152 void Ebml_WriteVoid(EbmlGlobal *glob, unsigned long vSize)
 153 {
 154     unsigned char tmp = 0;
 155     unsigned long i = 0;
 156
 157     Ebml_WriteID(glob, 0xEC);
 158     Ebml_WriteLen(glob, vSize);
 159
 160     for (i = 0; i < vSize; i++)
 161     {
 162         Ebml_Write(glob, &tmp, 1);
 163     }
 164 }
 165
 166 //TODO Serialize Date