Adding upstream version 86.0.1.upstream/86.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 80 insertions, 0 deletions

diff --git a/dom/media/gtest/AudioGenerator.h b/dom/media/gtest/AudioGenerator.h
new file mode 100644
index 0000000000..a528fbef08
--- /dev/null
+++ b/dom/media/gtest/AudioGenerator.h
@@ -0,0 +1,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_