Implement SetViewContents in the client lib

mojo/services/public/cpp/view_manager/lib/view_manager_synchronizer.cc

 // Copyright 2014 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 "mojo/services/public/cpp/view_manager/lib/view_manager_synchronizer.h"
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "mojo/public/cpp/shell/connect.h"
 #include "mojo/public/interfaces/shell/shell.mojom.h"
 #include "mojo/services/public/cpp/view_manager/lib/view_manager_private.h"
 #include "mojo/services/public/cpp/view_manager/lib/view_private.h"
 #include "mojo/services/public/cpp/view_manager/lib/view_tree_node_private.h"
 #include "mojo/services/public/cpp/view_manager/util.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/gfx/codec/png_codec.h"
 
 namespace mojo {
 namespace view_manager {
 
 uint32_t MakeTransportId(uint16_t connection_id, uint16_t local_id) {
   return (connection_id << 16) | local_id;
 }
 
 // Helper called to construct a local node/view object from transport data.
 ViewTreeNode* AddNodeToViewManager(ViewManager* manager,
@@ skipping 69 matching lines @@
     TYPE_DESTROY_VIEW,
     // Node creation and destruction.
     TYPE_CREATE_VIEW_TREE_NODE,
     TYPE_DESTROY_VIEW_TREE_NODE,
     // Modifications to the hierarchy (addition of or removal of nodes from a
     // parent.)
     TYPE_HIERARCHY,
     // View replacement.
     TYPE_SET_ACTIVE_VIEW,
     // Node bounds.
-    TYPE_SET_BOUNDS
+    TYPE_SET_BOUNDS,
+    // View contents
+    TYPE_SET_VIEW_CONTENTS
   };
 
   ViewManagerTransaction(TransactionType transaction_type,
                          ViewManagerSynchronizer* synchronizer)
       : transaction_type_(transaction_type),
         committed_(false),
         synchronizer_(synchronizer) {
   }
 
   // Overridden to perform transaction-specific commit actions.
@@ skipping 211 matching lines @@
   virtual void DoActionCompleted(bool success) OVERRIDE {
     // TODO(beng): recovery?
   }
 
   const TransportNodeId node_id_;
   const gfx::Rect bounds_;
 
   DISALLOW_COPY_AND_ASSIGN(SetBoundsTransaction);
 };
 
+class SetViewContentsTransaction : public ViewManagerTransaction {
+ public:
+  SetViewContentsTransaction(TransportViewId view_id,
+                             const SkBitmap& contents,
+                             ViewManagerSynchronizer* synchronizer)
+      : ViewManagerTransaction(TYPE_SET_VIEW_CONTENTS, synchronizer),
+        view_id_(view_id),
+        contents_(contents) {}
+  virtual ~SetViewContentsTransaction() {}
+
+ private:
+  // Overridden from ViewManagerTransaction:
+  virtual void DoCommit() OVERRIDE {
+    std::vector<unsigned char> data;
+    gfx::PNGCodec::EncodeBGRASkBitmap(contents_, false, &data);
+
+    void* memory = NULL;
+    ScopedSharedBufferHandle duped;
+    bool result = CreateMapAndDupSharedBuffer(data.size(),
+                                              &memory,
+                                              &shared_state_handle_,
+                                              &duped);
+    if (!result)
+      return;
+
+    memcpy(memory, &data[0], data.size());
+
+    AllocationScope scope;
+    service()->SetViewContents(view_id_, duped.Pass(), data.size(),
+                               ActionCompletedCallback());
+  }
+  virtual void DoActionCompleted(bool success) OVERRIDE {
+    // TODO(beng): recovery?
+  }
+
+  bool CreateMapAndDupSharedBuffer(size_t size,
+                                   void** memory,
+                                   ScopedSharedBufferHandle* handle,
+                                   ScopedSharedBufferHandle* duped) {
+    MojoResult result = CreateSharedBuffer(NULL, size, handle);
+    if (result != MOJO_RESULT_OK)
+      return false;
+    DCHECK(handle->is_valid());
+
+    result = DuplicateBuffer(handle->get(), NULL, duped);
+    if (result != MOJO_RESULT_OK)
+      return false;
+    DCHECK(duped->is_valid());
+
+    result = MapBuffer(
+        handle->get(), 0, size, memory, MOJO_MAP_BUFFER_FLAG_NONE);
+    if (result != MOJO_RESULT_OK)
+      return false;
+    DCHECK(*memory);
+
+    return true;
+  }
+
+  const TransportViewId view_id_;
+  const SkBitmap contents_;
+  ScopedSharedBufferHandle shared_state_handle_;
+
+  DISALLOW_COPY_AND_ASSIGN(SetViewContentsTransaction);
+};
+
 ViewManagerSynchronizer::ViewManagerSynchronizer(ViewManager* view_manager)
     : view_manager_(view_manager),
       connected_(false),
       connection_id_(0),
       next_id_(1),
       next_server_change_id_(0),
       sync_factory_(this),
       init_loop_(NULL) {
   ConnectTo(ViewManagerPrivate(view_manager_).shell(), "mojo:mojo_view_manager",
             &service_);
@@ skipping 82 matching lines @@
 }
 
 void ViewManagerSynchronizer::SetBounds(TransportNodeId node_id,
                                         const gfx::Rect& bounds) {
   DCHECK(connected_);
   pending_transactions_.push_back(
       new SetBoundsTransaction(node_id, bounds, this));
   Sync();
 }
 
+void ViewManagerSynchronizer::SetViewContents(TransportViewId view_id,
+                                              const SkBitmap& contents) {
+  DCHECK(connected_);
+  pending_transactions_.push_back(
+      new SetViewContentsTransaction(view_id, contents, this));
+  Sync();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // ViewManagerSynchronizer, IViewManagerClient implementation:
 
 void ViewManagerSynchronizer::OnConnectionEstablished(
     TransportConnectionId connection_id,
     TransportChangeId next_server_change_id,
     const Array<INode>& nodes) {
   connected_ = true;
   connection_id_ = connection_id;
   next_server_change_id_ = next_server_change_id;
@@ skipping 91 matching lines @@
 void ViewManagerSynchronizer::RemoveFromPendingQueue(
     ViewManagerTransaction* transaction) {
   DCHECK_EQ(transaction, pending_transactions_.front());
   pending_transactions_.erase(pending_transactions_.begin());
   if (pending_transactions_.empty() && !changes_acked_callback_.is_null())
     changes_acked_callback_.Run();
 }
 
 }  // namespace view_manager
 }  // namespace mojo