OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ | 5 #ifndef COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ |
6 #define COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ | 6 #define COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "base/macros.h" | 10 #include "base/macros.h" |
11 #include "base/memory/weak_ptr.h" | 11 #include "base/memory/ref_counted.h" |
12 #include "base/single_thread_task_runner.h" | 12 #include "base/single_thread_task_runner.h" |
13 #include "base/synchronization/waitable_event.h" | 13 #include "base/synchronization/waitable_event.h" |
| 14 #include "base/threading/thread_task_runner_handle.h" |
14 #include "gpu/command_buffer/client/gpu_memory_buffer_manager.h" | 15 #include "gpu/command_buffer/client/gpu_memory_buffer_manager.h" |
15 #include "gpu/ipc/host/gpu_memory_buffer_support.h" | 16 #include "gpu/ipc/host/gpu_memory_buffer_support.h" |
16 | 17 |
17 namespace ui { | 18 namespace ui { |
18 namespace mojom { | 19 namespace mojom { |
19 class GpuService; | 20 class GpuService; |
20 } | 21 } |
21 } | 22 } |
22 | 23 |
23 namespace viz { | 24 namespace viz { |
24 | 25 |
25 // This GpuMemoryBufferManager implementation is for [de]allocating gpu memory | 26 // This GpuMemoryBufferManager implementation is for [de]allocating gpu memory |
26 // from the gpu process over the mojom.GpuService api. | 27 // from the gpu process over the mojom.GpuService api. |
27 class ServerGpuMemoryBufferManager : public gpu::GpuMemoryBufferManager { | 28 class ServerGpuMemoryBufferManager |
| 29 : public gpu::GpuMemoryBufferManager, |
| 30 public base::RefCountedThreadSafe<ServerGpuMemoryBufferManager> { |
28 public: | 31 public: |
29 ServerGpuMemoryBufferManager(ui::mojom::GpuService* gpu_service, | 32 ServerGpuMemoryBufferManager( |
30 int client_id); | 33 ui::mojom::GpuService* gpu_service, |
31 ~ServerGpuMemoryBufferManager() override; | 34 int client_id, |
| 35 scoped_refptr<base::SingleThreadTaskRunner> task_runner = |
| 36 base::ThreadTaskRunnerHandle::Get()); |
32 | 37 |
33 void DestroyGpuMemoryBuffer(gfx::GpuMemoryBufferId id, | 38 void DestroyGpuMemoryBuffer(gfx::GpuMemoryBufferId id, |
34 int client_id, | 39 int client_id, |
35 const gpu::SyncToken& sync_token); | 40 const gpu::SyncToken& sync_token); |
36 | 41 |
37 void DestroyAllGpuMemoryBufferForClient(int client_id); | 42 void DestroyAllGpuMemoryBufferForClient(int client_id); |
38 | 43 |
39 void AllocateGpuMemoryBufferHandle( | 44 void AllocateGpuMemoryBufferHandle( |
40 gfx::GpuMemoryBufferId id, | 45 gfx::GpuMemoryBufferId id, |
41 int client_id, | 46 int client_id, |
42 const gfx::Size& size, | 47 const gfx::Size& size, |
43 gfx::BufferFormat format, | 48 gfx::BufferFormat format, |
44 gfx::BufferUsage usage, | 49 gfx::BufferUsage usage, |
45 gpu::SurfaceHandle surface_handle, | 50 gpu::SurfaceHandle surface_handle, |
46 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback); | 51 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback); |
47 | 52 |
48 // Overridden from gpu::GpuMemoryBufferManager: | 53 // Overridden from gpu::GpuMemoryBufferManager: |
49 std::unique_ptr<gfx::GpuMemoryBuffer> CreateGpuMemoryBuffer( | 54 std::unique_ptr<gfx::GpuMemoryBuffer> CreateGpuMemoryBuffer( |
50 const gfx::Size& size, | 55 const gfx::Size& size, |
51 gfx::BufferFormat format, | 56 gfx::BufferFormat format, |
52 gfx::BufferUsage usage, | 57 gfx::BufferUsage usage, |
53 gpu::SurfaceHandle surface_handle) override; | 58 gpu::SurfaceHandle surface_handle) override; |
54 void SetDestructionSyncToken(gfx::GpuMemoryBuffer* buffer, | 59 void SetDestructionSyncToken(gfx::GpuMemoryBuffer* buffer, |
55 const gpu::SyncToken& sync_token) override; | 60 const gpu::SyncToken& sync_token) override; |
56 | 61 |
| 62 void set_gpu_service(ui::mojom::GpuService* service) { |
| 63 gpu_service_ = service; |
| 64 } |
| 65 |
57 private: | 66 private: |
| 67 friend class base::RefCountedThreadSafe<ServerGpuMemoryBufferManager>; |
| 68 ~ServerGpuMemoryBufferManager() override; |
| 69 |
58 void OnGpuMemoryBufferAllocated( | 70 void OnGpuMemoryBufferAllocated( |
59 int client_id, | 71 int client_id, |
60 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback, | 72 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback, |
61 const gfx::GpuMemoryBufferHandle& handle); | 73 const gfx::GpuMemoryBufferHandle& handle); |
62 void SignalOnGpuMemoryBufferAllocated( | 74 void SignalOnGpuMemoryBufferAllocated( |
63 gfx::GpuMemoryBufferHandle* handle, | 75 std::unique_ptr<gfx::GpuMemoryBuffer>* buffer, |
64 base::WaitableEvent* wait_event, | 76 base::WaitableEvent* wait_event, |
| 77 const gfx::Size& size, |
| 78 gfx::BufferFormat format, |
| 79 gfx::BufferUsage usage, |
65 const gfx::GpuMemoryBufferHandle& allocated_handle); | 80 const gfx::GpuMemoryBufferHandle& allocated_handle); |
66 | 81 |
67 ui::mojom::GpuService* gpu_service_; | 82 ui::mojom::GpuService* gpu_service_; |
68 const int client_id_; | 83 const int client_id_; |
69 int next_gpu_memory_id_ = 1; | 84 int next_gpu_memory_id_ = 1; |
70 | 85 |
71 using NativeBuffers = | 86 using NativeBuffers = |
72 std::unordered_set<gfx::GpuMemoryBufferId, | 87 std::unordered_set<gfx::GpuMemoryBufferId, |
73 BASE_HASH_NAMESPACE::hash<gfx::GpuMemoryBufferId>>; | 88 BASE_HASH_NAMESPACE::hash<gfx::GpuMemoryBufferId>>; |
74 std::unordered_map<int, NativeBuffers> native_buffers_; | 89 std::unordered_map<int, NativeBuffers> native_buffers_; |
75 std::unordered_set<int> pending_buffers_; | 90 std::unordered_set<int> pending_buffers_; |
76 | 91 |
77 const gpu::GpuMemoryBufferConfigurationSet native_configurations_; | 92 const gpu::GpuMemoryBufferConfigurationSet native_configurations_; |
78 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 93 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
79 base::WeakPtrFactory<ServerGpuMemoryBufferManager> weak_factory_; | |
80 | 94 |
81 DISALLOW_COPY_AND_ASSIGN(ServerGpuMemoryBufferManager); | 95 DISALLOW_COPY_AND_ASSIGN(ServerGpuMemoryBufferManager); |
82 }; | 96 }; |
83 | 97 |
84 } // namespace viz | 98 } // namespace viz |
85 | 99 |
86 #endif // COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ | 100 #endif // COMPONENTS_VIZ_COMMON_SERVER_GPU_MEMORY_BUFFER_MANAGER_H_ |
OLD | NEW |