blob: 690185deb649a5a847ddc1f6fc0ff1bf3187aaa2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/*******************************************************************************
* 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
/**
* @file FFMPEGWriter.h
* @ingroup plugin
* The FFMPEGWriter class.
*/
#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
/**
* This class writes a sound file via ffmpeg.
*/
class AUD_PLUGIN_API FFMPEGWriter : public IWriter
{
private:
/**
* The current position in samples.
*/
int m_position;
/**
* The specification of the audio data.
*/
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;
/**
* The input buffer for the format converted data before encoding.
*/
Buffer m_input_buffer;
/**
* The buffer used for deinterleaving.
*/
Buffer m_deinterleave_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;
/**
* Whether the ouput has to be deinterleaved before writing.
*/
bool m_deinterleave;
/**
* Converter function.
*/
convert_f m_convert;
// delete copy constructor and operator=
FFMPEGWriter(const FFMPEGWriter&) = delete;
FFMPEGWriter& operator=(const FFMPEGWriter&) = delete;
/**
* Encodes to the output buffer.
*/
AUD_LOCAL void encode();
/**
* Finishes writing to the file.
*/
AUD_LOCAL void close();
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 Exception Thrown if the file specified does not exist or
* cannot be read with ffmpeg.
*/
FFMPEGWriter(std::string filename, DeviceSpecs specs, Container format, Codec codec, unsigned int bitrate);
/**
* Destroys the writer and closes the file.
*/
virtual ~FFMPEGWriter();
virtual int getPosition() const;
virtual DeviceSpecs getSpecs() const;
virtual void write(unsigned int length, sample_t* buffer);
};
AUD_NAMESPACE_END
|
