GPU Program Caching

gpu/command_buffer/service/memory_shader_cache.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/memory_shader_cache.h"
+#include "base/bind.h"
+#include "base/file_path.h"
+#include "net/disk_cache/disk_cache.h"
+#include "ui/gl/gl_bindings.h"
+
+using disk_cache::Backend;
+
+namespace gpu {
+
+//void thing(int a) {
+//  fprintf(stderr, "got %i from cache callback\n", a);
+//}
+
+MemoryShaderCache::MemoryShaderCache() {
+//  Backend* backend;
+//  const FilePath path("shaders");
+//  disk_cache::CreateCacheBackend(net::MEMORY_CACHE, path,
+//      16 * 1024 * 1024, false, NULL, NULL, &backend, base::Bind(&thing));
+//  cache_backend_ = scoped_ptr<Backend>(backend);
+}
+
+MemoryShaderCache::~MemoryShaderCache() { }
+
+bool MemoryShaderCache::isShaderCacheEnabled() {
+#if defined(OS_MACOSX)

[greggman, 2012/06/15 08:10:24]

You need to call glGetString and check for GL_ARB_get_progam_binary or
GL_OES_get_program_binary to decide if you can cache or not. Some non mac
platform might nt support program binary and mac might add it.

[dmurph, 2012/06/15 16:40:23]

Yeah, I ended up not using this method, I'll put those checks in
gpu_channel_manager so it can have the cache be NULL if it's not supported

+  return false;
+#else
+  return true;
+#endif
+}
+
+CompiledShaderStatus MemoryShaderCache::getShaderCompilationStatus(
+    const char* shader_src) {

[greggman, 2012/06/15 08:10:24]

would it be better if all of these functions took 'const std::string&
shader_src' instead of 'const char* shaders_src'? It seems like almost
everywhere you are calling it you're taking it from a std::string and then later
you're going to have to store the string requiring a strlen on a relatively long
string.

[dmurph, 2012/06/15 16:40:23]

Yeah, good idea :)

+  CachedShaderKey key(shader_src, "", "", 0, 0);
+  return shader_status_[key];

[greggman, 2012/06/15 08:10:24]

If you are just trying to find a key you should call find. using [] to access
stl based maps inserts the key if it doesn't exist. Or was that your intent?

[dmurph, 2012/06/15 16:40:23]

It wasn't, thanks for catching this, I'm not super familiar with c++ hash_map

+}
+
+void MemoryShaderCache::setShaderCompilationStatus(const char* shader_src,
+    CompiledShaderStatus status) {
+  CachedShaderKey key(shader_src, "", "", 0, 0);
+  shader_status_[key] = status;
+}
+
+LinkedProgramStatus MemoryShaderCache::getLinkedProgramStatus(
+    const char* untranslated_a, const char* untranslated_b) {
+  CachedProgramKey key(untranslated_a, untranslated_b, "", "", 0, 0);
+  return link_status_[key];
+}
+
+void MemoryShaderCache::setLinkedProgramStatus(const char* untranslated_a,
+    const char* untranslated_b, LinkedProgramStatus status) {
+  CachedProgramKey key(untranslated_a, untranslated_b, "", "", 0, 0);
+  link_status_[key] = status;
+}
+
+void MemoryShaderCache::getLinkedProgram(const char* untranslated_a,
+    const char* untranslated_b, GLsizei* length, GLenum* binaryFormat,
+    const GLvoid** binary) {
+  CachedProgramKey key(untranslated_a, untranslated_b, "", "", 0, 0);
+  StoreValue value = store_[key];
+  *length = value.length;
+  *binaryFormat = value.format;
+  *binary = value.data;
+}
+
+void MemoryShaderCache::setLinkedProgram(const char* untranslated_a,
+    const char* untranslated_b, GLsizei length, GLenum binaryFormat,
+    GLvoid* binary) {
+  CachedProgramKey key(untranslated_a, untranslated_b, "", "", 0, 0);
+  StoreValue value(length, binaryFormat, binary);
+  store_[key] = value;
+}
+
+} // namespace gpu