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