about summary refs log tree commit diff
path: root/library/stdarch/crates/intrinsic-test/src/common/write_file.rs
blob: 0ba3e829a6b80e32dac1ad8c97a9f7d16cfb1bbb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
use super::gen_c::create_c_test_program;
use super::gen_c::setup_c_file_paths;
use super::gen_rust::{create_rust_test_program, setup_rust_file_paths};
use super::intrinsic::IntrinsicDefinition;
use super::intrinsic_helpers::IntrinsicTypeDefinition;
use std::fs::File;
use std::io::Write;

pub fn write_file(filename: &String, code: String) {
    let mut file = File::create(filename).unwrap();
    file.write_all(code.into_bytes().as_slice()).unwrap();
}

pub fn write_c_testfiles<T: IntrinsicTypeDefinition + Sized>(
    intrinsics: &Vec<&dyn IntrinsicDefinition<T>>,
    target: &str,
    c_target: &str,
    headers: &[&str],
    notice: &str,
    arch_specific_definitions: &[&str],
) -> Vec<String> {
    let intrinsics_name_list = intrinsics
        .iter()
        .map(|i| i.name().clone())
        .collect::<Vec<_>>();
    let filename_mapping = setup_c_file_paths(&intrinsics_name_list);

    intrinsics.iter().for_each(|&i| {
        let c_code = create_c_test_program(
            i,
            headers,
            target,
            c_target,
            notice,
            arch_specific_definitions,
        );
        if let Some(filename) = filename_mapping.get(&i.name()) {
            write_file(filename, c_code)
        };
    });

    intrinsics_name_list
}

pub fn write_rust_testfiles<T: IntrinsicTypeDefinition>(
    intrinsics: Vec<&dyn IntrinsicDefinition<T>>,
    rust_target: &str,
    notice: &str,
    definitions: &str,
    cfg: &str,
) -> Vec<String> {
    let intrinsics_name_list = intrinsics
        .iter()
        .map(|i| i.name().clone())
        .collect::<Vec<_>>();
    let filename_mapping = setup_rust_file_paths(&intrinsics_name_list);

    intrinsics.iter().for_each(|&i| {
        let rust_code = create_rust_test_program(i, rust_target, notice, definitions, cfg);
        if let Some(filename) = filename_mapping.get(&i.name()) {
            write_file(filename, rust_code)
        }
    });

    intrinsics_name_list
}