You've already forked godot
mirror of
https://github.com/godotengine/godot.git
synced 2025-11-06 12:20:30 +00:00
Added constant support to shaders
Co-authored-by: DavidSichma <sichmada@gmail.com>
This commit is contained in:
@@ -132,6 +132,7 @@ const char *ShaderLanguage::token_names[TK_MAX] = {
|
||||
"TYPE_SAMPLERCUBE",
|
||||
"INTERPOLATION_FLAT",
|
||||
"INTERPOLATION_SMOOTH",
|
||||
"CONST",
|
||||
"PRECISION_LOW",
|
||||
"PRECISION_MID",
|
||||
"PRECISION_HIGH",
|
||||
@@ -271,6 +272,7 @@ const ShaderLanguage::KeyWord ShaderLanguage::keyword_list[] = {
|
||||
{ TK_TYPE_SAMPLERCUBE, "samplerCube" },
|
||||
{ TK_INTERPOLATION_FLAT, "flat" },
|
||||
{ TK_INTERPOLATION_SMOOTH, "smooth" },
|
||||
{ TK_CONST, "const" },
|
||||
{ TK_PRECISION_LOW, "lowp" },
|
||||
{ TK_PRECISION_MID, "mediump" },
|
||||
{ TK_PRECISION_HIGH, "highp" },
|
||||
@@ -920,6 +922,16 @@ bool ShaderLanguage::_find_identifier(const BlockNode *p_block, const Map<String
|
||||
return true;
|
||||
}
|
||||
|
||||
if (shader->constants.has(p_identifier)) {
|
||||
if (r_data_type) {
|
||||
*r_data_type = shader->constants[p_identifier].type;
|
||||
}
|
||||
if (r_type) {
|
||||
*r_type = IDENTIFIER_CONSTANT;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
for (int i = 0; i < shader->functions.size(); i++) {
|
||||
|
||||
if (!shader->functions[i].callable)
|
||||
@@ -2699,6 +2711,12 @@ bool ShaderLanguage::_validate_assign(Node *p_node, const Map<StringName, BuiltI
|
||||
return false;
|
||||
}
|
||||
|
||||
if (shader->constants.has(var->name)) {
|
||||
if (r_message)
|
||||
*r_message = RTR("Constants cannot be modified.");
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!(p_builtin_types.has(var->name) && p_builtin_types[var->name].constant)) {
|
||||
return true;
|
||||
}
|
||||
@@ -3993,7 +4011,7 @@ Error ShaderLanguage::_parse_block(BlockNode *p_block, const Map<StringName, Bui
|
||||
|
||||
} else {
|
||||
|
||||
//nothng else, so expression
|
||||
//nothing else, so expression
|
||||
_set_tkpos(pos); //rollback
|
||||
Node *expr = _parse_and_reduce_expression(p_block, p_builtin_types);
|
||||
if (!expr)
|
||||
@@ -4322,24 +4340,30 @@ Error ShaderLanguage::_parse_shader(const Map<StringName, FunctionInfo> &p_funct
|
||||
|
||||
} break;
|
||||
default: {
|
||||
//function
|
||||
//function or constant variable
|
||||
|
||||
bool is_constant = false;
|
||||
DataPrecision precision = PRECISION_DEFAULT;
|
||||
DataType type;
|
||||
StringName name;
|
||||
|
||||
if (tk.type == TK_CONST) {
|
||||
is_constant = true;
|
||||
tk = _get_token();
|
||||
}
|
||||
|
||||
if (is_token_precision(tk.type)) {
|
||||
precision = get_token_precision(tk.type);
|
||||
tk = _get_token();
|
||||
}
|
||||
|
||||
if (!is_token_datatype(tk.type)) {
|
||||
_set_error("Expected function, uniform or varying ");
|
||||
_set_error("Expected constant, function, uniform or varying ");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
if (!is_token_variable_datatype(tk.type)) {
|
||||
_set_error("Invalid data type for function return (samplers not allowed)");
|
||||
_set_error("Invalid data type for constants or function return (samplers not allowed)");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
@@ -4359,8 +4383,73 @@ Error ShaderLanguage::_parse_shader(const Map<StringName, FunctionInfo> &p_funct
|
||||
|
||||
tk = _get_token();
|
||||
if (tk.type != TK_PARENTHESIS_OPEN) {
|
||||
_set_error("Expected '(' after identifier");
|
||||
return ERR_PARSE_ERROR;
|
||||
if (type == TYPE_VOID) {
|
||||
_set_error("Expected '(' after function identifier");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
//variable
|
||||
|
||||
while (true) {
|
||||
ShaderNode::Constant constant;
|
||||
constant.type = type;
|
||||
constant.precision = precision;
|
||||
constant.initializer = NULL;
|
||||
|
||||
if (tk.type == TK_OP_ASSIGN) {
|
||||
|
||||
if (!is_constant) {
|
||||
_set_error("Expected 'const' keyword before constant definition");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
//variable created with assignment! must parse an expression
|
||||
Node *expr = _parse_and_reduce_expression(NULL, Map<StringName, BuiltInInfo>());
|
||||
if (!expr)
|
||||
return ERR_PARSE_ERROR;
|
||||
|
||||
if (expr->type != Node::TYPE_CONSTANT) {
|
||||
_set_error("Expected constant expression after '='");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
constant.initializer = static_cast<ConstantNode *>(expr);
|
||||
|
||||
if (type != expr->get_datatype()) {
|
||||
_set_error("Invalid assignment of '" + get_datatype_name(expr->get_datatype()) + "' to '" + get_datatype_name(type) + "'");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
tk = _get_token();
|
||||
} else {
|
||||
_set_error("Expected initialization of constant");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
shader->constants[name] = constant;
|
||||
if (tk.type == TK_COMMA) {
|
||||
tk = _get_token();
|
||||
if (tk.type != TK_IDENTIFIER) {
|
||||
_set_error("Expected identifier after type");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
name = tk.text;
|
||||
if (_find_identifier(NULL, Map<StringName, BuiltInInfo>(), name)) {
|
||||
_set_error("Redefinition of '" + String(name) + "'");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
|
||||
tk = _get_token();
|
||||
|
||||
} else if (tk.type == TK_SEMICOLON) {
|
||||
break;
|
||||
} else {
|
||||
_set_error("Expected ',' or ';' after constant");
|
||||
return ERR_PARSE_ERROR;
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
Map<StringName, BuiltInInfo> builtin_types;
|
||||
|
||||
Reference in New Issue
Block a user