authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2016-01-08 17:52:45-07:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2016-01-08 17:52:45-07:00
logd14a31100f3f4e7b8d43c8ad794a82da36532aa7
tree9cb70f9785427c24499dff91bc3383837c7de78d
parent2a8d6af7ba9dcea5f13e306f2d032f3f344950af

implement unknown size array indexing and slicing

5 files changed, 59 insertions(+), 23 deletions(-)

src/analyze.cpp+10-6
......@@ -1373,7 +1373,7 @@ static TypeTableEntry *analyze_slice_expr(CodeGen *g, ImportTableEntry *import,
13731373 array_type->data.structure.is_unknown_size_array)
13741374 {
13751375 return_type = get_unknown_size_array_type(g, import,
1376 array_type->data.structure.fields[0].type_entry,
1376 array_type->data.structure.fields[0].type_entry->data.pointer.child_type,
13771377 node->data.slice_expr.is_const);
13781378 } else {
13791379 add_node_error(g, node,
......@@ -1405,15 +1405,19 @@ static TypeTableEntry *analyze_array_access_expr(CodeGen *g, ImportTableEntry *i
14051405
14061406 TypeTableEntry *return_type;
14071407
1408 if (array_type->id == TypeTableEntryIdArray) {
1408 if (array_type->id == TypeTableEntryIdInvalid) {
1409 return_type = g->builtin_types.entry_invalid;
1410 } else if (array_type->id == TypeTableEntryIdArray) {
14091411 return_type = array_type->data.array.child_type;
14101412 } else if (array_type->id == TypeTableEntryIdPointer) {
14111413 return_type = array_type->data.pointer.child_type;
1414 } else if (array_type->id == TypeTableEntryIdStruct &&
1415 array_type->data.structure.is_unknown_size_array)
1416 {
1417 return_type = array_type->data.structure.fields[0].type_entry->data.pointer.child_type;
14121418 } else {
1413 if (array_type->id != TypeTableEntryIdInvalid) {
1414 add_node_error(g, node,
1415 buf_sprintf("array access of non-array type '%s'", buf_ptr(&array_type->name)));
1416 }
1419 add_node_error(g, node,
1420 buf_sprintf("array access of non-array type '%s'", buf_ptr(&array_type->name)));
14171421 return_type = g->builtin_types.entry_invalid;
14181422 }
14191423
src/codegen.cpp+39-2
......@@ -260,6 +260,15 @@ static LLVMValueRef gen_array_ptr(CodeGen *g, AstNode *node) {
260260 };
261261 add_debug_source_node(g, node);
262262 return LLVMBuildInBoundsGEP(g->builder, array_ptr, indices, 1, "");
263 } else if (type_entry->id == TypeTableEntryIdStruct) {
264 assert(type_entry->data.structure.is_unknown_size_array);
265 assert(LLVMGetTypeKind(LLVMTypeOf(array_ptr)) == LLVMPointerTypeKind);
266 assert(LLVMGetTypeKind(LLVMGetElementType(LLVMTypeOf(array_ptr))) == LLVMStructTypeKind);
267
268 add_debug_source_node(g, node);
269 LLVMValueRef ptr_ptr = LLVMBuildStructGEP(g->builder, array_ptr, 0, "");
270 LLVMValueRef ptr = LLVMBuildLoad(g->builder, ptr_ptr, "");
271 return LLVMBuildInBoundsGEP(g->builder, ptr, &subscript_value, 1, "");
263272 } else {
264273 zig_unreachable();
265274 }
......@@ -314,9 +323,9 @@ static LLVMValueRef gen_slice_expr(CodeGen *g, AstNode *node) {
314323 TypeTableEntry *array_type = get_expr_type(array_ref_node);
315324
316325 LLVMValueRef tmp_struct_ptr = node->codegen_node->data.struct_val_expr_node.ptr;
326 LLVMValueRef array_ptr = gen_array_base_ptr(g, array_ref_node);
317327
318328 if (array_type->id == TypeTableEntryIdArray) {
319 LLVMValueRef array_ptr = gen_array_base_ptr(g, array_ref_node);
320329 LLVMValueRef start_val = gen_expr(g, node->data.slice_expr.start);
321330 LLVMValueRef end_val;
322331 if (node->data.slice_expr.end) {
......@@ -343,7 +352,31 @@ static LLVMValueRef gen_slice_expr(CodeGen *g, AstNode *node) {
343352 zig_panic("TODO gen_slice_expr pointer");
344353 } else if (array_type->id == TypeTableEntryIdStruct) {
345354 assert(array_type->data.structure.is_unknown_size_array);
346 zig_panic("TODO gen_slice_expr unknown size array");
355 assert(LLVMGetTypeKind(LLVMTypeOf(array_ptr)) == LLVMPointerTypeKind);
356 assert(LLVMGetTypeKind(LLVMGetElementType(LLVMTypeOf(array_ptr))) == LLVMStructTypeKind);
357
358 LLVMValueRef start_val = gen_expr(g, node->data.slice_expr.start);
359 LLVMValueRef end_val;
360 if (node->data.slice_expr.end) {
361 end_val = gen_expr(g, node->data.slice_expr.end);
362 } else {
363 add_debug_source_node(g, node);
364 LLVMValueRef src_len_ptr = LLVMBuildStructGEP(g->builder, array_ptr, 1, "");
365 end_val = LLVMBuildLoad(g->builder, src_len_ptr, "");
366 }
367
368 add_debug_source_node(g, node);
369 LLVMValueRef src_ptr_ptr = LLVMBuildStructGEP(g->builder, array_ptr, 0, "");
370 LLVMValueRef src_ptr = LLVMBuildLoad(g->builder, src_ptr_ptr, "");
371 LLVMValueRef ptr_field_ptr = LLVMBuildStructGEP(g->builder, tmp_struct_ptr, 0, "");
372 LLVMValueRef slice_start_ptr = LLVMBuildInBoundsGEP(g->builder, src_ptr, &start_val, 1, "");
373 LLVMBuildStore(g->builder, slice_start_ptr, ptr_field_ptr);
374
375 LLVMValueRef len_field_ptr = LLVMBuildStructGEP(g->builder, tmp_struct_ptr, 1, "");
376 LLVMValueRef len_value = LLVMBuildSub(g->builder, end_val, start_val, "");
377 LLVMBuildStore(g->builder, len_value, len_field_ptr);
378
379 return tmp_struct_ptr;
347380 } else {
348381 zig_unreachable();
349382 }
......@@ -421,6 +454,10 @@ static LLVMValueRef gen_lvalue(CodeGen *g, AstNode *expr_node, AstNode *node,
421454 } else if (array_type->id == TypeTableEntryIdPointer) {
422455 *out_type_entry = array_type->data.pointer.child_type;
423456 target_ref = gen_array_ptr(g, node);
457 } else if (array_type->id == TypeTableEntryIdStruct) {
458 assert(array_type->data.structure.is_unknown_size_array);
459 *out_type_entry = array_type->data.structure.fields[0].type_entry->data.pointer.child_type;
460 target_ref = gen_array_ptr(g, node);
424461 } else {
425462 zig_unreachable();
426463 }
std/builtin.zig+1
......@@ -10,6 +10,7 @@ export fn memset(dest: &u8, c: u8, n: usize) -> &u8 {
1010 return dest;
1111}
1212
13// TODO annotate parameters with noalias
1314export fn memcpy(dest: &u8, src: &const u8, n: usize) -> &u8 {
1415 var index : #typeof(n) = 0;
1516 while (index != n) {
std/rand.zig+2-4
......@@ -45,8 +45,7 @@ pub struct Rand {
4545 var rand_val_array : [#sizeof(u32)]u8;
4646 *(rand_val_array.ptr as &u32) = r.get_u32();
4747 while (bytes_left > 0) {
48 // TODO array index operator so we can remove the .ptr
49 buf.ptr[buf.len - bytes_left] = rand_val_array[#sizeof(u32) - bytes_left];
48 buf[buf.len - bytes_left] = rand_val_array[#sizeof(u32) - bytes_left];
5049 bytes_left -= 1;
5150 }
5251 }
......@@ -88,8 +87,7 @@ pub struct Rand {
8887 fn get_bytes_aligned(r: &Rand, buf: []u8) -> usize {
8988 var bytes_left = buf.len;
9089 while (bytes_left >= 4) {
91 // TODO: array access so we can remove .ptr
92 *(&buf.ptr[buf.len - bytes_left] as &u32) = r.get_u32();
90 *(&buf[buf.len - bytes_left] as &u32) = r.get_u32();
9391 bytes_left -= #sizeof(u32);
9492 }
9593 return bytes_left;
std/std.zig+7-11
......@@ -26,7 +26,7 @@ pub fn fprint_str(fd: isize, str: []const u8) -> isize {
2626pub fn print_u64(x: u64) -> isize {
2727 // TODO use max_u64_base10_digits instead of hardcoding 20
2828 var buf: [20]u8;
29 const len = buf_print_u64(buf.ptr, x);
29 const len = buf_print_u64(buf, x);
3030 return write(stdout_fileno, buf.ptr, len);
3131}
3232
......@@ -35,7 +35,7 @@ pub fn print_u64(x: u64) -> isize {
3535pub fn print_i64(x: i64) -> isize {
3636 // TODO use max_u64_base10_digits instead of hardcoding 20
3737 var buf: [20]u8;
38 const len = buf_print_i64(buf.ptr, x);
38 const len = buf_print_i64(buf, x);
3939 return write(stdout_fileno, buf.ptr, len);
4040}
4141
......@@ -56,8 +56,7 @@ pub fn parse_u64(buf: []u8, radix: u8, result: &u64) -> bool {
5656
5757 var i : #typeof(buf.len) = 0;
5858 while (i < buf.len) {
59 // TODO array indexing operator
60 const c = buf.ptr[i];
59 const c = buf[i];
6160 const digit = char_to_digit(c);
6261
6362 if (digit > radix) {
......@@ -100,20 +99,16 @@ fn char_to_digit(c: u8) -> u8 {
10099
101100const max_u64_base10_digits: usize = 20;
102101
103// TODO use an array for out_buf instead of pointer. this should give bounds checking in
104// debug mode and length can get optimized out in release mode. requires array slicing syntax
105// for the buf_print_u64 call.
106fn buf_print_i64(out_buf: &u8, x: i64) -> usize {
102fn buf_print_i64(out_buf: []u8, x: i64) -> usize {
107103 if (x < 0) {
108104 out_buf[0] = '-';
109 return 1 + buf_print_u64(&out_buf[1], ((-(x + 1)) as u64) + 1);
105 return 1 + buf_print_u64(out_buf[1...], ((-(x + 1)) as u64) + 1);
110106 } else {
111107 return buf_print_u64(out_buf, x as u64);
112108 }
113109}
114110
115// TODO use an array for out_buf instead of pointer.
116fn buf_print_u64(out_buf: &u8, x: u64) -> usize {
111fn buf_print_u64(out_buf: []u8, x: u64) -> usize {
117112 var buf: [max_u64_base10_digits]u8;
118113 var a = x;
119114 var index = buf.len;
......@@ -130,6 +125,7 @@ fn buf_print_u64(out_buf: &u8, x: u64) -> usize {
130125 const len = buf.len - index;
131126
132127 // TODO memcpy intrinsic
128 // @memcpy(out_buf, buf, len);
133129 var i: usize = 0;
134130 while (i < len) {
135131 out_buf[i] = buf[index + i];