scrawl
9 years ago
parent
e647ee5424
commit
a9ad1b09e2
@ -0,0 +1,81 @@
|
|||||||
|
#include "shadermanager.hpp"
|
||||||
|
|
||||||
|
#include <fstream>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#include <boost/lexical_cast.hpp>
|
||||||
|
#include <boost/filesystem/path.hpp>
|
||||||
|
#include <boost/filesystem/fstream.hpp>
|
||||||
|
|
||||||
|
#include <OpenThreads/ScopedLock>
|
||||||
|
|
||||||
|
namespace Shader
|
||||||
|
{
|
||||||
|
|
||||||
|
void ShaderManager::setShaderPath(const std::string &path)
|
||||||
|
{
|
||||||
|
mPath = path;
|
||||||
|
}
|
||||||
|
|
||||||
|
osg::ref_ptr<osg::Shader> ShaderManager::getShader(const std::string &shaderTemplate, const ShaderManager::DefineMap &defines, osg::Shader::Type shaderType)
|
||||||
|
{
|
||||||
|
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(mMutex);
|
||||||
|
|
||||||
|
// read the template if we haven't already
|
||||||
|
TemplateMap::iterator templateIt = mShaderTemplates.find(shaderTemplate);
|
||||||
|
if (templateIt == mShaderTemplates.end())
|
||||||
|
{
|
||||||
|
boost::filesystem::path p = (boost::filesystem::path(mPath) / shaderTemplate);
|
||||||
|
boost::filesystem::ifstream stream;
|
||||||
|
stream.open(p);
|
||||||
|
if (stream.fail())
|
||||||
|
{
|
||||||
|
std::cerr << "Failed to open " << p.string() << std::endl;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
std::stringstream buffer;
|
||||||
|
buffer << stream.rdbuf();
|
||||||
|
|
||||||
|
templateIt = mShaderTemplates.insert(std::make_pair(shaderTemplate, buffer.str())).first;
|
||||||
|
}
|
||||||
|
|
||||||
|
ShaderMap::iterator shaderIt = mShaders.find(std::make_pair(shaderTemplate, defines));
|
||||||
|
if (shaderIt == mShaders.end())
|
||||||
|
{
|
||||||
|
std::string shaderSource = templateIt->second;
|
||||||
|
|
||||||
|
const char escapeCharacter = '@';
|
||||||
|
size_t foundPos = 0;
|
||||||
|
while ((foundPos = shaderSource.find(escapeCharacter)) != std::string::npos)
|
||||||
|
{
|
||||||
|
size_t endPos = shaderSource.find_first_of(" \n\r()[].;", foundPos);
|
||||||
|
if (endPos == std::string::npos)
|
||||||
|
{
|
||||||
|
std::cerr << "Unexpected EOF" << std::endl;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
std::string define = shaderSource.substr(foundPos+1, endPos - (foundPos+1));
|
||||||
|
DefineMap::const_iterator defineFound = defines.find(define);
|
||||||
|
if (defineFound == defines.end())
|
||||||
|
{
|
||||||
|
std::cerr << "Undefined " << define << " in shader " << shaderTemplate << std::endl;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
shaderSource.replace(foundPos, endPos-foundPos, defineFound->second);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
osg::ref_ptr<osg::Shader> shader (new osg::Shader(shaderType));
|
||||||
|
shader->setShaderSource(shaderSource);
|
||||||
|
// Assign a unique name to allow the SharedStateManager to compare shaders efficiently
|
||||||
|
static unsigned int counter = 0;
|
||||||
|
shader->setName(boost::lexical_cast<std::string>(counter++));
|
||||||
|
|
||||||
|
shaderIt = mShaders.insert(std::make_pair(std::make_pair(shaderTemplate, defines), shader)).first;
|
||||||
|
}
|
||||||
|
return shaderIt->second;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,49 @@
|
|||||||
|
#ifndef OPENMW_COMPONENTS_SHADERMANAGER_H
|
||||||
|
#define OPENMW_COMPONENTS_SHADERMANAGER_H
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include <osg/ref_ptr>
|
||||||
|
|
||||||
|
#include <osg/Shader>
|
||||||
|
|
||||||
|
#include <OpenThreads/Mutex>
|
||||||
|
|
||||||
|
namespace Shader
|
||||||
|
{
|
||||||
|
|
||||||
|
/// @brief Reads shader template files and turns them into a concrete shader, based on a list of define's.
|
||||||
|
/// @par Shader templates can get the value of a define with the syntax @define.
|
||||||
|
class ShaderManager
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
void setShaderPath(const std::string& path);
|
||||||
|
|
||||||
|
typedef std::map<std::string, std::string> DefineMap;
|
||||||
|
|
||||||
|
/// Create or retrieve a shader instance.
|
||||||
|
/// @param shaderTemplate The filename of the shader template.
|
||||||
|
/// @param defines Define values that can be retrieved by the shader template.
|
||||||
|
/// @param shaderType The type of shader (usually vertex or fragment shader).
|
||||||
|
/// @note May return NULL on failure.
|
||||||
|
/// @note Thread safe.
|
||||||
|
osg::ref_ptr<osg::Shader> getShader(const std::string& shaderTemplate, const DefineMap& defines, osg::Shader::Type shaderType);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::string mPath;
|
||||||
|
|
||||||
|
// <name, code>
|
||||||
|
typedef std::map<std::string, std::string> TemplateMap;
|
||||||
|
TemplateMap mShaderTemplates;
|
||||||
|
|
||||||
|
typedef std::pair<std::string, DefineMap> MapKey;
|
||||||
|
typedef std::map<MapKey, osg::ref_ptr<osg::Shader> > ShaderMap;
|
||||||
|
ShaderMap mShaders;
|
||||||
|
|
||||||
|
OpenThreads::Mutex mMutex;
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,195 @@
|
|||||||
|
#include "shadervisitor.hpp"
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#include <osg/Texture>
|
||||||
|
#include <osg/Material>
|
||||||
|
#include <osg/Geometry>
|
||||||
|
|
||||||
|
#include <osgUtil/TangentSpaceGenerator>
|
||||||
|
|
||||||
|
#include <boost/lexical_cast.hpp>
|
||||||
|
|
||||||
|
#include "shadermanager.hpp"
|
||||||
|
|
||||||
|
namespace Shader
|
||||||
|
{
|
||||||
|
|
||||||
|
ShaderVisitor::ShaderRequirements::ShaderRequirements()
|
||||||
|
: mColorMaterial(false)
|
||||||
|
, mVertexColorMode(GL_AMBIENT_AND_DIFFUSE)
|
||||||
|
, mTexStageRequiringTangents(-1)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ShaderVisitor::ShaderVisitor(ShaderManager& shaderManager, const std::string &defaultVsTemplate, const std::string &defaultFsTemplate)
|
||||||
|
: osg::NodeVisitor(TRAVERSE_ALL_CHILDREN)
|
||||||
|
, mShaderManager(shaderManager)
|
||||||
|
, mDefaultVsTemplate(defaultVsTemplate)
|
||||||
|
, mDefaultFsTemplate(defaultFsTemplate)
|
||||||
|
{
|
||||||
|
mRequirements.push_back(ShaderRequirements());
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::apply(osg::Node& node)
|
||||||
|
{
|
||||||
|
if (node.getStateSet())
|
||||||
|
{
|
||||||
|
pushRequirements();
|
||||||
|
applyStateSet(node.getStateSet());
|
||||||
|
traverse(node);
|
||||||
|
popRequirements();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
traverse(node);
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::applyStateSet(osg::StateSet* stateset)
|
||||||
|
{
|
||||||
|
const osg::StateSet::TextureAttributeList& texAttributes = stateset->getTextureAttributeList();
|
||||||
|
for(unsigned int unit=0;unit<texAttributes.size();++unit)
|
||||||
|
{
|
||||||
|
const osg::StateAttribute *attr = stateset->getTextureAttribute(unit, osg::StateAttribute::TEXTURE);
|
||||||
|
if (attr)
|
||||||
|
{
|
||||||
|
const osg::Texture* texture = attr->asTexture();
|
||||||
|
if (texture)
|
||||||
|
{
|
||||||
|
if (!texture->getName().empty())
|
||||||
|
{
|
||||||
|
mRequirements.back().mTextures[unit] = texture->getName();
|
||||||
|
if (texture->getName() == "normalMap")
|
||||||
|
{
|
||||||
|
mRequirements.back().mTexStageRequiringTangents = unit;
|
||||||
|
// normal maps are by default off since the FFP can't render them, now that we'll use shaders switch to On
|
||||||
|
stateset->setTextureMode(unit, GL_TEXTURE_2D, osg::StateAttribute::ON);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
std::cerr << "ShaderVisitor encountered unknown texture " << texture << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// remove state that has no effect when rendering with shaders
|
||||||
|
stateset->removeTextureAttribute(unit, osg::StateAttribute::TEXENV);
|
||||||
|
}
|
||||||
|
|
||||||
|
const osg::StateSet::AttributeList& attributes = stateset->getAttributeList();
|
||||||
|
for (osg::StateSet::AttributeList::const_iterator it = attributes.begin(); it != attributes.end(); ++it)
|
||||||
|
{
|
||||||
|
if (it->first.first == osg::StateAttribute::MATERIAL)
|
||||||
|
{
|
||||||
|
const osg::Material* mat = static_cast<const osg::Material*>(it->second.first.get());
|
||||||
|
mRequirements.back().mVertexColorMode = mat->getColorMode();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::pushRequirements()
|
||||||
|
{
|
||||||
|
mRequirements.push_back(mRequirements.back());
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::popRequirements()
|
||||||
|
{
|
||||||
|
mRequirements.pop_back();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::createProgram(const ShaderRequirements &reqs, osg::StateSet *stateset)
|
||||||
|
{
|
||||||
|
ShaderManager::DefineMap defineMap;
|
||||||
|
const char* defaultTextures[] = { "diffuseMap", "normalMap", "emissiveMap", "darkMap", "detailMap" };
|
||||||
|
for (unsigned int i=0; i<sizeof(defaultTextures)/sizeof(defaultTextures[0]); ++i)
|
||||||
|
{
|
||||||
|
defineMap[defaultTextures[i]] = "0";
|
||||||
|
defineMap[std::string(defaultTextures[i]) + std::string("UV")] = "0";
|
||||||
|
}
|
||||||
|
for (std::map<int, std::string>::const_iterator texIt = reqs.mTextures.begin(); texIt != reqs.mTextures.end(); ++texIt)
|
||||||
|
{
|
||||||
|
defineMap[texIt->second] = "1";
|
||||||
|
defineMap[texIt->second + std::string("UV")] = boost::lexical_cast<std::string>(texIt->first);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!reqs.mColorMaterial)
|
||||||
|
defineMap["colorMode"] = "0";
|
||||||
|
else
|
||||||
|
{
|
||||||
|
switch (reqs.mVertexColorMode)
|
||||||
|
{
|
||||||
|
default:
|
||||||
|
case GL_AMBIENT_AND_DIFFUSE:
|
||||||
|
defineMap["colorMode"] = "2";
|
||||||
|
break;
|
||||||
|
case GL_AMBIENT:
|
||||||
|
defineMap["colorMode"] = "1";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
osg::ref_ptr<osg::Shader> vertexShader (mShaderManager.getShader(mDefaultVsTemplate, defineMap, osg::Shader::VERTEX));
|
||||||
|
osg::ref_ptr<osg::Shader> fragmentShader (mShaderManager.getShader(mDefaultFsTemplate, defineMap, osg::Shader::FRAGMENT));
|
||||||
|
|
||||||
|
if (vertexShader && fragmentShader)
|
||||||
|
{
|
||||||
|
osg::ref_ptr<osg::Program> program (new osg::Program);
|
||||||
|
program->addShader(vertexShader);
|
||||||
|
program->addShader(fragmentShader);
|
||||||
|
|
||||||
|
stateset->setAttributeAndModes(program, osg::StateAttribute::ON);
|
||||||
|
|
||||||
|
for (std::map<int, std::string>::const_iterator texIt = reqs.mTextures.begin(); texIt != reqs.mTextures.end(); ++texIt)
|
||||||
|
{
|
||||||
|
stateset->addUniform(new osg::Uniform(texIt->second.c_str(), texIt->first), osg::StateAttribute::ON);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::apply(osg::Geometry& geometry)
|
||||||
|
{
|
||||||
|
bool needPop = (geometry.getStateSet() != NULL);
|
||||||
|
if (geometry.getStateSet())
|
||||||
|
{
|
||||||
|
pushRequirements();
|
||||||
|
applyStateSet(geometry.getStateSet());
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!mRequirements.empty())
|
||||||
|
{
|
||||||
|
const ShaderRequirements& reqs = mRequirements.back();
|
||||||
|
if (reqs.mTexStageRequiringTangents != -1)
|
||||||
|
{
|
||||||
|
osg::ref_ptr<osgUtil::TangentSpaceGenerator> generator (new osgUtil::TangentSpaceGenerator);
|
||||||
|
generator->generate(&geometry, reqs.mTexStageRequiringTangents);
|
||||||
|
|
||||||
|
geometry.setTexCoordArray(7, generator->getTangentArray(), osg::Array::BIND_PER_VERTEX);
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: find a better place for the stateset
|
||||||
|
createProgram(reqs, geometry.getOrCreateStateSet());
|
||||||
|
}
|
||||||
|
|
||||||
|
if (needPop)
|
||||||
|
popRequirements();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ShaderVisitor::apply(osg::Drawable& drawable)
|
||||||
|
{
|
||||||
|
// non-Geometry drawable (e.g. particle system)
|
||||||
|
bool needPop = (drawable.getStateSet() != NULL);
|
||||||
|
|
||||||
|
if (drawable.getStateSet())
|
||||||
|
{
|
||||||
|
pushRequirements();
|
||||||
|
applyStateSet(drawable.getStateSet());
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!mRequirements.empty())
|
||||||
|
{
|
||||||
|
// TODO: find a better place for the stateset
|
||||||
|
createProgram(mRequirements.back(), drawable.getOrCreateStateSet());
|
||||||
|
}
|
||||||
|
|
||||||
|
if (needPop)
|
||||||
|
popRequirements();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,54 @@
|
|||||||
|
#ifndef OPENMW_COMPONENTS_SHADERVISITOR_H
|
||||||
|
#define OPENMW_COMPONENTS_SHADERVISITOR_H
|
||||||
|
|
||||||
|
#include <osg/NodeVisitor>
|
||||||
|
|
||||||
|
namespace Shader
|
||||||
|
{
|
||||||
|
|
||||||
|
class ShaderManager;
|
||||||
|
|
||||||
|
/// @brief Adjusts the given subgraph to render using shaders.
|
||||||
|
class ShaderVisitor : public osg::NodeVisitor
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
ShaderVisitor(ShaderManager& shaderManager, const std::string& defaultVsTemplate, const std::string& defaultFsTemplate);
|
||||||
|
|
||||||
|
virtual void apply(osg::Node& node);
|
||||||
|
|
||||||
|
virtual void apply(osg::Drawable& drawable);
|
||||||
|
virtual void apply(osg::Geometry& geometry);
|
||||||
|
|
||||||
|
void applyStateSet(osg::StateSet* stateset);
|
||||||
|
|
||||||
|
void pushRequirements();
|
||||||
|
void popRequirements();
|
||||||
|
|
||||||
|
private:
|
||||||
|
ShaderManager& mShaderManager;
|
||||||
|
|
||||||
|
struct ShaderRequirements
|
||||||
|
{
|
||||||
|
ShaderRequirements();
|
||||||
|
|
||||||
|
// <texture stage, texture name>
|
||||||
|
std::map<int, std::string> mTextures;
|
||||||
|
|
||||||
|
bool mColorMaterial;
|
||||||
|
// osg::Material::ColorMode
|
||||||
|
int mVertexColorMode;
|
||||||
|
|
||||||
|
// -1 == no tangents required
|
||||||
|
int mTexStageRequiringTangents;
|
||||||
|
};
|
||||||
|
std::vector<ShaderRequirements> mRequirements;
|
||||||
|
|
||||||
|
std::string mDefaultVsTemplate;
|
||||||
|
std::string mDefaultFsTemplate;
|
||||||
|
|
||||||
|
void createProgram(const ShaderRequirements& reqs, osg::StateSet* stateset);
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,15 @@
|
|||||||
|
#version 120
|
||||||
|
|
||||||
|
#if @diffuseMap
|
||||||
|
uniform sampler2D diffuseMap;
|
||||||
|
varying vec2 diffuseMapUV;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
void main()
|
||||||
|
{
|
||||||
|
#if @diffuseMap
|
||||||
|
gl_FragData[0] = texture2D(diffuseMap, diffuseMapUV);
|
||||||
|
#else
|
||||||
|
gl_FragData[0] = vec4(1,1,1,1);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
#version 120
|
||||||
|
|
||||||
|
#if @diffuseMap
|
||||||
|
varying vec2 diffuseMapUV;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
void main(void)
|
||||||
|
{
|
||||||
|
gl_Position = gl_ModelViewProjectionMatrix * gl_Vertex;
|
||||||
|
|
||||||
|
#if @diffuseMap
|
||||||
|
diffuseMapUV = gl_MultiTexCoord@diffuseMapUV.xy;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue