summaryrefslogtreecommitdiffstats
path: root/libmpdemux/muxer_lavf.c
diff options
context:
space:
mode:
Diffstat (limited to 'libmpdemux/muxer_lavf.c')
-rw-r--r--libmpdemux/muxer_lavf.c408
1 files changed, 0 insertions, 408 deletions
diff --git a/libmpdemux/muxer_lavf.c b/libmpdemux/muxer_lavf.c
deleted file mode 100644
index a4113c7d9a..0000000000
--- a/libmpdemux/muxer_lavf.c
+++ /dev/null
@@ -1,408 +0,0 @@
-/*
- * This file is part of MPlayer.
- *
- * MPlayer is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * MPlayer 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License along
- * with MPlayer; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <sys/types.h>
-#include <inttypes.h>
-#include <limits.h>
-#include "config.h"
-#include "mp_msg.h"
-
-#include "aviheader.h"
-#include "ms_hdr.h"
-#include "av_opts.h"
-
-#include "stream/stream.h"
-#include "muxer.h"
-#include "demuxer.h"
-#include "stheader.h"
-#include "m_option.h"
-#include "libavformat/avformat.h"
-#include "libavutil/avstring.h"
-
-#include "mp_taglists.h"
-
-enum PixelFormat imgfmt2pixfmt(int fmt);
-
-extern char *info_name;
-extern char *info_artist;
-extern char *info_genre;
-extern char *info_subject;
-extern char *info_copyright;
-extern char *info_sourceform;
-extern char *info_comment;
-
-#define BIO_BUFFER_SIZE 32768
-
-typedef struct {
- //AVInputFormat *avif;
- AVFormatContext *oc;
- ByteIOContext *pb;
- int audio_streams;
- int video_streams;
- int64_t last_pts;
- uint8_t buffer[BIO_BUFFER_SIZE];
-} muxer_priv_t;
-
-typedef struct {
- int64_t last_pts;
- AVStream *avstream;
-} muxer_stream_priv_t;
-
-static char *conf_format = NULL;
-static int mux_rate= 0;
-static int mux_packet_size= 0;
-static float mux_preload= 0.5;
-static float mux_max_delay= 0.7;
-static char *mux_avopt = NULL;
-
-const m_option_t lavfopts_conf[] = {
- {"format", &(conf_format), CONF_TYPE_STRING, 0, 0, 0, NULL},
- {"muxrate", &mux_rate, CONF_TYPE_INT, CONF_RANGE, 0, INT_MAX, NULL},
- {"packetsize", &mux_packet_size, CONF_TYPE_INT, CONF_RANGE, 0, INT_MAX, NULL},
- {"preload", &mux_preload, CONF_TYPE_FLOAT, CONF_RANGE, 0, INT_MAX, NULL},
- {"delay", &mux_max_delay, CONF_TYPE_FLOAT, CONF_RANGE, 0, INT_MAX, NULL},
- {"o", &mux_avopt, CONF_TYPE_STRING, 0, 0, 0, NULL},
-
- {NULL, NULL, 0, 0, 0, 0, NULL}
-};
-
-static int mp_write(void *opaque, uint8_t *buf, int size)
-{
- muxer_t *muxer = opaque;
- return stream_write_buffer(muxer->stream, buf, size);
-}
-
-static int64_t mp_seek(void *opaque, int64_t pos, int whence)
-{
- muxer_t *muxer = opaque;
- if(whence == SEEK_CUR)
- {
- off_t cur = stream_tell(muxer->stream);
- if(cur == -1)
- return -1;
- pos += cur;
- }
- else if(whence == SEEK_END)
- {
- off_t size=0;
- if(stream_control(muxer->stream, STREAM_CTRL_GET_SIZE, &size) == STREAM_UNSUPPORTED || size < pos)
- return -1;
- pos = size - pos;
- }
- mp_msg(MSGT_MUXER, MSGL_DBG2, "SEEK %"PRIu64"\n", (int64_t)pos);
- if(!stream_seek(muxer->stream, pos))
- return -1;
- return 0;
-}
-
-
-static muxer_stream_t* lavf_new_stream(muxer_t *muxer, int type)
-{
- muxer_priv_t *priv = muxer->priv;
- muxer_stream_t *stream;
- muxer_stream_priv_t *spriv;
- AVCodecContext *ctx;
-
- if(!muxer || (type != MUXER_TYPE_VIDEO && type != MUXER_TYPE_AUDIO))
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "UNKNOWN TYPE %d\n", type);
- return NULL;
- }
-
- stream = calloc(1, sizeof(muxer_stream_t));
- if(!stream)
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "Could not allocate muxer_stream, EXIT.\n");
- return NULL;
- }
- muxer->streams[muxer->avih.dwStreams] = stream;
- stream->b_buffer = malloc(2048);
- if(!stream->b_buffer)
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "Could not allocate b_buffer, EXIT.\n");
- free(stream);
- return NULL;
- }
- stream->b_buffer_size = 2048;
- stream->b_buffer_ptr = 0;
- stream->b_buffer_len = 0;
-
- spriv = calloc(1, sizeof(muxer_stream_priv_t));
- if(!spriv)
- {
- free(stream);
- return NULL;
- }
- stream->priv = spriv;
-
- spriv->avstream = av_new_stream(priv->oc, 1);
- if(!spriv->avstream)
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "Could not allocate avstream, EXIT.\n");
- return NULL;
- }
- spriv->avstream->stream_copy = 1;
-
- ctx = spriv->avstream->codec;
- ctx->codec_id = muxer->avih.dwStreams;
- switch(type)
- {
- case MUXER_TYPE_VIDEO:
- ctx->codec_type = CODEC_TYPE_VIDEO;
- break;
- case MUXER_TYPE_AUDIO:
- ctx->codec_type = CODEC_TYPE_AUDIO;
- break;
- }
-
- muxer->avih.dwStreams++;
- stream->muxer = muxer;
- stream->type = type;
- mp_msg(MSGT_MUXER, MSGL_V, "ALLOCATED STREAM N. %d, type=%d\n", muxer->avih.dwStreams, type);
- return stream;
-}
-
-
-static void fix_parameters(muxer_stream_t *stream)
-{
- muxer_stream_priv_t *spriv = (muxer_stream_priv_t *) stream->priv;
- AVCodecContext *ctx;
-
- ctx = spriv->avstream->codec;
-
- ctx->bit_rate= stream->avg_rate;
- if(stream->wf && stream->wf->nAvgBytesPerSec && !ctx->bit_rate)
- ctx->bit_rate = stream->wf->nAvgBytesPerSec * 8;
- ctx->rc_buffer_size= stream->vbv_size;
- ctx->rc_max_rate= stream->max_rate;
-
- if(stream->type == MUXER_TYPE_AUDIO)
- {
- ctx->codec_id = mp_av_codec_get_id(mp_wav_taglists, stream->wf->wFormatTag);
-#if 0 //breaks aac in mov at least
- ctx->codec_tag = codec_get_wav_tag(ctx->codec_id);
-#endif
- mp_msg(MSGT_MUXER, MSGL_INFO, "AUDIO CODEC ID: %x, TAG: %x\n", ctx->codec_id, (uint32_t) ctx->codec_tag);
- ctx->sample_rate = stream->wf->nSamplesPerSec;
-// mp_msg(MSGT_MUXER, MSGL_INFO, "stream->h.dwSampleSize: %d\n", stream->h.dwSampleSize);
- ctx->channels = stream->wf->nChannels;
- if(stream->h.dwRate && (stream->h.dwScale * (int64_t)ctx->sample_rate) % stream->h.dwRate == 0)
- ctx->frame_size= (stream->h.dwScale * (int64_t)ctx->sample_rate) / stream->h.dwRate;
- mp_msg(MSGT_MUXER, MSGL_V, "MUXER_LAVF(audio stream) frame_size: %d, scale: %u, sps: %u, rate: %u, ctx->block_align = stream->wf->nBlockAlign; %d=%d stream->wf->nAvgBytesPerSec:%d\n",
- ctx->frame_size, stream->h.dwScale, ctx->sample_rate, stream->h.dwRate,
- ctx->block_align, stream->wf->nBlockAlign, stream->wf->nAvgBytesPerSec);
- ctx->block_align = stream->h.dwSampleSize;
- if(stream->wf+1 && stream->wf->cbSize)
- {
- ctx->extradata = av_malloc(stream->wf->cbSize);
- if(ctx->extradata != NULL)
- {
- ctx->extradata_size = stream->wf->cbSize;
- memcpy(ctx->extradata, stream->wf+1, ctx->extradata_size);
- }
- else
- mp_msg(MSGT_MUXER, MSGL_ERR, "MUXER_LAVF(audio stream) error! Could not allocate %d bytes for extradata.\n",
- stream->wf->cbSize);
- }
- }
- else if(stream->type == MUXER_TYPE_VIDEO)
- {
- ctx->codec_id = mp_av_codec_get_id(mp_bmp_taglists, stream->bih->biCompression);
- if(ctx->codec_id <= 0 || force_fourcc)
- ctx->codec_tag= stream->bih->biCompression;
- mp_msg(MSGT_MUXER, MSGL_INFO, "VIDEO CODEC ID: %d\n", ctx->codec_id);
- if (stream->imgfmt)
- ctx->pix_fmt = imgfmt2pixfmt(stream->imgfmt);
- ctx->width = stream->bih->biWidth;
- ctx->height = stream->bih->biHeight;
- ctx->bit_rate = 800000;
- ctx->time_base.den = stream->h.dwRate;
- ctx->time_base.num = stream->h.dwScale;
- if(stream->bih+1 && (stream->bih->biSize > sizeof(BITMAPINFOHEADER)))
- {
- ctx->extradata_size = stream->bih->biSize - sizeof(BITMAPINFOHEADER);
- ctx->extradata = av_malloc(ctx->extradata_size);
- if(ctx->extradata != NULL)
- memcpy(ctx->extradata, stream->bih+1, ctx->extradata_size);
- else
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "MUXER_LAVF(video stream) error! Could not allocate %d bytes for extradata.\n",
- ctx->extradata_size);
- ctx->extradata_size = 0;
- }
- }
- }
-}
-
-static void write_chunk(muxer_stream_t *stream, size_t len, unsigned int flags, double dts, double pts)
-{
- muxer_t *muxer = (muxer_t*) stream->muxer;
- muxer_priv_t *priv = (muxer_priv_t *) muxer->priv;
- muxer_stream_priv_t *spriv = (muxer_stream_priv_t *) stream->priv;
- AVPacket pkt;
-
- if(len)
- {
- av_init_packet(&pkt);
- pkt.size = len;
- pkt.stream_index= spriv->avstream->index;
- pkt.data = stream->buffer;
-
- if(flags & AVIIF_KEYFRAME)
- pkt.flags |= PKT_FLAG_KEY;
- else
- pkt.flags = 0;
-
- pkt.dts = (dts / av_q2d(priv->oc->streams[pkt.stream_index]->time_base) + 0.5);
- pkt.pts = (pts / av_q2d(priv->oc->streams[pkt.stream_index]->time_base) + 0.5);
-//fprintf(stderr, "%Ld %Ld id:%d tb:%f %f\n", pkt.dts, pkt.pts, pkt.stream_index, av_q2d(priv->oc->streams[pkt.stream_index]->time_base), stream->timer);
-
- if(av_interleaved_write_frame(priv->oc, &pkt) != 0) //av_write_frame(priv->oc, &pkt)
- {
- mp_msg(MSGT_MUXER, MSGL_ERR, "Error while writing frame.\n");
- }
- }
-
- return;
-}
-
-
-static void write_header(muxer_t *muxer)
-{
- muxer_priv_t *priv = (muxer_priv_t *) muxer->priv;
-
- mp_tmsg(MSGT_MUXER, MSGL_INFO, "Writing header...\n");
- av_write_header(priv->oc);
- muxer->cont_write_header = NULL;
-}
-
-
-static void write_trailer(muxer_t *muxer)
-{
- int i;
- muxer_priv_t *priv = (muxer_priv_t *) muxer->priv;
-
- mp_tmsg(MSGT_MUXER, MSGL_INFO, "Writing index...\n");
- av_write_trailer(priv->oc);
- for(i = 0; i < priv->oc->nb_streams; i++)
- {
- av_freep(&(priv->oc->streams[i]));
- }
-
- av_freep(&priv->oc->pb);
-
- av_free(priv->oc);
-}
-
-static void list_formats(void) {
- AVOutputFormat *fmt;
- mp_msg(MSGT_DEMUX, MSGL_INFO, "Available lavf output formats:\n");
- for (fmt = first_oformat; fmt; fmt = fmt->next)
- mp_msg(MSGT_DEMUX, MSGL_INFO, "%15s : %s\n", fmt->name, fmt->long_name);
-}
-
-extern char *out_filename;
-int muxer_init_muxer_lavf(muxer_t *muxer)
-{
- muxer_priv_t *priv;
- AVOutputFormat *fmt = NULL;
-
- av_register_all();
-
- if (conf_format && strcmp(conf_format, "help") == 0) {
- list_formats();
- return 0;
- }
-
- mp_msg(MSGT_MUXER, MSGL_WARN, "** MUXER_LAVF *****************************************************************\n");
- mp_msg(MSGT_MUXER, MSGL_WARN,
-"REMEMBER: MEncoder's libavformat muxing is presently broken and can generate\n"
-"INCORRECT files in the presence of B-frames. Moreover, due to bugs MPlayer\n"
-"will play these INCORRECT files as if nothing were wrong!\n"
-"*******************************************************************************\n");
-
- priv = calloc(1, sizeof(muxer_priv_t));
- if(priv == NULL)
- return 0;
-
- priv->oc = av_alloc_format_context();
- if(!priv->oc)
- {
- mp_msg(MSGT_MUXER, MSGL_FATAL, "Could not get format context.\n");
- goto fail;
- }
-
- if(conf_format)
- fmt = guess_format(conf_format, NULL, NULL);
- if(! fmt)
- fmt = guess_format(NULL, out_filename, NULL);
- if(! fmt)
- {
- mp_msg(MSGT_MUXER, MSGL_FATAL, "Cannot get specified format.\n");
- goto fail;
- }
- priv->oc->oformat = fmt;
-
-
- if(av_set_parameters(priv->oc, NULL) < 0)
- {
- mp_msg(MSGT_MUXER, MSGL_FATAL, "invalid output format parameters\n");
- goto fail;
- }
- priv->oc->packet_size= mux_packet_size;
- priv->oc->mux_rate= mux_rate;
- priv->oc->preload= (int)(mux_preload*AV_TIME_BASE);
- priv->oc->max_delay= (int)(mux_max_delay*AV_TIME_BASE);
- if (info_name)
- av_strlcpy(priv->oc->title , info_name, sizeof(priv->oc->title ));
- if (info_artist)
- av_strlcpy(priv->oc->author , info_artist, sizeof(priv->oc->author ));
- if (info_genre)
- av_strlcpy(priv->oc->genre , info_genre, sizeof(priv->oc->genre ));
- if (info_copyright)
- av_strlcpy(priv->oc->copyright, info_copyright, sizeof(priv->oc->copyright));
- if (info_comment)
- av_strlcpy(priv->oc->comment , info_comment, sizeof(priv->oc->comment ));
-
- if(mux_avopt){
- if(parse_avopts(priv->oc, mux_avopt) < 0){
- mp_msg(MSGT_MUXER,MSGL_ERR, "Your options /%s/ look like gibberish to me pal.\n", mux_avopt);
- goto fail;
- }
- }
-
- priv->oc->pb = av_alloc_put_byte(priv->buffer, BIO_BUFFER_SIZE, 1, muxer, NULL, mp_write, mp_seek);
- if ((muxer->stream->flags & MP_STREAM_SEEK) != MP_STREAM_SEEK)
- priv->oc->pb->is_streamed = 1;
-
- muxer->priv = (void *) priv;
- muxer->cont_new_stream = &lavf_new_stream;
- muxer->cont_write_chunk = &write_chunk;
- muxer->cont_write_header = &write_header;
- muxer->cont_write_index = &write_trailer;
- muxer->fix_stream_parameters = &fix_parameters;
- mp_msg(MSGT_MUXER, MSGL_INFO, "OK, exit.\n");
- return 1;
-
-fail:
- free(priv);
- return 0;
-}