1 /* 2 * Copyright 2010 Mozilla Foundation 3 * 4 * This program is made available under an ISC-style license. See the 5 * accompanying file LICENSE for details. 6 */ 7 #include <assert.h> 8 #include <stdarg.h> 9 #include <stdio.h> 10 #include <stdlib.h> 11 #include <stdint.h> 12 #include "nestegg/nestegg.h" 13 14 #undef DEBUG 15 #define SEEK_TEST 16 17 static int 18 stdio_read(void * p, size_t length, void * fp) 19 { 20 size_t r; 21 22 r = fread(p, length, 1, fp); 23 if (r == 0 && feof(fp)) 24 return 0; 25 return r == 0 ? -1 : 1; 26 } 27 28 static int 29 stdio_seek(int64_t offset, int whence, void * fp) 30 { 31 return fseek(fp, offset, whence); 32 } 33 34 static int64_t 35 stdio_tell(void * fp) 36 { 37 return ftell(fp); 38 } 39 40 static void 41 log_callback(nestegg * ctx, unsigned int severity, char const * fmt, ...) 42 { 43 va_list ap; 44 char const * sev = NULL; 45 46 #ifndef DEBUG 47 if (severity < NESTEGG_LOG_WARNING) 48 return; 49 #endif 50 51 switch (severity) { 52 case NESTEGG_LOG_DEBUG: 53 sev = "debug: "; 54 break; 55 case NESTEGG_LOG_WARNING: 56 sev = "warning: "; 57 break; 58 case NESTEGG_LOG_CRITICAL: 59 sev = "critical:"; 60 break; 61 default: 62 sev = "unknown: "; 63 } 64 65 fprintf(stderr, "%p %s ", (void *) ctx, sev); 66 67 va_start(ap, fmt); 68 vfprintf(stderr, fmt, ap); 69 va_end(ap); 70 71 fprintf(stderr, "\n"); 72 } 73 74 int 75 main(int argc, char * argv[]) 76 { 77 FILE * fp; 78 int r, type; 79 nestegg * ctx; 80 nestegg_audio_params aparams; 81 nestegg_packet * pkt; 82 nestegg_video_params vparams; 83 size_t length, size; 84 uint64_t duration, tstamp, pkt_tstamp; 85 unsigned char * codec_data, * ptr; 86 unsigned int cnt, i, j, track, tracks, pkt_cnt, pkt_track; 87 unsigned int data_items = 0; 88 nestegg_io io = { 89 stdio_read, 90 stdio_seek, 91 stdio_tell, 92 NULL 93 }; 94 95 if (argc != 2) 96 return EXIT_FAILURE; 97 98 fp = fopen(argv[1], "rb"); 99 if (!fp) 100 return EXIT_FAILURE; 101 102 io.userdata = fp; 103 104 ctx = NULL; 105 r = nestegg_init(&ctx, io, log_callback); 106 if (r != 0) 107 return EXIT_FAILURE; 108 109 nestegg_track_count(ctx, &tracks); 110 nestegg_duration(ctx, &duration); 111 #ifdef DEBUG 112 fprintf(stderr, "media has %u tracks and duration %fs\n", tracks, duration / 1e9); 113 #endif 114 115 for (i = 0; i < tracks; ++i) { 116 type = nestegg_track_type(ctx, i); 117 #ifdef DEBUG 118 fprintf(stderr, "track %u: type: %d codec: %d", i, 119 type, nestegg_track_codec_id(ctx, i)); 120 #endif 121 nestegg_track_codec_data_count(ctx, i, &data_items); 122 for (j = 0; j < data_items; ++j) { 123 nestegg_track_codec_data(ctx, i, j, &codec_data, &length); 124 #ifdef DEBUG 125 fprintf(stderr, " (%p, %u)", codec_data, (unsigned int) length); 126 #endif 127 } 128 if (type == NESTEGG_TRACK_VIDEO) { 129 nestegg_track_video_params(ctx, i, &vparams); 130 #ifdef DEBUG 131 fprintf(stderr, " video: %ux%u (d: %ux%u %ux%ux%ux%u)", 132 vparams.width, vparams.height, 133 vparams.display_width, vparams.display_height, 134 vparams.crop_top, vparams.crop_left, vparams.crop_bottom, vparams.crop_right); 135 #endif 136 } else if (type == NESTEGG_TRACK_AUDIO) { 137 nestegg_track_audio_params(ctx, i, &aparams); 138 #ifdef DEBUG 139 fprintf(stderr, " audio: %.2fhz %u bit %u channels", 140 aparams.rate, aparams.depth, aparams.channels); 141 #endif 142 } 143 #ifdef DEBUG 144 fprintf(stderr, "\n"); 145 #endif 146 } 147 148 #ifdef SEEK_TEST 149 #ifdef DEBUG 150 fprintf(stderr, "seek to middle\n"); 151 #endif 152 r = nestegg_track_seek(ctx, 0, duration / 2); 153 if (r == 0) { 154 #ifdef DEBUG 155 fprintf(stderr, "middle "); 156 #endif 157 r = nestegg_read_packet(ctx, &pkt); 158 if (r == 1) { 159 nestegg_packet_track(pkt, &track); 160 nestegg_packet_count(pkt, &cnt); 161 nestegg_packet_tstamp(pkt, &tstamp); 162 #ifdef DEBUG 163 fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 164 #endif 165 nestegg_free_packet(pkt); 166 } else { 167 #ifdef DEBUG 168 fprintf(stderr, "middle seek failed\n"); 169 #endif 170 } 171 } 172 173 #ifdef DEBUG 174 fprintf(stderr, "seek to ~end\n"); 175 #endif 176 r = nestegg_track_seek(ctx, 0, duration - (duration / 10)); 177 if (r == 0) { 178 #ifdef DEBUG 179 fprintf(stderr, "end "); 180 #endif 181 r = nestegg_read_packet(ctx, &pkt); 182 if (r == 1) { 183 nestegg_packet_track(pkt, &track); 184 nestegg_packet_count(pkt, &cnt); 185 nestegg_packet_tstamp(pkt, &tstamp); 186 #ifdef DEBUG 187 fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 188 #endif 189 nestegg_free_packet(pkt); 190 } else { 191 #ifdef DEBUG 192 fprintf(stderr, "end seek failed\n"); 193 #endif 194 } 195 } 196 197 #ifdef DEBUG 198 fprintf(stderr, "seek to ~start\n"); 199 #endif 200 r = nestegg_track_seek(ctx, 0, duration / 10); 201 if (r == 0) { 202 #ifdef DEBUG 203 fprintf(stderr, "start "); 204 #endif 205 r = nestegg_read_packet(ctx, &pkt); 206 if (r == 1) { 207 nestegg_packet_track(pkt, &track); 208 nestegg_packet_count(pkt, &cnt); 209 nestegg_packet_tstamp(pkt, &tstamp); 210 #ifdef DEBUG 211 fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 212 #endif 213 nestegg_free_packet(pkt); 214 } else { 215 #ifdef DEBUG 216 fprintf(stderr, "start seek failed\n"); 217 #endif 218 } 219 } 220 #endif 221 222 while (nestegg_read_packet(ctx, &pkt) > 0) { 223 nestegg_packet_track(pkt, &pkt_track); 224 nestegg_packet_count(pkt, &pkt_cnt); 225 nestegg_packet_tstamp(pkt, &pkt_tstamp); 226 227 #ifdef DEBUG 228 fprintf(stderr, "t %u pts %f frames %u: ", pkt_track, pkt_tstamp / 1e9, pkt_cnt); 229 #endif 230 231 for (i = 0; i < pkt_cnt; ++i) { 232 nestegg_packet_data(pkt, i, &ptr, &size); 233 #ifdef DEBUG 234 fprintf(stderr, "%u ", (unsigned int) size); 235 #endif 236 } 237 #ifdef DEBUG 238 fprintf(stderr, "\n"); 239 #endif 240 241 nestegg_free_packet(pkt); 242 } 243 244 nestegg_destroy(ctx); 245 fclose(fp); 246 247 return EXIT_SUCCESS; 248 } 249
