gcc/testsuite/jit.dg/test-accessing-union.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3
   4 #include "libgccjit.h"
   5
   6 #include "harness.h"
   7
   8 /* Quote from here in docs/topics/types.rst.  */
   9
  10 union int_or_float
  11 {
  12   int as_int;
  13   float as_float;
  14 };
  15
  16 void
  17 create_code (gcc_jit_context *ctxt, void *user_data)
  18 {
  19   /* Let's try to inject the equivalent of:
  20      float
  21      test_union (int i)
  22      {
  23         union int_or_float u;
  24         u.as_int = i;
  25         return u.as_float;
  26      }
  27   */
  28   gcc_jit_type *int_type =
  29     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
  30   gcc_jit_type *float_type =
  31     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_FLOAT);
  32   gcc_jit_field *as_int =
  33     gcc_jit_context_new_field (ctxt,
  34                                NULL,
  35                                int_type,
  36                                "as_int");
  37   gcc_jit_field *as_float =
  38     gcc_jit_context_new_field (ctxt,
  39                                NULL,
  40                                float_type,
  41                                "as_float");
  42   gcc_jit_field *fields[] = {as_int, as_float};
  43   gcc_jit_type *union_type =
  44     gcc_jit_context_new_union_type (ctxt, NULL,
  45                                     "int_or_float", 2, fields);
  46
  47   /* Build the test function.  */
  48   gcc_jit_param *param_i =
  49     gcc_jit_context_new_param (ctxt, NULL, int_type, "i");
  50   gcc_jit_function *test_fn =
  51     gcc_jit_context_new_function (ctxt, NULL,
  52                                   GCC_JIT_FUNCTION_EXPORTED,
  53                                   float_type,
  54                                   "test_union",
  55                                   1, &param_i,
  56                                   0);
  57
  58   gcc_jit_lvalue *u =
  59     gcc_jit_function_new_local (test_fn, NULL,
  60                                 union_type, "u");
  61
  62   gcc_jit_block *block = gcc_jit_function_new_block (test_fn, NULL);
  63
  64   /* u.as_int = i; */
  65   gcc_jit_block_add_assignment (
  66     block,
  67     NULL,
  68     /* "u.as_int = ..." */
  69     gcc_jit_lvalue_access_field (u,
  70                                  NULL,
  71                                  as_int),
  72     gcc_jit_param_as_rvalue (param_i));
  73
  74   /* return u.as_float; */
  75   gcc_jit_block_end_with_return (
  76     block, NULL,
  77     gcc_jit_rvalue_access_field (gcc_jit_lvalue_as_rvalue (u),
  78                                  NULL,
  79                                  as_float));
  80 }
  81
  82 /* Quote up to here in docs/topics/types.rst.  */
  83
  84 void
  85 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
  86 {
  87   typedef float (*fn_type) (int i);
  88   CHECK_NON_NULL (result);
  89
  90   fn_type test_union =
  91     (fn_type)gcc_jit_result_get_code (result, "test_union");
  92   CHECK_NON_NULL (test_union);
  93
  94   /* Call the JIT-generated function.  */
  95   float f_result = test_union (42);
  96
  97   union int_or_float u;
  98   u.as_float = f_result;
  99
 100   CHECK_VALUE (u.as_int, 42);
 101 }