void
ASTfunction_declaration::add_meta (ASTNode *meta)
{
for ( ; meta; meta = meta->nextptr()) {
ASSERT (meta->nodetype() == ASTNode::variable_declaration_node);
const ASTvariable_declaration *metavar = static_cast<const ASTvariable_declaration *>(meta);
Symbol *metasym = metavar->sym();
if (metasym->name() == "builtin") {
m_is_builtin = true;
if (func()->typespec().is_closure()) { // It is a builtin closure
// Force keyword arguments at the end
func()->argcodes(ustring(std::string(func()->argcodes().c_str()) + "."));
}
// For built-in functions, if any of the params are output,
// also automatically mark it as readwrite_special_case.
for (ASTNode *f = formals().get(); f; f = f->nextptr()) {
ASSERT (f->nodetype() == variable_declaration_node);
ASTvariable_declaration *v = (ASTvariable_declaration *)f;
if (v->is_output())
func()->readwrite_special_case (true);
}
}
else if (metasym->name() == "derivs")
func()->takes_derivs (true);
else if (metasym->name() == "printf_args")
func()->printf_args (true);
else if (metasym->name() == "texture_args")
func()->texture_args (true);
else if (metasym->name() == "rw")
func()->readwrite_special_case (true);
}
}
ASTfunction_declaration::ASTfunction_declaration (OSLCompilerImpl *comp,
TypeSpec type, ustring name,
ASTNode *form, ASTNode *stmts, ASTNode *meta)
: ASTNode (function_declaration_node, comp, 0, meta, form, stmts),
m_name(name), m_sym(NULL), m_is_builtin(false)
{
m_typespec = type;
Symbol *f = comp->symtab().clash (name);
if (f && f->symtype() != SymTypeFunction) {
error ("\"%s\" already declared in this scope as a ", name.c_str(),
f->typespec().string().c_str());
// FIXME -- print the file and line of the other definition
f = NULL;
}
// FIXME -- allow multiple function declarations, but only if they
// aren't the same polymorphic type.
if (name[0] == '_' && name[1] == '_' && name[2] == '_') {
error ("\"%s\" : sorry, can't start with three underscores",
name.c_str());
}
m_sym = new FunctionSymbol (name, type, this);
func()->nextpoly ((FunctionSymbol *)f);
std::string argcodes = oslcompiler->code_from_type (m_typespec);
for (ASTNode *arg = form; arg; arg = arg->nextptr()) {
const TypeSpec &t (arg->typespec());
if (t == TypeSpec() /* UNKNOWN */) {
m_typespec = TypeDesc::UNKNOWN;
return;
}
argcodes += oslcompiler->code_from_type (t);
ASSERT (arg->nodetype() == variable_declaration_node);
ASTvariable_declaration *v = (ASTvariable_declaration *)arg;
if (v->init())
v->error ("function parameter '%s' may not have a default initializer.",
v->name().c_str());
}
func()->argcodes (ustring (argcodes));
oslcompiler->symtab().insert (m_sym);
// Typecheck it right now, upon declaration
typecheck (typespec ());
}
ASTshader_declaration::ASTshader_declaration (OSLCompilerImpl *comp,
int stype, ustring name, ASTNode *form,
ASTNode *stmts, ASTNode *meta)
: ASTNode (shader_declaration_node, comp, stype, meta, form, stmts),
m_shadername(name)
{
// Double check some requirements of shader parameters
for (ASTNode *arg = form; arg; arg = arg->nextptr()) {
ASSERT (arg->nodetype() == variable_declaration_node);
ASTvariable_declaration *v = (ASTvariable_declaration *)arg;
if (! v->init())
v->error ("shader parameter '%s' MUST have a default initializer",
v->name().c_str());
if (v->is_output() && v->typespec().is_unsized_array())
v->error ("shader output parameter '%s' can't be unsized array",
v->name().c_str());
}
}