Oilpan: Move MediaStreamSource, MediaStreamComponent and MediaStreamDescriptor to oilpan's heap

Source/platform/mediastream/MediaStreamSource.cpp

 /*
  * Copyright (C) 2012 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
@@ skipping 16 matching lines @@
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 #include "platform/mediastream/MediaStreamSource.h"
 
 
 namespace blink {
 
-PassRefPtr<MediaStreamSource> MediaStreamSource::create(const String& id, Type type, const String& name, ReadyState readyState, bool requiresConsumer)
+// The disposer pattern actually makes the deletion of the extra data happen
+// earlier and not later. The disposer makes sure that the extra data is
+// destructed in weak processing which is run before sweeping and therefore
+// all the objects are still alive and can be touched.
+//
+// FIXME: Oilpan: This disposer pattern is duplicated in a lot of places.
+// We should create a good abstraction class for this and remove the code duplication.
+class MediaStreamSourceDisposer {
+public:
+    explicit MediaStreamSourceDisposer(MediaStreamSource& source) : m_source(source) { }
+    ~MediaStreamSourceDisposer()
+    {
+        m_source.dispose();
+    }
+
+private:
+    MediaStreamSource& m_source;
+};
+
+typedef HeapHashMap<WeakMember<MediaStreamSource>, OwnPtr<MediaStreamSourceDisposer> > SourceDisposers;
+
+static SourceDisposers& sourceDisposers()
 {
-    return adoptRef(new MediaStreamSource(id, type, name, readyState, requiresConsumer));
+    DEFINE_STATIC_LOCAL(Persistent<SourceDisposers>, disposers, (new SourceDisposers));
+    return *disposers;
+}
+
+MediaStreamSource* MediaStreamSource::create(const String& id, Type type, const String& name, ReadyState readyState, bool requiresConsumer)
+{
+    return new MediaStreamSource(id, type, name, readyState, requiresConsumer);
 }
 
 MediaStreamSource::MediaStreamSource(const String& id, Type type, const String& name, ReadyState readyState, bool requiresConsumer)
     : m_id(id)
     , m_type(type)
     , m_name(name)
     , m_readyState(readyState)
     , m_requiresConsumer(requiresConsumer)
 {
+    sourceDisposers().add(this, adoptPtr(new MediaStreamSourceDisposer(*this)));
+}
+
+void MediaStreamSource::dispose()
+{
+    m_extraData = nullptr;
 }
 
 void MediaStreamSource::setReadyState(ReadyState readyState)
 {
     if (m_readyState != ReadyStateEnded && m_readyState != readyState) {
         m_readyState = readyState;
-        for (Vector<Observer*>::iterator i = m_observers.begin(); i != m_observers.end(); ++i)
+        for (HeapHashSet<WeakMember<Observer> >::iterator i = m_observers.begin(); i != m_observers.end(); ++i)
             (*i)->sourceChangedState();
     }
 }
 
 void MediaStreamSource::addObserver(MediaStreamSource::Observer* observer)
 {
-    m_observers.append(observer);
-}
-
-void MediaStreamSource::removeObserver(MediaStreamSource::Observer* observer)
-{
-    size_t pos = m_observers.find(observer);
-    if (pos != kNotFound)
-        m_observers.remove(pos);
+    ASSERT(!m_observers.contains(observer));
+    m_observers.add(observer);
 }
 
 void MediaStreamSource::addAudioConsumer(AudioDestinationConsumer* consumer)
 {
     ASSERT(m_requiresConsumer);
     MutexLocker locker(m_audioConsumersLock);
     m_audioConsumers.add(consumer);
 }
 
 bool MediaStreamSource::removeAudioConsumer(AudioDestinationConsumer* consumer)
@@ skipping 16 matching lines @@
 }
 
 void MediaStreamSource::consumeAudio(AudioBus* bus, size_t numberOfFrames)
 {
     ASSERT(m_requiresConsumer);
     MutexLocker locker(m_audioConsumersLock);
     for (HeapHashSet<Member<AudioDestinationConsumer> >::iterator it = m_audioConsumers.begin(); it != m_audioConsumers.end(); ++it)
         (*it)->consumeAudio(bus, numberOfFrames);
 }
 
+void MediaStreamSource::trace(Visitor* visitor)
+{
+    visitor->trace(m_observers);
+    visitor->trace(m_audioConsumers);
+}
+
 } // namespace blink
+