| #include <stdlib.h> |
| #include <stdio.h> |
| #include <stddef.h> |
| |
| #include "libgccjit.h" |
| |
| #include "harness.h" |
| |
| #define ARRAY_SIZE (4) |
| |
| /* Verify that struct layout works properly when adding an array field. */ |
| struct array_holder |
| { |
| float m_before; |
| int m_ints[ARRAY_SIZE]; |
| float m_after; |
| }; |
| |
| void |
| create_code (gcc_jit_context *ctxt, void *user_data) |
| { |
| /* Let's try to inject the equivalent of: |
| |
| void |
| test_array (struct array_holder *ah) |
| { |
| ah->m_before = 4.0f; |
| for i in 0 to (ARRAY_SIZE - 1): |
| ah->m_ints[i] = (i * i); |
| ah->m_after = 2.0f; |
| } |
| */ |
| gcc_jit_type *void_type = |
| gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_VOID); |
| gcc_jit_type *float_type = |
| gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_FLOAT); |
| gcc_jit_type *int_type = |
| gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT); |
| |
| gcc_jit_field *field_m_before = |
| gcc_jit_context_new_field (ctxt, NULL, float_type, "m_before"); |
| gcc_jit_field *field_m_ints = |
| gcc_jit_context_new_field ( |
| ctxt, NULL, |
| gcc_jit_context_new_array_type (ctxt, NULL, int_type, ARRAY_SIZE), |
| "m_ints"); |
| gcc_jit_field *field_m_after = |
| gcc_jit_context_new_field (ctxt, NULL, float_type, "m_after"); |
| |
| gcc_jit_field *fields[] = { |
| field_m_before, |
| field_m_ints, |
| field_m_after, |
| }; |
| |
| gcc_jit_struct *struct_type = |
| gcc_jit_context_new_struct_type ( |
| ctxt, |
| NULL, |
| "array_holder", |
| 3, fields); |
| |
| gcc_jit_type *struct_ptr_type = |
| gcc_jit_type_get_pointer (gcc_jit_struct_as_type (struct_type)); |
| |
| /* Build the test_fn. */ |
| gcc_jit_param *param_ah = |
| gcc_jit_context_new_param (ctxt, NULL, struct_ptr_type, "ah"); |
| gcc_jit_function *func = |
| gcc_jit_context_new_function (ctxt, NULL, |
| GCC_JIT_FUNCTION_EXPORTED, |
| void_type, |
| "test_array", |
| 1, ¶m_ah, |
| 0); |
| |
| gcc_jit_block *initial = gcc_jit_function_new_block (func, "initial"); |
| gcc_jit_block *loop_test = gcc_jit_function_new_block (func, "loop_test"); |
| gcc_jit_block *loop_body = gcc_jit_function_new_block (func, "loop_body"); |
| gcc_jit_block *final = gcc_jit_function_new_block (func, "final"); |
| |
| /* "ah->m_before = 4.0f;" */ |
| gcc_jit_block_add_assignment ( |
| initial, NULL, |
| gcc_jit_rvalue_dereference_field ( |
| gcc_jit_param_as_rvalue (param_ah), NULL, field_m_before), |
| gcc_jit_context_new_rvalue_from_int (ctxt, float_type, 4)); |
| |
| gcc_jit_block_add_comment (initial, NULL, |
| "for i in 0 to (ARRAY_SIZE - 1):"); |
| gcc_jit_lvalue *i = |
| gcc_jit_function_new_local (func, NULL, int_type, "i"); |
| gcc_jit_block_add_assignment (initial, NULL, |
| i, |
| gcc_jit_context_zero (ctxt, int_type)); |
| |
| gcc_jit_block_end_with_jump (initial, NULL, loop_test); |
| |
| gcc_jit_block_end_with_conditional (loop_test, NULL, |
| gcc_jit_context_new_comparison ( |
| ctxt, NULL, |
| GCC_JIT_COMPARISON_LT, |
| gcc_jit_lvalue_as_rvalue (i), |
| gcc_jit_context_new_rvalue_from_int (ctxt, int_type, ARRAY_SIZE)), |
| loop_body, |
| final); |
| |
| gcc_jit_block_add_comment (loop_body, NULL, "ah->m_ints[i] = (i * i);"); |
| gcc_jit_block_add_assignment ( |
| loop_body, NULL, |
| gcc_jit_context_new_array_access ( |
| ctxt, NULL, |
| gcc_jit_lvalue_as_rvalue (gcc_jit_rvalue_dereference_field ( |
| gcc_jit_param_as_rvalue (param_ah), |
| NULL, |
| field_m_ints)), |
| gcc_jit_lvalue_as_rvalue (i)), |
| gcc_jit_context_new_binary_op ( |
| ctxt, NULL, |
| GCC_JIT_BINARY_OP_MULT, |
| int_type, |
| gcc_jit_lvalue_as_rvalue (i), |
| gcc_jit_lvalue_as_rvalue (i))); |
| |
| /* "i++" */ |
| gcc_jit_block_add_assignment_op ( |
| loop_body, NULL, |
| i, |
| GCC_JIT_BINARY_OP_PLUS, |
| gcc_jit_context_one (ctxt, int_type)); |
| |
| gcc_jit_block_end_with_jump (loop_body, NULL, loop_test); |
| |
| /* ah->m_after = 2.0f; */ |
| gcc_jit_block_add_assignment ( |
| final, NULL, |
| gcc_jit_rvalue_dereference_field ( |
| gcc_jit_param_as_rvalue (param_ah), NULL, field_m_after), |
| gcc_jit_context_new_rvalue_from_int (ctxt, float_type, 2)); |
| gcc_jit_block_end_with_void_return (final, NULL); |
| |
| } |
| |
| void |
| verify_code (gcc_jit_context *ctxt, gcc_jit_result *result) |
| { |
| typedef void (*fn_type) (struct array_holder *ah); |
| |
| CHECK_NON_NULL (result); |
| fn_type test_array = |
| (fn_type)gcc_jit_result_get_code (result, "test_array"); |
| CHECK_NON_NULL (test_array); |
| |
| struct array_holder ah; |
| memset (&ah, 0xf0, sizeof (ah)); |
| |
| test_array (&ah); |
| CHECK_VALUE (ah.m_before, 4.0f); |
| CHECK_VALUE (ah.m_ints[0], 0); |
| CHECK_VALUE (ah.m_ints[1], 1); |
| CHECK_VALUE (ah.m_ints[2], 4); |
| CHECK_VALUE (ah.m_ints[3], 9); |
| CHECK_VALUE (ah.m_after, 2.0f); |
| |
| } |