added status function
[libfirm] / ir / be / beprofile.c
1 /** vim: set sw=4 ts=4:
2  * @file   beprofile.c
3  * @date   2006-04-06
4  * @author Adam M. Szalkowski
5  * @cvs-id $Id$
6  *
7  * Code instrumentation and execution count profiling
8  *
9  * Copyright (C) 2006 Universitaet Karlsruhe
10  * Released under the GPL
11  */
12 #ifdef HAVE_CONFIG_H
13 #include "config.h"
14 #endif
15
16 #include <math.h>
17
18 #include "hashptr.h"
19 #include "debug.h"
20 #include "obst.h"
21 #include "set.h"
22 #include "list.h"
23 #include "pmap.h"
24
25 #include "entity.h"
26 #include "irprintf.h"
27 #include "irgwalk.h"
28 #include "irdump_t.h"
29 #include "irnode_t.h"
30 #include "ircons_t.h"
31 #include "irloop_t.h"
32 #include "iredges.h"
33 #include "execfreq.h"
34 #include "irvrfy.h"
35 #include "type.h"
36 #include "entity.h"
37
38 #include "be_t.h"
39 #include "belive_t.h"
40 #include "besched_t.h"
41 #include "beirgmod.h"
42 #include "bearch.h"
43 #include "beabi.h"
44 #include "benode_t.h"
45 #include "beutil.h"
46 #include "ircons.h"
47
48 #include "bechordal_t.h"
49
50 #ifdef WITH_LIBCORE
51 #include <libcore/lc_opts.h>
52 #include <libcore/lc_opts_enum.h>
53 #endif /* WITH_LIBCORE */
54
55 #include "beprofile.h"
56
57 typedef struct _block_id_walker_data_t {
58         tarval        **array;
59         unsigned int    id;
60         ir_node *symconst;
61 } block_id_walker_data_t;
62
63 typedef struct _execcount_t {
64         unsigned int block;
65         unsigned int count;
66 } execcount_t;
67
68 static int
69 cmp_execcount(const void * a, const void * b, size_t size)
70 {
71         return ((execcount_t*)a)->block != ((execcount_t*)b)->block;
72 }
73
74 static void
75 block_counter(ir_node * bb, void * data)
76 {
77         unsigned int  *count = data;
78         *count = *count + 1;
79
80 }
81
82 static unsigned int
83 count_blocks(ir_graph * irg)
84 {
85         unsigned int count = 0;
86
87         irg_block_walk_graph(irg, block_counter, NULL, &count);
88         return count;
89 }
90
91 /* keep the execcounts here because they are only read once per compiler run */
92 static set * profile = NULL;
93
94 /**
95  * Instrument a block with code needed for profiling
96  */
97 static void
98 instrument_block(ir_node * bb, ir_node * address, unsigned int id)
99 {
100         ir_graph *irg = get_irn_irg(bb);
101         ir_node *start_block = get_irg_start_block(irg);
102         ir_node  *load, *store, *offset, *add, *projm, *proji;
103         ir_node *cnst;
104
105         if(bb == start_block || bb == get_irg_end_block(irg))
106                 return;
107
108         cnst   = new_r_Const_long(irg, start_block, mode_Iu, get_mode_size_bytes(mode_Iu) * id);
109         offset = new_r_Add(irg, bb, address, cnst, mode_P);
110         load   = new_r_Load(irg, bb, new_NoMem(), offset, mode_Iu);
111         projm  = new_r_Proj(irg, bb, load, mode_M, pn_Load_M);
112         proji  = new_r_Proj(irg, bb, load, mode_Iu, pn_Load_res);
113         cnst   = new_r_Const_long(irg, start_block, mode_Iu, 1);
114         add    = new_r_Add(irg, bb, proji, cnst, mode_Iu);
115         store  = new_r_Store(irg, bb, projm, offset, add);
116         projm  = new_r_Proj(irg, bb, store, mode_M, pn_Store_M);
117         keep_alive(projm);
118 }
119
120 /**
121  * Generates a new irg which calls the initializer
122  */
123 static ir_graph *
124 gen_initializer_irg(entity * ent_filename, entity * bblock_id, entity * bblock_counts, int n_blocks)
125 {
126         ir_node *start_block;
127
128         ir_node   *ins[4];
129         ident     *name = new_id_from_str("__firmprof_initializer");
130         entity    *ent  = new_entity(get_glob_type(), name, new_type_method(name, 0, 0));
131         ir_node   *ret, *call, *symconst;
132         symconst_symbol sym;
133
134         ident     *init_name = new_id_from_str("__init_firmprof");
135         ir_type   *init_type = new_type_method(init_name, 4, 0);
136         ir_type   *uint, *uintptr, *string;
137         entity    *init_ent;
138         ir_graph  *irg;
139         ir_node   *bb;
140         ir_type   *empty_frame_type;
141
142         set_entity_ld_ident(ent, name);
143
144         uint    = new_type_primitive(new_id_from_str("__uint"), mode_Iu);
145         uintptr = new_type_pointer(new_id_from_str("__uintptr"), uint, mode_P);
146         string  = new_type_pointer(new_id_from_str("__charptr"), new_type_primitive(new_id_from_str("__char"), mode_Bs), mode_P);
147
148         set_method_param_type(init_type, 0, string);
149         set_method_param_type(init_type, 1, uintptr);
150         set_method_param_type(init_type, 2, uintptr);
151         set_method_param_type(init_type, 3, uint);
152         init_ent = new_entity(get_glob_type(), init_name, init_type);
153         set_entity_ld_ident(init_ent, init_name);
154
155         irg = new_ir_graph(ent, 0);
156         set_current_ir_graph(irg);
157         empty_frame_type = get_irg_frame_type(irg);
158         set_type_size_bytes(empty_frame_type, 0);
159
160         bb = get_cur_block();
161
162         start_block = get_irg_start_block(irg);
163
164         sym.entity_p = init_ent;
165         symconst     = new_r_SymConst(irg, start_block, sym, symconst_addr_ent);
166
167         sym.entity_p = ent_filename;
168         ins[0] = new_r_SymConst(irg, start_block, sym, symconst_addr_ent);
169         sym.entity_p = bblock_id;
170         ins[1] = new_r_SymConst(irg, start_block, sym, symconst_addr_ent);
171         sym.entity_p = bblock_counts;
172         ins[2] = new_r_SymConst(irg, start_block, sym, symconst_addr_ent);
173         ins[3] = new_r_Const_long(irg, start_block, mode_Iu, n_blocks);
174
175         call = new_r_Call( irg,
176                         bb,                                                     //ir_node *     block,
177                         get_irg_initial_mem(irg),       //ir_node *     store,
178                         symconst,                                       //ir_node *     callee,
179                         4,                                                      //int   arity,
180                         ins,                                            //ir_node **    in,
181                         init_type                                       //ir_type *     tp
182                         );
183
184         ret = new_r_Return ( irg,
185                         bb,                                                                             //ir_node *     block,
186                         new_r_Proj(irg, bb, call, mode_M, pn_Call_M_regular),   //ir_node *     store,
187                         0,                                                                              //int   arity,
188                         NULL                                                                    //ir_node **    in
189                         );
190
191         mature_immBlock(bb);
192
193         add_immBlock_pred(get_irg_end_block(irg), ret);
194         mature_immBlock(get_irg_end_block(irg));
195
196         irg_finalize_cons(irg);
197
198         return irg;
199 }
200
201 static void
202 block_id_walker(ir_node * bb, void * data)
203 {
204         block_id_walker_data_t *wd = data;
205
206         wd->array[wd->id] = new_tarval_from_long(get_irn_node_nr(bb), mode_Iu);
207         instrument_block(bb, wd->symconst, wd->id);
208         ++wd->id;
209 }
210
211 ir_graph *
212 be_profile_instrument(void)
213 {
214         ir_graph      *const_irg = get_const_code_irg();
215         ir_node       *const_block = get_irg_start_block(const_irg);
216         int            n, i;
217         unsigned int   n_blocks = 0;
218         entity        *bblock_id, *bblock_counts, *ent_filename;
219         ir_type       *array_type, *integer_type, *string_type, *character_type;
220         tarval       **tarval_array, **tarval_string;
221         char          *filename = "test.c"; //FIXME
222         int            filename_len = strlen(filename)+1;
223         ident         *cur_ident;
224
225         block_id_walker_data_t  wd;
226         symconst_symbol sym;
227
228         integer_type   = new_type_primitive(new_id_from_str("__uint"), mode_Iu);
229         array_type     = new_type_array(new_id_from_str("__block_info_array"), 1, integer_type);
230         set_array_bounds_int(array_type, 0, 0, n_blocks);
231
232         character_type = new_type_primitive(new_id_from_str("__char"), mode_Bs);
233         string_type    = new_type_array(new_id_from_str("__function_name"), 1, character_type);
234         set_array_bounds_int(string_type, 0, 0, filename_len);
235
236         cur_ident      = new_id_from_str("__BLOCK_IDS");
237         bblock_id      = new_entity(get_glob_type(), cur_ident, array_type);
238         set_entity_ld_ident(bblock_id, cur_ident);
239         set_entity_variability(bblock_id, variability_initialized);
240
241         cur_ident      = new_id_from_str("__BLOCK_COUNTS");
242         bblock_counts  = new_entity(get_glob_type(), cur_ident, array_type);
243         set_entity_ld_ident(bblock_counts, cur_ident);
244         set_entity_variability(bblock_counts, variability_initialized);
245
246         cur_ident      = new_id_from_str("__FUNCTION_NAME");
247         ent_filename   = new_entity(get_glob_type(), cur_ident, string_type);
248         set_entity_ld_ident(ent_filename, cur_ident);
249         set_entity_variability(ent_filename, variability_initialized);
250
251         for (n = get_irp_n_irgs() - 1; n >= 0; --n) {
252                 ir_graph *irg = get_irp_irg(n);
253
254                 n_blocks += count_blocks(irg);
255         }
256
257         /* initialize count array */
258         tarval_array = alloca(sizeof(*tarval_array) * n_blocks);
259         for (i = 0; i < n_blocks; ++i) {
260                 tarval_array[i] = get_tarval_null(mode_Iu);
261         }
262         set_array_entity_values(bblock_counts, tarval_array, n_blocks);
263
264         /* initialize function name string constant */
265         tarval_string = alloca(sizeof(*tarval_string) * (filename_len));
266         for (i = 0; i < filename_len; ++i) {
267                 tarval_string[i] = new_tarval_from_long(filename[i], mode_Bs);
268         }
269         set_array_entity_values(ent_filename, tarval_string, filename_len);
270
271
272         /* initialize block id array and instrument blocks */
273         wd.array = tarval_array;
274         wd.id    = 0;
275         for (n = get_irp_n_irgs() - 1; n >= 0; --n) {
276                 ir_graph *irg = get_irp_irg(n);
277
278                 /* generate a symbolic constant pointing to the count array */
279                 sym.entity_p = bblock_counts;
280                 wd.symconst  = new_r_SymConst(irg, get_irg_start_block(irg), sym, symconst_addr_ent);
281
282                 irg_block_walk_graph(irg, block_id_walker, NULL, &wd);
283         }
284         set_array_entity_values(bblock_id, tarval_array, n_blocks);
285
286         return gen_initializer_irg(ent_filename, bblock_id, bblock_counts, n_blocks);
287 }
288
289
290 /**
291  * Reads the corresponding profile info file if it exists and returns a
292  * profile info struct
293  */
294 void
295 be_profile_read(char * filename)
296 {
297         FILE   *f;
298         char    buf[8];
299         size_t  ret;
300
301         f = fopen(filename, "r");
302         if(f == NULL) {
303                 perror("opening of profile data failed");
304                 return;
305         }
306
307         /* check magic */
308         ret = fread(buf, 8, 1, f);
309         if(ret == 0 || strncmp(buf, "firmprof", 8) != 0) {
310                 return;
311         }
312
313         if(profile) be_profile_free();
314         profile = new_set(cmp_execcount, 16);
315
316         do {
317                 execcount_t  query;
318                 ret = fread(&query, sizeof(unsigned int), 2, f);
319
320                 if(ret != 2) break;
321
322                 set_insert(profile, &query, sizeof(query), query.block);
323         } while(1);
324
325         fclose(f);
326 }
327
328 /**
329  * Frees the profile info
330  */
331 void
332 be_profile_free(void)
333 {
334         if(profile)
335                 del_set(profile);
336 }
337
338 /**
339  * Tells whether profile module has aquired data
340  */
341 int
342 be_profile_has_data(void)
343 {
344         return (profile != NULL);
345 }
346
347 /**
348  * Get block execution count as determined be profiling
349  */
350 unsigned int
351 be_profile_get_block_execcount(const ir_node * block)
352 {
353         execcount_t *ec, query;
354
355         if(!profile)
356                 return 1;
357
358         query.block = get_irn_node_nr(block);
359         ec = set_find(profile, &query, sizeof(query), get_irn_node_nr(block));
360
361         if(ec) {
362                 return ec->count;
363         } else {
364                 return 1;
365         }
366 }