FFMPEGWriter.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright 2009-2016 Jörg Müller
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
 
#pragma once
 
#ifdef FFMPEG_PLUGIN
#define AUD_BUILD_PLUGIN
#endif
 
#include "respec/ConverterFunctions.h"
#include "util/Buffer.h"
#include "file/IWriter.h"
 
#include <string>
 
struct AVCodecContext;
extern "C" {
#include <libavformat/avformat.h>
}
 
AUD_NAMESPACE_BEGIN
 
class AUD_PLUGIN_API FFMPEGWriter : public IWriter
{
private:
    int m_position;
 
    DeviceSpecs m_specs;
 
    AVFormatContext* m_formatCtx;
 
    AVCodecContext* m_codecCtx;
 
    AVStream* m_stream;
 
    AVPacket* m_packet;
 
    AVFrame* m_frame;
 
    Buffer m_input_buffer;
 
    Buffer m_deinterleave_buffer;
 
    unsigned int m_input_samples;
 
    unsigned int m_input_size;
 
    bool m_deinterleave;
 
    convert_f m_convert;
 
    // delete copy constructor and operator=
    FFMPEGWriter(const FFMPEGWriter&) = delete;
    FFMPEGWriter& operator=(const FFMPEGWriter&) = delete;
 
    AUD_LOCAL void encode();
 
    AUD_LOCAL void close();
 
public:
    FFMPEGWriter(const std::string &filename, DeviceSpecs specs, Container format, Codec codec, unsigned int bitrate);
 
    virtual ~FFMPEGWriter();
 
    virtual int getPosition() const;
    virtual DeviceSpecs getSpecs() const;
    virtual void write(unsigned int length, sample_t* buffer);
};
 
AUD_NAMESPACE_END