OpusEncoder.h

Go to the documentation of this file.

/*************************************************************************
    OpusEncoder.h  -  sub encoder base class for Opus in an Ogg container
                             -------------------
    begin                : Thu Jan 03 2013
    copyright            : (C) 2013 by Thomas Eschenbacher
    email                : Thomas.Eschenbacher@gmx.de
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/

#ifndef OPUS_ENCODER_H
#define OPUS_ENCODER_H

#ifdef HAVE_OGG_OPUS
#include "config.h"

#include <ogg/ogg.h>
#include <opus/opus.h>
#include <opus/opus_multistream.h>

#include "libkwave/FileInfo.h"
#include "libkwave/MultiTrackSink.h"
#include "libkwave/VorbisCommentMap.h"
#include "libkwave/modules/SampleBuffer.h"

#include "OggSubEncoder.h"
#include "OpusHeader.h"

class QIODevice;
class QWidget;

namespace Kwave
{

    class ChannelMixer;
    class FileInfo;
    class MultiTrackReader;
    class StreamObject;

    class OpusEncoder: public Kwave::OggSubEncoder
    {
    public:

        OpusEncoder();

        virtual ~OpusEncoder() Q_DECL_OVERRIDE;

        virtual bool open(QWidget *widget,
                          const Kwave::FileInfo &info,
                          Kwave::MultiTrackReader &src) Q_DECL_OVERRIDE;

        virtual bool writeHeader(QIODevice &dst) Q_DECL_OVERRIDE;

        virtual bool encode(Kwave::MultiTrackReader &src,
                            QIODevice &dst) Q_DECL_OVERRIDE;

        virtual void close() Q_DECL_OVERRIDE;

    private:

        bool setupDownMix(QWidget *widget, unsigned int tracks, int bitrate);

        bool setupBitrate(QWidget *widget, unsigned int tracks);

        bool setupCodingRate(QWidget *widget, unsigned int tracks, double rate);

        bool setupEncoder(QWidget *widget, unsigned int tracks, double rate);

        bool setupBitrateMode(QWidget *widget);

        bool writeOpusHeader(QIODevice &dst);

        bool writeOpusTags(QIODevice &dst);

        bool writeOggPage(QIODevice &dst);

        unsigned int fillInBuffer(Kwave::MultiTrackReader &src);

    private:

        Kwave::VorbisCommentMap m_comments_map;

        Kwave::FileInfo m_info;

        ogg_stream_state m_os;

        ogg_page         m_og;

        ogg_packet       m_op;

        enum {
            DOWNMIX_OFF    = -1, 
            DOWNMIX_AUTO   =  0, 
            DOWNMIX_MONO   =  1, 
            DOWNMIX_STEREO =  2  
        } m_downmix;

        int m_bitrate;

        int m_coding_rate;

        unsigned int m_encoder_channels;

        Kwave::ChannelMixer *m_channel_mixer;

        Kwave::StreamObject *m_rate_converter;

        unsigned int m_frame_size;

        unsigned int m_extra_out;

        Kwave::opus_header_t m_opus_header;

        unsigned int m_max_frame_bytes;

        unsigned char *m_packet_buffer;

        OpusMSEncoder *m_encoder;

        float *m_encoder_input;

        Kwave::StreamObject *m_last_queue_element;

        Kwave::MultiTrackSink<Kwave::SampleBuffer, true> *m_buffer;
    };
}

#endif /* HAVE_OGG_OPUS */

#endif /* OPUS_ENCODER_H */

//***************************************************************************
//***************************************************************************