X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fprogramcompiler.cpp;h=2bb2694609f89277b9d36efc50a0df5de0354d34;hb=a209ca32465c61b38f3a57ee6577568f3994e8ab;hp=55f62ae93e16d954b846fa13262f9108220c2dda;hpb=78cec1556c85db6beb4f3d9f918b5a1f421719ef;p=libs%2Fgl.git diff --git a/source/programcompiler.cpp b/source/programcompiler.cpp index 55f62ae9..2bb26946 100644 --- a/source/programcompiler.cpp +++ b/source/programcompiler.cpp @@ -8,6 +8,26 @@ using namespace std; +namespace { + +const char builtins_src[] = + "////// vertex\n" + "out gl_PerVertex {\n" + " vec4 gl_Position;\n" + " float gl_ClipDistance[];\n" + "};" + "////// geometry\n" + "in gl_PerVertex {\n" + " vec4 gl_Position;\n" + " float gl_ClipDistance[];\n" + "} gl_in[];\n" + "out gl_PerVertex {\n" + " vec4 gl_Position;\n" + " float gl_ClipDistance[];\n" + "};\n"; + +} + namespace Msp { namespace GL { @@ -51,6 +71,35 @@ void ProgramCompiler::add_shaders(Program &program) program.bind_attribute(TEXCOORD4, "texcoord"); } +Module *ProgramCompiler::create_builtins_module() +{ + ProgramParser parser; + Module *module = new Module(parser.parse(builtins_src)); + for(list::iterator i=module->stages.begin(); i!=module->stages.end(); ++i) + { + VariableResolver resolver; + i->content.visit(resolver); + for(map::iterator j=i->content.variables.begin(); j!=i->content.variables.end(); ++j) + j->second->linked_declaration = j->second; + } + return module; +} + +Module &ProgramCompiler::get_builtins_module() +{ + static RefPtr builtins_module = create_builtins_module(); + return *builtins_module; +} + +Stage *ProgramCompiler::get_builtins(StageType type) +{ + Module &module = get_builtins_module(); + for(list::iterator i=module.stages.begin(); i!=module.stages.end(); ++i) + if(i->type==type) + return &*i; + return 0; +} + void ProgramCompiler::process() { for(list::iterator i=module->stages.begin(); i!=module->stages.end(); ++i) @@ -83,7 +132,7 @@ bool ProgramCompiler::optimize(Stage &stage) remover.to_remove = unused_locator.unused_nodes; remover.apply(stage); - return remover.n_removed; + return !unused_locator.unused_nodes.empty(); } void ProgramCompiler::inject_block(Block &target, const Block &source) @@ -242,7 +291,7 @@ void ProgramCompiler::Formatter::visit(VariableDeclaration &var) formatted += "const "; if(!var.sampling.empty()) formatted += format("%s ", var.sampling); - if(!var.interface.empty()) + if(!var.interface.empty() && var.interface!=block_interface) formatted += format("%s ", var.interface); formatted += format("%s %s", var.type, var.name); if(var.array) @@ -263,6 +312,7 @@ void ProgramCompiler::Formatter::visit(VariableDeclaration &var) void ProgramCompiler::Formatter::visit(InterfaceBlock &iface) { + SetForScope set(block_interface, iface.interface); formatted += format("%s %s\n", iface.interface, iface.name); iface.members.visit(*this); formatted += ';'; @@ -333,6 +383,17 @@ ProgramCompiler::VariableResolver::VariableResolver(): anonymous(false) { } +void ProgramCompiler::VariableResolver::apply(Stage &s) +{ + SetForScope set(stage, &s); + Stage *builtins = get_builtins(stage->type); + if(builtins) + blocks.push_back(&builtins->content); + stage->content.visit(*this); + if(builtins) + blocks.pop_back(); +} + void ProgramCompiler::VariableResolver::visit(Block &block) { blocks.push_back(&block); @@ -407,6 +468,9 @@ void ProgramCompiler::VariableResolver::visit(VariableDeclaration &var) var.type_declaration = j->second; } + if(!block_interface.empty() && var.interface.empty()) + var.interface = block_interface; + TraversingVisitor::visit(var); blocks.back()->variables[var.name] = &var; if(anonymous && blocks.size()>1) @@ -416,6 +480,7 @@ void ProgramCompiler::VariableResolver::visit(VariableDeclaration &var) void ProgramCompiler::VariableResolver::visit(InterfaceBlock &iface) { SetFlag set(anonymous); + SetForScope set2(block_interface, iface.interface); TraversingVisitor::visit(iface); } @@ -491,8 +556,12 @@ bool ProgramCompiler::InterfaceGenerator::generate_interface(VariableDeclaration iface_var->type = var.type; iface_var->type_declaration = var.type_declaration; iface_var->name = name; - iface_var->array = (var.array || (stage->type==GEOMETRY && iface=="in")); - iface_var->array_size = var.array_size; + if(stage->type==GEOMETRY) + iface_var->array = ((var.array && var.interface!="in") || iface=="in"); + else + iface_var->array = var.array; + if(iface_var->array) + iface_var->array_size = var.array_size; if(iface=="in") iface_var->linked_declaration = &var; iface_declarations[name] = iface_var; @@ -587,6 +656,24 @@ void ProgramCompiler::InterfaceGenerator::visit(Passthrough &pass) } } + if(stage->type==GEOMETRY) + { + VariableReference *ref = new VariableReference; + ref->name = "gl_in"; + + BinaryExpression *subscript = new BinaryExpression; + subscript->left = ref; + subscript->oper = "["; + subscript->right = pass.subscript; + subscript->after = "]"; + + MemberAccess *memacc = new MemberAccess; + memacc->left = subscript; + memacc->member = "gl_Position"; + + insert_assignment("gl_Position", memacc); + } + for(vector::const_iterator i=pass_vars.begin(); i!=pass_vars.end(); ++i) { string out_name = change_prefix((*i)->name, out_prefix); @@ -626,6 +713,7 @@ void ProgramCompiler::VariableRenamer::visit(VariableDeclaration &var) ProgramCompiler::UnusedVariableLocator::UnusedVariableLocator(): + aggregate(0), assignment(false), record_target(false), assignment_target(0), @@ -656,6 +744,10 @@ void ProgramCompiler::UnusedVariableLocator::visit(VariableReference &var) if(assignment && var.declaration==assignment_target) self_referencing = true; } + + map::iterator i = aggregates.find(var.declaration); + if(i!=aggregates.end()) + unused_nodes.erase(i->second); } void ProgramCompiler::UnusedVariableLocator::visit(MemberAccess &memacc) @@ -710,50 +802,50 @@ void ProgramCompiler::UnusedVariableLocator::visit(ExpressionStatement &expr) assignment = false; } +void ProgramCompiler::UnusedVariableLocator::visit(StructDeclaration &strct) +{ + SetForScope set(aggregate, &strct); + unused_nodes.insert(&strct); + TraversingVisitor::visit(strct); +} + void ProgramCompiler::UnusedVariableLocator::visit(VariableDeclaration &var) { - unused_nodes.insert(&var); - if(var.init_expression) + if(aggregate) + aggregates[&var] = aggregate; + else { - unused_nodes.insert(&*var.init_expression); - assignments[&var] = &*var.init_expression; + unused_nodes.insert(&var); + if(var.init_expression) + { + unused_nodes.insert(&*var.init_expression); + assignments[&var] = &*var.init_expression; + } } + unused_nodes.erase(var.type_declaration); TraversingVisitor::visit(var); } +void ProgramCompiler::UnusedVariableLocator::visit(InterfaceBlock &iface) +{ + SetForScope set(aggregate, &iface); + unused_nodes.insert(&iface); + TraversingVisitor::visit(iface); +} -ProgramCompiler::NodeRemover::NodeRemover(): - n_removed(0), - immutable_block(false), - remove_block(false) -{ } void ProgramCompiler::NodeRemover::visit(Block &block) { - remove_block = immutable_block; for(list >::iterator i=block.body.begin(); i!=block.body.end(); ) { - bool remove = to_remove.count(&**i); - if(!remove) - remove_block = false; (*i)->visit(*this); - - if(remove ? !immutable_block : remove_block) - { + if(to_remove.count(&**i)) block.body.erase(i++); - ++n_removed; - } else ++i; } } -void ProgramCompiler::NodeRemover::visit(StructDeclaration &strct) -{ - SetFlag set(immutable_block); - TraversingVisitor::visit(strct); -} - void ProgramCompiler::NodeRemover::visit(VariableDeclaration &var) { if(to_remove.count(&var)) @@ -767,11 +859,5 @@ void ProgramCompiler::NodeRemover::visit(VariableDeclaration &var) var.init_expression = 0; } -void ProgramCompiler::NodeRemover::visit(InterfaceBlock &iface) -{ - SetFlag set(immutable_block); - TraversingVisitor::visit(iface); -} - } // namespace GL } // namespace Msp