hvirtual/mplexlo/mplex.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4
   5 #include "libmpeg3.h"
   6 #include "mpeg3protos.h"
   7
   8 #define PACKET_SIZE 2048
   9
  10 typedef struct
  11 {
  12         int derivative;
  13         unsigned char packet_buffer[PACKET_SIZE];
  14         FILE *out_file;
  15 } multiplexer_t;
  16
  17 typedef struct
  18 {
  19         long bytes_decoded;
  20         long frames_decoded;
  21         long samples_decoded;
  22         mpeg3_t *file;
  23         FILE *raw_file;
  24         int stream_number;
  25         int end_of_data;
  26 } track_t;
  27
  28 int write_pack_header(unsigned char *ptr,
  29                 multiplexer_t *mplex,
  30                 float seconds,
  31                 int stream_id,
  32                 int ac3)
  33 {
  34         int packet_length;
  35 // PACK START CODE
  36         *ptr++ = 0x00;
  37         *ptr++ = 0x00;
  38         *ptr++ = 0x01;
  39         *ptr++ = 0xba;
  40
  41         if(mplex->derivative == 1)
  42         {
  43                 unsigned long timestamp = (unsigned long)(seconds * 90000);
  44                 *ptr = 0x20;
  45                 *ptr++ |= ((timestamp & 0xc0000000) >> 29) | 1;
  46                 *ptr++ =  (timestamp & 0x3fc00000) >> 22;
  47                 *ptr++ =  ((timestamp & 0x003f8000) >> 14) | 1;
  48                 *ptr++ =  (timestamp & 0x00007f80) >> 7;
  49                 *ptr++ =  ((timestamp & 0x0000007f) << 1) | 1;
  50
  51                 *ptr++ = 0;
  52                 *ptr++ = 0;
  53                 *ptr++ = 0;
  54         }
  55         else
  56         if(mplex->derivative == 2)
  57         {
  58                 *ptr = 0x40;
  59         }
  60
  61         *ptr++ = 0x00;
  62         *ptr++ = 0x00;
  63         *ptr++ = 0x01;
  64         if(ac3)
  65                 *ptr++ = 0xbd;
  66         else
  67                 *ptr++ = stream_id;
  68
  69 // Packet length
  70         packet_length = PACKET_SIZE - (ptr - mplex->packet_buffer) - 2;
  71         *ptr++ = (packet_length & 0xff00) >> 8;
  72         *ptr++ = packet_length & 0xff;
  73
  74 // Pts/dts flags
  75         *ptr++ = 0x0f;
  76
  77 // AC3 stream_id
  78         if(ac3)
  79         {
  80                 *ptr++ = stream_id;
  81                 *ptr++ = 0x00;
  82                 *ptr++ = 0x00;
  83                 *ptr++ = 0x00;
  84         }
  85
  86         return PACKET_SIZE - (ptr - mplex->packet_buffer);
  87 }
  88
  89 int write_packet(track_t *track,
  90                 float start_time,
  91                 float end_time,
  92                 multiplexer_t *mplex,
  93                 int stream_id,
  94                 int ac3)
  95 {
  96         int result = 0;
  97
  98         long bytes_needed = track->bytes_decoded - ftell(track->raw_file);
  99         long current_byte = 0;
 100
 101         while(current_byte < bytes_needed)
 102         {
 103 // Write packet header
 104                 float current_time = start_time + (float)current_byte / bytes_needed * (end_time - start_time);
 105                 int packet_bytes = write_pack_header(mplex->packet_buffer,
 106                         mplex,
 107                         current_time,
 108                         stream_id,
 109                         ac3);
 110                 unsigned char *ptr = mplex->packet_buffer + PACKET_SIZE - packet_bytes;
 111                 int bytes_read = fread(ptr, 1, packet_bytes, track->raw_file);
 112
 113                 result = !fwrite(mplex->packet_buffer, PACKET_SIZE, 1, mplex->out_file);
 114                 current_byte += packet_bytes;
 115         }
 116         return result;
 117 }
 118
 119 void skip_frame(mpeg3_demuxer_t *demuxer)
 120 {
 121         unsigned long header = 0;
 122         mpeg3_title_t *title = demuxer->titles[demuxer->current_title];
 123
 124         do{
 125                 header <<= 8;
 126                 header &= 0xffffffff;
 127                 header |= mpeg3io_read_char(title->fs);
 128         }while(header != MPEG3_PICTURE_START_CODE && !mpeg3io_eof(title->fs));
 129
 130 /*
 131  *      if(!mpeg3io_eof(title->fs))
 132  *              mpeg3io_seek_relative(title->fs, -4);
 133  */
 134         ((mpeg3_t*)demuxer->file)->last_type_read = 2;
 135 }
 136
 137 int main(int argc, char *argv[])
 138 {
 139         int streams = 0;
 140         int stream = 0;
 141         int result = 0;
 142         track_t *atracks[streams + 1];
 143         track_t *vtracks[streams + 1];
 144         int total_atracks = 0;
 145         int total_vtracks = 0;
 146         float frame_rate = 30000.0 / 1001.0;
 147         float sample_rate = 48000;
 148         long frames_decoded = 0;
 149         float *audio_temp = malloc(1);
 150         long audio_temp_allocated = 0;
 151         float current_time = 0, previous_time = 0;
 152         int video_completed = 0;
 153         int audio_completed = 0;
 154         int i;
 155         int ac3 = 0;
 156         multiplexer_t mplex;
 157         char **path;
 158         char *output_path;
 159         int old_percentage = 0;
 160
 161         path = malloc(sizeof(char*) * argc);
 162         mplex.derivative = 1;
 163         mplex.out_file = 0;
 164
 165         if(argc < 4)
 166         {
 167                 printf("Tiny MPLEX by Adam Williams\n");
 168                 printf("Usage: mplex [-a] <stream 1> <stream2> ... <output>\n");
 169                 printf("        -a use ac3 packet headers\n");
 170                 exit(1);
 171         }
 172
 173         for(i = 1; i < argc; i++)
 174         {
 175                 if(!strcmp(argv[i], "-a"))
 176                 {
 177                         ac3 = 1;
 178                 }
 179                 else
 180                 if(i == argc - 1)
 181                 {
 182                         output_path = argv[i];
 183                 }
 184                 else
 185                 {
 186                         path[stream] = malloc(strlen(argv[i]) + 1);
 187                         strcpy(path[stream], argv[i]);
 188                         streams++;
 189                         stream++;
 190                 }
 191         }
 192
 193 // Open files
 194         for(stream = 0; stream < streams; stream++)
 195         {
 196                 int is_audio, is_video;
 197                 mpeg3_t *file = mpeg3_open(path[stream]);
 198
 199                 if(!file)
 200                 {
 201                         printf("Couldn't open %s\n", path[stream]);
 202                         result = 1;
 203                 }
 204                 else
 205                 {
 206                         is_audio = mpeg3_has_audio(file);
 207                         is_video = mpeg3_has_video(file);
 208                         mpeg3_set_cpus(file, 2);
 209
 210                         if(is_audio && is_video)
 211                         {
 212                                 printf("%s: Can't multiplex a system stream.\n", path[stream]);
 213                                 result = 1;
 214                         }
 215                         else
 216                         if(is_audio)
 217                         {
 218                                 atracks[total_atracks] = calloc(1, sizeof(track_t));
 219                                 atracks[total_atracks]->file = file;
 220                                 sample_rate = mpeg3_sample_rate(file, 0);
 221                                 atracks[total_atracks]->raw_file = fopen(path[stream], "rb");
 222                                 atracks[total_atracks]->stream_number = total_atracks;
 223                                 total_atracks++;
 224                         }
 225                         else
 226                         if(is_video)
 227                         {
 228                                 vtracks[total_vtracks] = calloc(1, sizeof(track_t));
 229                                 vtracks[total_vtracks]->file = file;
 230                                 frame_rate = mpeg3_frame_rate(file, 0);
 231                                 vtracks[total_vtracks]->raw_file = fopen(path[stream], "rb");
 232                                 vtracks[total_vtracks]->stream_number = total_vtracks;
 233 // Get the first frame
 234                                 skip_frame(vtracks[total_vtracks]->file->vtrack[0]->demuxer);
 235                                 total_vtracks++;
 236                         }
 237                         else
 238                         {
 239                                 printf("%s: Has neither audio or video.\n", path[stream]);
 240                                 result = 1;
 241                         }
 242                 }
 243         }
 244
 245         if(!result)
 246         {
 247                 if(!total_vtracks)
 248                 {
 249                         printf("You must supply at least 1 video track.\n");
 250                         result = 1;
 251                 }
 252         }
 253
 254 // Open output
 255         if(!result)
 256                 if(!(mplex.out_file = fopen(output_path, "wb")))
 257                 {
 258                         printf("Couldn't open output file\n");
 259                         result = 1;
 260                 }
 261
 262 // Write multiplexed stream
 263         if(!result)
 264         {
 265                 while(!result && !(video_completed && audio_completed))
 266                 {
 267                         previous_time = current_time;
 268 // Want the frame to come before the audio that goes with it.
 269                         for(stream = 0; stream < total_vtracks && !result; stream++)
 270                         {
 271 // Decode a frame and write it
 272                                 track_t *track = vtracks[stream];
 273
 274                                 if(!track->end_of_data)
 275                                 {
 276                                         int percentage;
 277                                         skip_frame(track->file->vtrack[0]->demuxer);
 278                                         track->frames_decoded++;
 279                                         track->bytes_decoded = mpeg3demux_tell(track->file->vtrack[0]->demuxer);
 280                                         if(track->frames_decoded > frames_decoded)
 281                                         {
 282                                                 frames_decoded = track->frames_decoded;
 283                                                 current_time = (float)frames_decoded / frame_rate;
 284                                         }
 285
 286                                         result = write_packet(track,
 287                                                                         previous_time,
 288                                                                         current_time,
 289                                                                         &mplex,
 290                                                                         0xe0 | track->stream_number,
 291                                                                         0);
 292                                     track->end_of_data = mpeg3_end_of_video(track->file, 0);
 293                                         percentage = (int)(mpeg3_tell_percentage(track->file) * 100);
 294                                         if(percentage - old_percentage >= 1)
 295                                         {
 296                                                 printf("\t%d%% completed\r", percentage);
 297                                                 old_percentage = percentage;
 298                                         }
 299                                         fflush(stdout);
 300                                 }
 301                         }
 302
 303 // Decode audio until the last frame is covered
 304                         for(stream = 0; stream < total_atracks && !result; stream++)
 305                         {
 306                                 track_t *track = atracks[stream];
 307 //printf("mplex decode audio 1\n");
 308                                 if(!track->end_of_data &&
 309                                         (track->samples_decoded < current_time * sample_rate ||
 310                                                 video_completed))
 311                                 {
 312                                         if(!video_completed)
 313                                         {
 314                                                 long samples_needed = (long)(current_time * sample_rate) - track->samples_decoded;
 315                                                 if(audio_temp_allocated < samples_needed)
 316                                                 {
 317                                                         free(audio_temp);
 318                                                         audio_temp = malloc(sizeof(float) * samples_needed);
 319                                                         audio_temp_allocated = samples_needed;
 320                                                 }
 321
 322                                                 mpeg3_read_audio(track->file,
 323                                                         audio_temp,
 324                                                         0,
 325                                                         0,
 326                                                         samples_needed,
 327                                                         0);
 328                                                 track->bytes_decoded = mpeg3demux_tell(track->file->atrack[0]->demuxer);
 329                                                 if(!track->end_of_data) track->bytes_decoded -= 2048;
 330                                                 track->samples_decoded += samples_needed;
 331                                                 track->end_of_data = mpeg3_end_of_audio(track->file, 0);
 332                                         }
 333                                         else
 334                                         {
 335                                                 track->bytes_decoded = mpeg3demuxer_total_bytes(track->file->atrack[0]->demuxer);
 336                                                 track->end_of_data = 1;
 337                                         }
 338
 339 //printf("mplex decode audio 2\n");
 340                                         result = write_packet(track,
 341                                                 previous_time,
 342                                                 current_time,
 343                                                 &mplex,
 344                                                 ac3 ? track->stream_number : (0xc0 | track->stream_number),
 345                                                 ac3);
 346                                 }
 347                         }
 348
 349                         for(stream = 0; stream < total_vtracks; stream++)
 350                         {
 351                                 if(vtracks[stream]->end_of_data) video_completed++;
 352                         }
 353                         if(video_completed < total_vtracks) video_completed = 0;
 354
 355                         for(stream = 0; stream < total_atracks; stream++)
 356                         {
 357                                 if(atracks[stream]->end_of_data) audio_completed++;
 358                         }
 359                         if(audio_completed < total_atracks) audio_completed = 0;
 360                 }
 361         }
 362
 363 // Close streams
 364         for(stream = 0; stream < total_atracks; stream++)
 365         {
 366                 mpeg3_close(atracks[stream]->file);
 367                 fclose(atracks[stream]->raw_file);
 368                 free(atracks[stream]);
 369         }
 370
 371         for(stream = 0; stream < total_vtracks; stream++)
 372         {
 373                 mpeg3_close(vtracks[stream]->file);
 374                 fclose(vtracks[stream]->raw_file);
 375                 free(vtracks[stream]);
 376         }
 377
 378         if(mplex.out_file) fclose(mplex.out_file);
 379
 380         return result;
 381 }
 382
 383