src/third_party/libvpx/source/libvpx/third_party/libmkv/WebMElement.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 "EbmlBufferWriter.h"
  11 #include "EbmlIDs.h"
  12 #include "WebMElement.h"
  13 #include <stdio.h>
  14 #include "vpx/vpx_integer.h"
  15
  16 #define kVorbisPrivateMaxSize  4000
  17
  18 void writeHeader(EbmlGlobal *glob) {
  19   EbmlLoc start;
  20   Ebml_StartSubElement(glob, &start, EBML);
  21   Ebml_SerializeUnsigned(glob, EBMLVersion, 1);
  22   Ebml_SerializeUnsigned(glob, EBMLReadVersion, 1); // EBML Read Version
  23   Ebml_SerializeUnsigned(glob, EBMLMaxIDLength, 4); // EBML Max ID Length
  24   Ebml_SerializeUnsigned(glob, EBMLMaxSizeLength, 8); // EBML Max Size Length
  25   Ebml_SerializeString(glob, DocType, "webm"); // Doc Type
  26   Ebml_SerializeUnsigned(glob, DocTypeVersion, 2); // Doc Type Version
  27   Ebml_SerializeUnsigned(glob, DocTypeReadVersion, 2); // Doc Type Read Version
  28   Ebml_EndSubElement(glob, &start);
  29 }
  30
  31 void writeSimpleBlock(EbmlGlobal *glob, unsigned char trackNumber, short timeCode,
  32                       int isKeyframe, unsigned char lacingFlag, int discardable,
  33                       unsigned char *data, unsigned long dataLength) {
  34   Ebml_WriteID(glob, SimpleBlock);
  35   unsigned long blockLength = 4 + dataLength;
  36   blockLength |= 0x10000000; // TODO check length < 0x0FFFFFFFF
  37   Ebml_Serialize(glob, &blockLength, sizeof(blockLength), 4);
  38   trackNumber |= 0x80;  // TODO check track nubmer < 128
  39   Ebml_Write(glob, &trackNumber, 1);
  40   // Ebml_WriteSigned16(glob, timeCode,2); //this is 3 bytes
  41   Ebml_Serialize(glob, &timeCode, sizeof(timeCode), 2);
  42   unsigned char flags = 0x00 | (isKeyframe ? 0x80 : 0x00) | (lacingFlag << 1) | discardable;
  43   Ebml_Write(glob, &flags, 1);
  44   Ebml_Write(glob, data, dataLength);
  45 }
  46
  47 static uint64_t generateTrackID(unsigned int trackNumber) {
  48   uint64_t t = time(NULL) * trackNumber;
  49   uint64_t r = rand();
  50   r = r << 32;
  51   r +=  rand();
  52   uint64_t rval = t ^ r;
  53   return rval;
  54 }
  55
  56 void writeVideoTrack(EbmlGlobal *glob, unsigned int trackNumber,
  57                      int flagLacing, const char *codecId,
  58                      unsigned int pixelWidth, unsigned int pixelHeight,
  59                      double frameRate) {
  60   EbmlLoc start;
  61   Ebml_StartSubElement(glob, &start, TrackEntry);
  62   Ebml_SerializeUnsigned(glob, TrackNumber, trackNumber);
  63   uint64_t trackID = generateTrackID(trackNumber);
  64   Ebml_SerializeUnsigned(glob, TrackUID, trackID);
  65   Ebml_SerializeString(glob, CodecName, "VP8");  // TODO shouldn't be fixed
  66
  67   Ebml_SerializeUnsigned(glob, TrackType, 1); // video is always 1
  68   Ebml_SerializeString(glob, CodecID, codecId);
  69   {
  70     EbmlLoc videoStart;
  71     Ebml_StartSubElement(glob, &videoStart, Video);
  72     Ebml_SerializeUnsigned(glob, PixelWidth, pixelWidth);
  73     Ebml_SerializeUnsigned(glob, PixelHeight, pixelHeight);
  74     Ebml_SerializeFloat(glob, FrameRate, frameRate);
  75     Ebml_EndSubElement(glob, &videoStart); // Video
  76   }
  77   Ebml_EndSubElement(glob, &start); // Track Entry
  78 }
  79 void writeAudioTrack(EbmlGlobal *glob, unsigned int trackNumber,
  80                      int flagLacing, const char *codecId,
  81                      double samplingFrequency, unsigned int channels,
  82                      unsigned char *private, unsigned long privateSize) {
  83   EbmlLoc start;
  84   Ebml_StartSubElement(glob, &start, TrackEntry);
  85   Ebml_SerializeUnsigned(glob, TrackNumber, trackNumber);
  86   uint64_t trackID = generateTrackID(trackNumber);
  87   Ebml_SerializeUnsigned(glob, TrackUID, trackID);
  88   Ebml_SerializeUnsigned(glob, TrackType, 2); // audio is always 2
  89   // I am using defaults for thesed required fields
  90   /*  Ebml_SerializeUnsigned(glob, FlagEnabled, 1);
  91       Ebml_SerializeUnsigned(glob, FlagDefault, 1);
  92       Ebml_SerializeUnsigned(glob, FlagForced, 1);
  93       Ebml_SerializeUnsigned(glob, FlagLacing, flagLacing);*/
  94   Ebml_SerializeString(glob, CodecID, codecId);
  95   Ebml_SerializeData(glob, CodecPrivate, private, privateSize);
  96
  97   Ebml_SerializeString(glob, CodecName, "VORBIS");  // fixed for now
  98   {
  99     EbmlLoc AudioStart;
 100     Ebml_StartSubElement(glob, &AudioStart, Audio);
 101     Ebml_SerializeFloat(glob, SamplingFrequency, samplingFrequency);
 102     Ebml_SerializeUnsigned(glob, Channels, channels);
 103     Ebml_EndSubElement(glob, &AudioStart);
 104   }
 105   Ebml_EndSubElement(glob, &start);
 106 }
 107 void writeSegmentInformation(EbmlGlobal *ebml, EbmlLoc *startInfo, unsigned long timeCodeScale, double duration) {
 108   Ebml_StartSubElement(ebml, startInfo, Info);
 109   Ebml_SerializeUnsigned(ebml, TimecodeScale, timeCodeScale);
 110   Ebml_SerializeFloat(ebml, Segment_Duration, duration * 1000.0); // Currently fixed to using milliseconds
 111   Ebml_SerializeString(ebml, 0x4D80, "QTmuxingAppLibWebM-0.0.1");
 112   Ebml_SerializeString(ebml, 0x5741, "QTwritingAppLibWebM-0.0.1");
 113   Ebml_EndSubElement(ebml, startInfo);
 114 }
 115
 116 /*
 117 void Mkv_InitializeSegment(Ebml& ebml_out, EbmlLoc& ebmlLoc)
 118 {
 119     Ebml_StartSubElement(ebml_out, ebmlLoc, 0x18538067);
 120 }
 121
 122 void Mkv_InitializeSeek(Ebml& ebml_out, EbmlLoc& ebmlLoc)
 123 {
 124     Ebml_StartSubElement(ebml_out, ebmlLoc, 0x114d9b74);
 125 }
 126 void Mkv_WriteSeekInformation(Ebml& ebml_out, SeekStruct& seekInformation)
 127 {
 128     EbmlLoc ebmlLoc;
 129     Ebml_StartSubElement(ebml_out, ebmlLoc, 0x4dbb);
 130     Ebml_SerializeString(ebml_out, 0x53ab, seekInformation.SeekID);
 131     Ebml_SerializeUnsigned(ebml_out, 0x53ac, seekInformation.SeekPosition);
 132     Ebml_EndSubElement(ebml_out, ebmlLoc);
 133 }
 134
 135 void Mkv_WriteSegmentInformation(Ebml& ebml_out, SegmentInformationStruct& segmentInformation)
 136 {
 137     Ebml_SerializeUnsigned(ebml_out, 0x73a4, segmentInformation.segmentUID);
 138     if (segmentInformation.filename != 0)
 139         Ebml_SerializeString(ebml_out, 0x7384, segmentInformation.filename);
 140     Ebml_SerializeUnsigned(ebml_out, 0x2AD7B1, segmentInformation.TimecodeScale);
 141     Ebml_SerializeUnsigned(ebml_out, 0x4489, segmentInformation.Duration);
 142     // TODO date
 143     Ebml_SerializeWString(ebml_out, 0x4D80, L"MKVMUX");
 144     Ebml_SerializeWString(ebml_out, 0x5741, segmentInformation.WritingApp);
 145 }
 146
 147 void Mkv_InitializeTrack(Ebml& ebml_out, EbmlLoc& ebmlLoc)
 148 {
 149     Ebml_StartSubElement(ebml_out, ebmlLoc, 0x1654AE6B);
 150 }
 151
 152 static void Mkv_WriteGenericTrackData(Ebml& ebml_out, TrackStruct& track)
 153 {
 154     Ebml_SerializeUnsigned(ebml_out, 0xD7, track.TrackNumber);
 155     Ebml_SerializeUnsigned(ebml_out, 0x73C5, track.TrackUID);
 156     Ebml_SerializeUnsigned(ebml_out, 0x83, track.TrackType);
 157     Ebml_SerializeUnsigned(ebml_out, 0xB9, track.FlagEnabled ? 1 :0);
 158     Ebml_SerializeUnsigned(ebml_out, 0x88, track.FlagDefault ? 1 :0);
 159     Ebml_SerializeUnsigned(ebml_out, 0x55AA, track.FlagForced ? 1 :0);
 160     if (track.Language != 0)
 161         Ebml_SerializeString(ebml_out, 0x22B59C, track.Language);
 162     if (track.CodecID != 0)
 163         Ebml_SerializeString(ebml_out, 0x86, track.CodecID);
 164     if (track.CodecPrivate != 0)
 165         Ebml_SerializeData(ebml_out, 0x63A2, track.CodecPrivate, track.CodecPrivateLength);
 166     if (track.CodecName != 0)
 167         Ebml_SerializeWString(ebml_out, 0x258688, track.CodecName);
 168 }
 169
 170 void Mkv_WriteVideoTrack(Ebml& ebml_out, TrackStruct & track, VideoTrackStruct& video)
 171 {
 172     EbmlLoc trackHeadLoc, videoHeadLoc;
 173     Ebml_StartSubElement(ebml_out, trackHeadLoc, 0xAE);  // start Track
 174     Mkv_WriteGenericTrackData(ebml_out, track);
 175     Ebml_StartSubElement(ebml_out, videoHeadLoc, 0xE0);  // start Video
 176     Ebml_SerializeUnsigned(ebml_out, 0x9A, video.FlagInterlaced ? 1 :0);
 177     Ebml_SerializeUnsigned(ebml_out, 0xB0, video.PixelWidth);
 178     Ebml_SerializeUnsigned(ebml_out, 0xBA, video.PixelHeight);
 179     Ebml_SerializeUnsigned(ebml_out, 0x54B0, video.PixelDisplayWidth);
 180     Ebml_SerializeUnsigned(ebml_out, 0x54BA, video.PixelDisplayHeight);
 181     Ebml_SerializeUnsigned(ebml_out, 0x54B2, video.displayUnit);
 182     Ebml_SerializeFloat(ebml_out, 0x2383E3, video.FrameRate);
 183     Ebml_EndSubElement(ebml_out, videoHeadLoc);
 184     Ebml_EndSubElement(ebml_out, trackHeadLoc);
 185
 186 }
 187
 188 void Mkv_WriteAudioTrack(Ebml& ebml_out, TrackStruct & track, AudioTrackStruct& video)
 189 {
 190     EbmlLoc trackHeadLoc, audioHeadLoc;
 191     Ebml_StartSubElement(ebml_out, trackHeadLoc, 0xAE);
 192     Mkv_WriteGenericTrackData(ebml_out, track);
 193     Ebml_StartSubElement(ebml_out, audioHeadLoc, 0xE0);  // start Audio
 194     Ebml_SerializeFloat(ebml_out, 0xB5, video.SamplingFrequency);
 195     Ebml_SerializeUnsigned(ebml_out, 0x9F, video.Channels);
 196     Ebml_SerializeUnsigned(ebml_out, 0x6264, video.BitDepth);
 197     Ebml_EndSubElement(ebml_out, audioHeadLoc); // end audio
 198     Ebml_EndSubElement(ebml_out, trackHeadLoc);
 199 }
 200
 201 void Mkv_WriteEbmlClusterHead(Ebml& ebml_out,  EbmlLoc& ebmlLoc, ClusterHeadStruct & clusterHead)
 202 {
 203     Ebml_StartSubElement(ebml_out, ebmlLoc, 0x1F43B675);
 204     Ebml_SerializeUnsigned(ebml_out, 0x6264, clusterHead.TimeCode);
 205 }
 206
 207 void Mkv_WriteSimpleBlockHead(Ebml& ebml_out,  EbmlLoc& ebmlLoc, SimpleBlockStruct& block)
 208 {
 209     Ebml_StartSubElement(ebml_out, ebmlLoc, 0xA3);
 210     Ebml_Write1UInt(ebml_out, block.TrackNumber);
 211     Ebml_WriteSigned16(ebml_out,block.TimeCode);
 212     unsigned char flags = 0x00 | (block.iskey ? 0x80:0x00) | (block.lacing << 1) | block.discardable;
 213     Ebml_Write1UInt(ebml_out, flags);  // TODO this may be the wrong function
 214     Ebml_Serialize(ebml_out, block.data, block.dataLength);
 215     Ebml_EndSubElement(ebml_out,ebmlLoc);
 216 }
 217 */