roc/compiler/builtins/bitcode/src/lib.rs

// NOTE: Editing this file on its own does nothing! The procedure for
// incorporating changes here is in this crate' README.

#![crate_type = "lib"]
#![no_std]

mod libm;

/// TODO this is no longer used. Feel free to delete it the next time
/// we need to rebuild builtins.bc!
#[no_mangle]
pub fn i64_to_f64_(num: i64) -> f64 {
    num as f64
}

/// Adapted from Rust's core::num module, by the Rust core team,
/// licensed under the Apache License, version 2.0 - https://www.apache.org/licenses/LICENSE-2.0
///
/// Thank you, Rust core team!
#[no_mangle]
pub fn pow_int_(mut base: i64, mut exp: i64) -> i64 {
    let mut acc = 1;

    while exp > 1 {
        if (exp & 1) == 1 {
            acc *= base;
        }
        exp /= 2;
        base *= base;
    }

    // Deal with the final bit of the exponent separately, since
    // squaring the base afterwards is not necessary and may cause a
    // needless overflow.
    if exp == 1 {
        acc *= base;
    }

    acc
}

#[no_mangle]
pub fn str_split_<'a>(
    list: &'a mut [&'a [u8]],
    str_: &'a [u8],
    delimiter: &[u8],
) -> &'a [&'a [u8]] {
    let mut ret_list_index = 0;

    let str_len = str_.len();
    let delimiter_len = delimiter.len();

    let mut slice_start_index = 0;

    let mut str_index = 0;

    if str_len > delimiter_len {
        while str_index <= (str_len - delimiter_len) {
            let mut delimiter_index = 0;
            let mut matches_delimiter = true;

            while matches_delimiter && delimiter_index < delimiter_len {
                let delimiter_char = delimiter[delimiter_index];
                let str_char = str_[str_index + delimiter_index];

                if delimiter_char != str_char {
                    matches_delimiter = false;
                }

                delimiter_index += 1;
            }

            if matches_delimiter {
                list[ret_list_index] = &str_[slice_start_index..str_index];
                slice_start_index = str_index + delimiter_len;
                ret_list_index += 1;
                str_index += delimiter_len;
            } else {
                str_index += 1;
            }
        }
    }

    list[ret_list_index] = &str_[slice_start_index..str_len];

    list
}

#[no_mangle]
pub fn count_segments_(str: &[u8], delimiter: &[u8]) -> i64 {
    let mut count: i64 = 1;

    let str_len = str.len();
    let delimiter_len = delimiter.len();

    if str_len > delimiter_len {
        for str_index in 0..(str_len - delimiter_len) {
            let mut delimiter_index = 0;

            let mut matches_delimiter = true;

            while matches_delimiter && delimiter_index < delimiter_len {
                let delimiter_char = delimiter[delimiter_index];
                let str_char = str[str_index + delimiter_index];

                if delimiter_char != str_char {
                    matches_delimiter = false;
                }

                delimiter_index += 1;
            }

            if matches_delimiter {
                count += 1;
            }
        }
    }

    count
}

/// Adapted from Rust's core::num module, by the Rust core team,
/// licensed under the Apache License, version 2.0 - https://www.apache.org/licenses/LICENSE-2.0
///
/// Thank you, Rust core team!
#[no_mangle]
pub fn is_finite_(num: f64) -> bool {
    f64::is_finite(num)
}

#[no_mangle]
pub fn atan_(x: f64) -> f64 {
    libm::atan(x)
}