1 files changed, 261 insertions, 0 deletions
diff --git a/dom/canvas/WebGLMemoryTracker.cpp b/dom/canvas/WebGLMemoryTracker.cpp
new file mode 100644
index 000000000..8d3999f79
--- /dev/null
+++ b/dom/canvas/WebGLMemoryTracker.cpp
@@ -0,0 +1,261 @@
+/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/* 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 "WebGLMemoryTracker.h"
+
+#include "WebGLBuffer.h"
+#include "WebGLContext.h"
+#include "WebGLVertexAttribData.h"
+#include "WebGLProgram.h"
+#include "WebGLRenderbuffer.h"
+#include "WebGLShader.h"
+#include "WebGLTexture.h"
+#include "WebGLUniformLocation.h"
+
+namespace mozilla {
+
+NS_IMETHODIMP
+WebGLMemoryTracker::CollectReports(nsIHandleReportCallback* aHandleReport,
+                                   nsISupports* aData, bool)
+{
+    MOZ_COLLECT_REPORT(
+        "webgl-texture-memory", KIND_OTHER, UNITS_BYTES, GetTextureMemoryUsed(),
+        "Memory used by WebGL textures. The OpenGL implementation is free to "
+        "store these textures in either video memory or main memory. This "
+        "measurement is only a lower bound, actual memory usage may be higher "
+        "for example if the storage is strided.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-texture-count", KIND_OTHER, UNITS_COUNT, GetTextureCount(),
+        "Number of WebGL textures.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-buffer-memory", KIND_OTHER, UNITS_BYTES, GetBufferMemoryUsed(),
+        "Memory used by WebGL buffers. The OpenGL implementation is free to "
+        "store these buffers in either video memory or main memory. This "
+        "measurement is only a lower bound, actual memory usage may be higher "
+        "for example if the storage is strided.");
+
+    MOZ_COLLECT_REPORT(
+        "explicit/webgl/buffer-cache-memory", KIND_HEAP, UNITS_BYTES,
+        GetBufferCacheMemoryUsed(),
+        "Memory used by WebGL buffer caches. The WebGL implementation caches "
+        "the contents of element array buffers only. This adds up with the "
+        "'webgl-buffer-memory' value, but contrary to it, this one represents "
+        "bytes on the heap, not managed by OpenGL.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-buffer-count", KIND_OTHER, UNITS_COUNT, GetBufferCount(),
+        "Number of WebGL buffers.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-renderbuffer-memory", KIND_OTHER, UNITS_BYTES,
+        GetRenderbufferMemoryUsed(),
+        "Memory used by WebGL renderbuffers. The OpenGL implementation is free "
+        "to store these renderbuffers in either video memory or main memory. "
+        "This measurement is only a lower bound, actual memory usage may be "
+        "higher, for example if the storage is strided.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-renderbuffer-count", KIND_OTHER, UNITS_COUNT,
+        GetRenderbufferCount(),
+        "Number of WebGL renderbuffers.");
+
+    MOZ_COLLECT_REPORT(
+        "explicit/webgl/shader", KIND_HEAP, UNITS_BYTES, GetShaderSize(),
+        "Combined size of WebGL shader ASCII sources and translation logs "
+        "cached on the heap.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-shader-count", KIND_OTHER, UNITS_COUNT, GetShaderCount(),
+        "Number of WebGL shaders.");
+
+    MOZ_COLLECT_REPORT(
+        "webgl-context-count", KIND_OTHER, UNITS_COUNT, GetContextCount(),
+        "Number of WebGL contexts.");
+
+    return NS_OK;
+}
+
+NS_IMPL_ISUPPORTS(WebGLMemoryTracker, nsIMemoryReporter)
+
+StaticRefPtr<WebGLMemoryTracker> WebGLMemoryTracker::sUniqueInstance;
+
+WebGLMemoryTracker*
+WebGLMemoryTracker::UniqueInstance()
+{
+    if (!sUniqueInstance) {
+        sUniqueInstance = new WebGLMemoryTracker;
+        sUniqueInstance->InitMemoryReporter();
+    }
+    return sUniqueInstance;
+}
+
+WebGLMemoryTracker::WebGLMemoryTracker()
+{
+}
+
+void
+WebGLMemoryTracker::InitMemoryReporter()
+{
+    RegisterWeakMemoryReporter(this);
+}
+
+WebGLMemoryTracker::~WebGLMemoryTracker()
+{
+    UnregisterWeakMemoryReporter(this);
+}
+
+MOZ_DEFINE_MALLOC_SIZE_OF(WebGLBufferMallocSizeOf)
+
+int64_t
+WebGLMemoryTracker::GetBufferCacheMemoryUsed()
+{
+    const ContextsArrayType& contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLBuffer* buffer = contexts[i]->mBuffers.getFirst();
+             buffer;
+             buffer = buffer->getNext())
+        {
+            if (buffer->Content() == WebGLBuffer::Kind::ElementArray) {
+                result += buffer->SizeOfIncludingThis(WebGLBufferMallocSizeOf);
+            }
+        }
+    }
+    return result;
+}
+
+MOZ_DEFINE_MALLOC_SIZE_OF(WebGLShaderMallocSizeOf)
+
+int64_t
+WebGLMemoryTracker::GetShaderSize()
+{
+    const ContextsArrayType& contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLShader* shader = contexts[i]->mShaders.getFirst();
+             shader;
+             shader = shader->getNext())
+        {
+            result += shader->SizeOfIncludingThis(WebGLShaderMallocSizeOf);
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetTextureMemoryUsed()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLTexture* texture = contexts[i]->mTextures.getFirst();
+             texture;
+             texture = texture->getNext())
+        {
+            result += texture->MemoryUsage();
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetTextureCount()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLTexture* texture = contexts[i]->mTextures.getFirst();
+             texture;
+             texture = texture->getNext())
+        {
+            result++;
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetBufferMemoryUsed()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLBuffer* buffer = contexts[i]->mBuffers.getFirst();
+             buffer;
+             buffer = buffer->getNext())
+        {
+            result += buffer->ByteLength();
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetBufferCount()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLBuffer* buffer = contexts[i]->mBuffers.getFirst();
+             buffer;
+             buffer = buffer->getNext())
+        {
+            result++;
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetRenderbufferMemoryUsed()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLRenderbuffer* rb = contexts[i]->mRenderbuffers.getFirst();
+             rb;
+             rb = rb->getNext())
+        {
+            result += rb->MemoryUsage();
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetRenderbufferCount()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLRenderbuffer* rb = contexts[i]->mRenderbuffers.getFirst();
+             rb;
+             rb = rb->getNext())
+        {
+            result++;
+        }
+    }
+    return result;
+}
+
+/*static*/ int64_t
+WebGLMemoryTracker::GetShaderCount()
+{
+    const ContextsArrayType & contexts = Contexts();
+    int64_t result = 0;
+    for(size_t i = 0; i < contexts.Length(); ++i) {
+        for (const WebGLShader* shader = contexts[i]->mShaders.getFirst();
+             shader;
+             shader = shader->getNext())
+        {
+            result++;
+        }
+    }
+    return result;
+}
+
+} // namespace mozilla