OStreamVideoEncoder.hh

Go to the documentation of this file.

/*
 *  Copyright (C) Massimo Cora' 2006 <maxcvs@email.it>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU 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 Library General Public License for more details.
 *
 *  You should have received a copy of the GNU 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.
 */

#ifndef _OSTREAMVIDEOENCODER_H_
#define _OSTREAMVIDEOENCODER_H_

#include <cv.h>
#include <highgui.h>
#include <ffmpeg/avcodec.h>
#include <ffmpeg/avformat.h>

#include "OStreamConfig.hh"

class OStreamVideoEncoder
{
public:
                
        OStreamVideoEncoder( unsigned int io_buf_size, ostream_encoding_type encoding_type,
                                                 int video_width, int video_height, 
                                                 int fps = 25,
                                                 PixelFormat opencv_pix_fmt = PIX_FMT_BGR24 );
        virtual ~OStreamVideoEncoder();

        const unsigned char* encode_video_frame( IplImage *image, int *out_size );
        
private:                // functions

        bool encoder_initialize ();
        
        AVFrame* create_avframe( int pixel_format );
        
        AVStream* create_video_stream( AVFormatContext *io_fmt_context, int codec_id );

        bool open_codec( AVStream *stream );
        
private:                // data
        bool _initialized;

        AVFrame *_ready_to_encode_frame, *_opencv_frame;
        
        AVOutputFormat *_out_fmt;
        
        AVFormatContext *_out_fmt_context;
        
        AVStream *_video_stream;

        PixelFormat _opencv_pix_fmt;
        
        unsigned char* _internal_buffer;
        int _internal_buffer_size;
        
        int _encoding_type;

        int _video_height;
        int _video_width;
        int _fps;

};


#endif  //_OSTREAMVIDEOENCODER_H_

---