libmpeg3/mpeg3vtrack.c

   1 #include "libmpeg3.h"
   2 #include "mpeg3protos.h"
   3
   4 #include <stdlib.h>
   5
   6 mpeg3_vtrack_t* mpeg3_new_vtrack(mpeg3_t *file,
   7         int custom_id,
   8         mpeg3_demuxer_t *demuxer,
   9         int number)
  10 {
  11         int result = 0;
  12         mpeg3_vtrack_t *new_vtrack;
  13
  14         new_vtrack = calloc(1, sizeof(mpeg3_vtrack_t));
  15         new_vtrack->demuxer = mpeg3_new_demuxer(file, 0, 1, custom_id);
  16         new_vtrack->frame_cache = mpeg3_new_cache();
  17         if(file->seekable)
  18         {
  19                 mpeg3demux_copy_titles(new_vtrack->demuxer, demuxer);
  20         }
  21         new_vtrack->current_position = 0;
  22         new_vtrack->pid = custom_id;
  23
  24 // Copy pointers
  25         if(file->frame_offsets)
  26         {
  27                 new_vtrack->frame_offsets = file->frame_offsets[number];
  28                 new_vtrack->total_frame_offsets = file->total_frame_offsets[number];
  29                 new_vtrack->keyframe_numbers = file->keyframe_numbers[number];
  30                 new_vtrack->total_keyframe_numbers = file->total_keyframe_numbers[number];
  31                 new_vtrack->demuxer->stream_end = file->video_eof[number];
  32         }
  33
  34 /* Get information about the track here. */
  35         new_vtrack->video = mpeg3video_new(file,
  36                 new_vtrack);
  37
  38         if(!new_vtrack->video)
  39         {
  40 /* Failed */
  41                 mpeg3_delete_vtrack(file, new_vtrack);
  42                 new_vtrack = 0;
  43         }
  44
  45
  46         return new_vtrack;
  47 }
  48
  49 int mpeg3_delete_vtrack(mpeg3_t *file, mpeg3_vtrack_t *vtrack)
  50 {
  51         if(vtrack->video) mpeg3video_delete(vtrack->video);
  52         if(vtrack->demuxer) mpeg3_delete_demuxer(vtrack->demuxer);
  53         if(vtrack->private_offsets)
  54         {
  55                 if(vtrack->frame_offsets) free(vtrack->frame_offsets);
  56                 if(vtrack->keyframe_numbers) free(vtrack->keyframe_numbers);
  57         }
  58         mpeg3_delete_cache(vtrack->frame_cache);
  59
  60         int i;
  61         for(i = 0; i < vtrack->total_subtitles; i++)
  62         {
  63                 mpeg3_delete_subtitle(vtrack->subtitles[i]);
  64         }
  65         if(vtrack->subtitles) free(vtrack->subtitles);
  66         free(vtrack);
  67         return 0;
  68 }
  69
  70 static int last_keyframe = 0;
  71 void mpeg3_append_frame(mpeg3_vtrack_t *vtrack, int64_t offset, int is_keyframe)
  72 {
  73         if(vtrack->total_frame_offsets >= vtrack->frame_offsets_allocated)
  74         {
  75                 vtrack->frame_offsets_allocated =
  76                         MAX(vtrack->total_frame_offsets * 2, 1024);
  77                 vtrack->frame_offsets = realloc(vtrack->frame_offsets,
  78                         sizeof(int64_t) * vtrack->frame_offsets_allocated);
  79         }
  80
  81         vtrack->frame_offsets[vtrack->total_frame_offsets++] = offset;
  82
  83         if(is_keyframe)
  84         {
  85                 if(vtrack->total_keyframe_numbers >= vtrack->keyframe_numbers_allocated)
  86                 {
  87                         vtrack->keyframe_numbers_allocated =
  88                                 MAX(vtrack->total_keyframe_numbers * 2, 1024);
  89                         vtrack->keyframe_numbers = realloc(vtrack->keyframe_numbers,
  90                                 sizeof(int64_t) * vtrack->keyframe_numbers_allocated);
  91                 }
  92
  93 // Because the frame offsets are for the frame
  94 // after, this needs to take off one frame.
  95                 int corrected_frame = vtrack->total_frame_offsets - 2;
  96                 if(corrected_frame < 0) corrected_frame = 0;
  97                 vtrack->keyframe_numbers[vtrack->total_keyframe_numbers++] =
  98                         corrected_frame;
  99         }
 100
 101         vtrack->private_offsets = 1;
 102 }
 103
 104
 105
 106
 107
 108