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
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/speech/audio_encoder.h"
#include <stddef.h>
#include <memory>
#include "base/check_op.h"
#include "base/containers/heap_array.h"
#include "base/strings/string_number_conversions.h"
#include "components/speech/audio_buffer.h"
namespace {
const char kContentTypeFLAC[] = "audio/x-flac; rate=";
const int kFLACCompressionLevel = 0; // 0 for speed
FLAC__StreamEncoderWriteStatus WriteCallback(const FLAC__StreamEncoder* encoder,
const FLAC__byte buffer[],
size_t bytes,
unsigned samples,
unsigned current_frame,
void* client_data) {
AudioBuffer* encoded_audio_buffer = static_cast<AudioBuffer*>(client_data);
encoded_audio_buffer->Enqueue(buffer, bytes);
return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
}
} // namespace
AudioEncoder::AudioEncoder(int sampling_rate, int bits_per_sample)
: encoded_audio_buffer_(1), /* Byte granularity of encoded samples. */
encoder_(FLAC__stream_encoder_new()),
is_encoder_initialized_(false) {
FLAC__stream_encoder_set_channels(encoder_.get(), 1);
FLAC__stream_encoder_set_bits_per_sample(encoder_.get(), bits_per_sample);
FLAC__stream_encoder_set_sample_rate(encoder_.get(), sampling_rate);
FLAC__stream_encoder_set_compression_level(encoder_.get(),
kFLACCompressionLevel);
// Initializing the encoder will cause sync bytes to be written to
// its output stream, so we wait until the first call to Encode()
// before doing so.
}
AudioEncoder::~AudioEncoder() = default;
void AudioEncoder::Encode(const AudioChunk& raw_audio) {
DCHECK_EQ(raw_audio.bytes_per_sample(), 2);
if (!is_encoder_initialized_) {
const FLAC__StreamEncoderInitStatus encoder_status =
FLAC__stream_encoder_init_stream(encoder_.get(), WriteCallback, nullptr,
nullptr, nullptr,
&encoded_audio_buffer_);
DCHECK_EQ(encoder_status, FLAC__STREAM_ENCODER_INIT_STATUS_OK);
is_encoder_initialized_ = true;
}
// FLAC encoder wants samples as int32s.
const int num_samples = raw_audio.NumSamples();
base::HeapArray<FLAC__int32> flac_samples =
base::HeapArray<FLAC__int32>::Uninit(num_samples);
for (int i = 0; i < num_samples; ++i)
flac_samples[i] = static_cast<FLAC__int32>(raw_audio.GetSample16(i));
FLAC__int32* flac_samples_ptr = flac_samples.data();
FLAC__stream_encoder_process(encoder_.get(), &flac_samples_ptr, num_samples);
}
void AudioEncoder::Flush() {
FLAC__stream_encoder_finish(encoder_.get());
}
scoped_refptr<AudioChunk> AudioEncoder::GetEncodedDataAndClear() {
return encoded_audio_buffer_.DequeueAll();
}
std::string AudioEncoder::GetMimeType() {
return std::string(kContentTypeFLAC) +
base::NumberToString(
FLAC__stream_encoder_get_sample_rate(encoder_.get()));
}
int AudioEncoder::GetBitsPerSample() {
return FLAC__stream_encoder_get_bits_per_sample(encoder_.get());
}
|
