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
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at https://mozilla.org/MPL/2.0/. */
#ifndef DOM_MEDIA_GTEST_AUDIO_GENERATOR_H_
#define DOM_MEDIA_GTEST_AUDIO_GENERATOR_H_
#include "AudioSegment.h"
#include "prtime.h"
#include "SineWaveGenerator.h"
namespace mozilla {
template <typename Sample>
class AudioGenerator {
public:
AudioGenerator(uint32_t aChannels, uint32_t aSampleRate,
uint32_t aFrequency = 1000)
: mChannels(aChannels),
mSampleRate(aSampleRate),
mFrequency(aFrequency),
mGenerator(aSampleRate, aFrequency) {}
void Generate(mozilla::AudioSegment& aSegment, const uint32_t& aSamples) {
SetInterleaved(false);
CheckedInt<size_t> bufferSize(sizeof(Sample));
bufferSize *= aSamples;
RefPtr<SharedBuffer> buffer = SharedBuffer::Create(bufferSize);
Sample* dest = static_cast<Sample*>(buffer->Data());
mGenerator.generate(dest, aSamples);
AutoTArray<const Sample*, 1> channels;
for (uint32_t i = 0; i < mChannels; ++i) {
channels.AppendElement(dest);
}
aSegment.AppendFrames(buffer.forget(), channels, aSamples,
PRINCIPAL_HANDLE_NONE);
}
void GenerateInterleaved(Sample* aBuffer, const uint32_t& aFrames) {
SetInterleaved(true);
mGenerator.generate(aBuffer, aFrames * mChannels);
}
void SetInterleaved(bool aInterleaved) {
if (aInterleaved == mInterleaved) {
return;
}
mInterleaved = aInterleaved;
if (mInterleaved) {
TrackTicks offset = Offset();
mGenerator =
SineWaveGenerator<Sample>(mSampleRate, mFrequency, mChannels);
mGenerator.SetOffset(offset * mChannels);
} else {
TrackTicks offset = Offset();
mGenerator = SineWaveGenerator<Sample>(mSampleRate, mFrequency);
mGenerator.SetOffset(offset / mChannels);
}
}
void SetOffset(TrackTicks aFrames) { mGenerator.SetOffset(aFrames); }
TrackTicks Offset() const { return mGenerator.Offset(); }
static float Amplitude() { return SineWaveGenerator<Sample>::Amplitude(); }
const uint32_t mChannels;
const uint32_t mSampleRate;
const uint32_t mFrequency;
private:
bool mInterleaved = false;
mozilla::SineWaveGenerator<Sample> mGenerator;
};
} // namespace mozilla
#endif // DOM_MEDIA_GTEST_AUDIO_GENERATOR_H_
|
