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