Handle nil old data case

This commit is contained in:
Damian Tarnawski
2025-08-24 15:04:19 +02:00
parent ac4a89e765
commit 66f4c93420
+48 -41
View File
@@ -249,68 +249,73 @@ _reserve_soa :: proc(array: ^$T/#soa[dynamic]$E, capacity: int, zero_memory: boo
old_data := (^rawptr)(array)^ old_data := (^rawptr)(array)^
new_bytes, resize_err := array.allocator.procedure( if old_data != nil {
array.allocator.data, .Resize_Non_Zeroed, new_size, max_align, new_bytes, resize_err := array.allocator.procedure(
old_data, old_size, loc, array.allocator.data, .Resize_Non_Zeroed, new_size, max_align,
) old_data, old_size, loc,
new_data := raw_data(new_bytes) )
new_data := raw_data(new_bytes)
old_offset := 0 if resize_err == .None {
new_offset := 0
if resize_err == .None { footer.cap = capacity
footer.cap = capacity old_offset := 0
new_offset := 0
// Correct data memory // Correct data memory
// from: |x x y y z z _ _ _| // from: |x x y y z z _ _ _|
// to: |x x _ y y _ z z _| // to: |x x _ y y _ z z _|
// move old data to the end of the new allocation to avoid overlap // move old data to the end of the new allocation to avoid overlap
old_start := uintptr(new_data) + uintptr(new_size - old_size) old_start := uintptr(new_data) + uintptr(new_size - old_size)
mem_copy(rawptr(old_start), old_data, old_size) mem_copy(rawptr(old_start), new_data, old_size)
// now: |_ _ _ x x y y z z| // now: |_ _ _ x x y y z z|
for i in 0..<field_count { for i in 0..<field_count {
type := si.types[i].variant.(Type_Info_Multi_Pointer).elem type := si.types[i].variant.(Type_Info_Multi_Pointer).elem
old_offset = align_forward_int(old_offset, max_align) old_offset = align_forward_int(old_offset, max_align)
new_offset = align_forward_int(new_offset, max_align) new_offset = align_forward_int(new_offset, max_align)
new_data_elem := rawptr(uintptr(new_data) + uintptr(new_offset)) new_data_elem := rawptr(uintptr(new_data) + uintptr(new_offset))
old_data_elem := rawptr(old_start + uintptr(old_offset)) old_data_elem := rawptr(old_start + uintptr(old_offset))
old_size_elem := type.size * old_cap old_size_elem := type.size * old_cap
new_size_elem := type.size * capacity new_size_elem := type.size * capacity
mem_copy(new_data_elem, old_data_elem, old_size_elem) mem_copy(new_data_elem, old_data_elem, old_size_elem)
(^rawptr)(uintptr(array) + i*size_of(rawptr))^ = new_data_elem (^rawptr)(uintptr(array) + i*size_of(rawptr))^ = new_data_elem
if zero_memory { if zero_memory {
mem_zero(rawptr(uintptr(new_data_elem) + uintptr(old_size_elem)), new_size_elem - old_size_elem) mem_zero(rawptr(uintptr(new_data_elem) + uintptr(old_size_elem)), new_size_elem - old_size_elem)
}
old_offset += old_size_elem
new_offset += new_size_elem
} }
old_offset += old_size_elem return nil
new_offset += new_size_elem
} }
return nil if resize_err != .Mode_Not_Implemented {
return resize_err
}
} }
if resize_err != .Mode_Not_Implemented { new_bytes := array.allocator.procedure(
return resize_err
}
new_bytes = array.allocator.procedure(
array.allocator.data, .Alloc if zero_memory else .Alloc_Non_Zeroed, new_size, max_align, array.allocator.data, .Alloc if zero_memory else .Alloc_Non_Zeroed, new_size, max_align,
nil, old_size, loc, nil, old_size, loc,
) or_return ) or_return
new_data = raw_data(new_bytes) new_data := raw_data(new_bytes)
footer.cap = capacity footer.cap = capacity
old_offset := 0
new_offset := 0
// Correct data memory // Correct data memory
// from: |x x y y z z| ... |_ _ _ _ _ _ _ _ _| // from: |x x y y z z| ... |_ _ _ _ _ _ _ _ _|
// to: |x x _ y y _ z z _| // to: |x x _ y y _ z z _|
@@ -332,10 +337,12 @@ _reserve_soa :: proc(array: ^$T/#soa[dynamic]$E, capacity: int, zero_memory: boo
new_offset += type.size * capacity new_offset += type.size * capacity
} }
array.allocator.procedure( if old_data != nil {
array.allocator.data, .Free, 0, max_align, array.allocator.procedure(
old_data, old_size, loc, array.allocator.data, .Free, 0, max_align,
) or_return old_data, old_size, loc,
) or_return
}
return nil return nil
} }