3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.m0k.org/>.
5 It may be used under the terms of the GNU General Public License. */
14 struct hb_mux_object_s
26 uint64_t prev_chapter_tc;
28 uint16_t current_chapter;
31 /**********************************************************************
33 **********************************************************************
34 * Allocates hb_mux_data_t structures, create file and write headers
35 *********************************************************************/
36 static int MKVInit( hb_mux_object_t * m )
38 hb_job_t * job = m->job;
39 hb_title_t * title = job->title;
41 hb_mux_data_t * mux_data;
44 uint8_t default_track_flag = 1;
46 ogg_packet *ogg_headers[3];
47 mk_TrackConfig *track;
49 track = calloc(1, sizeof(mk_TrackConfig));
51 m->file = mk_createWriter(job->file, 1000000, 1);
54 mux_data = calloc(1, sizeof( hb_mux_data_t ) );
55 job->mux_data = mux_data;
57 track->trackType = MK_TRACK_VIDEO;
58 track->flagDefault = 1;
62 track->codecID = MK_VCODEC_MP4AVC;
63 /* Taken from x264 muxers.c */
64 avcC_len = 5 + 1 + 2 + job->config.h264.sps_length + 1 + 2 + job->config.h264.pps_length;
65 avcC = malloc(avcC_len);
70 avcC[1] = job->config.h264.sps[1]; /* AVCProfileIndication */
71 avcC[2] = job->config.h264.sps[2]; /* profile_compat */
72 avcC[3] = job->config.h264.sps[3]; /* AVCLevelIndication */
73 avcC[4] = 0xff; // nalu size length is four bytes
74 avcC[5] = 0xe1; // one sps
76 avcC[6] = job->config.h264.sps_length >> 8;
77 avcC[7] = job->config.h264.sps_length;
79 memcpy(avcC+8, job->config.h264.sps, job->config.h264.sps_length);
81 avcC[8+job->config.h264.sps_length] = 1; // one pps
82 avcC[9+job->config.h264.sps_length] = job->config.h264.pps_length >> 8;
83 avcC[10+job->config.h264.sps_length] = job->config.h264.pps_length;
85 memcpy( avcC+11+job->config.h264.sps_length, job->config.h264.pps, job->config.h264.pps_length );
86 track->codecPrivate = avcC;
87 track->codecPrivateSize = avcC_len;
92 case HB_VCODEC_FFMPEG:
93 track->codecID = MK_VCODEC_MP4ASP;
94 track->codecPrivate = job->config.mpeg4.bytes;
95 track->codecPrivateSize = job->config.mpeg4.length;
99 hb_log("muxmkv: Unknown video codec: %x", job->vcodec);
103 track->video.pixelWidth = job->width;
104 track->video.pixelHeight = job->height;
105 track->video.displayHeight = job->height;
108 track->video.displayWidth = job->width * ((double)job->pixel_aspect_width / (double)job->pixel_aspect_height);
112 track->video.displayWidth = job->width;
116 track->defaultDuration = (int64_t)(((float)job->vrate_base / (float)job->vrate) * 1000000000);
118 mux_data->track = mk_createTrack(m->file, track);
120 memset(track, 0, sizeof(mk_TrackConfig));
122 /* add the audio tracks */
123 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
125 audio = hb_list_item( title->list_audio, i );
126 mux_data = malloc( sizeof( hb_mux_data_t ) );
127 audio->mux_data = mux_data;
132 track->codecPrivate = NULL;
133 track->codecPrivateSize = 0;
134 track->codecID = MK_ACODEC_AC3;
137 track->codecPrivate = NULL;
138 track->codecPrivateSize = 0;
139 track->codecID = MK_ACODEC_MP3;
141 case HB_ACODEC_VORBIS:
147 track->codecID = MK_ACODEC_VORBIS;
148 cp_size = sizeof( char );
149 for (i = 0; i < 3; ++i)
151 ogg_headers[i] = (ogg_packet *)audio->config.vorbis.headers[i];
152 ogg_headers[i]->packet = (unsigned char *)&audio->config.vorbis.headers[i] + sizeof( ogg_packet );
153 cp_size += (sizeof( char ) * ((ogg_headers[i]->bytes / 255) + 1)) + ogg_headers[i]->bytes;
154 /* This will be too big, but it doesn't matter, as we only need it to be big enough. */
156 cp = track->codecPrivate = calloc(1, cp_size);
158 for (i = 0; i < 2; ++i)
160 for (j = ogg_headers[i]->bytes; j >= 255; j -= 255)
166 for(i = 0; i < 3; ++i)
168 memcpy(cp + offset, ogg_headers[i]->packet, ogg_headers[i]->bytes);
169 offset += ogg_headers[i]->bytes;
171 track->codecPrivateSize = offset;
175 track->codecPrivate = audio->config.aac.bytes;
176 track->codecPrivateSize = audio->config.aac.length;
177 track->codecID = MK_ACODEC_AAC;
181 hb_log("muxmkv: Unknown audio codec: %x", job->acodec);
185 if (default_track_flag)
187 track->flagDefault = 1;
188 default_track_flag = 0;
191 track->trackType = MK_TRACK_AUDIO;
192 track->language = audio->iso639_2;
193 track->audio.samplingFreq = (float)job->arate;
194 track->audio.channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(audio->amixdown);
195 // track->defaultDuration = job->arate * 1000;
196 mux_data->track = mk_createTrack(m->file, track);
197 if (job->acodec == HB_ACODEC_VORBIS && track->codecPrivate != NULL)
198 free(track->codecPrivate);
201 mk_writeHeader( m->file, "HandBrake " HB_VERSION);
210 static int MKVMux( hb_mux_object_t * m, hb_mux_data_t * mux_data,
213 hb_job_t * job = m->job;
214 hb_title_t * title = job->title;
215 uint64_t timecode = 0;
216 hb_chapter_t *chapter_data;
217 char tmp_buffer[1024];
218 char *string = tmp_buffer;
219 if (mux_data == job->mux_data)
222 /* Where does the 11130 come from? I had to calculate it from the actual
223 * and the observed duration of the file. Otherwise the timecodes come
224 * out way too small, and you get a 2hr movie that plays in .64 sec. */
225 if ((job->vcodec == HB_VCODEC_X264) && (buf->frametype & HB_FRAME_REF))
227 timecode = (buf->start + (buf->renderOffset - 1000000)) * 11130;
231 timecode = buf->start * 11130;
234 if (job->chapter_markers && (buf->new_chap || timecode == 0))
236 /* Make sure we're not writing a chapter that has 0 length */
237 if (mux_data->prev_chapter_tc != timecode)
239 chapter_data = hb_list_item( title->list_chapter, mux_data->current_chapter );
240 tmp_buffer[0] = '\0';
242 if( chapter_data != NULL )
244 string = chapter_data->title;
247 if( strlen(string) == 0 || strlen(string) >= 1024 )
249 snprintf( tmp_buffer, 1023, "Chapter %02i", mux_data->current_chapter++ );
252 mk_createChapterSimple(m->file, mux_data->prev_chapter_tc, timecode, string);
254 mux_data->prev_chapter_tc = timecode;
257 if (buf->stop * 11130 > mux_data->max_tc)
258 mux_data->max_tc = buf->stop * 11130;
263 timecode = buf->start * 11130;
264 if (job->acodec == HB_ACODEC_VORBIS)
266 /* ughhh, vorbis is a pain :( */
269 op = (ogg_packet *)buf->data;
270 op->packet = buf->data + sizeof( ogg_packet );
271 mk_startFrame(m->file, mux_data->track);
272 mk_addFrameData(m->file, mux_data->track, op->packet, op->bytes);
273 mk_setFrameFlags(m->file, mux_data->track, timecode, 1);
278 mk_startFrame(m->file, mux_data->track);
279 mk_addFrameData(m->file, mux_data->track, buf->data, buf->size);
280 mk_setFrameFlags(m->file, mux_data->track, timecode,
281 ((job->vcodec == HB_VCODEC_X264 && mux_data == job->mux_data) ? (buf->frametype == HB_FRAME_IDR) : ((buf->frametype & HB_FRAME_KEY) != 0)) );
285 static int MKVEnd( hb_mux_object_t * m )
287 hb_job_t *job = m->job;
288 hb_mux_data_t *mux_data = job->mux_data;
289 hb_title_t *title = job->title;
290 hb_chapter_t *chapter_data = hb_list_item( title->list_chapter, mux_data->current_chapter );
291 char tmp_buffer[1024];
292 char *string = tmp_buffer;
294 if(job->chapter_markers)
296 tmp_buffer[0] = '\0';
298 if( chapter_data != NULL )
300 string = chapter_data->title;
303 if( strlen(string) == 0 || strlen(string) >= 1024 )
305 snprintf( tmp_buffer, 1023, "Chapter %02i", mux_data->current_chapter );
308 mk_createChapterSimple(m->file, mux_data->prev_chapter_tc, mux_data->max_tc, string);
313 // TODO: Free what we alloc'd
318 hb_mux_object_t * hb_mux_mkv_init( hb_job_t * job )
320 hb_mux_object_t * m = calloc( sizeof( hb_mux_object_t ), 1 );