1 files changed, 150 insertions, 0 deletions
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