summaryrefslogtreecommitdiff
path: root/libavwrapper
diff options
context:
space:
mode:
authorTim Redfern <tim@herge.(none)>2013-05-18 17:34:53 +0100
committerTim Redfern <tim@herge.(none)>2013-05-18 17:34:53 +0100
commitbcd9eeed1f60c594da11d93c1b52c4ec192280ce (patch)
treeaf760f3e2a1cefc3abcc177f9b5c75a5722c7d67 /libavwrapper
parentfad2c7236e70faf0669dde3e6cbd51986fdbc6ee (diff)
dependencies installer
Diffstat (limited to 'libavwrapper')
-rw-r--r--libavwrapper/Makefile251
-rwxr-xr-xlibavwrapper/libavwrapper.cpp755
-rwxr-xr-xlibavwrapper/libavwrapper.h150
-rwxr-xr-xlibavwrapper/unit_test.cpp14
-rwxr-xr-xlibavwrapper/unit_test.h3
5 files changed, 1173 insertions, 0 deletions
diff --git a/libavwrapper/Makefile b/libavwrapper/Makefile
new file mode 100644
index 0000000..ded20b6
--- /dev/null
+++ b/libavwrapper/Makefile
@@ -0,0 +1,251 @@
+# The pre-processor and compiler options.
+
+#http://docs.gstreamer.com/display/GstSDK/Installing+on+Linux
+
+#MY_CFLAGS = -fpermissive -std=c++11 -Wno-error -I /opt/gstreamer-sdk/include/gstreamer-0.10/ -I /opt/gstreamer-sdk/include/glib-2.0 -I /opt/gstreamer-sdk/lib/glib-2.0/include -I /opt/gstreamer-sdk/include/libxml2 $(shell pkg-config gstreamer-0.10 gstreamer-video-0.10 gstreamer-base-0.10 --cflags)
+#MY_CFLAGS = -fpermissive -std=c++11 -Wno-error $(shell pkg-config gstreamer-0.10 gstreamer-video-0.10 gstreamer-base-0.10 --cflags)
+
+# -I ../ffmpeg
+
+# The linker options.libgstaasinklibgstaasink.so
+MY_LIBS = -L /usr/local/lib -lswscale -lavcodec -lavformat -lavfilter -lavdevice -lavutil
+#MY_LIBS = -lPocoNet -lPocoXML -lPocoUtil -lPocoFoundation -lvamp-hostsdk -lsndfile -L /usr/local/lib -lswscale -lavcodec -lavformat -lavfilter -lavdevice -lavutil $(shell pkg-config gstreamer-0.10 gstreamer-video-0.10 gstreamer-base-0.10 --libs)
+# -lgstreamer-0.10 -lgstreamer-video-0.10 -lgstreamer-base-0.10 -lglib-2.0 -lgstapp-0.10
+#MY_LIBS = ../libavcodec/ffmpeg/libavcodec/libavcodec.a ../libavcodec/ffmpeg/libavutil/libavutil.a ../libavcodec/ffmpeg/libavformat/libavformat.a ../libavcodec/ffmpeg/libavfilter/libavfilter.a ../libavcodec/ffmpeg/libavdevice/libavdevice.a -lPocoNet -lPocoXML -lPocoUtil -lPocoFoundation -lvamp-hostsdk
+#GAH! HARD!
+
+# The pre-processor options used by the cpp (man cpp for more).
+CPPFLAGS = -Wall
+
+# The options used in linking as well as in any direct use of ld.
+LDFLAGS =
+
+# The directories in which source files reside.
+# If not specified, only the current directory will be serached.
+SRCDIRS =
+
+# The executable file name.
+# If not specified, current directory name or `a.out' will be used.
+PROGRAM =
+
+## Implicit Section: change the following only when necessary.
+##==========================================================================
+
+# The source file types (headers excluded).
+# .c indicates C source files, and others C++ ones.
+SRCEXTS = .c .C .cc .cpp .CPP .c++ .cxx .cp
+
+# The header file types.
+HDREXTS = .h .H .hh .hpp .HPP .h++ .hxx .hp
+
+# The pre-processor and compiler options.
+# Users can override those variables from the command line.
+CFLAGS = -g -O2
+CXXFLAGS=
+
+# The C program compiler.
+#CC = gcc
+
+# The C++ program compiler.
+#CXX = g++
+
+# Un-comment the following line to compile C programs as C++ ones.
+#CC = $(CXX)
+
+# The command used to delete file.
+RM = rm -f
+
+ETAGS = etags
+ETAGSFLAGS =
+
+CTAGS = ctags
+CTAGSFLAGS =
+
+## Stable Section: usually no need to be changed. But you can add more.
+##==========================================================================
+SHELL = /bin/sh
+EMPTY =
+SPACE = $(EMPTY) $(EMPTY)
+ifeq ($(PROGRAM),)
+ CUR_PATH_NAMES = $(subst /,$(SPACE),$(subst $(SPACE),_,$(CURDIR)))
+ PROGRAM = $(word $(words $(CUR_PATH_NAMES)),$(CUR_PATH_NAMES))
+ ifeq ($(PROGRAM),)
+ PROGRAM = a.out
+ endif
+endif
+ifeq ($(SRCDIRS),)
+ SRCDIRS = .
+endif
+SOURCES = $(foreach d,$(SRCDIRS),$(wildcard $(addprefix $(d)/*,$(SRCEXTS))))
+HEADERS = $(foreach d,$(SRCDIRS),$(wildcard $(addprefix $(d)/*,$(HDREXTS))))
+SRC_CXX = $(filter-out %.c,$(SOURCES))
+OBJS = $(addsuffix .o, $(basename $(SOURCES)))
+DEPS = $(OBJS:.o=.d)
+
+## Define some useful variables.
+DEP_OPT = $(shell if `$(CC) --version | grep "GCC" >/dev/null`; then \
+ echo "-MM -MP"; else echo "-M"; fi )
+DEPEND = $(CC) $(DEP_OPT) $(MY_CFLAGS) $(CFLAGS) $(CPPFLAGS)
+DEPEND.d = $(subst -g ,,$(DEPEND))
+COMPILE.c = $(CC) $(MY_CFLAGS) $(CFLAGS) $(CPPFLAGS) -c
+COMPILE.cxx = $(CXX) $(MY_CFLAGS) $(CXXFLAGS) $(CPPFLAGS) -c
+LINK.c = $(CC) $(MY_CFLAGS) $(CFLAGS) $(CPPFLAGS) $(LDFLAGS)
+LINK.cxx = $(CXX) $(MY_CFLAGS) $(CXXFLAGS) $(CPPFLAGS) $(LDFLAGS)
+
+.PHONY: all objs tags ctags clean distclean help show install
+
+# Delete the default suffixes
+.SUFFIXES:
+
+all: Release
+
+Clang: CXX = clang
+
+Clang: $(PROGRAM)
+
+Release: CXXFLAGS += -O2
+
+Release: $(PROGRAM)
+
+Debug: CXXFLAGS += -g3
+
+Debug: $(PROGRAM)
+
+prefix=/usr/local
+
+Install: rotord
+ strip rotord
+ install -m 0755 rotord $(prefix)/bin
+
+
+# Rules for creating dependency files (.d).
+#------------------------------------------
+
+%.d:%.c
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.C
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.cc
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.cpp
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.CPP
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.c++
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.cp
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+%.d:%.cxx
+ @echo -n $(dir $<) > $@
+ @$(DEPEND.d) $< >> $@
+
+# Rules for generating object files (.o).
+#----------------------------------------
+objs:$(OBJS)
+
+%.o:%.c
+ $(COMPILE.c) $< -o $@
+
+%.o:%.C
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.cc
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.cpp
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.CPP
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.c++
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.cp
+ $(COMPILE.cxx) $< -o $@
+
+%.o:%.cxx
+ $(COMPILE.cxx) $< -o $@
+
+# Rules for generating the tags.
+#-------------------------------------
+tags: $(HEADERS) $(SOURCES)
+ $(ETAGS) $(ETAGSFLAGS) $(HEADERS) $(SOURCES)
+
+ctags: $(HEADERS) $(SOURCES)
+ $(CTAGS) $(CTAGSFLAGS) $(HEADERS) $(SOURCES)
+
+# Rules for generating the executable.
+#-------------------------------------
+$(PROGRAM):$(OBJS)
+ifeq ($(SRC_CXX),) # C program
+ $(LINK.c) $(OBJS) $(MY_LIBS) -o $@
+ @echo Type ./$@ to execute the program.
+else # C++ program
+ $(LINK.cxx) $(OBJS) $(MY_LIBS) -o $@
+ @echo Type ./$@ to execute the program.
+endif
+
+ifndef NODEP
+ifneq ($(DEPS),)
+ sinclude $(DEPS)
+endif
+endif
+
+
+clean:
+ $(RM) $(OBJS) $(PROGRAM) $(PROGRAM).exe
+
+distclean: clean
+ $(RM) $(DEPS) TAGS
+
+# Show help.
+help:
+ @echo 'Generic Makefile for C/C++ Programs (gcmakefile) version 0.5'
+ @echo 'Copyright (C) 2007, 2008 whyglinux <whyglinux@hotmail.com>'
+ @echo
+ @echo 'Usage: make [TARGET]'
+ @echo 'TARGETS:'
+ @echo ' all (=make) compile and link.'
+ @echo ' NODEP=yes make without generating dependencies.'
+ @echo ' objs compile only (no linking).'
+ @echo ' tags create tags for Emacs editor.'
+ @echo ' ctags create ctags for VI editor.'
+ @echo ' clean clean objects and the executable file.'
+ @echo ' distclean clean objects, the executable and dependencies.'
+ @echo ' show show variables (for debug use only).'
+ @echo ' help print this message.'
+ @echo
+ @echo 'Report bugs to <whyglinux AT gmail DOT com>.'
+
+# Show variables (for debug use only.)
+show:
+ @echo 'PROGRAM :' $(PROGRAM)
+ @echo 'SRCDIRS :' $(SRCDIRS)
+ @echo 'HEADERS :' $(HEADERS)
+ @echo 'SOURCES :' $(SOURCES)
+ @echo 'SRC_CXX :' $(SRC_CXX)
+ @echo 'OBJS :' $(OBJS)
+ @echo 'DEPS :' $(DEPS)
+ @echo 'DEPEND :' $(DEPEND)
+ @echo 'COMPILE.c :' $(COMPILE.c)
+ @echo 'COMPILE.cxx :' $(COMPILE.cxx)
+ @echo 'link.c :' $(LINK.c)
+ @echo 'link.cxx :' $(LINK.cxx)
+
+## End of the Makefile ## Suggestions are welcome ## All rights reserved ##
+#############################################################################
diff --git a/libavwrapper/libavwrapper.cpp b/libavwrapper/libavwrapper.cpp
new file mode 100755
index 0000000..5e1a50d
--- /dev/null
+++ b/libavwrapper/libavwrapper.cpp
@@ -0,0 +1,755 @@
+ #include "libavwrapper.h"
+
+
+extern "C"
+{
+#include <libswscale/swscale.h>
+}
+
+/*
+#include <QNetworkReply>
+#include <QNetworkRequest>
+#include <QEventLoop>
+#include <QFileInfo>
+#include <QMutexLocker>
+#include <QDebug>
+*/
+
+#include <stdexcept>
+#include <iostream>
+#include <cassert>
+
+using namespace std;
+
+// Translated to C++ by Christopher Bruns May 2012
+// from ffmeg_adapt.c in whisk package by Nathan Clack, Mark Bolstadt, Michael Meeuwisse
+
+//QMutex decoder::mutex;
+
+// Avoid link error on some macs
+#ifdef __APPLE__
+extern "C" {
+#include <stdlib.h>
+#include <errno.h>
+// #include "compiler/compiler.h"
+
+/*
+ * Darwin doesn't have posix_memalign(), provide a private
+ * weak alternative
+ */
+ /*
+int __weak posix_memalign(void **ptr, size_t align, size_t size)
+{
+ if (*ptr)
+ return 0;
+
+ return ENOMEM;
+}
+*/
+}
+#endif
+
+// Custom read function so FFMPEG does not need to read from a local file by name.
+// But rather from a stream derived from a URL or whatever.
+extern "C" {
+
+int readFunction(void* opaque, uint8_t* buf, int buf_size)
+{
+ //QIODevice* stream = (QIODevice*)opaque;
+ ifstream* stream = (ifstream*)opaque;
+ //int numBytes =
+ stream->read((char*)buf, (streamsize)buf_size);
+ return stream->gcount(); //?? is this right
+ //numBytes; //TODO work out
+}
+
+// http://cdry.wordpress.com/2009/09/09/using-custom-io-callbacks-with-ffmpeg/
+int64_t seekFunction(void* opaque, int64_t offset, int whence)
+{
+ //QIODevice* stream = (QIODevice*)opaque;
+ ifstream* stream = (ifstream*)opaque;
+ if (stream == NULL)
+ return -1;
+ else if (whence == AVSEEK_SIZE)
+ return -1; // "size of my handle in bytes"
+ //else if (stream->isSequential())
+ // return -1; // cannot seek a sequential stream //presume this would be certain kind of network stream
+ else if (whence == SEEK_CUR) { // relative to start of file
+ if (! stream->seekg(offset,ios_base::cur)) //stream->pos() + offset) )
+ return -1;
+ }
+ else if (whence == SEEK_END) { // relative to end of file
+ assert(offset < 0);
+ if (! stream->seekg(offset,ios_base::end)) //stream->size() + offset) )
+ return -1;
+ }
+ else if (whence == SEEK_SET) { // relative to start of file
+ if (! stream->seekg(offset) )
+ return -1;
+ }
+ else {
+ assert(false);
+ }
+ return stream->tellg();
+}
+
+}
+
+
+/////////////////////////////
+// AVPacketWrapper methods //
+/////////////////////////////
+
+
+class AVPacketWrapper
+{
+public:
+ AVPacketWrapper();
+ virtual ~AVPacketWrapper();
+ void free();
+
+ AVPacket packet;
+};
+
+
+AVPacketWrapper::AVPacketWrapper()
+{
+ packet.destruct = NULL;
+}
+
+/* virtual */
+AVPacketWrapper::~AVPacketWrapper()
+{
+ free();
+}
+
+void AVPacketWrapper::free()
+{
+ av_free_packet(&packet);
+}
+
+
+/////////////////////////
+// decoder methods //
+/////////////////////////
+
+libav::decoder::decoder(PixelFormat pixelFormat)
+ : isOpen(false)
+{
+ initialize();
+ format = pixelFormat;
+}
+
+/*
+decoder::decoder(QUrl url, PixelFormat pixelFormat)
+ : isOpen(false)
+{
+ //QMutexLocker lock(&decoder::mutex);
+ initialize();
+ format = pixelFormat;
+ isOpen = open(url, pixelFormat);
+}
+*/
+
+/* virtual */
+libav::decoder::~decoder()
+{
+ //QMutexLocker lock(&decoder::mutex);
+ if (NULL != Sctx) {
+ sws_freeContext(Sctx);
+ Sctx = NULL;
+ }
+ if (NULL != pRaw) {
+ av_free(pRaw);
+ pRaw = NULL;
+ }
+ if (NULL != pFrameRGB) {
+ av_free(pFrameRGB);
+ pFrameRGB = NULL;
+ }
+ if (NULL != pCtx) {
+ avcodec_close(pCtx);
+ pCtx = NULL;
+ }
+ if (NULL != container) {
+ avformat_close_input(&container);
+ container = NULL;
+ }
+ if (NULL != buffer) {
+ av_free(buffer);
+ buffer = NULL;
+ }
+ if (NULL != blank) {
+ av_free(blank);
+ blank = NULL;
+ }
+ /*
+ if (NULL != avioContext) {
+ av_free(avioContext);
+ avioContext = NULL;
+ }
+ */
+ //QNetworkreply
+ //if (reply != NULL) {
+ // reply->deleteLater();
+ // reply = NULL;
+ //}
+ // Don't need to free pCodec?
+}
+/*
+bool decoder::open(QUrl url, enum PixelFormat formatParam)
+{
+ if (url.isEmpty())
+ return false;
+
+ // Is the movie source a local file?
+ if (url.host() == "localhost")
+ url.setHost("");
+ QString fileName = url.toLocalFile();
+ if ( (! fileName.isEmpty())
+ && (QFileInfo(fileName).exists()) )
+ {
+ // return open(fileName, formatParam); // for testing only
+
+ // Yes, the source is a local file
+ fileStream.setFileName(fileName);
+ // qDebug() << fileName;
+ if (! fileStream.open(QIODevice::ReadOnly))
+ return false;
+ return open(fileStream, fileName, formatParam);
+ }
+
+ // ...No, the source is not a local file
+ if (url.host() == "")
+ url.setHost("localhost");
+ fileName = url.path();
+
+ // http://stackoverflow.com/questions/9604633/reading-a-file-located-in-memory-with-libavformat
+ // Load from URL
+ QEventLoop loop; // for synchronous url fetch http://stackoverflow.com/questions/5486090/qnetworkreply-wait-for-finished
+ QObject::connect(&networkManager, SIGNAL(finished(QNetworkReply*)),
+ &loop, SLOT(quit()));
+ QNetworkRequest request = QNetworkRequest(url);
+ // qDebug() << "networkManager" << __FILE__ << __LINE__;
+ reply = networkManager.get(request);
+ loop.exec();
+ if (reply->error() != QNetworkReply::NoError) {
+ // qDebug() << reply->error();
+ reply->deleteLater();
+ reply = NULL;
+ return false;
+ }
+ QIODevice * stream = reply;
+ // Mpeg needs seekable device, so create in-memory buffer if necessary
+ if (stream->isSequential()) {
+ byteArray = stream->readAll();
+ fileBuffer.setBuffer(&byteArray);
+ fileBuffer.open(QIODevice::ReadOnly);
+ if (! fileBuffer.seek(0))
+ return false;
+ stream = &fileBuffer;
+ assert(! stream->isSequential());
+ }
+ bool result = open(*stream, fileName, formatParam);
+ return result;
+}
+
+bool decoder::open(QIODevice& fileStream, QString& fileName, enum PixelFormat formatParam)
+{
+ // http://stackoverflow.com/questions/9604633/reading-a-file-located-in-memory-with-libavformat
+ // I think AVIOContext is the trick used to redivert the input stream
+ ioBuffer = (unsigned char *)av_malloc(ioBufferSize + FF_INPUT_BUFFER_PADDING_SIZE); // can get av_free()ed by libav
+ avioContext = avio_alloc_context(ioBuffer, ioBufferSize, 0, (void*)(&fileStream), &readFunction, NULL, &seekFunction);
+ container = avformat_alloc_context();
+ container->pb = avioContext;
+
+ // Open file, check usability
+ std::string fileNameStd = fileName.toStdString();
+ if (!avtry( avformat_open_input(&container, fileNameStd.c_str(), NULL, NULL), fileNameStd ))
+ return false;
+ return openUsingInitializedContainer(formatParam);
+}
+*/
+// file name based method for historical continuity
+bool libav::decoder::open(char* fileName, enum PixelFormat formatParam){
+
+ if (!avtry( avformat_open_input(&container, fileName, NULL, NULL), string(fileName) ))
+ return false;
+ return openUsingInitializedContainer(formatParam);
+}
+bool libav::decoder::open(string& fileName, enum PixelFormat formatParam)
+{
+ // Open file, check usability
+
+ if (!avtry( avformat_open_input(&container, fileName.c_str(), NULL, NULL), fileName ))
+ return false;
+ return openUsingInitializedContainer(formatParam);
+}
+
+
+bool libav::decoder::openUsingInitializedContainer(enum PixelFormat formatParam)
+{
+ format = formatParam;
+ sc = getNumberOfChannels();
+
+ if (!avtry( avformat_find_stream_info(container, NULL), "Cannot find stream information." ))
+ return false;
+ if (!avtry( videoStream=av_find_best_stream(container, AVMEDIA_TYPE_VIDEO, -1, -1, &pCodec, 0), "Cannot find a video stream." ))
+ return false;
+ pCtx=container->streams[videoStream]->codec;
+ width = pCtx->width;
+ height = pCtx->height;
+ if (!avtry( avcodec_open2(pCtx, pCodec, NULL), "Cannot open video decoder." ))
+ return false;
+
+ /* Frame rate fix for some codecs */
+ if( pCtx->time_base.num > 1000 && pCtx->time_base.den == 1 )
+ pCtx->time_base.den = 1000;
+
+ /* Compute the total number of frames in the file */
+ /* duration is in microsecs */
+ numFrames = (int)(( container->duration / (double)AV_TIME_BASE ) * pCtx->time_base.den + 0.5);
+
+ /* Get framebuffers */
+ if (! (pRaw = avcodec_alloc_frame()) )
+ throw std::runtime_error("");
+ if (! (pFrameRGB = avcodec_alloc_frame()) )
+ throw std::runtime_error("");
+
+ /* Create data buffer */
+ if (format == PIX_FMT_NONE) {
+ numBytes = 0;
+ buffer = NULL;
+ blank = NULL;
+ pFrameRGB = NULL;
+ Sctx = NULL;
+ }
+ else {
+ numBytes = avpicture_get_size( format, pCtx->width, pCtx->height ); // RGB24 format
+ if (! (buffer = (uint8_t*)av_malloc(numBytes + FF_INPUT_BUFFER_PADDING_SIZE)) ) // RGB24 format
+ throw std::runtime_error("");
+ if (! (blank = (uint8_t*)av_mallocz(avpicture_get_size(pCtx->pix_fmt,width,height))) ) // native codec format
+ throw std::runtime_error("");
+
+ /* Init buffers */
+ avpicture_fill( (AVPicture * ) pFrameRGB, buffer, format,
+ pCtx->width, pCtx->height );
+
+ /* Init scale & convert */
+ if (! (Sctx=sws_getContext(
+ pCtx->width,
+ pCtx->height,
+ pCtx->pix_fmt,
+ width,
+ height,
+ format,
+ SWS_POINT, // fastest?
+ NULL,NULL,NULL)) )
+ throw std::runtime_error("");
+ }
+
+ /* Give some info on stderr about the file & stream */
+ //dump_format(container, 0, fname, 0);
+
+ previousFrameIndex = -1;
+ return true;
+}
+
+bool libav::decoder::fetchFrame(int targetFrameIndex)
+{
+ if ((targetFrameIndex < 0) || (targetFrameIndex > numFrames))
+ return false;
+ if (targetFrameIndex == (previousFrameIndex + 1)) {
+ if (! readNextFrame(targetFrameIndex))
+ return false;
+ }
+ else
+ if (seekToFrame(targetFrameIndex) < 0)
+ return false;
+ previousFrameIndex = targetFrameIndex;
+ return true;
+}
+
+// \returns current frame on success, otherwise -1
+int libav::decoder::seekToFrame(int targetFrameIndex)
+{
+ int64_t duration = container->streams[videoStream]->duration;
+ int64_t ts = av_rescale(duration,targetFrameIndex,numFrames);
+ int64_t tol = av_rescale(duration,1,2*numFrames);
+ if ( (targetFrameIndex < 0) || (targetFrameIndex >= numFrames) ) {
+ return -1;
+ }
+ int result = avformat_seek_file( container, //format context
+ videoStream,//stream id
+ 0, //min timestamp
+ ts, //target timestamp
+ ts, //max timestamp
+ 0); //AVSEEK_FLAG_ANY),//flags
+ if (result < 0)
+ return -1;
+
+ avcodec_flush_buffers(pCtx);
+ if (! readNextFrame(targetFrameIndex))
+ return -1;
+
+ return targetFrameIndex;
+}
+
+bool libav::decoder::readNextFrame(int targetFrameIndex)
+{
+ AVPacket packet = {0};
+ av_init_packet(&packet);
+ bool result = readNextFrameWithPacket(targetFrameIndex, packet, pRaw);
+ av_free_packet(&packet);
+ return result;
+}
+
+// WARNING this method can raise an exception
+bool libav::decoder::readNextFrameWithPacket(int targetFrameIndex, AVPacket& packet, AVFrame* pYuv)
+{
+ int finished = 0;
+ do {
+ finished = 0;
+ av_free_packet(&packet);
+ int result;
+ if (!avtry(av_read_frame( container, &packet ), "Failed to read frame"))
+ return false; // !!NOTE: see docs on packet.convergence_duration for proper seeking
+ if( packet.stream_index != videoStream ) /* Is it what we're trying to parse? */
+ continue;
+ if (!avtry(avcodec_decode_video2( pCtx, pYuv, &finished, &packet ), "Failed to decode video"))
+ return false;
+ // handle odd cases and debug
+ if((pCtx->codec_id==CODEC_ID_RAWVIDEO) && !finished)
+ {
+ avpicture_fill( (AVPicture * ) pYuv, blank, pCtx->pix_fmt,width, height ); // set to blank frame
+ finished = 1;
+ }
+#if 0 // very useful for debugging
+ cout << "Packet - pts:" << (int)packet.pts;
+ cout << " dts:" << (int)packet.dts;
+ cout << " - flag: " << packet.flags;
+ cout << " - finished: " << finished;
+ cout << " - Frame pts:" << (int)pYuv->pts;
+ cout << " " << (int)pYuv->best_effort_timestamp;
+ cout << endl;
+ /* printf("Packet - pts:%5d dts:%5d (%5d) - flag: %1d - finished: %3d - Frame pts:%5d %5d\n",
+ (int)packet.pts,(int)packet.dts,
+ packet.flags,finished,
+ (int)pYuv->pts,(int)pYuv->best_effort_timestamp); */
+#endif
+ if(!finished) {
+ if (packet.pts == AV_NOPTS_VALUE)
+ throw std::runtime_error("");
+ if (packet.size == 0) // packet.size==0 usually means EOF
+ break;
+ }
+ } while ( (!finished) || (pYuv->best_effort_timestamp < targetFrameIndex));
+
+ av_free_packet(&packet);
+
+ if (format != PIX_FMT_NONE) {
+ sws_scale(Sctx, // sws context
+ pYuv->data, // src slice
+ pYuv->linesize, // src stride
+ 0, // src slice origin y
+ pCtx->height, // src slice height
+ pFrameRGB->data, // dst
+ pFrameRGB->linesize ); // dst stride
+ }
+
+ previousFrameIndex = targetFrameIndex;
+ return true;
+}
+
+uint8_t libav::decoder::getPixelIntensity(int x, int y, Channel c) const
+{
+ return *(pFrameRGB->data[0] + y * pFrameRGB->linesize[0] + x * sc + c);
+}
+
+int libav::decoder::getNumberOfFrames() const { return numFrames; }
+
+int libav::decoder::getWidth() const { return width; }
+
+int libav::decoder::getHeight() const { return height; }
+
+int libav::decoder::getNumberOfChannels() const
+{
+ switch(format)
+ {
+ case PIX_FMT_BGRA:
+ return 4;
+ break;
+ case PIX_FMT_RGB24:
+ return 3;
+ break;
+ case PIX_FMT_GRAY8:
+ return 1;
+ break;
+ default:
+ return 0;
+ break;
+ }
+ return 0;
+}
+
+void libav::decoder::initialize()
+{
+ Sctx = NULL;
+ pRaw = NULL;
+ pFrameRGB = NULL;
+ pCtx = NULL;
+ container = NULL;
+ buffer = NULL;
+ blank = NULL;
+ pCodec = NULL;
+ format = PIX_FMT_NONE;
+ //network stuff
+ //reply = NULL;
+ //ioBuffer = NULL;
+ //avioContext = NULL;
+ decoder::maybeInitFFMpegLib();
+}
+
+void libav::decoder::maybeInitFFMpegLib()
+{
+ if (decoder::b_is_one_time_inited)
+ return;
+ av_register_all();
+ avcodec_register_all();
+ avformat_network_init();
+ decoder::b_is_one_time_inited = true;
+}
+
+bool libav::decoder::avtry(int result, const std::string& msg) {
+ if ((result < 0) && (result != AVERROR_EOF)) {
+ char buf[1024];
+ av_strerror(result, buf, sizeof(buf));
+ std::string message = std::string("libav::Error: ") + msg + buf;
+ //qDebug() << QString(message.c_str());
+ cerr<<message<<endl;
+ return false;
+ }
+ return true;
+}
+
+bool libav::decoder::b_is_one_time_inited = false;
+
+
+
+///////////////////////////
+// encoder methods //
+///////////////////////////
+
+
+libav::encoder::encoder(const char * file_name, int width, int height, enum AVCodecID codec_id)
+ : picture_yuv(NULL)
+ , picture_rgb(NULL)
+ , container(NULL)
+{
+ if (0 != (width % 2))
+ cerr << "WARNING: Video width is not a multiple of 2" << endl;
+ if (0 != (height % 2))
+ cerr << "WARNING: Video height is not a multiple of 2" << endl;
+
+ decoder::maybeInitFFMpegLib();
+
+ container = avformat_alloc_context();
+ if (NULL == container)
+ throw std::runtime_error("Unable to allocate format context");
+
+ AVOutputFormat * fmt = av_guess_format(NULL, file_name, NULL);
+ if (!fmt)
+ fmt = av_guess_format("mpeg", NULL, NULL);
+ if (!fmt)
+ throw std::runtime_error("Unable to deduce video format");
+ container->oformat = fmt;
+
+ fmt->video_codec = codec_id;
+ // fmt->video_codec = CODEC_ID_H264; // fails to write
+
+ AVStream * video_st = avformat_new_stream(container, NULL);
+
+ pCtx = video_st->codec;
+ pCtx->codec_id = fmt->video_codec;
+ pCtx->codec_type = AVMEDIA_TYPE_VIDEO;
+ // resolution must be a multiple of two
+ pCtx->width = width;
+ pCtx->height = height;
+
+ // bit_rate determines image quality
+ pCtx->bit_rate = width * height * 4; // ?
+ // pCtx->qmax = 50; // no effect?
+
+ // "high quality" parameters from http://www.cs.ait.ac.th/~on/mplayer/pl/menc-feat-enc-libavcodec.html
+ // vcodec=mpeg4:mbd=2:mv0:trell:v4mv:cbp:last_pred=3:predia=2:dia=2:vmax_b_frames=2:vb_strategy=1:precmp=2:cmp=2:subcmp=2:preme=2:vme=5:naq:qns=2
+ if (false) // does not help
+ // if (pCtx->codec_id == CODEC_ID_MPEG4)
+ {
+ pCtx->mb_decision = 2;
+ pCtx->last_predictor_count = 3;
+ pCtx->pre_dia_size = 2;
+ pCtx->dia_size = 2;
+ pCtx->max_b_frames = 2;
+ pCtx->b_frame_strategy = 2;
+ pCtx->trellis = 2;
+ pCtx->compression_level = 2;
+ pCtx->global_quality = 300;
+ pCtx->pre_me = 2;
+ pCtx->mv0_threshold = 1;
+ // pCtx->quantizer_noise_shaping = 2; // deprecated
+ // TODO
+ }
+
+ pCtx->time_base = (AVRational){1, 25};
+ // pCtx->time_base = (AVRational){1, 10};
+ pCtx->gop_size = 12; // emit one intra frame every twelve frames
+ // pCtx->max_b_frames = 0;
+ pCtx->pix_fmt = PIX_FMT_YUV420P;
+ if (fmt->flags & AVFMT_GLOBALHEADER)
+ pCtx->flags |= CODEC_FLAG_GLOBAL_HEADER;
+
+ if (pCtx->codec_id == CODEC_ID_H264)
+ {
+ // http://stackoverflow.com/questions/3553003/encoding-h-264-with-libavcodec-x264
+ pCtx->coder_type = 1; // coder = 1
+ pCtx->flags|=CODEC_FLAG_LOOP_FILTER; // flags=+loop
+ pCtx->me_cmp|= 1; // cmp=+chroma, where CHROMA = 1
+ // pCtx->partitions|=X264_PART_I8X8+X264_PART_I4X4+X264_PART_P8X8+X264_PART_B8X8; // partitions=+parti8x8+parti4x4+partp8x8+partb8x8
+ pCtx->me_method=ME_HEX; // me_method=hex
+ pCtx->me_subpel_quality = 7; // subq=7
+ pCtx->me_range = 16; // me_range=16
+ pCtx->gop_size = 250; // g=250
+ pCtx->keyint_min = 25; // keyint_min=25
+ pCtx->scenechange_threshold = 40; // sc_threshold=40
+ pCtx->i_quant_factor = 0.71; // i_qfactor=0.71
+ pCtx->b_frame_strategy = 1; // b_strategy=1
+ pCtx->qcompress = 0.6; // qcomp=0.6
+ pCtx->qmin = 10; // qmin=10
+ pCtx->qmax = 51; // qmax=51
+ pCtx->max_qdiff = 4; // qdiff=4
+ pCtx->max_b_frames = 3; // bf=3
+ pCtx->refs = 3; // refs=3
+ // pCtx->directpred = 1; // directpred=1
+ pCtx->trellis = 1; // trellis=1
+ // pCtx->flags2|=CODEC_FLAG2_BPYRAMID+CODEC_FLAG2_MIXED_REFS+CODEC_FLAG2_WPRED+CODEC_FLAG2_8X8DCT+CODEC_FLAG2_FASTPSKIP; // flags2=+bpyramid+mixed_refs+wpred+dct8x8+fastpskip
+ // pCtx->weighted_p_pred = 2; // wpredp=2
+ // libx264-main.ffpreset preset
+ // pCtx->flags2|=CODEC_FLAG2_8X8DCT;
+ // pCtx->flags2^=CODEC_FLAG2_8X8DCT; // flags2=-dct8x8
+ }
+
+ AVCodec * codec = avcodec_find_encoder(pCtx->codec_id);
+ if (NULL == codec)
+ throw std::runtime_error("Unable to find Mpeg4 codec");
+ if (codec->pix_fmts)
+ pCtx->pix_fmt = codec->pix_fmts[0];
+ {
+ //QMutexLocker lock(&decoder::mutex);
+ if (avcodec_open2(pCtx, codec, NULL) < 0)
+ throw std::runtime_error("Error opening codec");
+ }
+
+ /* Get framebuffers */
+ if (! (picture_yuv = avcodec_alloc_frame()) ) // final frame format
+ throw std::runtime_error("");
+ if (! (picture_rgb = avcodec_alloc_frame()) ) // rgb version I can understand easily
+ throw std::runtime_error("");
+ /* the image can be allocated by any means and av_image_alloc() is
+ * just the most convenient way if av_malloc() is to be used */
+ if ( av_image_alloc(picture_yuv->data, picture_yuv->linesize,
+ pCtx->width, pCtx->height, pCtx->pix_fmt, 1) < 0 )
+ throw std::runtime_error("Error allocating YUV frame buffer");
+ if ( av_image_alloc(picture_rgb->data, picture_rgb->linesize,
+ pCtx->width, pCtx->height, PIX_FMT_RGB24, 1) < 0 )
+ throw std::runtime_error("Error allocating RGB frame buffer");
+
+ /* Init scale & convert */
+ if (! (Sctx=sws_getContext(
+ width,
+ height,
+ PIX_FMT_RGB24,
+ pCtx->width,
+ pCtx->height,
+ pCtx->pix_fmt,
+ SWS_BICUBIC,NULL,NULL,NULL)) )
+ throw std::runtime_error("");
+
+ /* open the output file */
+ if (!(fmt->flags & AVFMT_NOFILE))
+ {
+ //QMutexLocker lock(&decoder::mutex);
+ if (avio_open(&container->pb, file_name, AVIO_FLAG_WRITE) < 0)
+ throw std::runtime_error("Error opening output video file");
+ }
+ avformat_write_header(container, NULL);
+}
+
+void libav::encoder::setPixelIntensity(int x, int y, int c, uint8_t value)
+{
+ uint8_t * ptr = picture_rgb->data[0] + y * picture_rgb->linesize[0] + x * 3 + c;
+ *ptr = value;
+}
+
+void libav::encoder::write_frame()
+{
+ // convert from RGB24 to YUV
+ sws_scale(Sctx, // sws context
+ picture_rgb->data, // src slice
+ picture_rgb->linesize, // src stride
+ 0, // src slice origin y
+ pCtx->height, // src slice height
+ picture_yuv->data, // dst
+ picture_yuv->linesize ); // dst stride
+
+ /* encode the image */
+ // use non-deprecated avcodec_encode_video2(...)
+ AVPacket packet;
+ av_init_packet(&packet);
+ packet.data = NULL;
+ packet.size = 0;
+
+ int got_packet;
+ int ret = avcodec_encode_video2(pCtx,
+ &packet,
+ picture_yuv,
+ &got_packet);
+ if (ret < 0)
+ throw std::runtime_error("Video encoding failed");
+ if (got_packet)
+ {
+ // std::cout << "encoding frame" << std::endl;
+ int result = av_write_frame(container, &packet);
+ av_destruct_packet(&packet);
+ }
+}
+
+/* virtual */
+libav::encoder::~encoder()
+{
+ int result = av_write_frame(container, NULL); // flush
+ result = av_write_trailer(container);
+ {
+ //QMutexLocker lock(&decoder::mutex);
+ avio_close(container->pb);
+ }
+ for (int i = 0; i < container->nb_streams; ++i)
+ av_freep(container->streams[i]);
+ av_free(container);
+ container = NULL;
+
+ {
+ //QMutexLocker lock(&decoder::mutex);
+ avcodec_close(pCtx);
+ }
+ av_free(pCtx);
+ pCtx = NULL;
+ av_free(picture_yuv->data[0]);
+ av_free(picture_yuv);
+ picture_yuv = NULL;
+ av_free(picture_rgb->data[0]);
+ av_free(picture_rgb);
+ picture_rgb = NULL;
+}
+
+
+
diff --git a/libavwrapper/libavwrapper.h b/libavwrapper/libavwrapper.h
new file mode 100755
index 0000000..449c2f2
--- /dev/null
+++ b/libavwrapper/libavwrapper.h
@@ -0,0 +1,150 @@
+ #ifndef libavwrapper_H
+#define libavwrapper_H
+
+/*
+ * libavwrapper.h
+ * May 2012 Christopher Bruns
+ * The libavwrapper class is a C++ wrapper around the poorly documented
+ * libavcodec movie API used by ffmpeg. I made extensive use of Nathan
+ * Clack's implemention in the whisk project.
+ *
+ * The libavwrapper.h and libavwrapper.cpp files depend only on the libavcodec
+ * and allied sets of libraries. To compartmentalize and reduce dependencies
+ * I placed the Vaa3d specific use of this class into a separate set of
+ * source files: loadV3dFFMpeg.h/cpp
+ */
+
+
+#ifndef UINT64_C
+#define UINT64_C(c) (c ## ULL)
+#endif
+
+
+
+extern "C" {
+#include <libavcodec/avcodec.h>
+#include <libavformat/avformat.h>
+#include <libavutil/pixfmt.h>
+#include <libavutil/opt.h>
+#include <libavutil/imgutils.h>
+
+#include <libswscale/swscale.h> //?
+}
+
+/*
+#include <QFile>
+#include <QNetworkAccessManager>
+#include <QMutex>
+#include <QUrl>
+#include <QBuffer>
+*/
+
+
+#include <string>
+#include <stdexcept>
+#include <iostream>
+#include <fstream>
+
+namespace libav {
+
+// Translated to C++ by Christopher Bruns May 2012
+// from ffmeg_adapt.c in whisk package by Nathan Clack, Mark Bolstadt, Michael Meeuwisse
+ class decoder
+ {
+ public:
+ enum Channel {
+ RED = 0,
+ GRAY = 0,
+ GREEN = 1,
+ BLUE = 2,
+ ALPHA = 3
+ };
+
+ // Some libavcodec calls are not reentrant
+ //static QMutex mutex;
+ static void maybeInitFFMpegLib();
+
+ decoder(PixelFormat pixelFormat=PIX_FMT_RGB24);
+ //decoder(QUrl url, PixelFormat pixelFormat=PIX_FMT_RGB24);
+ virtual ~decoder();
+ //bool open(QUrl url, enum PixelFormat formatParam = PIX_FMT_RGB24);
+ //bool open(QIODevice& fileStream, QString& fileName, enum PixelFormat formatParam = PIX_FMT_RGB24);
+ uint8_t getPixelIntensity(int x, int y, Channel c = GRAY) const;
+ bool fetchFrame(int targetFrameIndex = 0);
+ int getNumberOfFrames() const;
+ int getWidth() const;
+ int getHeight() const;
+ int getNumberOfChannels() const;
+ bool readNextFrame(int targetFrameIndex = 0);
+ bool readNextFrameWithPacket(int targetFrameIndex, AVPacket& packet, AVFrame* pYuv);
+ int seekToFrame(int targetFrameIndex = 0);
+
+ // make certain members public, for use by Fast3DTexture class
+ AVFrame *pFrameRGB;
+ AVFrame *pRaw;
+ AVFormatContext *container;
+ AVCodecContext *pCtx;
+ int videoStream;
+ int previousFrameIndex;
+ bool isOpen;
+
+ bool open(std::string& fileName, enum PixelFormat formatParam = PIX_FMT_RGB24);
+ bool open(char* fileName, enum PixelFormat formatParam = PIX_FMT_RGB24);
+
+ protected:
+ static bool b_is_one_time_inited;
+
+ void initialize();
+
+ bool openUsingInitializedContainer(enum PixelFormat formatParam = PIX_FMT_RGB24 );
+ static bool avtry(int result, const std::string& msg);
+
+ AVCodec *pCodec;
+ uint8_t *buffer,
+ *blank;
+ //struct
+ SwsContext *Sctx;
+ int width, height;
+ PixelFormat format;
+ size_t numBytes;
+ int numFrames;
+ int sc; // number of color channels
+
+ // For loading from URL
+ /*
+ static const int ioBufferSize = 32768;
+ unsigned char * ioBuffer;
+ QNetworkAccessManager networkManager;
+ AVIOContext* avioContext;
+ QFile fileStream;
+ QNetworkReply* reply;
+ QBuffer fileBuffer;
+ QByteArray byteArray;
+ */
+ };
+
+
+ // TODO - finish refactoring based on
+ // http://svn.gnumonks.org/trunk/21c3-video/ffmpeg/ffmpeg-0.4.9-pre1/output_example.c
+ class encoder
+ {
+ public:
+ //typedef encoder::Channel Channel;
+
+ encoder(const char * file_name, int width, int height, enum AVCodecID codec_id = CODEC_ID_MPEG4);
+ virtual ~encoder();
+ void setPixelIntensity(int x, int y, int c, uint8_t value);
+ void write_frame();
+
+ protected:
+ AVFormatContext *container;
+ AVCodecContext *pCtx;
+ AVFrame *picture_yuv;
+ AVFrame *picture_rgb;
+ struct SwsContext *Sctx;
+ };
+}
+
+
+
+#endif // libavwrapper_H
diff --git a/libavwrapper/unit_test.cpp b/libavwrapper/unit_test.cpp
new file mode 100755
index 0000000..00070c8
--- /dev/null
+++ b/libavwrapper/unit_test.cpp
@@ -0,0 +1,14 @@
+#include "unit_test.h"
+
+using namespace std;
+
+int main(int argc, char** argv)
+{
+ libav::decoder decoder;
+
+ decoder.open("newsins1_360.mp4");
+ cerr<<"opened movie file, "<<decoder.getWidth()<<"x"<<decoder.getHeight()<<", "<<decoder.getNumberOfFrames()<<" frames, "<<decoder.getNumberOfChannels()<<" channels"<<endl;
+
+ cout << "hello, world!"<<endl;
+ return 1;
+}
diff --git a/libavwrapper/unit_test.h b/libavwrapper/unit_test.h
new file mode 100755
index 0000000..41f5554
--- /dev/null
+++ b/libavwrapper/unit_test.h
@@ -0,0 +1,3 @@
+#include "libavwrapper.h"
+
+#include <iostream> \ No newline at end of file