forked from bartvdbraak/blender
161 lines
3.5 KiB
C++
161 lines
3.5 KiB
C++
/*
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* Copyright 2009-2011 Jörg Hermann Müller
|
|
*
|
|
* This file is part of AudaSpace.
|
|
*
|
|
* Audaspace 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.
|
|
*
|
|
* AudaSpace 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 Audaspace; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*/
|
|
|
|
/** \file audaspace/ffmpeg/AUD_FFMPEGWriter.h
|
|
* \ingroup audffmpeg
|
|
*/
|
|
|
|
|
|
#ifndef __AUD_FFMPEGWRITER_H__
|
|
#define __AUD_FFMPEGWRITER_H__
|
|
|
|
#include "AUD_ConverterFunctions.h"
|
|
#include "AUD_Buffer.h"
|
|
#include "AUD_IWriter.h"
|
|
|
|
#include <string>
|
|
|
|
struct AVCodecContext;
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
}
|
|
|
|
/**
|
|
* This class writes a sound file via ffmpeg.
|
|
*/
|
|
class AUD_FFMPEGWriter : public AUD_IWriter
|
|
{
|
|
private:
|
|
/**
|
|
* The current position in samples.
|
|
*/
|
|
int m_position;
|
|
|
|
/**
|
|
* The specification of the audio data.
|
|
*/
|
|
AUD_DeviceSpecs m_specs;
|
|
|
|
/**
|
|
* The AVFormatContext structure for using ffmpeg.
|
|
*/
|
|
AVFormatContext* m_formatCtx;
|
|
|
|
/**
|
|
* The AVCodecContext structure for using ffmpeg.
|
|
*/
|
|
AVCodecContext* m_codecCtx;
|
|
|
|
/**
|
|
* The AVOutputFormat structure for using ffmpeg.
|
|
*/
|
|
AVOutputFormat* m_outputFmt;
|
|
|
|
/**
|
|
* The AVStream structure for using ffmpeg.
|
|
*/
|
|
AVStream* m_stream;
|
|
|
|
/**
|
|
* Frame sent to the encoder.
|
|
*/
|
|
AVFrame *m_frame;
|
|
|
|
/**
|
|
* PTS of next frame to write.
|
|
*/
|
|
int m_frame_pts;
|
|
|
|
/**
|
|
* Number of bytes per sample.
|
|
*/
|
|
int m_sample_size;
|
|
|
|
/**
|
|
* Need to de-interleave audio for planar sample formats.
|
|
*/
|
|
bool m_deinterleave;
|
|
|
|
AUD_Buffer m_deinterleave_buffer;
|
|
|
|
/**
|
|
* The input buffer for the format converted data before encoding.
|
|
*/
|
|
AUD_Buffer m_input_buffer;
|
|
|
|
/**
|
|
* The output buffer for the encoded audio data.
|
|
*/
|
|
AUD_Buffer m_output_buffer;
|
|
|
|
/**
|
|
* The count of input samples we have so far.
|
|
*/
|
|
unsigned int m_input_samples;
|
|
|
|
/**
|
|
* The count of input samples necessary to encode a packet.
|
|
*/
|
|
unsigned int m_input_size;
|
|
|
|
/**
|
|
* Converter function.
|
|
*/
|
|
AUD_convert_f m_convert;
|
|
|
|
// hide copy constructor and operator=
|
|
AUD_FFMPEGWriter(const AUD_FFMPEGWriter&);
|
|
AUD_FFMPEGWriter& operator=(const AUD_FFMPEGWriter&);
|
|
|
|
/**
|
|
* Encodes to the output buffer.
|
|
* \param data Pointer to the data to encode.
|
|
*/
|
|
void encode(sample_t* data);
|
|
|
|
public:
|
|
/**
|
|
* Creates a new writer.
|
|
* \param filename The path to the file to be read.
|
|
* \param specs The file's audio specification.
|
|
* \param format The file's container format.
|
|
* \param codec The codec used for encoding the audio data.
|
|
* \param bitrate The bitrate for encoding.
|
|
* \exception AUD_Exception Thrown if the file specified does not exist or
|
|
* cannot be read with ffmpeg.
|
|
*/
|
|
AUD_FFMPEGWriter(std::string filename, AUD_DeviceSpecs specs, AUD_Container format, AUD_Codec codec, unsigned int bitrate);
|
|
|
|
/**
|
|
* Destroys the writer and closes the file.
|
|
*/
|
|
virtual ~AUD_FFMPEGWriter();
|
|
|
|
virtual int getPosition() const;
|
|
virtual AUD_DeviceSpecs getSpecs() const;
|
|
virtual void write(unsigned int length, sample_t* buffer);
|
|
};
|
|
|
|
#endif //__AUD_FFMPEGWRITER_H__
|