DevTools: [HeapProfiler] Provide API for heap objects tracking info.

Source/bindings/v8/ScriptProfiler.cpp

 /*
  * Copyright (c) 2011, 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:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 25 matching lines @@
 #include "V8Node.h"
 #include "bindings/v8/RetainedDOMInfo.h"
 #include "bindings/v8/ScriptObject.h"
 #include "bindings/v8/V8Binding.h"
 #include "bindings/v8/V8DOMWrapper.h"
 #include "bindings/v8/WrapperTypeInfo.h"
 #include "core/dom/WebCoreMemoryInstrumentation.h"
 #include "core/inspector/BindingVisitors.h"
 
 #include <v8-profiler.h>
+#include <v8.h>
 
 #include "wtf/ThreadSpecific.h"
 
 namespace WebCore {
 
 typedef HashMap<String, double> ProfileNameIdleTimeMap;
 
 void ScriptProfiler::start(ScriptState* state, const String& title)
 {
     ProfileNameIdleTimeMap* profileNameIdleTimeMap = ScriptProfiler::currentProfileNameIdleTimeMap();
@@ skipping 136 matching lines @@
         }
         return 0;
     }
 
 private:
     Vector<CString> m_strings;
 };
 
 } // namespace
 
+void ScriptProfiler::startTrackingHeapObjects()
+{
+    v8::Isolate::GetCurrent()->GetHeapProfiler()->StartTrackingHeapObjects();
+}
+
+namespace {
+
+class HeapStatsStream : public v8::OutputStream {
+public:
+    HeapStatsStream(ScriptProfiler::OutputStream* stream) : m_stream(stream) { }
+    virtual void EndOfStream() OVERRIDE { }
+
+    virtual WriteResult WriteAsciiChunk(char* data, int size) OVERRIDE
+    {
+        ASSERT(false);
+        return kAbort;
+    }
+
+    virtual WriteResult WriteHeapStatsChunk(v8::HeapStatsUpdate* updateData, int count) OVERRIDE
+    {
+        Vector<uint32_t> rawData(count * 3);
+        for (int i = 0; i < count; ++i) {
+            int offset = i * 3;
+            rawData[offset] = updateData[i].index;
+            rawData[offset + 1] = updateData[i].count;
+            rawData[offset + 2] = updateData[i].size;
+        }
+        m_stream->write(rawData.data(), rawData.size());
+        return kContinue;
+    }
+
+private:
+    ScriptProfiler::OutputStream* m_stream;
+};
+
+}
+
+unsigned ScriptProfiler::requestHeapStatsUpdate(ScriptProfiler::OutputStream* stream)
+{
+    HeapStatsStream heapStatsStream(stream);
+    return v8::Isolate::GetCurrent()->GetHeapProfiler()->GetHeapStats(&heapStatsStream);
+}
+
+void ScriptProfiler::stopTrackingHeapObjects()
+{
+    v8::Isolate::GetCurrent()->GetHeapProfiler()->StopTrackingHeapObjects();
+}
+
 // FIXME: This method should receive a ScriptState, from which we should retrieve an Isolate.
 PassRefPtr<ScriptHeapSnapshot> ScriptProfiler::takeHeapSnapshot(const String& title, HeapSnapshotProgress* control)
 {
     v8::Isolate* isolate = v8::Isolate::GetCurrent();
     v8::HeapProfiler* profiler = isolate->GetHeapProfiler();
     if (!profiler)
         return 0;
     v8::HandleScope handleScope(isolate);
     ASSERT(control);
     ActivityControlAdapter adapter(control);
@@ skipping 101 matching lines @@
 }
 
 ProfileNameIdleTimeMap* ScriptProfiler::currentProfileNameIdleTimeMap()
 {
     AtomicallyInitializedStatic(WTF::ThreadSpecific<ProfileNameIdleTimeMap>*, map = new WTF::ThreadSpecific<ProfileNameIdleTimeMap>);
     return *map;
 }
 
 } // namespace WebCore