3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.fr/>.
5 It may be used under the terms of the GNU General Public License. */
14 /* Scale factor to apply to timecodes to convert from HandBrake's
15 * 1/90000s to nanoseconds as expected by libmkv */
16 #define TIMECODE_SCALE 1000000000 / 90000
18 struct hb_mux_object_s
30 uint64_t prev_chapter_tc;
31 uint16_t current_chapter;
35 /**********************************************************************
37 **********************************************************************
38 * Allocates hb_mux_data_t structures, create file and write headers
39 *********************************************************************/
40 static int MKVInit( hb_mux_object_t * m )
42 hb_job_t * job = m->job;
43 hb_title_t * title = job->title;
45 hb_mux_data_t * mux_data;
48 uint8_t default_track_flag = 1;
50 ogg_packet *ogg_headers[3];
51 mk_TrackConfig *track;
53 track = calloc(1, sizeof(mk_TrackConfig));
55 m->file = mk_createWriter(job->file, 1000000, 1);
59 hb_error( "Could not create output file, Disk Full?" );
66 mux_data = calloc(1, sizeof( hb_mux_data_t ) );
67 job->mux_data = mux_data;
69 track->trackType = MK_TRACK_VIDEO;
70 track->flagDefault = 1;
71 track->flagEnabled = 1;
75 track->codecID = MK_VCODEC_MP4AVC;
76 /* Taken from x264 muxers.c */
77 avcC_len = 5 + 1 + 2 + job->config.h264.sps_length + 1 + 2 + job->config.h264.pps_length;
78 avcC = malloc(avcC_len);
83 avcC[1] = job->config.h264.sps[1]; /* AVCProfileIndication */
84 avcC[2] = job->config.h264.sps[2]; /* profile_compat */
85 avcC[3] = job->config.h264.sps[3]; /* AVCLevelIndication */
86 avcC[4] = 0xff; // nalu size length is four bytes
87 avcC[5] = 0xe1; // one sps
89 avcC[6] = job->config.h264.sps_length >> 8;
90 avcC[7] = job->config.h264.sps_length;
92 memcpy(avcC+8, job->config.h264.sps, job->config.h264.sps_length);
94 avcC[8+job->config.h264.sps_length] = 1; // one pps
95 avcC[9+job->config.h264.sps_length] = job->config.h264.pps_length >> 8;
96 avcC[10+job->config.h264.sps_length] = job->config.h264.pps_length;
98 memcpy( avcC+11+job->config.h264.sps_length, job->config.h264.pps, job->config.h264.pps_length );
99 track->codecPrivate = avcC;
100 track->codecPrivateSize = avcC_len;
105 case HB_VCODEC_FFMPEG:
106 track->codecID = MK_VCODEC_MP4ASP;
107 track->codecPrivate = job->config.mpeg4.bytes;
108 track->codecPrivateSize = job->config.mpeg4.length;
110 case HB_VCODEC_THEORA:
113 uint64_t cp_size = 0;
114 track->codecID = MK_VCODEC_THEORA;
115 uint64_t header_sizes[3];
116 for (i = 0; i < 3; ++i)
118 ogg_headers[i] = (ogg_packet *)job->config.theora.headers[i];
119 ogg_headers[i]->packet = (unsigned char *)&job->config.theora.headers[i] + sizeof( ogg_packet );
120 header_sizes[i] = ogg_headers[i]->bytes;
122 track->codecPrivate = mk_laceXiph(header_sizes, 2, &cp_size);
123 track->codecPrivate = realloc(track->codecPrivate, cp_size + ogg_headers[0]->bytes + ogg_headers[1]->bytes + ogg_headers[2]->bytes);
124 for(i = 0; i < 3; ++i)
126 memcpy(track->codecPrivate + cp_size, ogg_headers[i]->packet, ogg_headers[i]->bytes);
127 cp_size += ogg_headers[i]->bytes;
129 track->codecPrivateSize = cp_size;
134 hb_error("muxmkv: Unknown video codec: %x", job->vcodec);
138 track->extra.video.pixelWidth = job->width;
139 track->extra.video.pixelHeight = job->height;
140 track->extra.video.displayHeight = job->height;
141 if( job->anamorphic.mode )
143 track->extra.video.displayWidth = job->width * ((double)job->anamorphic.par_width / (double)job->anamorphic.par_height);
147 track->extra.video.displayWidth = job->width;
151 track->defaultDuration = (int64_t)(((float)job->vrate_base / (float)job->vrate) * 1000000000);
153 mux_data->track = mk_createTrack(m->file, track);
155 memset(track, 0, sizeof(mk_TrackConfig));
157 /* add the audio tracks */
158 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
160 audio = hb_list_item( title->list_audio, i );
161 mux_data = malloc( sizeof( hb_mux_data_t ) );
162 audio->priv.mux_data = mux_data;
164 mux_data->codec = audio->config.out.codec;
166 switch (audio->config.out.codec)
169 track->codecPrivate = NULL;
170 track->codecPrivateSize = 0;
171 track->codecID = MK_ACODEC_DTS;
174 track->codecPrivate = NULL;
175 track->codecPrivateSize = 0;
176 track->codecID = MK_ACODEC_AC3;
179 track->codecPrivate = NULL;
180 track->codecPrivateSize = 0;
181 track->codecID = MK_ACODEC_MP3;
183 case HB_ACODEC_VORBIS:
186 uint64_t cp_size = 0;
187 track->codecID = MK_ACODEC_VORBIS;
188 uint64_t header_sizes[3];
189 for (i = 0; i < 3; ++i)
191 ogg_headers[i] = (ogg_packet *)audio->priv.config.vorbis.headers[i];
192 ogg_headers[i]->packet = (unsigned char *)&audio->priv.config.vorbis.headers[i] + sizeof( ogg_packet );
193 header_sizes[i] = ogg_headers[i]->bytes;
195 track->codecPrivate = mk_laceXiph(header_sizes, 2, &cp_size);
196 track->codecPrivate = realloc(track->codecPrivate, cp_size + ogg_headers[0]->bytes + ogg_headers[1]->bytes + ogg_headers[2]->bytes);
197 for(i = 0; i < 3; ++i)
199 memcpy(track->codecPrivate + cp_size, ogg_headers[i]->packet, ogg_headers[i]->bytes);
200 cp_size += ogg_headers[i]->bytes;
202 track->codecPrivateSize = cp_size;
206 track->codecPrivate = audio->priv.config.aac.bytes;
207 track->codecPrivateSize = audio->priv.config.aac.length;
208 track->codecID = MK_ACODEC_AAC;
212 hb_error("muxmkv: Unknown audio codec: %x", audio->config.out.codec);
216 if (default_track_flag)
218 track->flagDefault = 1;
219 default_track_flag = 0;
223 track->flagDefault = 0;
225 track->flagEnabled = 1;
226 track->trackType = MK_TRACK_AUDIO;
227 track->language = audio->config.lang.iso639_2;
228 track->extra.audio.samplingFreq = (float)audio->config.out.samplerate;
229 if (audio->config.out.codec == HB_ACODEC_AC3 ||
230 audio->config.out.codec == HB_ACODEC_DCA)
232 track->extra.audio.channels = HB_INPUT_CH_LAYOUT_GET_DISCRETE_COUNT(audio->config.in.channel_layout);
236 track->extra.audio.channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(audio->config.out.mixdown);
238 // track->defaultDuration = job->arate * 1000;
239 mux_data->track = mk_createTrack(m->file, track);
240 if (audio->config.out.codec == HB_ACODEC_VORBIS && track->codecPrivate != NULL)
241 free(track->codecPrivate);
244 if( mk_writeHeader( m->file, "HandBrake " HB_PROJECT_VERSION) < 0 )
246 hb_error( "Failed to write to output file, disk full?");
257 static int MKVMux( hb_mux_object_t * m, hb_mux_data_t * mux_data,
260 ogg_packet *op = NULL;
261 hb_job_t * job = m->job;
262 hb_title_t * title = job->title;
263 uint64_t timecode = 0;
264 hb_chapter_t *chapter_data;
265 char tmp_buffer[1024];
266 char *string = tmp_buffer;
268 if (mux_data == job->mux_data)
271 timecode = buf->start * TIMECODE_SCALE;
273 if (job->chapter_markers && (buf->new_chap || timecode == 0))
275 /* Make sure we're not writing a chapter that has 0 length */
276 if (mux_data->prev_chapter_tc != timecode)
280 mux_data->current_chapter = buf->new_chap - 2;
282 chapter_data = hb_list_item( title->list_chapter,
283 mux_data->current_chapter++ );
284 tmp_buffer[0] = '\0';
286 if( chapter_data != NULL )
288 string = chapter_data->title;
291 if( strlen(string) == 0 || strlen(string) >= 1024 )
293 snprintf( tmp_buffer, 1023, "Chapter %02i", mux_data->current_chapter );
296 mk_createChapterSimple(m->file, mux_data->prev_chapter_tc, mux_data->prev_chapter_tc, string);
298 mux_data->prev_chapter_tc = timecode;
301 if (job->vcodec == HB_VCODEC_THEORA)
303 /* ughhh, theora is a pain :( */
304 op = (ogg_packet *)buf->data;
305 op->packet = buf->data + sizeof( ogg_packet );
306 if (mk_startFrame(m->file, mux_data->track) < 0)
308 hb_error( "Failed to write frame to output file, Disk Full?" );
311 mk_addFrameData(m->file, mux_data->track, op->packet, op->bytes);
312 mk_setFrameFlags(m->file, mux_data->track, timecode, 1);
319 timecode = buf->start * TIMECODE_SCALE;
320 if (mux_data->codec == HB_ACODEC_VORBIS)
322 /* ughhh, vorbis is a pain :( */
323 op = (ogg_packet *)buf->data;
324 op->packet = buf->data + sizeof( ogg_packet );
325 if (mk_startFrame(m->file, mux_data->track))
327 hb_error( "Failed to write frame to output file, Disk Full?" );
330 mk_addFrameData(m->file, mux_data->track, op->packet, op->bytes);
331 mk_setFrameFlags(m->file, mux_data->track, timecode, 1);
336 if( mk_startFrame(m->file, mux_data->track) < 0)
338 hb_error( "Failed to write frame to output file, Disk Full?" );
341 mk_addFrameData(m->file, mux_data->track, buf->data, buf->size);
342 mk_setFrameFlags(m->file, mux_data->track, timecode,
343 ((job->vcodec == HB_VCODEC_X264 && mux_data == job->mux_data) ? (buf->frametype == HB_FRAME_IDR) : ((buf->frametype & HB_FRAME_KEY) != 0)) );
347 static int MKVEnd( hb_mux_object_t * m )
349 hb_job_t *job = m->job;
350 hb_mux_data_t *mux_data = job->mux_data;
351 hb_title_t *title = job->title;
352 hb_chapter_t *chapter_data;
353 char tmp_buffer[1024];
354 char *string = tmp_buffer;
359 * We must have failed to create the file in the first place.
364 chapter_data = hb_list_item( title->list_chapter, mux_data->current_chapter++ );
366 if(job->chapter_markers)
368 tmp_buffer[0] = '\0';
370 if( chapter_data != NULL )
372 string = chapter_data->title;
375 if( strlen(string) == 0 || strlen(string) >= 1024 )
377 snprintf( tmp_buffer, 1023, "Chapter %02i", mux_data->current_chapter );
380 mk_createChapterSimple(m->file, mux_data->prev_chapter_tc, mux_data->prev_chapter_tc, string);
383 if( title->metadata )
385 hb_metadata_t *md = title->metadata;
387 hb_deep_log( 2, "Writing Metadata to output file...");
388 mk_createTagSimple( m->file, MK_TAG_TITLE, md->name );
389 mk_createTagSimple( m->file, "ARTIST", md->artist );
390 mk_createTagSimple( m->file, "COMPOSER", md->composer );
391 mk_createTagSimple( m->file, MK_TAG_SYNOPSIS, md->comment );
392 mk_createTagSimple( m->file, "DATE_RELEASED", md->release_date );
393 // mk_createTagSimple( m->file, "", md->album );
394 mk_createTagSimple( m->file, MK_TAG_GENRE, md->genre );
397 if( mk_close(m->file) < 0 )
399 hb_error( "Failed to flush the last frame and close the output file, Disk Full?" );
403 // TODO: Free what we alloc'd
408 hb_mux_object_t * hb_mux_mkv_init( hb_job_t * job )
410 hb_mux_object_t * m = calloc( sizeof( hb_mux_object_t ), 1 );