4720bef1841b64db5de7910a169f5520ab2eca20
[blender-staging.git] / source / gameengine / VideoTexture / VideoFFmpeg.h
1 /* $Id$
2 -----------------------------------------------------------------------------
3 This source file is part of VideoTexture library
4
5 Copyright (c) 2007 The Zdeno Ash Miklas
6
7 This program is free software; you can redistribute it and/or modify it under
8 the terms of the GNU Lesser General Public License as published by the Free Software
9 Foundation; either version 2 of the License, or (at your option) any later
10 version.
11
12 This program is distributed in the hope that it will be useful, but WITHOUT
13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public License along with
17 this program; if not, write to the Free Software Foundation, Inc., 59 Temple
18 Place - Suite 330, Boston, MA 02111-1307, USA, or go to
19 http://www.gnu.org/copyleft/lesser.txt.
20 -----------------------------------------------------------------------------
21 */
22 #if !defined VIDEOFFMPEG_H
23 #define VIDEOFFMPEG_H
24
25 #ifdef WITH_FFMPEG
26 extern "C" {
27 #include <ffmpeg/avformat.h>
28 #include <ffmpeg/avcodec.h>
29 #include <ffmpeg/rational.h>
30 #include <ffmpeg/swscale.h>
31 }
32
33 #if LIBAVFORMAT_VERSION_INT < (49 << 16)
34 #define FFMPEG_OLD_FRAME_RATE 1
35 #else
36 #define FFMPEG_CODEC_IS_POINTER 1
37 #endif
38
39 #if LIBAVFORMAT_VERSION_INT >= (52 << 16)
40 #define FFMPEG_PB_IS_POINTER 1
41 #endif
42
43 #ifdef FFMPEG_CODEC_IS_POINTER
44 static inline AVCodecContext* get_codec_from_stream(AVStream* stream)
45 {
46         return stream->codec;
47 }
48 #else
49 static inline AVCodecContext* get_codec_from_stream(AVStream* stream)
50 {
51         return &stream->codec;
52 }
53 #endif
54
55 #include "VideoBase.h"
56
57
58 // type VideoFFmpeg declaration
59 class VideoFFmpeg : public VideoBase
60 {
61 public:
62         /// constructor
63         VideoFFmpeg (HRESULT * hRslt);
64         /// destructor
65         virtual ~VideoFFmpeg ();
66
67         /// set initial parameters
68         void initParams (short width, short height, float rate);
69         /// open video file
70         virtual void openFile (char * file);
71         /// open video capture device
72         virtual void openCam (char * driver, short camIdx);
73
74         /// release video source
75         virtual bool release (void);
76
77         /// play video
78         virtual bool play (void);
79         /// stop/pause video
80         virtual bool stop (void);
81
82         /// set play range
83         virtual void setRange (double start, double stop);
84         /// set frame rate
85         virtual void setFrameRate (float rate);
86         // some specific getters and setters
87         int getPreseek(void) { return m_preseek; }
88         void setPreseek(int preseek) { if (preseek >= 0) m_preseek = preseek; }
89         bool getDeinterlace(void) { return m_deinterlace; }
90         void setDeinterlace(bool deinterlace) { m_deinterlace = deinterlace; }
91
92 protected:
93
94         // format and codec information
95         AVCodec *m_codec;
96         AVFormatContext *m_formatCtx;
97         AVCodecContext *m_codecCtx;
98         // raw frame extracted from video file
99         AVFrame *m_frame;
100         // deinterlaced frame if codec requires it
101         AVFrame *m_frameDeinterlaced;
102         // decoded RGB24 frame if codec requires it
103         AVFrame *m_frameBGR;
104         // conversion from raw to RGB is done with sws_scale
105         struct SwsContext *m_imgConvertCtx;
106         // should the codec be deinterlaced?
107         bool m_deinterlace;
108         // number of frame of preseek
109         int m_preseek;
110         // order number of stream holding the video in format context
111         int m_videoStream;
112
113         // the actual frame rate
114         double m_baseFrameRate;
115
116         /// last displayed frame
117         long m_lastFrame;
118
119         /// current file pointer position in file expressed in frame number
120         long m_curPosition;
121
122         /// time of video play start
123         double m_startTime;
124
125         /// width of capture in pixel
126         short m_captWidth;
127         
128         /// height of capture in pixel
129         short m_captHeight;
130
131         /// frame rate of capture in frames per seconds
132         float m_captRate;
133
134         /// image calculation
135         virtual void calcImage (unsigned int texId);
136
137         /// load frame from video
138         void loadFrame (void);
139
140         /// set actual position
141         void setPositions (void);
142
143         /// get actual framerate
144         double actFrameRate (void) { return m_frameRate * m_baseFrameRate; }
145
146         /// common function to video file and capture
147         int openStream(const char *filename, AVInputFormat *inputFormat, AVFormatParameters *formatParams);
148
149         /// check if a frame is available and load it in pFrame, return true if a frame could be retrieved
150         bool grabFrame(long frame);
151
152         /// return the frame in RGB24 format, the image data is found in AVFrame.data[0]
153         AVFrame* getFrame(void) { return m_frameBGR; }
154 };
155
156 inline VideoFFmpeg * getFFmpeg (PyImage * self) 
157 {
158         return static_cast<VideoFFmpeg*>(self->m_image); 
159 }
160
161 #endif  //WITH_FFMPEG
162
163 #endif