1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <string.h>
4
5 #include "libgccjit++.h"
6
7 #include "harness.h"
8
9 /* Quote from here in docs/cp/topics/functions.rst. */
10
11 void
create_code(gcc_jit_context * c_ctxt,void * user_data)12 create_code (gcc_jit_context *c_ctxt, void *user_data)
13 {
14 /* Let's try to inject the equivalent of:
15 int
16 test_switch (int x)
17 {
18 switch (x)
19 {
20 case 0 ... 5:
21 return 3;
22
23 case 25 ... 27:
24 return 4;
25
26 case -42 ... -17:
27 return 83;
28
29 case 40:
30 return 8;
31
32 default:
33 return 10;
34 }
35 }
36 */
37 gccjit::context ctxt (c_ctxt);
38 gccjit::type t_int = ctxt.get_type (GCC_JIT_TYPE_INT);
39 gccjit::type return_type = t_int;
40 gccjit::param x = ctxt.new_param (t_int, "x");
41 std::vector <gccjit::param> params;
42 params.push_back (x);
43 gccjit::function func = ctxt.new_function (GCC_JIT_FUNCTION_EXPORTED,
44 return_type,
45 "test_switch",
46 params, 0);
47
48 gccjit::block b_initial = func.new_block ("initial");
49
50 gccjit::block b_default = func.new_block ("default");
51 gccjit::block b_case_0_5 = func.new_block ("case_0_5");
52 gccjit::block b_case_25_27 = func.new_block ("case_25_27");
53 gccjit::block b_case_m42_m17 = func.new_block ("case_m42_m17");
54 gccjit::block b_case_40 = func.new_block ("case_40");
55
56 std::vector <gccjit::case_> cases;
57 cases.push_back (ctxt.new_case (ctxt.new_rvalue (t_int, 0),
58 ctxt.new_rvalue (t_int, 5),
59 b_case_0_5));
60 cases.push_back (ctxt.new_case (ctxt.new_rvalue (t_int, 25),
61 ctxt.new_rvalue (t_int, 27),
62 b_case_25_27));
63 cases.push_back (ctxt.new_case (ctxt.new_rvalue (t_int, -42),
64 ctxt.new_rvalue (t_int, -17),
65 b_case_m42_m17));
66 cases.push_back (ctxt.new_case (ctxt.new_rvalue (t_int, 40),
67 ctxt.new_rvalue (t_int, 40),
68 b_case_40));
69 b_initial.end_with_switch (x,
70 b_default,
71 cases);
72
73 b_case_0_5.end_with_return (ctxt.new_rvalue (t_int, 3));
74 b_case_25_27.end_with_return (ctxt.new_rvalue (t_int, 4));
75 b_case_m42_m17.end_with_return (ctxt.new_rvalue (t_int, 83));
76 b_case_40.end_with_return (ctxt.new_rvalue (t_int, 8));
77 b_default.end_with_return (ctxt.new_rvalue (t_int, 10));
78 }
79
80 /* Quote up to here in docs/cp/topics/functions.rst. */
81
82 static int
c_test_switch(int x)83 c_test_switch (int x)
84 {
85 switch (x)
86 {
87 case 0 ... 5:
88 return 3;
89 case 25 ... 27:
90 return 4;
91 case -42 ... -17:
92 return 83;
93 case 40:
94 return 8;
95 default:
96 return 10;
97 }
98 }
99
100 void
verify_code(gcc_jit_context * ctxt,gcc_jit_result * result)101 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
102 {
103 typedef int (*test_switch_type) (int);
104 CHECK_NON_NULL (result);
105 test_switch_type test_switch =
106 (test_switch_type)gcc_jit_result_get_code (result, "test_switch");
107 CHECK_NON_NULL (test_switch);
108
109 int i;
110
111 for (i = -255; i < 255; i++)
112 {
113 int val = test_switch (i);
114 int exp = c_test_switch (i);
115 if (val != exp)
116 fail ("test_switch (%i) returned: %i; expected; %i", i, val, exp);
117 }
118 }
119