FFMPEGWriter.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright 2009-2015 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;

    AVOutputFormat* m_outputFmt;

    AVStream* m_stream;

    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(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