---

AVISource.h

Go to the documentation of this file.

/*************************************************************************
 *
 * Source code for Real Time Image Library (libRTImage) 
 *
 * Leeds Vision Group give permission for this code to be copied, modified 
 * and distributed within the University of Leeds subject to the following
 * conditions:-
 *
 * - The code is not to be used for commercial gain.
 * - The code and use thereof will be attributed to the authors where
 *   appropriate (including demonstrations which rely on it's use).
 * - All modified, distributions of the source files will retain this header.
 *
 ****************************************************************************/

#ifndef AVI_SOURCE_H
#define AVI_SOURCE_H

#include "Image.h"
#include "ImageSource.h"
#include <avcodec.h>
#include <avformat.h>

class AVISource : public ImageSource
{
protected:
/* Protected Members */
    AVFormatContext *pFormatCtx;
    int             i;
    int             videoStream;
    AVCodecContext  *pCodecCtx;         
    AVCodec         *pCodec;            
    AVFrame         *pFrame;            
    AVFrame         *pFrameRGB;
    int             numBytes;
    uint8_t         *buffer;

//NOTE: typedef unsigned char  uint8_t;


        unsigned char** buff;           

        unsigned int    width;          
        unsigned int    height;         
        long            frames;         
        float           framerate;      
public:
/* Public Methods */

        AVISource(char* path);

        ~AVISource();

        /* Extraction operators */
        
        virtual ImageSource& operator >> (ImageRGB&);
        virtual ImageSource& operator >> (ImageGrey&);
        virtual ImageSource& operator >> (Image&);

        void get_size(unsigned int&, unsigned int&);

        int get_no_frames() ;

        int set_frame(long frame_no) ;
        void get_framerate(float&);

        bool eof(void) const;
};

#endif  // AVI_SOURCE_H

---