Add per thread heap enabled MarkingVisitorImpl

third_party/WebKit/Source/platform/heap/MarkingVisitorImpl.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef MarkingVisitorImpl_h
 #define MarkingVisitorImpl_h
 
 #include "platform/heap/Heap.h"
 #include "platform/heap/ThreadState.h"
 #include "platform/heap/Visitor.h"
 #include "wtf/Allocator.h"
 #include "wtf/Functional.h"
 #include "wtf/HashFunctions.h"
 #include "wtf/Locker.h"
 #include "wtf/TypeTraits.h"
 
 namespace blink {
 
-template <typename Derived>
+template <typename Derived, bool PerThreadHeapEnabled>
+struct HeapAccessor {
+    static ThreadHeap* heap(Derived* visitor);
+};
+
+template<typename Derived>
+struct HeapAccessor<Derived, true> {
+    static ThreadHeap* heap(Derived* visitor)
+    {
+        return &visitor->heap();
+    }
+};
+
+template<typename Derived>
+struct HeapAccessor<Derived, false> {
+    static ThreadHeap* heap(Derived* visitor)
+    {
+        return ThreadHeap::mainThreadHeap();
+    }
+};
+
+template <typename Derived, bool PerThreadHeapEnabled>
 class MarkingVisitorImpl {
     USING_FAST_MALLOC(MarkingVisitorImpl);
 protected:
     inline void markHeader(HeapObjectHeader* header, const void* objectPointer, TraceCallback callback)
     {
         ASSERT(header);
         ASSERT(objectPointer);
         if (!toDerived()->shouldMarkObject(objectPointer))
             return;
 
         // If you hit this ASSERT, it means that there is a dangling pointer
         // from a live thread heap to a dead thread heap.  We must eliminate
         // the dangling pointer.
         // Release builds don't have the ASSERT, but it is OK because
         // release builds will crash in the following header->isMarked()
         // because all the entries of the orphaned arenas are zapped.
         ASSERT(!pageFromObject(objectPointer)->orphaned());
 
         if (header->isMarked())
             return;
 
         ASSERT(ThreadState::current()->isInGC());
         ASSERT(toDerived()->getMarkingMode() != Visitor::WeakProcessing);
 
         header->mark();
 
         if (callback)
-            ThreadHeap::mainThreadHeap()->pushTraceCallback(const_cast<void*>(objectPointer), callback);
+            getThreadHeap()->pushTraceCallback(const_cast<void*>(objectPointer), callback);
     }
 
     inline void mark(const void* objectPointer, TraceCallback callback)
     {
         if (!objectPointer)
             return;
         HeapObjectHeader* header = HeapObjectHeader::fromPayload(objectPointer);
         markHeader(header, header->payload(), callback);
     }
 
     inline void registerDelayedMarkNoTracing(const void* objectPointer)
     {
         ASSERT(toDerived()->getMarkingMode() != Visitor::WeakProcessing);
-        ThreadHeap::mainThreadHeap()->pushPostMarkingCallback(const_cast<void*>(objectPointer), &markNoTracingCallback);
+        getThreadHeap()->pushPostMarkingCallback(const_cast<void*>(objectPointer), &markNoTracingCallback);
     }
 
     inline void registerWeakMembers(const void* closure, const void* objectPointer, WeakCallback callback)
     {
         ASSERT(toDerived()->getMarkingMode() != Visitor::WeakProcessing);
         // We don't want to run weak processings when taking a snapshot.
         if (toDerived()->getMarkingMode() == Visitor::SnapshotMarking)
             return;
-        ThreadHeap::mainThreadHeap()->pushThreadLocalWeakCallback(const_cast<void*>(closure), const_cast<void*>(objectPointer), callback);
+        getThreadHeap()->pushThreadLocalWeakCallback(const_cast<void*>(closure), const_cast<void*>(objectPointer), callback);
     }
 
     inline void registerWeakTable(const void* closure, EphemeronCallback iterationCallback, EphemeronCallback iterationDoneCallback)
     {
         ASSERT(toDerived()->getMarkingMode() != Visitor::WeakProcessing);
-        ThreadHeap::mainThreadHeap()->registerWeakTable(const_cast<void*>(closure), iterationCallback, iterationDoneCallback);
+        getThreadHeap()->registerWeakTable(const_cast<void*>(closure), iterationCallback, iterationDoneCallback);
     }
 
 #if ENABLE(ASSERT)
     inline bool weakTableRegistered(const void* closure)
     {
-        return ThreadHeap::mainThreadHeap()->weakTableRegistered(closure);
+        return getThreadHeap()->weakTableRegistered(closure);
     }
 #endif
 
     inline bool ensureMarked(const void* objectPointer)
     {
         if (!objectPointer)
             return false;
         if (!toDerived()->shouldMarkObject(objectPointer))
             return false;
 #if ENABLE(ASSERT)
@@ skipping 10 matching lines @@
         header->mark();
 #endif
         return true;
     }
 
     Derived* toDerived()
     {
         return static_cast<Derived*>(this);
     }
 
+    ThreadHeap* getThreadHeap()
+    {
+        return HeapAccessor<Derived, PerThreadHeapEnabled>::heap(toDerived());
+    }
+
 protected:
     inline void registerWeakCellWithCallback(void** cell, WeakCallback callback)
     {
         ASSERT(toDerived()->getMarkingMode() != Visitor::WeakProcessing);
         // We don't want to run weak processings when taking a snapshot.
         if (toDerived()->getMarkingMode() == Visitor::SnapshotMarking)
             return;
-        ThreadHeap::mainThreadHeap()->pushGlobalWeakCallback(cell, callback);
+        getThreadHeap()->pushGlobalWeakCallback(cell, callback);
     }
 
 private:
     static void markNoTracingCallback(Visitor* visitor, void* object)
     {
         visitor->markNoTracing(object);
     }
 };
 
 } // namespace blink
 
 #endif