Implement suspend/resume for AudioContext

Source/modules/webaudio/AudioContext.cpp

 /*
  * Copyright (C) 2010, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1.  Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 12 matching lines @@
  */
 
 #include "config.h"
 
 #if ENABLE(WEB_AUDIO)
 
 #include "modules/webaudio/AudioContext.h"
 
 #include "bindings/core/v8/ExceptionMessages.h"
 #include "bindings/core/v8/ExceptionState.h"
+#include "bindings/core/v8/ScriptState.h"
+#include "core/dom/DOMException.h"
 #include "core/dom/Document.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/html/HTMLMediaElement.h"
 #include "core/inspector/ScriptCallStack.h"
 #include "modules/mediastream/MediaStream.h"
 #include "modules/webaudio/AnalyserNode.h"
 #include "modules/webaudio/AudioBuffer.h"
 #include "modules/webaudio/AudioBufferCallback.h"
 #include "modules/webaudio/AudioBufferSourceNode.h"
 #include "modules/webaudio/AudioListener.h"
@@ skipping 49 matching lines @@
     return audioContext;
 }
 
 // Constructor for rendering to the audio hardware.
 AudioContext::AudioContext(Document* document)
     : ActiveDOMObject(document)
     , m_isStopScheduled(false)
     , m_isCleared(false)
     , m_isInitialized(false)
     , m_destinationNode(nullptr)
+    , m_isResolvingResumePromises(false)
     , m_automaticPullNodesNeedUpdating(false)
     , m_connectionCount(0)
     , m_audioThread(0)
     , m_isOfflineContext(false)
+    , m_contextState(Suspended)
 {
     m_destinationNode = DefaultAudioDestinationNode::create(this);
 
     initialize();
 #if DEBUG_AUDIONODE_REFERENCES
     fprintf(stderr, "%p: AudioContext::AudioContext() #%u\n", this, AudioContext::s_hardwareContextCount);
 #endif
 }
 
 // Constructor for offline (non-realtime) rendering.
 AudioContext::AudioContext(Document* document, unsigned numberOfChannels, size_t numberOfFrames, float sampleRate)
     : ActiveDOMObject(document)
     , m_isStopScheduled(false)
     , m_isCleared(false)
     , m_isInitialized(false)
     , m_destinationNode(nullptr)
+    , m_isResolvingResumePromises(false)
     , m_automaticPullNodesNeedUpdating(false)
     , m_connectionCount(0)
     , m_audioThread(0)
     , m_isOfflineContext(true)
+    , m_contextState(Suspended)
 {
     // Create a new destination for offline rendering.
     m_renderTarget = AudioBuffer::create(numberOfChannels, numberOfFrames, sampleRate);
     if (m_renderTarget.get())
         m_destinationNode = OfflineAudioDestinationNode::create(this, m_renderTarget.get());
 
     initialize();
 }
 
 AudioContext::~AudioContext()
 {
 #if DEBUG_AUDIONODE_REFERENCES
     fprintf(stderr, "%p: AudioContext::~AudioContext()\n", this);
 #endif
     // AudioNodes keep a reference to their context, so there should be no way to be in the destructor if there are still AudioNodes around.
     ASSERT(!m_isInitialized);
     ASSERT(!m_referencedNodes.size());
     ASSERT(!m_finishedNodes.size());
     ASSERT(!m_automaticPullNodes.size());
     if (m_automaticPullNodesNeedUpdating)
         m_renderingAutomaticPullNodes.resize(m_automaticPullNodes.size());
     ASSERT(!m_renderingAutomaticPullNodes.size());
+    ASSERT(!m_resumePromises.size());
 }
 
 void AudioContext::initialize()
 {
     if (isInitialized())
         return;
 
     FFTFrame::initialize();
     m_listener = AudioListener::create();
 
     if (m_destinationNode.get()) {
         m_destinationNode->initialize();
 
         if (!isOfflineContext()) {
             // This starts the audio thread. The destination node's provideInput() method will now be called repeatedly to render audio.
             // Each time provideInput() is called, a portion of the audio stream is rendered. Let's call this time period a "render quantum".
             // NOTE: for now default AudioContext does not need an explicit startRendering() call from JavaScript.
             // We may want to consider requiring it for symmetry with OfflineAudioContext.
-            m_destinationNode->startRendering();
+            startRendering();
             ++s_hardwareContextCount;
         }
 
         m_isInitialized = true;
     }
 }
 
 void AudioContext::clear()
 {
     // We need to run disposers before destructing m_contextGraphMutex.
@@ skipping 345 matching lines @@
         exceptionState.throwDOMException(
             IndexSizeError,
             "length of imaginary array (" + String::number(imag->length())
             + ") exceeds allowed maximum of 4096");
         return 0;
     }
 
     return PeriodicWave::create(sampleRate(), real->view(), imag->view());
 }
 
+String AudioContext::state() const
+{
+    // These strings had better match the strings for AudioContextState in AudioContext.idl.
+    switch (m_contextState) {
+    case Suspended:
+        return "suspended";
+    case Running:
+        return "running";
+    case Closed:
+        return "closed";
+    }
+    ASSERT_NOT_REACHED();
+    return "";
+}
+
+void AudioContext::setContextState(AudioContextState newState)
+{
+    // Validate the transitions
+    switch (newState) {
+    case Suspended:
+        ASSERT(m_contextState == Running);
+        break;
+    case Running:
+        ASSERT(m_contextState == Suspended);
+        break;
+    case Closed:
+        ASSERT(m_contextState != Closed);
+        break;
+    }
+
+    m_contextState = newState;
+}
+
+ScriptPromise AudioContext::suspendContext(ScriptState* scriptState)
+{
+    ASSERT(isMainThread());
+    AutoLocker locker(this);
+
+    RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState);
+
+    ScriptPromise promise = resolver->promise();
+
+    if (isOfflineContext()) {
+        resolver->reject(
+            DOMException::create(
+                InvalidStateError,
+                "cannot suspend an OfflineAudioContext"));
+    } else {
+        // Save the promise which will get resolved at the end of the rendering quantum.
+        m_suspendPromises.append(resolver);
+    }
+
+    return promise;
+}
+
+ScriptPromise AudioContext::resumeContext(ScriptState* scriptState)
+{
+    ASSERT(isMainThread());
+    AutoLocker locker(this);
+
+    RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState);
+
+    ScriptPromise promise = resolver->promise();
+
+    if (isOfflineContext()) {
+        resolver->reject(
+            DOMException::create(
+                InvalidStateError,
+                "cannot resume an OfflineAudioContext"));
+    } else {
+        // Restart the destination node to pull on the audio graph.
+        if (m_destinationNode)
+            startRendering();
+
+        // Save the promise which will get resolved when the destination node starts pulling on the
+        // graph again.
+        m_resumePromises.append(resolver);
+    }
+
+    return promise;
+}
+
 void AudioContext::notifyNodeFinishedProcessing(AudioNode* node)
 {
     ASSERT(isAudioThread());
     m_finishedNodes.append(node);
 }
 
 void AudioContext::derefFinishedSourceNodes()
 {
     ASSERT(isGraphOwner());
     ASSERT(isAudioThread());
@@ skipping 86 matching lines @@
     // It's OK if the tryLock() fails, we'll just take slightly longer to pick up the changes.
     if (tryLock()) {
         // Update the channel count mode.
         updateChangedChannelCountMode();
 
         // Fixup the state of any dirty AudioSummingJunctions and AudioNodeOutputs.
         handleDirtyAudioSummingJunctions();
         handleDirtyAudioNodeOutputs();
 
         updateAutomaticPullNodes();
+        resolvePromisesForResume();
+
         unlock();
     }
 }
 
 void AudioContext::handlePostRenderTasks()
 {
     ASSERT(isAudioThread());
 
     // Must use a tryLock() here too.  Don't worry, the lock will very rarely be contended and this method is called frequently.
     // The worst that can happen is that there will be some nodes which will take slightly longer than usual to be deleted or removed
     // from the render graph (in which case they'll render silence).
     if (tryLock()) {
         // Update the channel count mode.
         updateChangedChannelCountMode();
 
         // Take care of AudioNode tasks where the tryLock() failed previously.
         handleDeferredAudioNodeTasks();
 
         // Dynamically clean up nodes which are no longer needed.
         derefFinishedSourceNodes();
 
         // Fixup the state of any dirty AudioSummingJunctions and AudioNodeOutputs.
         handleDirtyAudioSummingJunctions();
         handleDirtyAudioNodeOutputs();
 
         updateAutomaticPullNodes();
+        resolvePromisesForSuspend();
+
         unlock();
     }
 }
 
 void AudioContext::handleDeferredAudioNodeTasks()
 {
     ASSERT(isAudioThread() && isGraphOwner());
 
     for (unsigned i = 0; i < m_deferredBreakConnectionList.size(); ++i)
         m_deferredBreakConnectionList[i]->breakConnectionWithLock();
@@ skipping 119 matching lines @@
 }
 
 void AudioContext::processAutomaticPullNodes(size_t framesToProcess)
 {
     ASSERT(isAudioThread());
 
     for (unsigned i = 0; i < m_renderingAutomaticPullNodes.size(); ++i)
         m_renderingAutomaticPullNodes[i]->processIfNecessary(framesToProcess);
 }
 
+void AudioContext::resolvePromisesForResumeOnMainThread()
+{
+    ASSERT(isMainThread());
+    AutoLocker locker(this);
+
+    for (unsigned k = 0; k < m_resumePromises.size(); ++k) {
+        if (m_contextState == Closed) {
+            m_resumePromises[k]->reject(
+                DOMException::create(InvalidStateError, "Cannot resume a context that has been released"));
+        } else {
+            m_resumePromises[k]->resolve();
+        }
+    }
+
+    m_resumePromises.clear();
+    m_isResolvingResumePromises = false;
+}
+
+void AudioContext::resolvePromisesForResume()
+{
+    // This runs inside the AudioContext's lock when handling pre-render tasks.
+    ASSERT(isAudioThread());
+    ASSERT(isGraphOwner());
+
+    // Resolve any pending promises created by resume(). Only do this we if haven't already started
+    // resolving these promises. This gets called very often and it takes some time to resolve the
+    // promises in the main thread.
+    if (!m_isResolvingResumePromises && m_resumePromises.size() > 0) {
+        m_isResolvingResumePromises = true;
+        callOnMainThread(bind(&AudioContext::resolvePromisesForResumeOnMainThread, this));
+    }
+}
+
+void AudioContext::resolvePromisesForSuspendOnMainThread()
+{
+    ASSERT(isMainThread());
+    AutoLocker locker(this);
+
+    // We can stop rendering now.
+    if (m_destinationNode)
+        stopRendering();
+
+    for (unsigned k = 0; k < m_suspendPromises.size(); ++k) {
+        if (m_contextState == Closed) {
+            m_suspendPromises[k]->reject(
+                DOMException::create(InvalidStateError, "Cannot suspend a context that has been released"));
+        } else {
+            m_suspendPromises[k]->resolve();
+        }
+    }
+
+    m_suspendPromises.clear();
+}
+
+void AudioContext::resolvePromisesForSuspend()
+{
+    // This runs inside the AudioContext's lock when handling pre-render tasks.
+    ASSERT(isAudioThread());
+    ASSERT(isGraphOwner());
+
+    // Resolve any pending promises created by suspend()
+    if (m_suspendPromises.size() > 0)
+        callOnMainThread(bind(&AudioContext::resolvePromisesForSuspendOnMainThread, this));
+
+}
+
 const AtomicString& AudioContext::interfaceName() const
 {
     return EventTargetNames::AudioContext;
 }
 
 ExecutionContext* AudioContext::executionContext() const
 {
     return m_isStopScheduled ? 0 : ActiveDOMObject::executionContext();
 }
 
 void AudioContext::startRendering()
 {
-    destination()->startRendering();
+    // This is called for both online and offline contexts.
+    ASSERT(isMainThread());
+    ASSERT(m_destinationNode);
+
+    if (m_contextState == Suspended) {
+        destination()->startRendering();
+        setContextState(Running);
+    }
+}
+
+void AudioContext::stopRendering()
+{
+    ASSERT(isMainThread());
+    ASSERT(m_destinationNode);
+    ASSERT(!isOfflineContext());
+
+    if (m_contextState == Running) {
+        destination()->stopRendering();
+        setContextState(Suspended);
+    }
 }
 
 void AudioContext::fireCompletionEvent()
 {
     ASSERT(isMainThread());
     if (!isMainThread())
         return;
 
     AudioBuffer* renderedBuffer = m_renderTarget.get();
 
+    setContextState(Closed);
+
     ASSERT(renderedBuffer);
     if (!renderedBuffer)
         return;
 
     // Avoid firing the event if the document has already gone away.
     if (executionContext()) {
         // Call the offline rendering completion event listener.
         dispatchEvent(OfflineAudioCompletionEvent::create(renderedBuffer));
     }
 }
@@ skipping 29 matching lines @@
 
     for (HashSet<AudioNode*>::iterator k = m_deferredCountModeChange.begin(); k != m_deferredCountModeChange.end(); ++k)
         (*k)->updateChannelCountMode();
 
     m_deferredCountModeChange.clear();
 }
 
 } // namespace blink
 
 #endif // ENABLE(WEB_AUDIO)