KviOggTheoraEncoder.h

Go to the documentation of this file.

#ifndef _KVIOGGTHEORAENCODER_H_
#define _KVIOGGTHEORAENCODER_H_

//=============================================================================
//
//   File : KviOggTheoraEncoder.h
//   Creation date : Wed Dec 29 2010 15:51:21 CEST by Elvio Basello
//
//   This file is part of the KVIrc IRC client distribution
//   Copyright (C) 2010 Elvio Basello (hellvis69 at gmail dot com)
//
//   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 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
//
//=============================================================================

#ifndef COMPILE_DISABLE_OGG_THEORA

#include "kvi_settings.h"
#include "KviOggTheoraGeometry.h"

#include "theora/theoraenc.h"

#include <QColor>

class KviDataBuffer;

#define YUV444_BPP 3

class KVILIB_API KviOggTheoraEncoder
{
public:
        KviOggTheoraEncoder(KviDataBuffer * stream, int iWidth = 320, int iHeight = 240, int iFpsN = 5, int iFpsD = 1, int iParN = 4, int iParD = 3);

        virtual ~KviOggTheoraEncoder();

private:
        KviOggTheoraGeometry geometry; 
        KviDataBuffer * m_pStream;     
        quint8 * videoYuv;             
        ogg_int64_t text_sofar; 
        int frame_state;             
        unsigned char * yuvframe[3]; 
        size_t y4m_dst_buf_sz;      
        size_t y4m_dst_buf_read_sz; 
        size_t y4m_aux_buf_sz;      
        size_t y4m_aux_buf_read_sz; 
        ogg_stream_state zo; 
        ogg_stream_state to; 
        ogg_page og;         
        ogg_packet op;       
        th_enc_ctx * td; 
        th_info ti;      
        th_comment tc;   
        int textflag;  
        int videoflag; 
public:
        void addVideoFrame(QRgb * rgb32, int videoSize);

        void addTextFrame(unsigned char * textPkt, int textSize);

private:
        int fetch_and_process_video(quint8 * videoYuv, ogg_page * videopage, ogg_stream_state * to, th_enc_ctx * td, int videoflag);

        int fetch_and_process_video_packet(quint8 * videoYuv, th_enc_ctx * td, ogg_packet * op);

        static int ilog(unsigned _v);
};

#endif // COMPILE_DISABLE_OGG_THEORA
#endif // _KVIOGGTHEORAENCODER_H_