Add StreamTexture resgistration for the Browser

gpu/ipc/service/stream_texture_android.cc

 // Copyright (c) 2013 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/ipc/service/stream_texture_android.h"
 
 #include <string.h>
 
 #include "base/bind.h"
 #include "base/strings/stringize_macros.h"
 #include "gpu/command_buffer/service/context_group.h"
 #include "gpu/command_buffer/service/context_state.h"
 #include "gpu/command_buffer/service/gles2_cmd_decoder.h"
 #include "gpu/command_buffer/service/texture_manager.h"
+#include "gpu/ipc/common/android/surface_texture_manager.h"
 #include "gpu/ipc/common/android/surface_texture_peer.h"
 #include "gpu/ipc/common/gpu_messages.h"
 #include "gpu/ipc/service/gpu_channel.h"
 #include "ui/gfx/geometry/size.h"
 #include "ui/gl/gl_context.h"
 #include "ui/gl/gl_helper.h"
 #include "ui/gl/scoped_binders.h"
 #include "ui/gl/scoped_make_current.h"
 
 namespace gpu {
@@ skipping 42 matching lines @@
       owner_stub_(owner_stub),
       route_id_(route_id),
       has_listener_(false),
       texture_id_(texture_id),
       framebuffer_(0),
       vertex_shader_(0),
       fragment_shader_(0),
       program_(0),
       vertex_buffer_(0),
       u_xform_location_(-1),
+      registered_surface_texture_id(0),
       weak_factory_(this) {
   owner_stub->AddDestructionObserver(this);
   memset(current_matrix_, 0, sizeof(current_matrix_));
   owner_stub->channel()->AddRoute(route_id, owner_stub->stream_id(), this);
   surface_texture_->SetFrameAvailableCallback(base::Bind(
       &StreamTexture::OnFrameAvailable, weak_factory_.GetWeakPtr()));
 }
 
 StreamTexture::~StreamTexture() {
   if (owner_stub_) {
     owner_stub_->RemoveDestructionObserver(this);
     owner_stub_->channel()->RemoveRoute(route_id_);
   }
+
+  if (registered_surface_texture_id) {
+    SurfaceTextureManager::GetInstance()->UnregisterBrowserSurfaceTexture(
+        registered_surface_texture_id);
+  }
 }
 
 // gpu::gles2::GLStreamTextureMatrix implementation
 void StreamTexture::GetTextureMatrix(float xform[16]) {
   if (surface_texture_) {
     UpdateTexImage();
     surface_texture_->GetTransformMatrix(current_matrix_);
   }
   memcpy(xform, current_matrix_, sizeof(current_matrix_));
   YInvertMatrix(xform);
@@ skipping 118 matching lines @@
 }
 
 unsigned StreamTexture::GetInternalFormat() {
   return GL_RGBA;
 }
 
 bool StreamTexture::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(StreamTexture, message)
     IPC_MESSAGE_HANDLER(GpuStreamTextureMsg_StartListening, OnStartListening)
+    IPC_MESSAGE_HANDLER(GpuStreamTextureMsg_RegisterStreamTexture,
+                        OnRegisterStreamTexture)
     IPC_MESSAGE_HANDLER(GpuStreamTextureMsg_EstablishPeer, OnEstablishPeer)
     IPC_MESSAGE_HANDLER(GpuStreamTextureMsg_SetSize, OnSetSize)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   DCHECK(handled);
   return handled;
 }
 
 void StreamTexture::OnStartListening() {
   DCHECK(!has_listener_);
   has_listener_ = true;
 }
 
 void StreamTexture::OnEstablishPeer(int32_t primary_id, int32_t secondary_id) {
   if (!owner_stub_)
     return;
 
   base::ProcessHandle process = owner_stub_->channel()->GetClientPID();
 
   SurfaceTexturePeer::GetInstance()->EstablishSurfaceTexturePeer(
       process, surface_texture_, primary_id, secondary_id);
 }
 
+void StreamTexture::OnRegisterStreamTexture(int32_t surface_texture_id) {
+  if (!owner_stub_)
+    return;
+
+  DCHECK(!registered_surface_texture_id);
+  registered_surface_texture_id = surface_texture_id;
+
+  SurfaceTextureManager::GetInstance()->RegisterBrowserSurfaceTexture(
+      surface_texture_id, surface_texture_.get());
+}
+
 bool StreamTexture::BindTexImage(unsigned target) {
   NOTREACHED();
   return false;
 }
 
 void StreamTexture::ReleaseTexImage(unsigned target) {
   NOTREACHED();
 }
 
 bool StreamTexture::CopyTexSubImage(unsigned target,
@@ skipping 11 matching lines @@
   return false;
 }
 
 void StreamTexture::OnMemoryDump(base::trace_event::ProcessMemoryDump* pmd,
                                  uint64_t process_tracing_id,
                                  const std::string& dump_name) {
   // TODO(ericrk): Add OnMemoryDump for GLImages. crbug.com/514914
 }
 
 }  // namespace gpu