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         if(file->seekable)
  17         {
  18                 mpeg3demux_copy_titles(new_vtrack->demuxer, demuxer);
  19         }
  20         new_vtrack->current_position = 0;
  21         new_vtrack->pid = custom_id;
  22
  23 // Copy pointers
  24         if(file->frame_offsets)
  25         {
  26                 new_vtrack->frame_offsets = file->frame_offsets[number];
  27                 new_vtrack->total_frame_offsets = file->total_frame_offsets[number];
  28                 new_vtrack->keyframe_numbers = file->keyframe_numbers[number];
  29                 new_vtrack->total_keyframe_numbers = file->total_keyframe_numbers[number];
  30                 new_vtrack->demuxer->stream_end = file->video_eof[number];
  31         }
  32
  33 /* Get information about the track here. */
  34         new_vtrack->video = mpeg3video_new(file,
  35                 new_vtrack);
  36         if(!new_vtrack->video)
  37         {
  38 /* Failed */
  39                 mpeg3_delete_vtrack(file, new_vtrack);
  40                 new_vtrack = 0;
  41         }
  42
  43
  44         return new_vtrack;
  45 }
  46
  47 int mpeg3_delete_vtrack(mpeg3_t *file, mpeg3_vtrack_t *vtrack)
  48 {
  49         if(vtrack->video) mpeg3video_delete(vtrack->video);
  50         if(vtrack->demuxer) mpeg3_delete_demuxer(vtrack->demuxer);
  51         if(vtrack->private_offsets)
  52         {
  53                 if(vtrack->frame_offsets) free(vtrack->frame_offsets);
  54                 if(vtrack->keyframe_numbers) free(vtrack->keyframe_numbers);
  55         }
  56         free(vtrack);
  57         return 0;
  58 }
  59
  60 void mpeg3_append_frame(mpeg3_vtrack_t *vtrack, int64_t offset, int is_keyframe)
  61 {
  62         if(vtrack->total_frame_offsets >= vtrack->frame_offsets_allocated)
  63         {
  64                 vtrack->frame_offsets_allocated =
  65                         MAX(vtrack->total_frame_offsets * 2, 1024);
  66                 vtrack->frame_offsets = realloc(vtrack->frame_offsets,
  67                         sizeof(int64_t) * vtrack->frame_offsets_allocated);
  68         }
  69
  70         vtrack->frame_offsets[vtrack->total_frame_offsets++] = offset;
  71
  72         if(is_keyframe)
  73         {
  74                 if(vtrack->total_keyframe_numbers >= vtrack->keyframe_numbers_allocated)
  75                 {
  76                         vtrack->keyframe_numbers_allocated =
  77                                 MAX(vtrack->total_keyframe_numbers * 2, 1024);
  78                         vtrack->keyframe_numbers = realloc(vtrack->keyframe_numbers,
  79                                 sizeof(int64_t) * vtrack->keyframe_numbers_allocated);
  80                 }
  81
  82 // Because the frame offsets are for the frame
  83 // after, this needs to take off one frame.
  84                 int corrected_frame = vtrack->total_frame_offsets - 2;
  85                 if(corrected_frame < 0) corrected_frame = 0;
  86                 vtrack->keyframe_numbers[vtrack->total_keyframe_numbers++] =
  87                         corrected_frame;
  88         }
  89
  90         vtrack->private_offsets = 1;
  91 }
  92
  93
  94
  95
  96
  97