language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-29 06:44:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Have examples with zig hosts use roc_alloc etc

Browse source
This commit is contained in:
Richard Feldman 2021-05-22 22:56:28 -04:00
parent 2a68c0c2cd
commit 175115ee4e
5 changed files with 85 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
examples/benchmarks/platform/host.zig
View file

@ -29,6 +29,22 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
extern fn roc__mainForHost_1_Fx_size() i64;
extern fn roc__mainForHost_1_Fx_result_size() i64;
extern fn malloc(size: usize) callconv(.C) ?*c_void;
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
return malloc(size) orelse unreachable;
}
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
}
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
}
const Unit = extern struct {};
pub export fn main() u8 {

21
examples/custom-malloc/platform/host.zig
View file

@ -28,6 +28,10 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
extern fn roc__mainForHost_1_Fx_size() i64;
extern fn roc__mainForHost_1_Fx_result_size() i64;
extern fn malloc(size: usize) callconv(.C) ?*c_void;
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
const Unit = extern struct {};
pub export fn main() u8 {
@ -60,28 +64,33 @@ pub export fn main() u8 {
return 0;
}
pub export fn malloc(size: usize) callconv(.C) ?*c_void {
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
const stdout = std.io.getStdOut().writer();
const allocator = testing.allocator;
// Perform the actual malloc
const startNs = std.time.nanoTimestamp();
const ptr = allocator.alignedAlloc(u8, 16, size) catch unreachable;
const ptr = malloc(size) orelse unreachable;
const endNs = std.time.nanoTimestamp();
const totalMs = @divTrunc(endNs - startNs, 1000);
stdout.print("\x1B[36m{} | \x1B[39m Custom malloc allocated {} bytes in {} ms!\n", .{startNs, size, totalMs}) catch unreachable;
return @ptrCast(?*c_void, ptr);
return ptr;
}
pub export fn free(c_ptr: *u128) callconv(.C) void {
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
}
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
const stdout = std.io.getStdOut().writer();
const allocator = testing.allocator;
// Perform the actual free
const startNs = std.time.nanoTimestamp();
allocator.destroy(c_ptr);
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
const endNs = std.time.nanoTimestamp();
const totalMs = @divTrunc(endNs - startNs, 1000);
@ -149,7 +158,7 @@ pub export fn roc_fx_readAllUtf8(rocPath: RocStr) callconv(.C) ReadResult {
};
var str_ptr = @ptrCast([*]u8, content);
var roc_str3 = RocStr.init(testing.allocator, str_ptr, content.len);
var roc_str3 = RocStr.init(str_ptr, content.len);
return .{ .bytes = roc_str3, .errno = 0 };
}

20
examples/hello-world/platform/host.zig
View file

@ -19,6 +19,24 @@ comptime {
}
}
extern fn malloc(size: usize) callconv(.C) ?*c_void;
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
// TODO orelse cxa_throw instead of orelse unreachable
return malloc(size) orelse unreachable;
}
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
// TODO orelse cxa_throw instead of orelse unreachable
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
}
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
}
const mem = std.mem;
const Allocator = mem.Allocator;
@ -45,7 +63,7 @@ pub export fn main() i32 {
// stdout the result
stdout.print("{}\n", .{callresult.content.asSlice()}) catch unreachable;
callresult.content.deinit(std.heap.c_allocator);
callresult.content.deinit();
// end time
var ts2: std.os.timespec = undefined;

17
examples/quicksort/platform/host.zig
View file

@ -22,6 +22,23 @@ const Allocator = mem.Allocator;
extern fn roc__mainForHost_1_exposed(RocList, *RocCallResult) void;
extern fn malloc(size: usize) callconv(.C) ?*c_void;
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
return malloc(size) orelse unreachable;
}
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
}
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
}
// warning! the array is currently stack-allocated so don't make this too big
const NUM_NUMS = 100;

19
examples/task/platform/host.zig
View file

@ -28,6 +28,23 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
extern fn roc__mainForHost_1_Fx_size() i64;
extern fn roc__mainForHost_1_Fx_result_size() i64;
extern fn malloc(size: usize) callconv(.C) ?*c_void;
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
return malloc(size) orelse unreachable;
}
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
}
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
}
const Unit = extern struct {};
pub export fn main() u8 {
@ -120,7 +137,7 @@ pub export fn roc_fx_readAllUtf8(rocPath: RocStr) callconv(.C) ReadResult {
};
var str_ptr = @ptrCast([*]u8, content);
var roc_str3 = RocStr.init(testing.allocator, str_ptr, content.len);
var roc_str3 = RocStr.init(str_ptr, content.len);
return .{ .bytes = roc_str3, .errno = 0 };
}