1 files changed, 81 insertions, 0 deletions
diff --git a/dom/media/webaudio/ChannelSplitterNode.cpp b/dom/media/webaudio/ChannelSplitterNode.cpp
new file mode 100644
index 000000000..34a414d16
--- /dev/null
+++ b/dom/media/webaudio/ChannelSplitterNode.cpp
@@ -0,0 +1,81 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=2 sw=2 sts=2 et cindent: */
+/* 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 http://mozilla.org/MPL/2.0/. */
+
+#include "mozilla/dom/ChannelSplitterNode.h"
+#include "mozilla/dom/ChannelSplitterNodeBinding.h"
+#include "AudioNodeEngine.h"
+#include "AudioNodeStream.h"
+
+namespace mozilla {
+namespace dom {
+
+NS_IMPL_ISUPPORTS_INHERITED0(ChannelSplitterNode, AudioNode)
+
+class ChannelSplitterNodeEngine final : public AudioNodeEngine
+{
+public:
+  explicit ChannelSplitterNodeEngine(ChannelSplitterNode* aNode)
+    : AudioNodeEngine(aNode)
+  {
+    MOZ_ASSERT(NS_IsMainThread());
+  }
+
+  void ProcessBlocksOnPorts(AudioNodeStream* aStream,
+                            const OutputChunks& aInput,
+                            OutputChunks& aOutput,
+                            bool* aFinished) override
+  {
+    MOZ_ASSERT(aInput.Length() == 1, "Should only have one input port");
+
+    aOutput.SetLength(OutputCount());
+    for (uint16_t i = 0; i < OutputCount(); ++i) {
+      if (i < aInput[0].ChannelCount()) {
+        // Split out existing channels
+        aOutput[i].AllocateChannels(1);
+        AudioBlockCopyChannelWithScale(
+            static_cast<const float*>(aInput[0].mChannelData[i]),
+            aInput[0].mVolume,
+            aOutput[i].ChannelFloatsForWrite(0));
+      } else {
+        // Pad with silent channels if needed
+        aOutput[i].SetNull(WEBAUDIO_BLOCK_SIZE);
+      }
+    }
+  }
+
+  size_t SizeOfIncludingThis(MallocSizeOf aMallocSizeOf) const override
+  {
+    return aMallocSizeOf(this) + SizeOfExcludingThis(aMallocSizeOf);
+  }
+};
+
+ChannelSplitterNode::ChannelSplitterNode(AudioContext* aContext,
+                                         uint16_t aOutputCount)
+  : AudioNode(aContext,
+              2,
+              ChannelCountMode::Max,
+              ChannelInterpretation::Speakers)
+  , mOutputCount(aOutputCount)
+{
+  mStream = AudioNodeStream::Create(aContext,
+                                    new ChannelSplitterNodeEngine(this),
+                                    AudioNodeStream::NO_STREAM_FLAGS,
+                                    aContext->Graph());
+}
+
+ChannelSplitterNode::~ChannelSplitterNode()
+{
+}
+
+JSObject*
+ChannelSplitterNode::WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto)
+{
+  return ChannelSplitterNodeBinding::Wrap(aCx, this, aGivenProto);
+}
+
+} // namespace dom
+} // namespace mozilla
+