ShaderInfoCache.h

// Copyright (c) 2013 The Foundry Visionmongers Ltd. All Rights Reserved.

#ifndef FNRENDERERINFO_SHADERINFOCACHE_H
#define FNRENDERERINFO_SHADERINFOCACHE_H

#include <FnAttribute/FnAttribute.h>
#include <FnRendererInfo/FnRendererInfoAPI.h>

#include <map>

#include <iostream>

namespace Foundry
{
namespace Katana
{
namespace RendererInfo
{
    template<typename TShaderInfoType>
    class FNRENDERERINFO_API ShaderInfoCache
    {
    public:

        typedef typename std::map<std::string, TShaderInfoType>::const_iterator Iterator;

        bool isEmpty() const;

        void flush();

        void addShaderInfo(const std::string& shaderName, const TShaderInfoType& shaderInfo);

        const TShaderInfoType& getShaderInfo(const std::string& shaderName) const;

        Iterator begin() const;

        Iterator end() const;

        void setNullValue(const TShaderInfoType& nullValue);

    private:

        typedef std::map<std::string, TShaderInfoType> ShaderInfoMap;

        TShaderInfoType _nullValue;

        ShaderInfoMap _shaderInfo;
    };
template<typename TShaderInfoType>
bool ShaderInfoCache<TShaderInfoType>::isEmpty() const
{
    return _shaderInfo.empty();
}

template<typename TShaderInfoType>
void ShaderInfoCache<TShaderInfoType>::flush()
{
    _shaderInfo.clear();
}

template<typename TShaderInfoType>
void ShaderInfoCache<TShaderInfoType>::addShaderInfo(const std::string& shaderName,
                                                     const TShaderInfoType& shaderInfo)
{
    if (!shaderName.empty())
    {
        _shaderInfo[shaderName] = shaderInfo;
    }
}

template<typename TShaderInfoType>
const TShaderInfoType& ShaderInfoCache<TShaderInfoType>::getShaderInfo(const std::string& shaderName) const
{
    // Get the shader's TShaderInfoType inside _shaderInfo
    typename ShaderInfoMap::const_iterator iter=_shaderInfo.find(shaderName);
    if (iter == _shaderInfo.end())
    {
        return _nullValue;
    }

    return (*iter).second;
}

template<typename TShaderInfoType>
typename ShaderInfoCache<TShaderInfoType>::Iterator ShaderInfoCache<TShaderInfoType>::begin() const
{
    return _shaderInfo.begin();
}

template<typename TShaderInfoType>
typename ShaderInfoCache<TShaderInfoType>::Iterator ShaderInfoCache<TShaderInfoType>::end() const
{
    return _shaderInfo.end();
}

template<typename TShaderInfoType>
void ShaderInfoCache<TShaderInfoType>::setNullValue(const TShaderInfoType& nullValue)
{
    _nullValue = nullValue;
}

}
}
}

namespace FnKat = Foundry::Katana;

#endif