1/* Copyright (c) 2013-2015 Jeffrey Pfau
  2 *
  3 * This Source Code Form is subject to the terms of the Mozilla Public
  4 * License, v. 2.0. If a copy of the MPL was not distributed with this
  5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
  6#ifndef FFMPEG_ENCODER
  7#define FFMPEG_ENCODER
  8
  9#include <mgba-util/common.h>
 10
 11CXX_GUARD_START
 12
 13#include <mgba/internal/gba/gba.h>
 14
 15#include <libavformat/avformat.h>
 16#include <libavcodec/version.h>
 17
 18// Version 57.16 in FFmpeg
 19#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 37, 100)
 20#define FFMPEG_USE_PACKETS
 21#endif
 22
 23// Version 57.15 in libav
 24#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 35, 0)
 25#define FFMPEG_USE_NEW_BSF
 26#endif
 27
 28// Version 57.14 in libav
 29#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 48, 0)
 30#define FFMPEG_USE_CODECPAR
 31#endif
 32
 33#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 8, 0)
 34#define FFMPEG_USE_PACKET_UNREF
 35#endif
 36
 37struct FFmpegEncoder {
 38	struct mAVStream d;
 39	struct AVFormatContext* context;
 40
 41	unsigned audioBitrate;
 42	const char* audioCodec;
 43
 44	unsigned videoBitrate;
 45	const char* videoCodec;
 46
 47	const char* containerFormat;
 48
 49	struct AVCodecContext* audio;
 50	enum AVSampleFormat sampleFormat;
 51	int sampleRate;
 52	uint16_t* audioBuffer;
 53	size_t audioBufferSize;
 54	uint16_t* postaudioBuffer;
 55	size_t postaudioBufferSize;
 56	AVFrame* audioFrame;
 57	size_t currentAudioSample;
 58	int64_t currentAudioFrame;
 59	int64_t nextAudioPts; // TODO (0.6): Remove
 60#ifdef USE_LIBAVRESAMPLE
 61	struct AVAudioResampleContext* resampleContext;
 62#else
 63	struct SwrContext* resampleContext;
 64#endif
 65#ifdef FFMPEG_USE_NEW_BSF
 66	struct AVBSFContext* absf; // Needed for AAC in MP4
 67#else
 68	struct AVBitStreamFilterContext* absf; // Needed for AAC in MP4
 69#endif
 70	struct AVStream* audioStream;
 71
 72	struct AVCodecContext* video;
 73	enum AVPixelFormat pixFormat;
 74	struct AVFrame* videoFrame;
 75	int width;
 76	int height;
 77	int iwidth;
 78	int iheight;
 79	unsigned frameCycles;
 80	unsigned cycles;
 81	int64_t currentVideoFrame;
 82	struct SwsContext* scaleContext;
 83	struct AVStream* videoStream;
 84};
 85
 86void FFmpegEncoderInit(struct FFmpegEncoder*);
 87bool FFmpegEncoderSetAudio(struct FFmpegEncoder*, const char* acodec, unsigned abr);
 88bool FFmpegEncoderSetVideo(struct FFmpegEncoder*, const char* vcodec, unsigned vbr);
 89bool FFmpegEncoderSetContainer(struct FFmpegEncoder*, const char* container);
 90void FFmpegEncoderSetDimensions(struct FFmpegEncoder*, int width, int height);
 91bool FFmpegEncoderVerifyContainer(struct FFmpegEncoder*);
 92bool FFmpegEncoderOpen(struct FFmpegEncoder*, const char* outfile);
 93void FFmpegEncoderClose(struct FFmpegEncoder*);
 94bool FFmpegEncoderIsOpen(struct FFmpegEncoder*);
 95
 96void FFmpegEncoderSetInputFrameRate(struct FFmpegEncoder*, unsigned numerator, unsigned denominator);
 97
 98CXX_GUARD_END
 99
100#endif