gcc/testsuite/jit.dg/test-fibonacci.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4
   5 #include "libgccjit.h"
   6
   7 #include "harness.h"
   8
   9 void
  10 create_code (gcc_jit_context *ctxt, void *user_data)
  11 {
  12   const int FIRST_LINE = __LINE__ + 4;
  13   /* Let's try to inject the equivalent of:
  14 0000000001111111111222222222233333333334444444444555555555566666666667
  15 1234567890123456789012345678901234567890123456789012345678901234567890
  16 FIRST_LINE + 0: int
  17 FIRST_LINE + 1: my_fibonacci (int x)
  18 FIRST_LINE + 2: {
  19 FIRST_LINE + 3:   if (x < 2)
  20 FIRST_LINE + 4:     return x;
  21 FIRST_LINE + 5:   else
  22 FIRST_LINE + 6:     return my_fibonacci (x - 1) + my_fibonacci (x - 2);
  23 FIRST_LINE + 7: }
  24 0000000001111111111222222222233333333334444444444555555555566666666667
  25 1234567890123456789012345678901234567890123456789012345678901234567890
  26
  27      where the source locations are set up to point to the commented-out
  28      code above.
  29      It should therefore be possible to step through the generated code
  30      in the debugger, stepping through the above commented-out code
  31      fragement.
  32    */
  33   gcc_jit_type *the_type =
  34     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
  35   gcc_jit_type *return_type = the_type;
  36
  37   gcc_jit_param *x =
  38     gcc_jit_context_new_param (
  39       ctxt,
  40       gcc_jit_context_new_location (
  41         ctxt, __FILE__, FIRST_LINE + 1, 35),
  42       the_type, "x");
  43   gcc_jit_param *params[1] = {x};
  44   gcc_jit_function *func =
  45     gcc_jit_context_new_function (ctxt,
  46                                   gcc_jit_context_new_location (
  47                                     ctxt, __FILE__, FIRST_LINE, 17),
  48                                   GCC_JIT_FUNCTION_EXPORTED,
  49                                   return_type,
  50                                   "my_fibonacci",
  51                                   1, params, 0);
  52
  53   gcc_jit_block *initial =
  54     gcc_jit_function_new_block (func, "initial");
  55   gcc_jit_block *on_true =
  56     gcc_jit_function_new_block (func, "on_true");
  57   gcc_jit_block *on_false =
  58     gcc_jit_function_new_block (func, "on_false");
  59
  60   /* if (x < 2) */
  61   gcc_jit_block_end_with_conditional (
  62     initial,
  63     gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 3, 19),
  64     gcc_jit_context_new_comparison (
  65       ctxt,
  66       gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 3, 25),
  67       GCC_JIT_COMPARISON_LT,
  68       gcc_jit_param_as_rvalue (x),
  69       gcc_jit_context_new_rvalue_from_int (
  70         ctxt,
  71         the_type,
  72         2)),
  73     on_true,
  74     on_false);
  75
  76   /* true branch: */
  77   /* return x */
  78   gcc_jit_block_end_with_return (
  79     on_true,
  80     gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 4, 21),
  81     gcc_jit_param_as_rvalue (x));
  82
  83   /* false branch: */
  84   gcc_jit_rvalue *x_minus_1 =
  85     gcc_jit_context_new_binary_op (
  86       ctxt,
  87       gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 44),
  88       GCC_JIT_BINARY_OP_MINUS, the_type,
  89       gcc_jit_param_as_rvalue (x),
  90       gcc_jit_context_new_rvalue_from_int (
  91         ctxt,
  92         the_type,
  93         1));
  94   gcc_jit_rvalue *x_minus_2 =
  95     gcc_jit_context_new_binary_op (
  96       ctxt,
  97       gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 67),
  98       GCC_JIT_BINARY_OP_MINUS, the_type,
  99       gcc_jit_param_as_rvalue (x),
 100       gcc_jit_context_new_rvalue_from_int (
 101         ctxt,
 102         the_type,
 103         2));
 104   gcc_jit_block_end_with_return (
 105     on_false,
 106     gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 21),
 107     gcc_jit_context_new_binary_op (
 108       ctxt,
 109       gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 49),
 110       GCC_JIT_BINARY_OP_PLUS, the_type,
 111       /* my_fibonacci (x - 1) */
 112       gcc_jit_context_new_call (
 113         ctxt,
 114         gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 28),
 115         func,
 116         1, &x_minus_1),
 117       /* my_fibonacci (x - 2) */
 118       gcc_jit_context_new_call (
 119         ctxt,
 120         gcc_jit_context_new_location (ctxt, __FILE__, FIRST_LINE + 6, 51),
 121         func,
 122         1, &x_minus_2)));
 123 }
 124
 125 void
 126 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
 127 {
 128   typedef int (*my_fibonacci_fn_type) (int);
 129   CHECK_NON_NULL (result);
 130   my_fibonacci_fn_type my_fibonacci =
 131     (my_fibonacci_fn_type)gcc_jit_result_get_code (result, "my_fibonacci");
 132   CHECK_NON_NULL (my_fibonacci);
 133   int val = my_fibonacci (10);
 134   note ("my_fibonacci returned: %d", val);
 135   CHECK_VALUE (val, 55);
 136 }