libmpeg3/mpeg3dump.c

   1 #include "libmpeg3.h"
   2 #include "mpeg3protos.h"
   3 #include <stdlib.h>
   4 #include <string.h>
   5
   6 #define BUFSIZE 65536
   7
   8 void test_32bit_overflow(char *outfile, int (*out_counter), FILE *(*out))
   9 {
  10         if(ftell(*out) > 0x7f000000)
  11         {
  12                 char string[1024];
  13                 fclose(*out);
  14                 sprintf(string, "%s%03d", outfile, ++(*out_counter));
  15                 (*out) = fopen(string, "wb");
  16         }
  17 }
  18
  19 int main(int argc, char *argv[])
  20 {
  21         mpeg3_t *file;
  22         int i, j, result = 0;
  23         unsigned char *output, **output_rows;
  24         int out_counter = 0;
  25         float *audio_output_f;
  26         unsigned char *audio_output_i;
  27         long total_samples = 0;
  28         FILE *out;
  29         char outfile[1024];
  30         int decompress_audio = 0, decompress_video = 0;
  31         int audio_track = 0;
  32 /* Print cell offsets */
  33         int print_offsets = 0;
  34         int print_pids = 1;
  35
  36         outfile[0] = 0;
  37         if(argc < 2)
  38         {
  39                 printf(
  40 "Dump information or extract audio to a 24 bit pcm file.\n"
  41 "Example: dump -a0 outputfile.pcm take1.vob\n"
  42                 );
  43                 exit(1);
  44         }
  45
  46         for(i = 1; i < argc; i++)
  47         {
  48                 if(!strncmp(argv[i], "-a", 2))
  49                 {
  50 // Check for track number
  51                         if(strlen(argv[i]) > 2)
  52                         {
  53                                 audio_track = atol(argv[i] + 2);
  54                         }
  55
  56 // Check for filename
  57                         if(i + 1 < argc)
  58                         {
  59                                 strcpy(outfile, argv[++i]);
  60                                 decompress_audio = 1;
  61                         }
  62                         else
  63                         {
  64                                 fprintf(stderr, "-a must be paired with a filename.\n");
  65                                 exit(1);
  66                         }
  67
  68 // Check if file exists
  69                         if(out = fopen(outfile, "r"))
  70                         {
  71                                 fprintf(stderr, "%s exists.\n", outfile);
  72                                 exit(1);
  73                         }
  74                 }
  75         }
  76
  77         file = mpeg3_open(argv[argc - 1]);
  78         if(outfile[0])
  79         {
  80                 out = fopen(outfile, "wb");
  81         }
  82
  83
  84
  85
  86
  87
  88
  89
  90         if(file)
  91         {
  92
  93 // Audio streams
  94                 fprintf(stderr, "total_astreams=%d\n", mpeg3_total_astreams(file));
  95
  96                 for(i = 0; i < mpeg3_total_astreams(file); i++)
  97                 {
  98                         fprintf(stderr, "  Stream 0x%04x: channels=%d rate=%d samples=%ld format=%s\n",
  99                                 file->atrack[i]->demuxer->astream,
 100                                 mpeg3_audio_channels(file, i),
 101                                 mpeg3_sample_rate(file, i),
 102                                 mpeg3_audio_samples(file, i),
 103                                 mpeg3_audio_format(file, i));
 104
 105                         if(print_offsets)
 106                         {
 107                                 fprintf(stderr, "total_sample_offsets=%d\n", file->atrack[i]->total_sample_offsets);
 108                                 for(j = 0; j < file->atrack[i]->total_sample_offsets; j++)
 109                                 {
 110                                         fprintf(stderr, "%llx ", file->atrack[i]->sample_offsets[j]);
 111                                         if(j > 0 && !(j % 8)) fprintf(stderr, "\n");
 112                                 }
 113                                 fprintf(stderr, "\n");
 114                         }
 115                 }
 116
 117
 118
 119 // Video streams
 120                 fprintf(stderr, "total_vstreams=%d\n", mpeg3_total_vstreams(file));
 121
 122                 for(i = 0; i < mpeg3_total_vstreams(file); i++)
 123                 {
 124                         fprintf(stderr, "  Stream 0x%04x: w=%d h=%d framerate=%0.3f frames=%ld coding=%s\n",
 125                                 file->vtrack[i]->demuxer->vstream,
 126                                 mpeg3_video_width(file, i),
 127                                 mpeg3_video_height(file, i),
 128                                 mpeg3_frame_rate(file, i),
 129                                 mpeg3_video_frames(file, i),
 130                                 mpeg3_colormodel(file, i) == MPEG3_YUV420P ? "420" : "422");
 131
 132                         if(print_offsets)
 133                         {
 134                                 fprintf(stderr, "total_frame_offsets=%d\n", file->vtrack[i]->total_frame_offsets);
 135                                 for(j = 0; j < file->vtrack[i]->total_frame_offsets; j++)
 136                                 {
 137                                         fprintf(stderr, "%d=%llx ", j, file->vtrack[i]->frame_offsets[j]);
 138                                         if(j > 0 && !(j % 8)) fprintf(stderr, "\n");
 139                                 }
 140                                 fprintf(stderr, "\n");
 141
 142                                 fprintf(stderr, "total_keyframe_numbers=%d\n", file->vtrack[i]->total_keyframe_numbers);
 143                                 for(j = 0; j < file->vtrack[i]->total_keyframe_numbers; j++)
 144                                 {
 145                                         fprintf(stderr, "%lld ", file->vtrack[i]->keyframe_numbers[j]);
 146                                         if(j > 0 && !(j % 8)) fprintf(stderr, "\n");
 147                                 }
 148                                 fprintf(stderr, "\n");
 149                         }
 150                 }
 151
 152
 153 // Titles
 154                 fprintf(stderr, "total_titles=%d\n", file->demuxer->total_titles);
 155                 for(i = 0; i < file->demuxer->total_titles; i++)
 156                 {
 157                         fprintf(stderr, "  Title path=%s total_bytes=%llx cell_table_size=%d\n",
 158                                 file->demuxer->titles[i]->fs->path,
 159                                 file->demuxer->titles[i]->total_bytes,
 160                                 file->demuxer->titles[i]->cell_table_size);
 161
 162                         if(print_offsets)
 163                         {
 164                                 for(j = 0; j < file->demuxer->titles[i]->cell_table_size; j++)
 165                                         fprintf(stderr, "    Cell: %llx-%llx %llx-%llx program=%d\n",
 166                                                 file->demuxer->titles[i]->cell_table[j].program_start,
 167                                                 file->demuxer->titles[i]->cell_table[j].program_end,
 168                                                 file->demuxer->titles[i]->cell_table[j].title_start,
 169                                                 file->demuxer->titles[i]->cell_table[j].title_end,
 170                                                 file->demuxer->titles[i]->cell_table[j].program);
 171                         }
 172                 }
 173
 174
 175
 176 // Pids
 177                 if(print_pids)
 178                 {
 179                         mpeg3_demuxer_t *demuxer = file->demuxer;
 180                         printf("Total PIDs=%d\n", demuxer->total_pids);
 181                         for(i = 0; i < demuxer->total_pids; i++)
 182                         {
 183                                 printf("0x%04x ", demuxer->pid_table[i]);
 184                         }
 185                         printf("\n");
 186                 }
 187
 188
 189
 190 // Write audio
 191                 if(decompress_audio)
 192                 {
 193                         mpeg3_set_cpus(file, 2);
 194                         audio_output_f = malloc(BUFSIZE * sizeof(float));
 195                         audio_output_i = malloc(BUFSIZE * 3 * mpeg3_audio_channels(file, audio_track));
 196
 197 //printf("%d\n", mpeg3_end_of_audio(file, audio_track));
 198                         while(!mpeg3_end_of_audio(file, audio_track) && !result)
 199                         {
 200                                 test_32bit_overflow(outfile, &out_counter, &out);
 201
 202                                 for(i = 0; i < mpeg3_audio_channels(file, audio_track); i++)
 203                                 {
 204                                         if(i == 0)
 205                                                 result = mpeg3_read_audio(file,
 206                                                         audio_output_f,
 207                                                         0,
 208                                                         i,
 209                                                         BUFSIZE,
 210                                                         audio_track);
 211                                         else
 212                                                 result = mpeg3_reread_audio(file,
 213                                                         audio_output_f,      /* Pointer to pre-allocated buffer of floats */
 214                                                         0,      /* Pointer to pre-allocated buffer of int16's */
 215                                                         i,          /* Channel to decode */
 216                                                         BUFSIZE,         /* Number of samples to decode */
 217                                                         audio_track);
 218
 219                                         for(j = 0; j < BUFSIZE; j++)
 220                                         {
 221                                                 int sample = audio_output_f[j] * 0x7fffff;
 222                                                 unsigned char *output_i = audio_output_i + j * 3 * mpeg3_audio_channels(file, audio_track) + i * 3;
 223                                                 if(sample > 0x7fffff)
 224                                                         sample = 0x7fffff;
 225                                                 else
 226                                                 if(sample < -0x7fffff)
 227                                                         sample = -0x7fffff;
 228                                                 *output_i++ = (sample & 0xff0000) >> 16;
 229                                                 *output_i++ = (sample & 0xff00) >> 8;
 230                                                 *output_i = sample & 0xff;
 231                                         }
 232
 233                                 }
 234
 235                                 result = !fwrite(audio_output_i, BUFSIZE * 3 * mpeg3_audio_channels(file, audio_track), 1, out);
 236                         }
 237                 }
 238
 239 /*
 240  *              audio_output_i = malloc(BUFSIZE * 2 * mpeg3_audio_channels(file, 0));
 241  *              mpeg3_seek_percentage(file, 0.1);
 242  *              result = mpeg3_read_audio(file, 0, audio_output_i, 1, BUFSIZE, 0);
 243  */
 244
 245 /*
 246  *              output = malloc(mpeg3_video_width(file, 0) * mpeg3_video_height(file, 0) * 3 + 4);
 247  *              output_rows = malloc(sizeof(unsigned char*) * mpeg3_video_height(file, 0));
 248  *              for(i = 0; i < mpeg3_video_height(file, 0); i++)
 249  *                      output_rows[i] = &output[i * mpeg3_video_width(file, 0) * 3];
 250  * printf("dump 1\n");
 251  *              mpeg3_seek_percentage(file, 0.375);
 252  *              result = mpeg3_read_frame(file,
 253  *                                      output_rows,
 254  *                                      0,
 255  *                                      0,
 256  *                                      mpeg3_video_width(file, 0),
 257  *                                      mpeg3_video_height(file, 0),
 258  *                                      mpeg3_video_width(file, 0),
 259  *                                      mpeg3_video_height(file, 0),
 260  *                                      MPEG3_RGB888,
 261  *                                      0);
 262  * printf("dump 2\n");
 263  */
 264
 265                 mpeg3_close(file);
 266         }
 267         return 0;
 268 }