forked from bartvdbraak/blender
ce1625ebc0
The FFmpeg library allows to load image files. Although it is possible to load images using the VideoFFmpeg class, it is not very efficient. The new class VideoTexture.ImageFFmpeg is dedicated to image management. Constructor: ----------- VideoTexture.ImageFFmpeg('image_file_name') Opens the file but does not load the texture yet. The file name can also be a network address. It can also be a video file name; in that case only the first image is loaded. Methods: ------- refresh(True) Loads the image to texture. You just need to call it once, the file is automatically closed after that and calling refresh() again will have no effect. reload('new_file_name') Reloads the image (if new_file_name is omitted) or loads a new image. The file is opened but the texture is not updated yet, you need to call refresh() once to load the texture. Attributes: ---------- status returns the image status: 2 : file opened, texture not loaded 3 : file closed, texture loaded image returns the image data as a string of RGBA pixel size returns the image size [x,y] scale get/set the scale flag. If the scale flag is False, the image is rescale to texture format using gluScaleImage() function, slow but good quality. If the scale flag is True, the image is rescaled using a fast but less accurate algorithm. flip get/set Y-flip flag. Set to True by default as FFmpeg always provides the image upside down filter get/set filter(s) on the image. Example:
171 lines
4.6 KiB
C++
171 lines
4.6 KiB
C++
/* $Id$
|
|
-----------------------------------------------------------------------------
|
|
This source file is part of VideoTexture library
|
|
|
|
Copyright (c) 2007 The Zdeno Ash Miklas
|
|
|
|
This program is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU Lesser General Public License as published by the Free Software
|
|
Foundation; either version 2 of the License, or (at your option) any later
|
|
version.
|
|
|
|
This program 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 Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License along with
|
|
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
|
|
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
|
|
http://www.gnu.org/copyleft/lesser.txt.
|
|
-----------------------------------------------------------------------------
|
|
*/
|
|
#if !defined VIDEOFFMPEG_H
|
|
#define VIDEOFFMPEG_H
|
|
|
|
#ifdef WITH_FFMPEG
|
|
extern "C" {
|
|
#include <ffmpeg/avformat.h>
|
|
#include <ffmpeg/avcodec.h>
|
|
#include <ffmpeg/rational.h>
|
|
#include <ffmpeg/swscale.h>
|
|
}
|
|
|
|
#if LIBAVFORMAT_VERSION_INT < (49 << 16)
|
|
#define FFMPEG_OLD_FRAME_RATE 1
|
|
#else
|
|
#define FFMPEG_CODEC_IS_POINTER 1
|
|
#endif
|
|
|
|
#if LIBAVFORMAT_VERSION_INT >= (52 << 16)
|
|
#define FFMPEG_PB_IS_POINTER 1
|
|
#endif
|
|
|
|
#ifdef FFMPEG_CODEC_IS_POINTER
|
|
static inline AVCodecContext* get_codec_from_stream(AVStream* stream)
|
|
{
|
|
return stream->codec;
|
|
}
|
|
#else
|
|
static inline AVCodecContext* get_codec_from_stream(AVStream* stream)
|
|
{
|
|
return &stream->codec;
|
|
}
|
|
#endif
|
|
|
|
#include "VideoBase.h"
|
|
|
|
|
|
// type VideoFFmpeg declaration
|
|
class VideoFFmpeg : public VideoBase
|
|
{
|
|
public:
|
|
/// constructor
|
|
VideoFFmpeg (HRESULT * hRslt);
|
|
/// destructor
|
|
virtual ~VideoFFmpeg ();
|
|
|
|
/// set initial parameters
|
|
void initParams (short width, short height, float rate, bool image=false);
|
|
/// open video/image file
|
|
virtual void openFile (char * file);
|
|
/// open video capture device
|
|
virtual void openCam (char * driver, short camIdx);
|
|
|
|
/// release video source
|
|
virtual bool release (void);
|
|
|
|
/// play video
|
|
virtual bool play (void);
|
|
/// stop/pause video
|
|
virtual bool stop (void);
|
|
|
|
/// set play range
|
|
virtual void setRange (double start, double stop);
|
|
/// set frame rate
|
|
virtual void setFrameRate (float rate);
|
|
// some specific getters and setters
|
|
int getPreseek(void) { return m_preseek; }
|
|
void setPreseek(int preseek) { if (preseek >= 0) m_preseek = preseek; }
|
|
bool getDeinterlace(void) { return m_deinterlace; }
|
|
void setDeinterlace(bool deinterlace) { m_deinterlace = deinterlace; }
|
|
char *getImageName(void) { return (m_isImage) ? m_imageName.Ptr() : NULL; }
|
|
|
|
protected:
|
|
|
|
// format and codec information
|
|
AVCodec *m_codec;
|
|
AVFormatContext *m_formatCtx;
|
|
AVCodecContext *m_codecCtx;
|
|
// raw frame extracted from video file
|
|
AVFrame *m_frame;
|
|
// deinterlaced frame if codec requires it
|
|
AVFrame *m_frameDeinterlaced;
|
|
// decoded RGB24 frame if codec requires it
|
|
AVFrame *m_frameRGB;
|
|
// conversion from raw to RGB is done with sws_scale
|
|
struct SwsContext *m_imgConvertCtx;
|
|
// should the codec be deinterlaced?
|
|
bool m_deinterlace;
|
|
// number of frame of preseek
|
|
int m_preseek;
|
|
// order number of stream holding the video in format context
|
|
int m_videoStream;
|
|
|
|
// the actual frame rate
|
|
double m_baseFrameRate;
|
|
|
|
/// last displayed frame
|
|
long m_lastFrame;
|
|
|
|
/// current file pointer position in file expressed in frame number
|
|
long m_curPosition;
|
|
|
|
/// time of video play start
|
|
double m_startTime;
|
|
|
|
/// width of capture in pixel
|
|
short m_captWidth;
|
|
|
|
/// height of capture in pixel
|
|
short m_captHeight;
|
|
|
|
/// frame rate of capture in frames per seconds
|
|
float m_captRate;
|
|
|
|
/// is file an image?
|
|
bool m_isImage;
|
|
|
|
/// keep last image name
|
|
STR_String m_imageName;
|
|
|
|
/// image calculation
|
|
virtual void calcImage (unsigned int texId);
|
|
|
|
/// load frame from video
|
|
void loadFrame (void);
|
|
|
|
/// set actual position
|
|
void setPositions (void);
|
|
|
|
/// get actual framerate
|
|
double actFrameRate (void) { return m_frameRate * m_baseFrameRate; }
|
|
|
|
/// common function to video file and capture
|
|
int openStream(const char *filename, AVInputFormat *inputFormat, AVFormatParameters *formatParams);
|
|
|
|
/// check if a frame is available and load it in pFrame, return true if a frame could be retrieved
|
|
bool grabFrame(long frame);
|
|
|
|
/// return the frame in RGB24 format, the image data is found in AVFrame.data[0]
|
|
AVFrame* getFrame(void) { return m_frameRGB; }
|
|
};
|
|
|
|
inline VideoFFmpeg * getFFmpeg (PyImage * self)
|
|
{
|
|
return static_cast<VideoFFmpeg*>(self->m_image);
|
|
}
|
|
|
|
#endif //WITH_FFMPEG
|
|
|
|
#endif
|