add @compile_var builtin and "is_big_endian" compile var

master
Andrew Kelley 2016-02-02 21:46:53 -07:00
parent a50474e7cf
commit b04e64d300
4 changed files with 55 additions and 1 deletions

View File

@ -998,6 +998,7 @@ enum BuiltinFnId {
BuiltinFnIdCInclude,
BuiltinFnIdCDefine,
BuiltinFnIdCUndef,
BuiltinFnIdCompileVar,
};
struct BuiltinFnEntry {
@ -1058,6 +1059,7 @@ struct CodeGen {
LLVMTargetDataRef target_data_ref;
unsigned pointer_size_bytes;
bool is_big_endian;
bool is_static;
bool strip_debug_symbols;
bool have_exported_main;

View File

@ -3896,7 +3896,7 @@ static TypeTableEntry *analyze_builtin_fn_call_expr(CodeGen *g, ImportTableEntry
uint64_t big_c = char_val->data.x_bignum.data.x_uint;
assert(big_c <= UINT8_MAX);
uint8_t c = big_c;
buf_appendf(context->c_import_buf, "%c", c);
buf_append_char(context->c_import_buf, c);
}
buf_appendf(context->c_import_buf, ">\n");
@ -3907,6 +3907,50 @@ static TypeTableEntry *analyze_builtin_fn_call_expr(CodeGen *g, ImportTableEntry
case BuiltinFnIdCUndef:
zig_panic("TODO");
case BuiltinFnIdCompileVar:
{
AstNode **str_node = node->data.fn_call_expr.params.at(0)->parent_field;
TypeTableEntry *str_type = get_unknown_size_array_type(g, g->builtin_types.entry_u8, true);
TypeTableEntry *resolved_type = analyze_expression(g, import, context, str_type, *str_node);
if (resolved_type->id == TypeTableEntryIdInvalid) {
return resolved_type;
}
ConstExprValue *const_str_val = &get_resolved_expr(*str_node)->const_val;
if (!const_str_val->ok) {
add_node_error(g, *str_node, buf_sprintf("@compile_var requires constant expression"));
return g->builtin_types.entry_void;
}
ConstExprValue *ptr_field = const_str_val->data.x_struct.fields[0];
uint64_t len = ptr_field->data.x_ptr.len;
Buf var_name = BUF_INIT;
buf_resize(&var_name, 0);
for (uint64_t i = 0; i < len; i += 1) {
ConstExprValue *char_val = ptr_field->data.x_ptr.ptr[i];
uint64_t big_c = char_val->data.x_bignum.data.x_uint;
assert(big_c <= UINT8_MAX);
uint8_t c = big_c;
buf_append_char(&var_name, c);
}
ConstExprValue *const_val = &get_resolved_expr(node)->const_val;
const_val->depends_on_compile_var = true;
if (buf_eql_str(&var_name, "is_big_endian")) {
return resolve_expr_const_val_as_bool(g, node, g->is_big_endian);
} else {
add_node_error(g, *str_node,
buf_sprintf("unrecognized compile variable: '%s'", buf_ptr(&var_name)));
return g->builtin_types.entry_invalid;
}
break;
}
}
zig_unreachable();
}

View File

@ -304,6 +304,7 @@ static LLVMValueRef gen_builtin_fn_call_expr(CodeGen *g, AstNode *node) {
case BuiltinFnIdMinValue:
case BuiltinFnIdMaxValue:
case BuiltinFnIdMemberCount:
case BuiltinFnIdCompileVar:
// caught by constant expression eval codegen
zig_unreachable();
}
@ -3221,6 +3222,7 @@ static void define_builtin_fns(CodeGen *g) {
create_builtin_fn_with_arg_count(g, BuiltinFnIdCInclude, "c_include", 1);
create_builtin_fn_with_arg_count(g, BuiltinFnIdCDefine, "c_define", 2);
create_builtin_fn_with_arg_count(g, BuiltinFnIdCUndef, "c_undef", 1);
create_builtin_fn_with_arg_count(g, BuiltinFnIdCompileVar, "compile_var", 1);
}
@ -3267,6 +3269,7 @@ static void init(CodeGen *g, Buf *source_path) {
g->pointer_size_bytes = LLVMPointerSize(g->target_data_ref);
g->is_big_endian = (LLVMByteOrder(g->target_data_ref) == LLVMBigEndian);
g->builder = LLVMCreateBuilder();
g->dbuilder = LLVMZigCreateDIBuilder(g->module, true);

View File

@ -2033,6 +2033,11 @@ fn func() -> bogus {}
)SOURCE", 2,
".tmp_source.zig:3:1: error: redefinition of 'func'",
".tmp_source.zig:2:14: error: use of undeclared identifier 'bogus'");
add_compile_fail_case("bogus compile var", R"SOURCE(
const x = @compile_var("bogus");
)SOURCE", 1, ".tmp_source.zig:2:24: error: unrecognized compile variable: 'bogus'");
}
//////////////////////////////////////////////////////////////////////////////