Add CommandBuffer::SetGetBuffer

gpu/command_buffer/service/command_buffer_service.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "gpu/command_buffer/service/command_buffer_service.h"
 
 #include <limits>
 
 #include "base/callback.h"
 #include "base/process_util.h"
 #include "gpu/command_buffer/common/cmd_buffer_common.h"
 
 using ::base::SharedMemory;
 
 namespace gpu {
 
 CommandBufferService::CommandBufferService()
-    : num_entries_(0),
+    : ring_buffer_id_(-1),
+      num_entries_(0),
       get_offset_(0),
       put_offset_(0),
       token_(0),
       generation_(0),
       error_(error::kNoError) {
   // Element zero is always NULL.
   registered_objects_.push_back(Buffer());
 }
 
 CommandBufferService::~CommandBufferService() {
-  delete ring_buffer_.shared_memory;
-
   for (size_t i = 0; i < registered_objects_.size(); ++i) {
     if (registered_objects_[i].shared_memory)
       delete registered_objects_[i].shared_memory;
   }
 }
 
-bool CommandBufferService::Initialize(int32 size) {
-  // Fail if already initialized.
-  if (ring_buffer_.shared_memory) {
-    LOG(ERROR) << "Failed because already initialized.";
-    return false;
-  }
-
-  if (size <= 0 || size > kMaxCommandBufferSize) {
-    LOG(ERROR) << "Failed because command buffer size was invalid.";
-    return false;
-  }
-
-  num_entries_ = size / sizeof(CommandBufferEntry);
-
-  SharedMemory shared_memory;
-  if (!shared_memory.CreateAnonymous(size)) {
-    LOG(ERROR) << "Failed to create shared memory for command buffer.";
-    return true;
-  }
-
-  return Initialize(&shared_memory, size);
-}
-
-bool CommandBufferService::Initialize(base::SharedMemory* buffer, int32 size) {
-  // Fail if already initialized.
-  if (ring_buffer_.shared_memory) {
-    LOG(ERROR) << "Failed because already initialized.";
-    return false;
-  }
-
-  base::SharedMemoryHandle shared_memory_handle;
-  if (!buffer->ShareToProcess(base::GetCurrentProcessHandle(),
-                              &shared_memory_handle)) {
-    LOG(ERROR) << "Failed to duplicate command buffer shared memory handle.";
-    return false;
-  }
-
-  ring_buffer_.shared_memory = new base::SharedMemory(shared_memory_handle,
-                                                      false);
-  if (!ring_buffer_.shared_memory->Map(size)) {
-    LOG(ERROR) << "Failed because ring buffer could not be created or mapped ";
-    delete ring_buffer_.shared_memory;
-    ring_buffer_.shared_memory = NULL;
-    return false;
-  }
-
-  ring_buffer_.ptr = ring_buffer_.shared_memory->memory();
-  ring_buffer_.size = size;
-  num_entries_ = size / sizeof(CommandBufferEntry);
-
+bool CommandBufferService::Initialize() {
   return true;
 }
 
-Buffer CommandBufferService::GetRingBuffer() {
-  return ring_buffer_;
-}
-
 CommandBufferService::State CommandBufferService::GetState() {
   State state;
   state.num_entries = num_entries_;
   state.get_offset = get_offset_;
   state.put_offset = put_offset_;
   state.token = token_;
   state.error = error_;
   state.context_lost_reason = context_lost_reason_;
   state.generation = ++generation_;
 
@@ skipping 26 matching lines @@
     return;
   }
 
   put_offset_ = put_offset;
 
   if (put_offset_change_callback_.get()) {
     put_offset_change_callback_->Run();
   }
 }
 
+void CommandBufferService::SetGetBuffer(int32 shm_id) {
+  DCHECK_EQ(-1, ring_buffer_id_);
+  DCHECK_EQ(put_offset_, get_offset_);  // Only if it's empty.
+  ring_buffer_ = GetTransferBuffer(shm_id);
+  DCHECK(ring_buffer_.ptr);
+  ring_buffer_id_ = shm_id;
+  num_entries_ = ring_buffer_.size / sizeof(CommandBufferEntry);
+  put_offset_ = 0;
+  SetGetOffset(0);
+}
+
 void CommandBufferService::SetGetOffset(int32 get_offset) {
   DCHECK(get_offset >= 0 && get_offset < num_entries_);
   get_offset_ = get_offset;
 }
 
 int32 CommandBufferService::CreateTransferBuffer(size_t size,
                                                  int32 id_request) {
   SharedMemory buffer;
   if (!buffer.CreateAnonymous(size))
     return -1;
@@ skipping 69 matching lines @@
   if (handle <= 0)
     return;
 
   if (static_cast<size_t>(handle) >= registered_objects_.size())
     return;
 
   delete registered_objects_[handle].shared_memory;
   registered_objects_[handle] = Buffer();
   unused_registered_object_elements_.insert(handle);
 
+  if (handle == ring_buffer_id_) {
+    ring_buffer_id_ = -1;
+    ring_buffer_ = Buffer();
+    num_entries_ = 0;
+    get_offset_ = 0;
+    put_offset_ = 0;
+  }
+
   // Remove all null objects from the end of the vector. This allows the vector
   // to shrink when, for example, all objects are unregistered. Note that this
   // loop never removes element zero, which is always NULL.
   while (registered_objects_.size() > 1 &&
       !registered_objects_.back().shared_memory) {
     registered_objects_.pop_back();
     unused_registered_object_elements_.erase(
         static_cast<int32>(registered_objects_.size()));
   }
 }
@@ skipping 29 matching lines @@
     Callback0::Type* callback) {
   put_offset_change_callback_.reset(callback);
 }
 
 void CommandBufferService::SetParseErrorCallback(
     Callback0::Type* callback) {
   parse_error_callback_.reset(callback);
 }
 
 }  // namespace gpu