?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
r125: This commit was manufactured by cvs2svn to create tag 'r1_1_7-last'.
[cinelerra_cv/mob.git] / hvirtual / libmpeg3 / mpeg3toc.c
blobb7ced9d1f9cbf2045229459231e47532784336db
1 #include "libmpeg3.h"
2 #include "mpeg3protos.h"
3
4 #include <stdint.h>
5 #include <stdio.h>
6 #include <string.h>
7 #include <stdlib.h>
8 #include <sys/stat.h>
9
10
11
12
13 #define INIT_VECTORS(data, size, allocation, tracks) \
14 { \
15 int k; \
16 data = calloc(1, sizeof(int64_t*) * tracks); \
17 size = calloc(1, sizeof(int*) * tracks); \
18 allocation = calloc(1, sizeof(int*) * tracks); \
19 \
20 for(k = 0; k < tracks; k++) \
21 { \
22 allocation[k] = 0x100; \
23 data[k] = calloc(1, sizeof(int64_t) * allocation[k]); \
24 } \
25 }
26
27 #if 1
28 #define APPEND_VECTOR(data, size, allocation, track, value) \
29 { \
30 uint64_t **track_data = &(data)[(track)]; \
31 int *track_allocation = &(allocation)[(track)]; \
32 int *track_size = &(size)[(track)]; \
33 \
34 if(!(*track_data) || (*track_allocation) <= (*track_size)) \
35 { \
36 int64_t *new_data = calloc(1, sizeof(int64_t) * (*track_allocation) * 2); \
37 \
38 if((*track_data)) \
39 { \
40 memcpy(new_data, (*track_data), sizeof(int64_t) * (*track_allocation)); \
41 free((*track_data)); \
42 } \
43 (*track_allocation) *= 2; \
44 (*track_data) = new_data; \
45 } \
46 \
47 (*track_data)[(*track_size)++] = (value); \
48 }
49 #else
50 #define APPEND_VECTOR(data, size, allocation, track, value) \
51 ;
52 #endif
53
54 #define DELETE_VECTORS(data, size, allocation, tracks) \
55 { \
56 int k; \
57 for(k = 0; k < tracks; k++) if(data[k]) free(data[k]); \
58 free(data); \
59 }
60
61
62
63
64
65
66 #define PUT_INT32(x) \
67 { \
68 if(MPEG3_LITTLE_ENDIAN) \
69 { \
70 fputc(((unsigned char*)&x)[3], output); \
71 fputc(((unsigned char*)&x)[2], output); \
72 fputc(((unsigned char*)&x)[1], output); \
73 fputc(((unsigned char*)&x)[0], output); \
74 } \
75 else \
76 { \
77 fputc(((unsigned char*)&x)[0], output); \
78 fputc(((unsigned char*)&x)[1], output); \
79 fputc(((unsigned char*)&x)[2], output); \
80 fputc(((unsigned char*)&x)[3], output); \
81 } \
82 }
83
84
85
86
87 #define PUT_INT64(x) \
88 { \
89 if(MPEG3_LITTLE_ENDIAN) \
90 { \
91 fputc(((unsigned char*)&x)[7], output); \
92 fputc(((unsigned char*)&x)[6], output); \
93 fputc(((unsigned char*)&x)[5], output); \
94 fputc(((unsigned char*)&x)[4], output); \
95 fputc(((unsigned char*)&x)[3], output); \
96 fputc(((unsigned char*)&x)[2], output); \
97 fputc(((unsigned char*)&x)[1], output); \
98 fputc(((unsigned char*)&x)[0], output); \
99 } \
100 else \
101 { \
102 fwrite(&x, 1, 8, output); \
103 } \
104 }
105
106
107
108
109
110 int main(int argc, char *argv[])
111 {
112 struct stat st;
113
114 if(argc < 3)
115 {
116 fprintf(stderr, "Create a table of contents for a DVD or mpeg stream.\n"
117 " Usage: mpeg3toc <path> <output>\n"
118 "\n"
119 " The path should be absolute unless you plan\n"
120 " to always run your movie editor from the same directory\n"
121 " as the filename. For renderfarms the filesystem prefix\n"
122 " should be / and the movie directory mounted under the same\n"
123 " directory on each node.\n"
124 "Example: mpeg3toc /cd2/video_ts/vts_01_0.ifo titanic.toc\n");
125 exit(1);
126 }
127
128
129
130 stat(argv[1], &st);
131
132 if(!st.st_size)
133 {
134 fprintf(stderr, "%s is 0 length. Skipping\n", argv[1]);
135 }
136 else
137 {
138 int i, j, l;
139 int64_t size;
140 int vtracks;
141 int atracks;
142 mpeg3_t *input;
143 uint64_t **frame_offsets;
144 uint64_t **keyframe_numbers;
145 uint64_t **sample_offsets;
146 int *total_frame_offsets;
147 int *total_sample_offsets;
148 int *total_keyframe_numbers;
149 int *frame_offset_allocation;
150 int *sample_offset_allocation;
151 int *keyframe_numbers_allocation;
152 int done = 0;
153 double sample_rate;
154 double frame_rate;
155 FILE *output;
156 int total_samples = 0;
157 int total_frames = 0;
158 int rewind = 1;
159
160 //printf(__FUNCTION__ " 1\n");
161 input = mpeg3_open(argv[1]);
162 output = fopen(argv[2], "w");
163
164 //printf(__FUNCTION__ " 2\n");
165 vtracks = mpeg3_total_vstreams(input);
166 atracks = mpeg3_total_astreams(input);
167
168 if(atracks) sample_rate = mpeg3_sample_rate(input, 0);
169 if(vtracks) frame_rate = mpeg3_frame_rate(input, 0);
170
171 //printf(__FUNCTION__ " 3\n");
172 // Handle titles
173 INIT_VECTORS(frame_offsets, total_frame_offsets, frame_offset_allocation, vtracks);
174 INIT_VECTORS(keyframe_numbers, total_keyframe_numbers, keyframe_numbers_allocation, vtracks);
175 INIT_VECTORS(sample_offsets, total_sample_offsets, sample_offset_allocation, atracks);
176
177 //printf(__FUNCTION__ " 4\n");
178 while(!done)
179 {
180 int sample_count = MPEG3_AUDIO_CHUNKSIZE;
181 int frame_count;
182 int have_audio = 0;
183 int have_video = 0;
184 int64_t title_number = 0;
185
186 // Store current position and read sample_count from each atrack
187 for(j = 0; j < atracks; j++)
188 {
189 //printf(__FUNCTION__ " 3 %d\n", total_sample_offsets[j]);
190 if(rewind)
191 {
192 mpeg3_demuxer_t *demuxer = input->atrack[j]->demuxer;
193 mpeg3demux_seek_byte(demuxer, 0);
194 }
195
196 if(!mpeg3_end_of_audio(input, j))
197 {
198 // Don't want to maintain separate vectors for offset and title.
199 title_number = mpeg3demux_tell_title(input->atrack[j]->demuxer);
200 int64_t position = mpeg3demux_tell(input->atrack[j]->demuxer);
201 int64_t result;
202 if(position < MPEG3_IO_SIZE) position = MPEG3_IO_SIZE;
203 result = (title_number << 56) | (position - MPEG3_IO_SIZE);
204
205 have_audio = 1;
206 APPEND_VECTOR(sample_offsets,
207 total_sample_offsets,
208 sample_offset_allocation,
209 j,
210 result);
211
212
213 //printf(__FUNCTION__ " 6 %d\n", j);
214 // Throw away samples
215 mpeg3_read_audio(input,
216 0,
217 0,
218 0,
219 sample_count, /* Number of samples to decode */
220 j);
221
222 //printf(__FUNCTION__ " 7 %d\n", total_sample_offsets[j]);
223 }
224
225 if(j == atracks - 1)
226 {
227 total_samples += sample_count;
228 printf("Audio: title=%lld total_samples=%d ", title_number, total_samples);
229 }
230 }
231
232 //printf(__FUNCTION__ " 8\n");
233 if(have_audio)
234 {
235 frame_count =
236 (int)((double)total_samples / sample_rate * frame_rate + 0.5) -
237 total_frames;
238 }
239 else
240 {
241 frame_count = 1;
242 }
243
244 //printf(__FUNCTION__ " 9 %d\n", vtracks);
245 for(j = 0; j < vtracks; j++)
246 {
247 mpeg3video_t *video = input->vtrack[j]->video;
248 mpeg3_demuxer_t *demuxer = input->vtrack[j]->demuxer;
249 if(rewind)
250 {
251 mpeg3demux_seek_byte(demuxer, 0);
252 }
253
254
255 for(l = 0; l < frame_count; l++)
256 {
257 if(!mpeg3_end_of_video(input, j))
258 {
259 int64_t title_number = mpeg3demux_tell_title(demuxer);
260 int64_t position = mpeg3demux_tell(demuxer);
261 int64_t result;
262 uint32_t code = 0;
263 int got_top = 0;
264 int got_bottom = 0;
265 int got_keyframe = 0;
266 int fields = 0;
267
268 if(position < MPEG3_IO_SIZE) position = MPEG3_IO_SIZE;
269 result = (title_number << 56) | (position - MPEG3_IO_SIZE);
270 have_video = 1;
271
272
273
274 // Store offset of every frame in table
275 APPEND_VECTOR(frame_offsets,
276 total_frame_offsets,
277 frame_offset_allocation,
278 j,
279 result);
280
281
282
283 // Search for next frame start.
284 if(total_frame_offsets[j] == 1)
285 {
286 // Assume first frame is an I-frame and put its number in the keyframe number
287 // table.
288 APPEND_VECTOR(keyframe_numbers,
289 total_keyframe_numbers,
290 keyframe_numbers_allocation,
291 j,
292 0);
293
294
295
296 // Skip the first frame.
297 mpeg3video_get_header(video, 0);
298 video->current_repeat += 100;
299 }
300
301
302
303
304 // Get next frame
305 do
306 {
307 mpeg3video_get_header(video, 0);
308 video->current_repeat += 100;
309
310 if(video->pict_struct == TOP_FIELD)
311 {
312 got_top = 1;
313 }
314 else
315 if(video->pict_struct == BOTTOM_FIELD)
316 {
317 got_bottom = 1;
318 }
319 else
320 if(video->pict_struct == FRAME_PICTURE)
321 {
322 got_top = got_bottom = 1;
323 }
324 fields++;
325
326 // The way we do it, the I frames have the top field but both the I frame and
327 // subsequent P frame make the keyframe.
328 if(video->pict_type == I_TYPE)
329 got_keyframe = 1;
330 //printf(__FUNCTION__ " 10 %d %d %d\n", video->pict_type, got_top, got_bottom);
331 }while(!mpeg3_end_of_video(input, j) &&
332 !got_bottom &&
333 total_frame_offsets[j] > 1);
334
335
336
337
338 // Store number of a keyframe in the keyframe number table
339 if(got_keyframe)
340 APPEND_VECTOR(keyframe_numbers,
341 total_keyframe_numbers,
342 keyframe_numbers_allocation,
343 j,
344 total_frame_offsets[j] - 1);
345
346 if(j == vtracks - 1 && l == frame_count - 1)
347 {
348 total_frames += frame_count;
349 printf("Video: title=%lld total_frames=%d ", title_number, total_frames);
350 }
351 }
352 }
353 }
354
355 if(!have_audio && !have_video) done = 1;
356
357 printf("\r");
358 fflush(stdout);
359 /*
360 * if(total_frames > 10000)
361 * {
362 * printf("\n");
363 * return 0;
364 * }
365 */
366
367
368
369 rewind = 0;
370 }
371
372
373
374 // Write file type
375 fputc('T', output);
376 fputc('O', output);
377 fputc('C', output);
378 fputc(' ', output);
379
380
381 if(input->is_program_stream)
382 {
383 fputc(FILE_TYPE_PROGRAM, output);
384 }
385 else
386 if(input->is_transport_stream)
387 {
388 fputc(FILE_TYPE_TRANSPORT, output);
389 }
390 else
391 if(input->is_audio_stream)
392 {
393 fputc(FILE_TYPE_AUDIO, output);
394 }
395 else
396 if(input->is_video_stream)
397 {
398 fputc(FILE_TYPE_VIDEO, output);
399 }
400
401 // Write stream ID's
402 // Only program and transport streams have these
403 for(i = 0; i < MPEG3_MAX_STREAMS; i++)
404 {
405 if(input->demuxer->astream_table[i])
406 {
407 fputc(STREAM_AUDIO, output);
408 PUT_INT32(i);
409 PUT_INT32(input->demuxer->astream_table[i]);
410 }
411
412 if(input->demuxer->vstream_table[i])
413 {
414 fputc(STREAM_VIDEO, output);
415 PUT_INT32(i);
416 PUT_INT32(input->demuxer->vstream_table[i]);
417 }
418 }
419
420
421 // Write titles
422 for(i = 0; i < input->demuxer->total_titles; i++)
423 {
424 // Path
425 fputc(TITLE_PATH, output);
426 fprintf(output, input->demuxer->titles[i]->fs->path);
427 fputc(0, output);
428 // Total bytes
429 PUT_INT64(input->demuxer->titles[i]->total_bytes);
430 // Byte offsets of cells
431 PUT_INT32(input->demuxer->titles[i]->timecode_table_size);
432 for(j = 0; j < input->demuxer->titles[i]->timecode_table_size; j++)
433 {
434 PUT_INT64(input->demuxer->titles[i]->timecode_table[j].start_byte);
435 PUT_INT64(input->demuxer->titles[i]->timecode_table[j].end_byte);
436 PUT_INT32(input->demuxer->titles[i]->timecode_table[j].program);
437 }
438 }
439
440
441
442
443
444
445
446
447 fputc(ATRACK_COUNT, output);
448 PUT_INT32(atracks);
449
450 fputc(VTRACK_COUNT, output);
451 PUT_INT32(vtracks);
452
453 // Audio streams
454 for(j = 0; j < atracks; j++)
455 {
456 int channels = mpeg3_audio_channels(input, j);
457 PUT_INT32(channels);
458 PUT_INT32(total_sample_offsets[j]);
459 for(i = 0; i < total_sample_offsets[j]; i++)
460 {
461 PUT_INT64(sample_offsets[j][i]);
462 //printf("Audio: offset=%016llx\n", sample_offsets[j][i]);
463 }
464 }
465
466 // Video streams
467 for(j = 0; j < vtracks; j++)
468 {
469 PUT_INT32(total_frame_offsets[j]);
470 for(i = 0; i < total_frame_offsets[j]; i++)
471 {
472 PUT_INT64(frame_offsets[j][i]);
473 //printf("Video: offset=%016llx\n", frame_offsets[j][i]);
474 }
475
476 PUT_INT32(total_keyframe_numbers[j]);
477 for(i = 0; i < total_keyframe_numbers[j]; i++)
478 {
479 PUT_INT64(keyframe_numbers[j][i]);
480 //printf("Video: keyframe=%lld\n", keyframe_numbers[j][i]);
481 }
482 }
483
484
485
486
487 DELETE_VECTORS(frame_offsets, total_frame_offsets, frame_offset_allocation, vtracks);
488 DELETE_VECTORS(keyframe_numbers, total_keyframe_numbers, keyframe_numbers_allocation, vtracks);
489 DELETE_VECTORS(sample_offsets, total_sample_offsets, sample_offset_allocation, atracks);
490
491
492 mpeg3_close(input);
493 fclose(output);
494 }
495
496
497
498
499 return 0;
500 }
mob repository