kc3-lang/angle/src/libGLESv2/renderer/ShaderCache.h

Branch : - branch - main - tag -

• Show log

  Commit

• Author : daniel@transgaming.com
  Date : 2012-10-31 19:55:33
  Hash : 76d3e6e0
  Message : Rename renderer namespace to rx Trac #21999 Author: Shannon Woods Signed-off-by: Daniel Koch git-svn-id: https://angleproject.googlecode.com/svn/branches/dx11proto@1379 736b8ea6-26fd-11df-bfd4-992fa37f6226

• src/libGLESv2/renderer/ShaderCache.h

• //
  // Copyright (c) 2012 The ANGLE Project Authors. All rights reserved.
  // Use of this source code is governed by a BSD-style license that can be
  // found in the LICENSE file.
  //
  
  // ShaderCache: Defines rx::ShaderCache, a cache of Direct3D shader objects
  // keyed by their byte code.
  
  #ifndef LIBGLESV2_RENDERER_SHADER_CACHE_H_
  #define LIBGLESV2_RENDERER_SHADER_CACHE_H_
  
  #include <d3d9.h>
  
  #ifdef _MSC_VER
  #include <hash_map>
  #else
  #include <unordered_map>
  #endif
  
  #include "common/debug.h"
  
  namespace rx
  {
  template <typename ShaderObject>
  class ShaderCache
  {
    public:
      ShaderCache() : mDevice(NULL)
      {
      }
  
      ~ShaderCache()
      {
          // Call clear while the device is still valid.
          ASSERT(mMap.empty());
      }
  
      void initialize(IDirect3DDevice9* device)
      {
          mDevice = device;
      }
  
      ShaderObject *create(const DWORD *function, size_t length)
      {
          std::string key(reinterpret_cast<const char*>(function), length);
          typename Map::iterator it = mMap.find(key);
          if (it != mMap.end())
          {
              it->second->AddRef();
              return it->second;
          }
  
          ShaderObject *shader;
          HRESULT result = createShader(function, &shader);
          if (FAILED(result))
          {
              return NULL;
          }
  
          // Random eviction policy.
          if (mMap.size() >= kMaxMapSize)
          {
              mMap.begin()->second->Release();
              mMap.erase(mMap.begin());
          }
  
          shader->AddRef();
          mMap[key] = shader;
  
          return shader;
      }
  
      void clear()
      {
          for (typename Map::iterator it = mMap.begin(); it != mMap.end(); ++it)
          {
              it->second->Release();
          }
  
          mMap.clear();
      }
  
    private:
      DISALLOW_COPY_AND_ASSIGN(ShaderCache);
  
      const static size_t kMaxMapSize = 100;
  
      HRESULT createShader(const DWORD *function, IDirect3DVertexShader9 **shader)
      {
          return mDevice->CreateVertexShader(function, shader);
      }
  
      HRESULT createShader(const DWORD *function, IDirect3DPixelShader9 **shader)
      {
          return mDevice->CreatePixelShader(function, shader);
      }
  
  #ifndef HASH_MAP
  # ifdef _MSC_VER
  #  define HASH_MAP stdext::hash_map
  # else
  #  define HASH_MAP std::unordered_map
  # endif
  #endif
  
      typedef HASH_MAP<std::string, ShaderObject*> Map;
      Map mMap;
  
      IDirect3DDevice9 *mDevice;
  };
  
  typedef ShaderCache<IDirect3DVertexShader9> VertexShaderCache;
  typedef ShaderCache<IDirect3DPixelShader9> PixelShaderCache;
  
  }
  
  #endif   // LIBGLESV2_RENDERER_SHADER_CACHE_H_