GPU Program Caching

gpu/command_buffer/service/program_manager.cc

 // Copyright (c) 2012 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/program_manager.h"
 
 #include <algorithm>
 #include <set>
 
 #include "base/basictypes.h"
+#include "base/bind.h"
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/string_number_conversions.h"
+#include "net/base/net_log.h"
 #include "gpu/command_buffer/common/gles2_cmd_format.h"
 #include "gpu/command_buffer/common/gles2_cmd_utils.h"
 #include "gpu/command_buffer/service/gles2_cmd_decoder.h"
 #include "gpu/command_buffer/service/gpu_switches.h"
 
 namespace gpu {
 namespace gles2 {
 
 static int ShaderTypeToIndex(GLenum shader_type) {
   switch (shader_type) {
@@ skipping 221 matching lines @@
 void ProgramManager::ProgramInfo::ExecuteBindAttribLocationCalls() {
   for (std::map<std::string, GLint>::const_iterator it =
            bind_attrib_location_map_.begin();
        it != bind_attrib_location_map_.end(); ++it) {
     const std::string* mapped_name = GetAttribMappedName(it->first);
     if (mapped_name && *mapped_name != it->first)
       glBindAttribLocation(service_id_, it->second, mapped_name->c_str());
   }
 }
 
-bool ProgramManager::ProgramInfo::Link() {
+void ProgramManager::DoCompileShader(ShaderManager::ShaderInfo* info,
+                                     ShaderTranslator* translator,
+                                     FeatureInfo* feature_info) {
+    // Translate GL ES 2.0 shader to Desktop GL shader and pass that to
+    // glShaderSource and then glCompileShader.
+    const char* shader_src = info->source() ? info->source()->c_str() : "";
+    if (translator) {
+      if (!translator->Translate(shader_src)) {
+        info->SetStatus(false, translator->info_log(), NULL);
+        if(program_cache_) {
+          program_cache_->SetShaderCompilationStatus(
+              *info->source(),
+              ProgramCache::COMPILATION_UNKNOWN);
+        }
+        return;
+      }
+      shader_src = translator->translated_shader();
+      if (!feature_info->feature_flags().angle_translated_shader_source)
+        info->UpdateTranslatedSource(shader_src);
+    }
+
+    glShaderSource(info->service_id(), 1, &shader_src, NULL);
+    glCompileShader(info->service_id());
+    if (feature_info->feature_flags().angle_translated_shader_source) {
+      GLint max_len = 0;
+      glGetShaderiv(info->service_id(),
+                    GL_TRANSLATED_SHADER_SOURCE_LENGTH_ANGLE,
+                    &max_len);
+      scoped_array<char> temp(new char[max_len]);
+      GLint len = 0;
+      glGetTranslatedShaderSourceANGLE(
+          info->service_id(), max_len, &len, temp.get());
+      DCHECK(max_len == 0 || len < max_len);
+      DCHECK(len == 0 || temp[len] == '\0');
+      info->UpdateTranslatedSource(temp.get());
+    }
+
+    GLint status = GL_FALSE;
+    glGetShaderiv(info->service_id(),  GL_COMPILE_STATUS, &status);
+    if (status) {
+      info->SetStatus(true, "", translator);
+      if(program_cache_) {
+        program_cache_->SetShaderCompilationStatus(
+            *info->source(),
+            ProgramCache::COMPILATION_SUCCEEDED);
+      }
+    } else {
+      // We cannot reach here if we are using the shader translator.
+      // All invalid shaders must be rejected by the translator.
+      // All translated shaders must compile.
+      LOG_IF(ERROR, translator)
+          << "Shader translator allowed/produced an invalid shader.";
+      GLint max_len = 0;
+      glGetShaderiv(info->service_id(), GL_INFO_LOG_LENGTH, &max_len);
+      scoped_array<char> temp(new char[max_len]);
+      GLint len = 0;
+      glGetShaderInfoLog(info->service_id(), max_len, &len, temp.get());
+      DCHECK(max_len == 0 || len < max_len);
+      DCHECK(len == 0 || temp[len] == '\0');
+      info->SetStatus(false, std::string(temp.get(), len).c_str(), NULL);
+      if(program_cache_) {
+        program_cache_->SetShaderCompilationStatus(
+            *info->source(),
+            ProgramCache::COMPILATION_UNKNOWN);
+      }
+    }
+}
+
+bool ProgramManager::ProgramInfo::Link(ShaderManager* manager,
+                                       ShaderTranslator* vertex_translator,
+                                       ShaderTranslator* fragment_shader,
+                                       FeatureInfo* feature_info) {
   ClearLinkStatus();
   if (!CanLink()) {
     set_log_info("missing shaders");
     return false;
   }
   if (DetectAttribLocationBindingConflicts()) {
     set_log_info("glBindAttribLocation() conflicts");
     return false;
   }
   ExecuteBindAttribLocationCalls();
-  glLinkProgram(service_id());
+
+  bool link = true;
+  ProgramCache* cache = manager_->program_cache_;
+  const std::string* shader_a = attached_shaders_[0]->source();
+  const std::string* shader_b = attached_shaders_[1]->source();
+  if(cache) {
+    ProgramCache::LinkedProgramStatus status = cache->GetLinkedProgramStatus(
+                                   *shader_a,
+                                   *shader_b,
+                                   &bind_attrib_location_map_);
+    switch(status) {
+      case ProgramCache::LINK_UNKNOWN:
+        fprintf(stderr,
+                "program %i not linked yet compiling and linking\n",
+                service_id());
+        // compile our shaders + attach
+        if (attached_shaders_[0]->IsPendingCacheMissCompilation()) {
+          manager_->DoCompileShader(attached_shaders_[0],
+                                    vertex_translator,
+                                    feature_info);
+          AttachShader(manager, attached_shaders_[0]);
+          glAttachShader(service_id(), attached_shaders_[0]->service_id());
+          attached_shaders_[0]->SetPendingCompilation(false);
+        }
+        if(attached_shaders_[1]->IsPendingCacheMissCompilation()) {
+          manager_->DoCompileShader(attached_shaders_[1],
+                                    fragment_shader,
+                                    feature_info);
+          AttachShader(manager, attached_shaders_[1]);
+          glAttachShader(service_id(), attached_shaders_[1]->service_id());
+          attached_shaders_[1]->SetPendingCompilation(false);
+        }
+        link = true;
+        break;
+      case ProgramCache::LINK_SUCCEEDED:
+        fprintf(stderr, "## program %i linked and cached!\n", service_id());
+        cache->LoadLinkedProgram(service_id(),
+                                 attached_shaders_[0],
+                                 attached_shaders_[1],
+                                 &bind_attrib_location_map_);
+        link = false;
+        break;
+    }
+  }
+
+  if(link) {
+    glLinkProgram(service_id());
+  }
+
   GLint success = 0;
   glGetProgramiv(service_id(), GL_LINK_STATUS, &success);
   if (success == GL_TRUE) {
     Update();
+    if(cache && link) {
+      fprintf(stderr, "program %i just linked, caching\n", service_id());
+      cache->SaveLinkedProgram(service_id(),
+                               attached_shaders_[0],
+                               attached_shaders_[1],
+                               &bind_attrib_location_map_);
+
+    }
   } else {
     UpdateLogInfo();
+    if(cache) {
+      fprintf(stderr, "program %i not linked correctly\n", service_id());
+    }
   }
   return success == GL_TRUE;
 }
 
 void ProgramManager::ProgramInfo::Validate() {
   if (!IsValid()) {
     set_log_info("program not linked");
     return;
   }
   glValidateProgram(service_id());
@@ skipping 392 matching lines @@
     manager_->StopTracking(this);
     manager_ = NULL;
   }
 }
 
 // TODO(gman): make this some kind of random number. Base::RandInt is not
 // callable because of the sandbox. What matters is that it's possibly different
 // by at least 1 bit each time chrome is run.
 static int uniform_random_offset_ = 3;
 
-ProgramManager::ProgramManager()
+ProgramManager::ProgramManager(base::WeakPtr<ProgramCache> program_cache)
     : uniform_swizzle_(uniform_random_offset_++ % 15),
       program_info_count_(0),
       have_context_(true),
       disable_workarounds_(
           CommandLine::ForCurrentProcess()->HasSwitch(
-              switches::kDisableGpuDriverBugWorkarounds)) {
-}
+              switches::kDisableGpuDriverBugWorkarounds)),
+      program_cache_(program_cache){ }
 
 ProgramManager::~ProgramManager() {
   DCHECK(program_infos_.empty());
 }
 
 void ProgramManager::Destroy(bool have_context) {
   have_context_ = have_context;
   program_infos_.clear();
 }
 
@@ skipping 25 matching lines @@
   for (ProgramInfoMap::const_iterator it = program_infos_.begin();
        it != program_infos_.end(); ++it) {
     if (it->second->service_id() == service_id) {
       *client_id = it->first;
       return true;
     }
   }
   return false;
 }
 
+ProgramCache* ProgramManager::GetProgramCache() const {
+  return program_cache_;
+}
+
 bool ProgramManager::IsOwned(ProgramManager::ProgramInfo* info) {
   for (ProgramInfoMap::iterator it = program_infos_.begin();
        it != program_infos_.end(); ++it) {
     if (it->second.get() == info) {
       return true;
     }
   }
   return false;
 }
 
@@ skipping 65 matching lines @@
 }
 
 GLint ProgramManager::UnswizzleLocation(GLint v) const {
   return v < 0 ? v : Swizzle(v - uniform_swizzle_);
 }
 
 }  // namespace gles2
 }  // namespace gpu