gpu/command_buffer/service/transfer_buffer_manager.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef GPU_COMMAND_BUFFER_SERVICE_TRANSFER_BUFFER_MANAGER_H_
   6 #define GPU_COMMAND_BUFFER_SERVICE_TRANSFER_BUFFER_MANAGER_H_
   7
   8 #include <set>
   9 #include <vector>
  10
  11 #include "base/basictypes.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/containers/hash_tables.h"
  14 #include "base/memory/ref_counted.h"
  15 #include "base/trace_event/memory_dump_provider.h"
  16 #include "gpu/command_buffer/common/command_buffer_shared.h"
  17
  18 namespace gpu {
  19 namespace gles2 {
  20 class MemoryTracker;
  21 }
  22
  23 class GPU_EXPORT TransferBufferManagerInterface :
  24     public base::RefCounted<TransferBufferManagerInterface> {
  25  public:
  26   virtual bool RegisterTransferBuffer(int32 id,
  27                                       scoped_ptr<BufferBacking> buffer) = 0;
  28   virtual void DestroyTransferBuffer(int32 id) = 0;
  29   virtual scoped_refptr<Buffer> GetTransferBuffer(int32 id) = 0;
  30
  31  protected:
  32   friend class base::RefCounted<TransferBufferManagerInterface>;
  33
  34   virtual ~TransferBufferManagerInterface();
  35 };
  36
  37 class GPU_EXPORT TransferBufferManager
  38     : public TransferBufferManagerInterface,
  39       public base::trace_event::MemoryDumpProvider {
  40  public:
  41   explicit TransferBufferManager(gles2::MemoryTracker* memory_tracker);
  42
  43   // Overridden from base::trace_event::MemoryDumpProvider:
  44   bool OnMemoryDump(const base::trace_event::MemoryDumpArgs& args,
  45                     base::trace_event::ProcessMemoryDump* pmd) override;
  46
  47   bool Initialize();
  48   bool RegisterTransferBuffer(
  49       int32 id,
  50       scoped_ptr<BufferBacking> buffer_backing) override;
  51   void DestroyTransferBuffer(int32 id) override;
  52   scoped_refptr<Buffer> GetTransferBuffer(int32 id) override;
  53
  54  private:
  55   ~TransferBufferManager() override;
  56
  57   typedef base::hash_map<int32, scoped_refptr<Buffer> > BufferMap;
  58   BufferMap registered_buffers_;
  59   size_t shared_memory_bytes_allocated_;
  60   gles2::MemoryTracker* memory_tracker_;
  61
  62   DISALLOW_COPY_AND_ASSIGN(TransferBufferManager);
  63 };
  64
  65 }  // namespace gpu
  66
  67 #endif  // GPU_COMMAND_BUFFER_SERVICE_TRANSFER_BUFFER_MANAGER_H_