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