hvirtual/libmpeg3/mpeg3vtrack.c

Go to the documentation of this file.

#include "libmpeg3.h"
#include "mpeg3protos.h"

#include <stdlib.h>

mpeg3_vtrack_t* mpeg3_new_vtrack(mpeg3_t *file, 
        int custom_id, 
        mpeg3_demuxer_t *demuxer,
        int number)
{
        int result = 0;
        mpeg3_vtrack_t *new_vtrack;

        new_vtrack = calloc(1, sizeof(mpeg3_vtrack_t));
        new_vtrack->demuxer = mpeg3_new_demuxer(file, 0, 1, custom_id);
        new_vtrack->frame_cache = mpeg3_new_cache();
        if(file->seekable)
        {
                mpeg3demux_copy_titles(new_vtrack->demuxer, demuxer);
        }
        new_vtrack->current_position = 0;
        new_vtrack->pid = custom_id;

// Copy pointers
        if(file->frame_offsets)
        {
                new_vtrack->frame_offsets = file->frame_offsets[number];
                new_vtrack->total_frame_offsets = file->total_frame_offsets[number];
                new_vtrack->keyframe_numbers = file->keyframe_numbers[number];
                new_vtrack->total_keyframe_numbers = file->total_keyframe_numbers[number];
                new_vtrack->demuxer->stream_end = file->video_eof[number];
        }

/* Get information about the track here. */
        new_vtrack->video = mpeg3video_new(file, 
                new_vtrack);

        if(!new_vtrack->video)
        {
/* Failed */
                mpeg3_delete_vtrack(file, new_vtrack);
                new_vtrack = 0;
        }


        return new_vtrack;
}

int mpeg3_delete_vtrack(mpeg3_t *file, mpeg3_vtrack_t *vtrack)
{
        if(vtrack->video) mpeg3video_delete(vtrack->video);
        if(vtrack->demuxer) mpeg3_delete_demuxer(vtrack->demuxer);
        if(vtrack->private_offsets)
        {
                if(vtrack->frame_offsets) free(vtrack->frame_offsets);
                if(vtrack->keyframe_numbers) free(vtrack->keyframe_numbers);
        }
        mpeg3_delete_cache(vtrack->frame_cache);

        int i;
        for(i = 0; i < vtrack->total_subtitles; i++)
        {
                mpeg3_delete_subtitle(vtrack->subtitles[i]);
        }
        if(vtrack->subtitles) free(vtrack->subtitles);
        free(vtrack);
        return 0;
}

static int last_keyframe = 0;
void mpeg3_append_frame(mpeg3_vtrack_t *vtrack, int64_t offset, int is_keyframe)
{
        if(vtrack->total_frame_offsets >= vtrack->frame_offsets_allocated)
        {
                vtrack->frame_offsets_allocated = 
                        MAX(vtrack->total_frame_offsets * 2, 1024);
                vtrack->frame_offsets = realloc(vtrack->frame_offsets,
                        sizeof(int64_t) * vtrack->frame_offsets_allocated);
        }

        vtrack->frame_offsets[vtrack->total_frame_offsets++] = offset;

        if(is_keyframe)
        {
                if(vtrack->total_keyframe_numbers >= vtrack->keyframe_numbers_allocated)
                {
                        vtrack->keyframe_numbers_allocated = 
                                MAX(vtrack->total_keyframe_numbers * 2, 1024);
                        vtrack->keyframe_numbers = realloc(vtrack->keyframe_numbers,
                                sizeof(int64_t) * vtrack->keyframe_numbers_allocated);
                }

// Because the frame offsets are for the frame
// after, this needs to take off one frame.
                int corrected_frame = vtrack->total_frame_offsets - 2;
                if(corrected_frame < 0) corrected_frame = 0;
                vtrack->keyframe_numbers[vtrack->total_keyframe_numbers++] = 
                        corrected_frame;
        }

        vtrack->private_offsets = 1;
}

---