#include <algorithm>
#include <cstring>
+#include <set>
+#include <msp/gl/extensions/arb_uniform_buffer_object.h>
#include <msp/strings/format.h>
#include <msp/strings/utils.h>
#include "extension.h"
const ProgramBuilder::VariableDefinition ProgramBuilder::standard_variables[] =
{
{ FRAGMENT, "gl_FragColor", "vec4", "frag_color", "g" },
- { FRAGMENT, "frag_color", "vec4", "color_base", "!t" },
- { FRAGMENT, "frag_color", "vec4", "tex_sample", "!l!s!mt" },
- { FRAGMENT, "frag_color", "vec4", "tex_sample*color_base", "l|s|mt" },
- { FRAGMENT, "color_base", "vec4", "vec4(1.0)", "!l!s!m" },
- { FRAGMENT, "color_base", "vec4", "color", "!l!sm" },
- { FRAGMENT, "color_base", "vec4", "vec4(vec3(l_shadow), 1.0)", "!ls!m" },
- { FRAGMENT, "color_base", "vec4", "color*vec4(vec3(l_shadow), 1.0)", "!lsm" },
- { FRAGMENT, "color_base", "vec4", "vec4(rgb_light_env, 1.0)", "l!m" },
- { FRAGMENT, "color_base", "vec4", "vec4(rgb_light_env, material.diffuse.a)", "lm" },
- { FRAGMENT, "rgb_light_env", "vec3", "rgb_light_full+reflect_sample.rgb*reflectivity", "e" },
- { FRAGMENT, "rgb_light_env", "vec3", "rgb_light_full", "!e" },
- { FRAGMENT, "rgb_light_full", "vec3", "rgb_light_shadow+ambient_product_diffuse.rgb", "m" },
- { FRAGMENT, "rgb_light_full", "vec3", "rgb_light_shadow", "!m" },
- { FRAGMENT, "ambient_product_diffuse", "vec4", "gl_FrontLightModelProduct.sceneColor", "g" },
- { FRAGMENT, "ambient_product_diffuse", "vec4", "ambient_color*material.diffuse", "!g" },
- { FRAGMENT, "rgb_light_shadow", "vec3", "rgb_light*l_shadow", "s" },
- { FRAGMENT, "rgb_light_shadow", "vec3", "rgb_light", "!s" },
- { FRAGMENT, "rgb_light[i]", "vec3", "vec3(l_diffuse[i])", "!m!p" },
- { FRAGMENT, "rgb_light[i]", "vec3", "vec3(l_diffuse[i]+l_specular[i])", "!mp" },
- { FRAGMENT, "rgb_light[i]", "vec3", "l_diffuse[i]*light_product_diffuse[i].rgb", "m!p" },
- { FRAGMENT, "rgb_light[i]", "vec3", "l_diffuse[i]*light_product_diffuse[i].rgb+l_specular[i]*light_product_specular[i].rgb", "mp" },
- { FRAGMENT, "light_product_diffuse[i]", "vec4", "gl_FrontLightProduct[i].diffuse", "g" },
- { FRAGMENT, "light_product_diffuse[i]", "vec4", "light_sources[i].diffuse*material.diffuse", "!g" },
- { FRAGMENT, "light_product_specular[i]", "vec4", "gl_FrontLightProduct[i].specular", "g" },
- { FRAGMENT, "light_product_specular[i]", "vec4", "light_sources[i].specular*material.specular", "!g" },
+ { FRAGMENT, "frag_color", "vec4", "basic_color", "!e!l" },
+ { FRAGMENT, "frag_color", "vec4", "vec4(rgb_surface, surface_alpha)", "!el" },
+ { FRAGMENT, "frag_color", "vec4", "vec4(rgb_surface+rgb_reflection, surface_alpha)", "e" },
+
+ { FRAGMENT, "rgb_reflection", "vec3", "reflect_sample.rgb*reflectivity", 0 },
{ FRAGMENT, "reflect_sample", "vec4", "textureCube(environment, env_reflect_dir)", 0 },
{ FRAGMENT, "env_reflect_dir", "vec3", "env_eye_matrix*eye_reflect_dir", 0 },
{ FRAGMENT, "eye_reflect_dir", "vec3", "eye_tbn_matrix*tbn_reflect_dir", "n" },
- { FRAGMENT, "zzz_reflect_dir", "vec3", "reflect(zzz_incident_dir, n_zzz_normal)", 0 },
+
+ { FRAGMENT, "rgb_surface", "vec3", "rgb_unlit_surface", "!l" },
+ { FRAGMENT, "rgb_surface", "vec3", "rgb_lit_surface", "l" },
+ { FRAGMENT, "rgb_unlit_surface", "vec3", "basic_color.rgb", "!s" },
+ { FRAGMENT, "rgb_unlit_surface", "vec3", "basic_color.rgb*l_shadow", "s" },
+ { FRAGMENT, "rgb_lit_surface", "vec3", "rgb_illumination_diffuse", "!m!t" },
+ { FRAGMENT, "rgb_lit_surface", "vec3", "rgb_illumination_diffuse*tex_sample.rgb", "!mt" },
+ { FRAGMENT, "rgb_lit_surface", "vec3", "rgb_surface_ambient+rgb_surface_diffuse", "m!p" },
+ { FRAGMENT, "rgb_lit_surface", "vec3", "rgb_surface_ambient+rgb_surface_diffuse+rgb_surface_specular", "mp" },
+
+ { FRAGMENT, "surface_alpha", "float", "basic_color.a", "!m" },
+ { FRAGMENT, "surface_alpha", "float", "basic_color.a", "m!l" },
+ { FRAGMENT, "surface_alpha", "float", "material.diffuse.a", "ml!t" },
+ { FRAGMENT, "surface_alpha", "float", "material.diffuse.a*tex_sample.a", "mlt" },
+ { FRAGMENT, "basic_color", "vec4", "vec4(1.0)", "!m!t" },
+ { FRAGMENT, "basic_color", "vec4", "tex_sample", "!mt" },
+ { FRAGMENT, "basic_color", "vec4", "color", "m!t" },
+ { FRAGMENT, "basic_color", "vec4", "color*tex_sample", "mt" },
+
+ { FRAGMENT, "rgb_surface_ambient", "vec3", "rgb_illumination_ambient*material.ambient.rgb", "!t" },
+ { FRAGMENT, "rgb_surface_ambient", "vec3", "rgb_illumination_ambient*material.ambient.rgb*tex_sample.rgb", "t" },
+ { FRAGMENT, "rgb_illumination_ambient", "vec3", "ambient_color.rgb", "!y" },
+ { FRAGMENT, "rgb_illumination_ambient", "vec3", "ambient_color.rgb+l_skylight*sky_color.rgb", "y" },
+ { FRAGMENT, "l_skylight", "float", "dot(n_zzz_normal, zzz_sky_dir)*0.5+0.5", 0 },
+
+ { FRAGMENT, "rgb_surface_specular", "vec3", "rgb_illumination_specular*material.specular.rgb", 0 },
+ { FRAGMENT, "rgb_illumination_specular", "vec3", "rgb_illumination_specular_direct", "!y|e" },
+ { FRAGMENT, "rgb_illumination_specular", "vec3", "rgb_illumination_specular_direct+l_sky_specular*sky_color.rgb", "y!e" },
+ { FRAGMENT, "rgb_illumination_specular_direct", "vec3", "rgb_light_specular", "!s" },
+ { FRAGMENT, "rgb_illumination_specular_direct", "vec3", "rgb_light_specular*l_shadow", "s" },
+ { FRAGMENT, "rgb_light_specular[i]", "vec3", "l_specular[i]*light_sources[i].specular.rgb", 0 },
+ { FRAGMENT, "l_sky_specular", "float", "pow((1.0-pow(clamp(dot(zzz_reflect_dir, zzz_sky_dir)-horizon_limit, -1.0, 0.0), 2.0)), material.shininess/2.0)", 0 },
+ { FRAGMENT, "l_specular[i]", "float", "pow(max(dot(n_zzz_half_vec[i], n_zzz_normal), 0.0), material.shininess)", 0 },
+
+ { FRAGMENT, "rgb_surface_diffuse", "vec3", "rgb_illumination_diffuse*material.diffuse.rgb", "!t" },
+ { FRAGMENT, "rgb_surface_diffuse", "vec3", "rgb_illumination_diffuse*material.diffuse.rgb*tex_sample.rgb", "t" },
+ { FRAGMENT, "rgb_illumination_diffuse", "vec3", "rgb_light_diffuse", "!s" },
+ { FRAGMENT, "rgb_illumination_diffuse", "vec3", "rgb_light_diffuse*l_shadow", "s" },
+ { FRAGMENT, "rgb_light_diffuse[i]", "vec3", "l_diffuse[i]*light_sources[i].diffuse.rgb", 0 },
+ { FRAGMENT, "l_diffuse[i]", "float", "max(dot(n_zzz_normal, n_zzz_light_dir[i]), 0.0)", 0 },
+
{ FRAGMENT, "l_shadow", "float", "mix(1.0, shadow_sample, shadow_darkness)", 0 },
{ FRAGMENT, "shadow_sample", "float", "shadow2D(shadow, shd_vertex).r", 0 },
- { FRAGMENT, "l_diffuse[i]", "float", "max(dot(n_zzz_normal, n_zzz_light_dir[i]), 0.0)", 0 },
- { FRAGMENT, "l_specular[i]", "float", "pow(max(dot(n_zzz_half_vec[i], n_zzz_normal), 0.0), material.shininess)", 0 },
+
+ { FRAGMENT, "zzz_reflect_dir", "vec3", "reflect(zzz_incident_dir, n_zzz_normal)", 0 },
{ FRAGMENT, "n_zzz_half_vec[i]", "vec3", "normalize(zzz_light_dir[i]-zzz_incident_dir)", 0 },
{ FRAGMENT, "n_zzz_light_dir[i]", "vec3", "normalize(zzz_light_dir[i])", 0 },
{ FRAGMENT, "n_tbn_normal", "vec3", "normal_sample*2.0-1.0", "n" },
{ FRAGMENT, "tex_sample", "vec4", "texture2D(texture, texture_coord)", 0 },
{ VERTEX, "gl_Position", "vec4", "projection_matrix*eye_vertex", 0 },
- { VERTEX, "shd_vertex", "vec3", "vec3(dot(eye_vertex, gl_EyePlaneS[shadow_unit]), dot(eye_vertex, gl_EyePlaneT[shadow_unit]), dot(eye_vertex, gl_EyePlaneR[shadow_unit]))", "g" },
- { VERTEX, "shd_vertex", "vec3", "(shd_eye_matrix*eye_vertex).xyz", "!g" },
+ { VERTEX, "shd_vertex", "vec3", "(shd_eye_matrix*eye_vertex).xyz", 0 },
+ { VERTEX, "tbn_sky_dir", "vec3", "eye_sky_dir*eye_tbn_matrix", "n" },
{ VERTEX, "tbn_light_dir[i]", "vec3", "eye_light_dir[i]*eye_tbn_matrix", 0 },
{ VERTEX, "eye_light_dir[i]", "vec3", "normalize(eye_light_position[i].xyz-eye_vertex.xyz*eye_light_position[i].w)", 0 },
{ VERTEX, "eye_light_position[i]", "vec4", "light_sources[i].position", 0 },
{ ATTRIBUTE, "tangent", "vec3", 0, 0 },
{ ATTRIBUTE, "binormal", "vec3", 0, 0 },
- { UNIFORM, "shadow_unit", "int", 0, 0 },
{ UNIFORM, "texture", "sampler2D", 0, 0 },
{ UNIFORM, "shadow", "sampler2DShadow", 0, 0 },
- { UNIFORM, "shadow_darkness", "float", 0, 0 },
+ { UNIFORM, "ShadowMap::shadow_darkness", "float", 0, 0 },
{ UNIFORM, "normalmap", "sampler2D", 0, 0 },
{ UNIFORM, "environment", "samplerCube", 0, 0 },
- { UNIFORM, "env_eye_matrix", "mat3", 0, 0 },
- { UNIFORM, "reflectivity", "float", 0, 0 },
- { UNIFORM, "eye_obj_matrix", "mat4", "gl_ModelViewMatrix", 0 },
- { UNIFORM, "eye_obj_normal_matrix", "mat3", "gl_NormalMatrix", 0 },
- { UNIFORM, "projection_matrix", "mat4", "gl_ProjectionMatrix", 0 },
- { UNIFORM, "shd_eye_matrix", "mat4", 0, 0 },
- { UNIFORM, "light_sources", "struct { vec4 position; vec4 diffuse; vec4 specular; }[MAX_LIGHTS]", "gl_LightSource[i]", 0 },
- { UNIFORM, "ambient_color", "vec4", 0, 0 },
- { UNIFORM, "material", "struct { vec4 ambient; vec4 diffuse; vec4 specular; float shininess; }", "gl_FrontMaterial", 0 },
+ { UNIFORM, "EnvMap::env_eye_matrix", "mat3", 0, 0 },
+ { UNIFORM, "Material::reflectivity", "float", 0, 0 },
+ { UNIFORM, "Transform::eye_obj_matrix", "mat4", "gl_ModelViewMatrix", 0 },
+ { UNIFORM, "Transform::eye_obj_normal_matrix", "mat3", "gl_NormalMatrix", 0 },
+ { UNIFORM, "Transform::projection_matrix", "mat4", "gl_ProjectionMatrix", 0 },
+ { UNIFORM, "ShadowMap::shd_eye_matrix", "mat4", 0, 0 },
+ { UNIFORM, "Lighting::light_sources", "LightSourceParameters[MAX_LIGHTS]", "gl_LightSource[i]", 0 },
+ { UNIFORM, "Lighting::ambient_color", "vec4", "gl_LightModel.ambient", 0 },
+ { UNIFORM, "Lighting::sky_color", "vec4", 0, 0 },
+ { UNIFORM, "Lighting::eye_sky_dir", "vec3", 0, 0 },
+ { UNIFORM, "Lighting::horizon_limit", "float", 0, 0 },
+ { UNIFORM, "Material::material", "MaterialParameters", "gl_FrontMaterial", 0 },
+
+ { TYPE, "LightSourceParameters", "struct { vec4 position; vec4 diffuse; vec4 specular; }", "gl_LightSourceParameters", 0 },
+ { TYPE, "MaterialParameters", "struct { vec4 ambient; vec4 diffuse; vec4 specular; float shininess; }", "gl_MaterialParameters", 0 },
// Terminator entry
{ NO_SCOPE, 0, 0, 0, 0 }
};
-const char ProgramBuilder::interfaces[] = { 0, 0, 0, 'v', 0 };
+const char ProgramBuilder::interfaces[] = { 0, 0, 0, 0, 'v', 0 };
ProgramBuilder::ProgramBuilder(const StandardFeatures &f):
features(f),
continue;
}
+ if(def->scope==TYPE)
+ {
+ for(list<ShaderVariable *>::iterator j=resolved_vars.begin(); j!=resolved_vars.end(); ++j)
+ if(!(*j)->type && name_match(def->name, (*j)->variable->type))
+ (*j)->resolve_type(*def);
+
+ continue;
+ }
+
+ const char *def_uq_name = unqualified_name(def->name);
+
// See if this variable can satisfy any unresolved variables
ShaderVariable *last_resolved = 0;
for(list<ShaderVariable>::iterator j=variables.begin(); j!=variables.end(); ++j)
if(j->variable)
continue;
- if(!name_match(def->name, j->resolved_name.c_str()))
+ if(!name_match(def_uq_name, j->resolved_name.c_str()))
continue;
if(last_resolved)
if(!features.legacy)
{
- prog.bind_fragment_data(0, "frag_color");
+ // OpenGL ES does not support binding fragment shader outputs
+ if(get_gl_api()!=OPENGL_ES2)
+ prog.bind_fragment_data(0, "frag_color");
+
prog.bind_attribute(VERTEX4, "vertex");
if(features.lighting)
prog.bind_attribute(NORMAL3, "normal");
{
string source;
+ bool use_blocks = !features.legacy && ARB_uniform_buffer_object;
+
if(!features.legacy)
- source += "#version 130\n";
+ {
+ if(get_gl_api()==OPENGL_ES2)
+ {
+ if(use_blocks)
+ source += "#version 300 es\n";
+ }
+ else
+ {
+ source += "#version 130\n";
+ if(use_blocks)
+ source += "#extension GL_ARB_uniform_buffer_object: require\n";
+ }
+ }
+ set<const VariableDefinition *> declared_types;
+ set<string> uniform_blocks;
for(list<ShaderVariable *>::const_iterator i=variables.begin(); i!=variables.end(); ++i)
if((*i)->variable->scope==UNIFORM && (*i)->is_referenced_from(scope) && !(*i)->inlined)
- source += format("uniform %s;\n", (*i)->create_declaration());
+ {
+ if((*i)->type && !declared_types.count((*i)->type))
+ {
+ source += format("%s;\n", (*i)->create_type_declaration());
+ declared_types.insert((*i)->type);
+ }
+
+ if(!(*i)->resolved_block.empty() && use_blocks)
+ uniform_blocks.insert((*i)->resolved_block);
+ else
+ source += format("uniform %s;\n", (*i)->create_declaration());
+ }
+
+ for(set<string>::const_iterator i=uniform_blocks.begin(); i!=uniform_blocks.end(); ++i)
+ {
+ source += format("uniform %s\n{\n", *i);
+ for(list<ShaderVariable *>::const_iterator j=variables.begin(); j!=variables.end(); ++j)
+ if((*j)->resolved_block==*i)
+ source += format("\t%s;\n", (*j)->create_declaration());
+ source += "};\n";
+ }
/* Interface variables need to have global declarations. */
for(list<ShaderVariable *>::const_iterator i=variables.begin(); i!=variables.end(); ++i)
return cond;
}
+const char *ProgramBuilder::unqualified_name(const char *name)
+{
+ for(const char *p=name; *p; ++p)
+ if(*p==':' && *++p==':')
+ name = p+1;
+ return name;
+}
+
ProgramBuilder::MatchType ProgramBuilder::name_match(const char *n1, const char *n2, const char **space)
{
int i = 0;
material(false),
lighting(false),
max_lights(1),
+ skylight(false),
specular(false),
normalmap(false),
shadow(false),
if(lighting)
{
flags += 'l';
+ if(skylight)
+ flags += 'y';
if(specular)
flags += 'p';
if(normalmap)
ProgramBuilder::ShaderVariable::ShaderVariable(const std::string &n):
name(n),
variable(0),
+ type(0),
resolved_name(n),
fuzzy_space(name.find("zzz")!=string::npos),
array_sum(false),
{
variable = &var;
const char *space = 0;
- MatchType match = name_match(var.name, resolved_name.c_str(), &space);
+ const char *var_uq_name = unqualified_name(variable->name);
+ MatchType match = name_match(var_uq_name, resolved_name.c_str(), &space);
+
+ if(var_uq_name!=variable->name)
+ resolved_block.assign(variable->name, var_uq_name-2);
if(match==FUZZY)
resolve_space(string(space, 3));
var.referenced_by.insert(var.referenced_by.end(), referenced_by.begin(), referenced_by.end());
}
+void ProgramBuilder::ShaderVariable::resolve_type(const VariableDefinition &var)
+{
+ type = &var;
+}
+
void ProgramBuilder::ShaderVariable::resolve_space(const string &space)
{
if(fuzzy_space)
return static_cast<InterfaceFlags>(flags);
}
+string ProgramBuilder::ShaderVariable::create_type_declaration() const
+{
+ if(!type)
+ throw logic_error("no type");
+
+ if(!strncmp(type->type, "struct", 6))
+ {
+ const char *brace = strchr(type->type, '{');
+ if(brace)
+ return format("struct %s %s", type->name, brace);
+ }
+
+ throw invalid_variable_definition("invalid typedef");
+}
+
string ProgramBuilder::ShaderVariable::create_declaration(char interface, bool loop) const
{
if(variable->scope==UNIFORM && !array_subscript.empty())
add("normalmap", &StandardFeatures::normalmap);
add("reflection", &StandardFeatures::reflection);
add("shadow", &StandardFeatures::shadow);
+ add("skylight", &StandardFeatures::skylight);
add("specular", &StandardFeatures::specular);
add("texture", &StandardFeatures::texture);
}