view filmstripenc.c @ 5895:d8031a05ceec libavformat

Fix updating condition for the probe_size variable in the internal loop of ff_probe_input_buffer(), making sure that probe_size is always set to probe_max_size in the last iteration. Also make the function return an error if we get to the max probe length and still cannot figure out what the format is. Patch by Micah Galizia micahgalizia A gmail D com.
author stefano
date Fri, 26 Mar 2010 01:12:14 +0000
parents ba6b1ac2bab6
children 178de7695c6c
line wrap: on
line source

/*
 * Adobe Filmstrip muxer
 * Copyright (c) 2010 Peter Ross
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file libavformat/filmstripenc.c
 * Adobe Filmstrip muxer
 */

#include "libavutil/intreadwrite.h"
#include "avformat.h"

#define RAND_TAG MKBETAG('R','a','n','d')

typedef struct {
    int nb_frames;
} FilmstripMuxContext;

static int write_header(AVFormatContext *s)
{
    if (s->streams[0]->codec->pix_fmt != PIX_FMT_RGBA) {
        av_log(s, AV_LOG_ERROR, "only PIX_FMT_RGBA is supported\n");
        return AVERROR_INVALIDDATA;
    }
    return 0;
}

static int write_packet(AVFormatContext *s, AVPacket *pkt)
{
    FilmstripMuxContext *film = s->priv_data;
    put_buffer(s->pb, pkt->data, pkt->size);
    film->nb_frames++;
    return 0;
}

static int write_trailer(AVFormatContext *s)
{
    FilmstripMuxContext *film = s->priv_data;
    ByteIOContext *pb = s->pb;
    AVStream *st = s->streams[0];
    int i;

    put_be32(pb, RAND_TAG);
    put_be32(pb, film->nb_frames);
    put_be16(pb, 0);  // packing method
    put_be16(pb, 0);  // reserved
    put_be16(pb, st->codec->width);
    put_be16(pb, st->codec->height);
    put_be16(pb, 0);  // leading
    put_be16(pb, 1/av_q2d(st->codec->time_base));
    for (i = 0; i < 16; i++)
        put_byte(pb, 0x00);  // reserved
    put_flush_packet(pb);
    return 0;
}

AVOutputFormat filmstrip_muxer = {
    "filmstrip",
    NULL_IF_CONFIG_SMALL("Adobe Filmstrip"),
    NULL,
    "flm",
    sizeof(FilmstripMuxContext),
    CODEC_ID_NONE,
    CODEC_ID_RAWVIDEO,
    write_header,
    write_packet,
    write_trailer,
};