language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-30 23:31:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

implement startsWith builtin using Zig

Browse source
This commit is contained in:
Sébastien Besnier 2020-11-22 21:48:42 +01:00
parent b615b0127c
commit b36c2bf499
5 changed files with 130 additions and 113 deletions
Download patch file Download diff file Expand all files Collapse all files

112
compiler/gen/src/llvm/build_str.rs
View file

@ -2,8 +2,7 @@ use crate::llvm::build::{
call_bitcode_fn, call_void_bitcode_fn, ptr_from_symbol, Env, InPlace, Scope,
};
use crate::llvm::build_list::{
allocate_list, build_basic_phi2, empty_list, incrementing_elem_loop, incrementing_index_loop,
load_list_ptr, store_list,
allocate_list, build_basic_phi2, empty_list, incrementing_elem_loop, load_list_ptr, store_list,
};
use crate::llvm::convert::collection;
use inkwell::builder::Builder;
@ -677,104 +676,37 @@ pub fn str_starts_with<'a, 'ctx, 'env>(
_inplace: InPlace,
scope: &Scope<'a, 'ctx>,
parent: FunctionValue<'ctx>,
first_str_symbol: Symbol,
second_str_symbol: Symbol,
str_symbol: Symbol,
prefix_symbol: Symbol,
) -> BasicValueEnum<'ctx> {
let builder = env.builder;
let ctx = env.context;
let prefix_str_ptr = ptr_from_symbol(scope, first_str_symbol);
let str_ptr = ptr_from_symbol(scope, second_str_symbol);
let bool_wrapper_type = BasicTypeEnum::IntType(ctx.bool_type());
let str_ptr = ptr_from_symbol(scope, str_symbol);
let prefix_ptr = ptr_from_symbol(scope, prefix_symbol);
let ret_type = BasicTypeEnum::IntType(ctx.bool_type());
load_str(
env,
parent,
*prefix_str_ptr,
bool_wrapper_type,
|prefix_str_ptr, prefix_str_len, _prefix_str_smallness| {
*str_ptr,
ret_type,
|str_bytes_ptr, str_len, _str_smallness| {
load_str(
env,
parent,
*str_ptr,
bool_wrapper_type,
|input_str_ptr, input_str_len, _second_str_smallness| {
let return_false = || ctx.bool_type().const_zero().into();
let if_input_str_is_longer_or_equal_to_prefix = builder.build_int_compare(
IntPredicate::UGE,
input_str_len,
prefix_str_len,
"str_longer_than_prefix",
);
let check_if_str_starts_with_prefix = || {
// Loop over prefix and compare each character to the one from the input
// string at the same index
// If they are different - return false
incrementing_index_loop(
builder,
ctx,
parent,
prefix_str_len,
"starts_with_loop",
|index| {
// The pointer to the element in the list
let prefix_ptr = unsafe {
builder.build_in_bounds_gep(
prefix_str_ptr,
&[index],
"prefix_index",
)
};
let input_ptr = unsafe {
builder.build_in_bounds_gep(
input_str_ptr,
&[index],
"input_str_index",
)
};
let prefix_char = builder
.build_load(prefix_ptr, "get_prefix_char")
.into_int_value();
let input_char = builder
.build_load(input_ptr, "get_input_char")
.into_int_value();
let comparison = builder.build_int_compare(
IntPredicate::EQ,
prefix_char,
input_char,
"prefix_char_equals_to_input_char",
);
let condition_block =
ctx.append_basic_block(parent, "condition_block");
// Empty block to continue the loop if the comparison is true
let then_block = ctx.append_basic_block(parent, "then_block");
//
let else_block = ctx.append_basic_block(parent, "else_block");
// Build the condition_block
builder.position_at_end(condition_block);
builder
.build_conditional_branch(comparison, then_block, else_block);
// Build the else_block
builder.position_at_end(else_block);
// Do an early return
builder.build_return(Some(&ctx.bool_type().const_zero()));
// Return true by default
// builder.build_return(Some(&ctx.bool_type().const_int(1, false)));
},
);
};
build_basic_phi2(
*prefix_ptr,
ret_type,
|prefix_bytes_ptr, prefix_len, _prefix_smallness| {
call_bitcode_fn(
env,
parent,
if_input_str_is_longer_or_equal_to_prefix,
check_if_str_starts_with_prefix,
return_false,
bool_wrapper_type,
&[
BasicValueEnum::PointerValue(str_bytes_ptr),
BasicValueEnum::IntValue(str_len),
BasicValueEnum::PointerValue(prefix_bytes_ptr),
BasicValueEnum::IntValue(prefix_len),
],
&bitcode::STR_STARTS_WITH,
)
},
)