Line data    Source code

       1             : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
       2             : /* This Source Code Form is subject to the terms of the Mozilla Public
       3             :  * License, v. 2.0. If a copy of the MPL was not distributed with this
       4             :  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
       5             : 
       6             : #ifndef WEBGL_MEMORY_TRACKER_H_
       7             : #define WEBGL_MEMORY_TRACKER_H_
       8             : 
       9             : #include "mozilla/StaticPtr.h"
      10             : #include "nsIMemoryReporter.h"
      11             : 
      12             : namespace mozilla {
      13             : 
      14             : class WebGLContext;
      15             : 
      16             : class WebGLMemoryTracker : public nsIMemoryReporter
      17             : {
      18             :     NS_DECL_THREADSAFE_ISUPPORTS
      19             :     NS_DECL_NSIMEMORYREPORTER
      20             : 
      21             :     WebGLMemoryTracker();
      22             :     static StaticRefPtr<WebGLMemoryTracker> sUniqueInstance;
      23             : 
      24             :     // Here we store plain pointers, not RefPtrs: we don't want the
      25             :     // WebGLMemoryTracker unique instance to keep alive all
      26             :     // WebGLContexts ever created.
      27             :     typedef nsTArray<const WebGLContext*> ContextsArrayType;
      28             :     ContextsArrayType mContexts;
      29             : 
      30             :     void InitMemoryReporter();
      31             : 
      32             :     static WebGLMemoryTracker* UniqueInstance();
      33             : 
      34           0 :     static ContextsArrayType& Contexts() { return UniqueInstance()->mContexts; }
      35             : 
      36             :     friend class WebGLContext;
      37             : 
      38             :   public:
      39             : 
      40           0 :     static void AddWebGLContext(const WebGLContext* c) {
      41           0 :         Contexts().AppendElement(c);
      42           0 :     }
      43             : 
      44           0 :     static void RemoveWebGLContext(const WebGLContext* c) {
      45           0 :         ContextsArrayType & contexts = Contexts();
      46           0 :         contexts.RemoveElement(c);
      47           0 :         if (contexts.IsEmpty()) {
      48           0 :             sUniqueInstance = nullptr;
      49             :         }
      50           0 :     }
      51             : 
      52             :   private:
      53             :     virtual ~WebGLMemoryTracker();
      54             : 
      55             :     static int64_t GetTextureMemoryUsed();
      56             : 
      57             :     static int64_t GetTextureCount();
      58             : 
      59             :     static int64_t GetBufferMemoryUsed();
      60             : 
      61             :     static int64_t GetBufferCacheMemoryUsed();
      62             : 
      63             :     static int64_t GetBufferCount();
      64             : 
      65             :     static int64_t GetRenderbufferMemoryUsed();
      66             : 
      67             :     static int64_t GetRenderbufferCount();
      68             : 
      69             :     static int64_t GetShaderSize();
      70             : 
      71             :     static int64_t GetShaderCount();
      72             : 
      73           0 :     static int64_t GetContextCount() {
      74           0 :         return Contexts().Length();
      75             :     }
      76             : };
      77             : 
      78             : } // namespace mozilla
      79             : 
      80             : #endif // WEBGL_MEMORY_TRACKER_H_