2 * This file is part of libFirm.
3 * Copyright (C) 2012 University of Karlsruhe.
8 * @brief Paint chordal graphs.
9 * @author Sebastian Hack
21 #include "iredges_t.h"
26 #include "bechordal_t.h"
28 #include "bechordal_draw.h"
32 be_chordal_env_t *env;
38 #define decl_self(type, from) \
39 type *self = (type *) from
41 static void set_color(plotter_t *_self, const color_t *color)
43 decl_self(base_plotter_t, _self);
47 static const color_t *get_color(const plotter_t *_self)
49 decl_self(const base_plotter_t, _self);
53 static void set_width(plotter_t *_self, int width)
55 decl_self(base_plotter_t, _self);
59 static int get_width(const plotter_t *_self)
61 decl_self(const base_plotter_t, _self);
65 static void plotter_default_free(plotter_t *self)
79 | _ \/ ___| | _ \| | ___ | |_| |_ ___ _ __
80 | |_) \___ \ | |_) | |/ _ \| __| __/ _ \ '__|
81 | __/ ___) | | __/| | (_) | |_| || __/ |
82 |_| |____/ |_| |_|\___/ \__|\__\___|_|
86 static void ps_begin(plotter_t *_self, const rect_t *vis)
89 decl_self(ps_plotter_t, _self);
91 f = self->f = fopen(self->filename, "wt");
92 fprintf(f, "%%!PS-Adobe-2.0\n");
93 fprintf(f, "%%%%BoundingBox: %d %d %d %d\n", vis->x, vis->y, vis->w, vis->h);
96 static void ps_setcolor(plotter_t *_self, const color_t *color)
98 decl_self(ps_plotter_t, _self);
99 set_color(_self, color);
101 fprintf(self->f, "%.2f %.2f %.2f setrgbcolor\n",
102 color->r, color->g, color->b);
105 static void ps_line(plotter_t *_self, int x1, int y1, int x2, int y2)
107 decl_self(ps_plotter_t, _self);
109 fprintf(self->f, "%d %d moveto\n", x1, y1);
110 fprintf(self->f, "%d %d lineto\n", x2, y2);
111 fprintf(self->f, "stroke\n");
114 static void ps_box(plotter_t *_self, const rect_t *rect)
116 decl_self(ps_plotter_t, _self);
118 fprintf(self->f, "%d %d %d %d rectstroke\n",
119 rect->x, rect->y, rect->w, rect->h);
122 static void ps_text(plotter_t *_self, int x, int y, const char *str)
124 decl_self(ps_plotter_t, _self);
126 fprintf(self->f, "%d %d moveto\n", x, y);
127 fprintf(self->f, "(%s) show\n", str);
130 static void ps_finish(plotter_t *_self)
132 decl_self(ps_plotter_t, _self);
136 static const plotter_if_t ps_plotter_vtab = {
149 plotter_t *new_plotter_ps(const char *filename)
151 ps_plotter_t *ps_plotter = XMALLOC(ps_plotter_t);
152 plotter_t *p = (plotter_t *) ps_plotter;
154 ps_plotter->filename = filename;
155 p->vtab = &ps_plotter_vtab;
159 extern void plotter_free(plotter_t *self)
161 self->vtab->free(self);
165 const draw_chordal_opts_t draw_chordal_def_opts = {
166 10, 10, 30, 8, 10, 10
169 typedef struct draw_chordal_env_t {
170 const be_chordal_env_t *chordal_env;
171 const arch_register_class_t *cls;
174 const draw_chordal_opts_t *opts;
176 } draw_chordal_env_t;
183 #define doz(a, b) MAX((a) - (b), 0)
185 static void block_dims_walker(ir_node *block, void *data)
187 draw_chordal_env_t *env = (draw_chordal_env_t*)data;
188 struct list_head *head = get_block_border_head(env->chordal_env, block);
189 const draw_chordal_opts_t *opts = env->opts;
190 struct block_dims *dims = OALLOCZ(&env->obst, struct block_dims);
192 unsigned max_color = 0;
193 unsigned max_step = 0;
194 foreach_border_head(head, b) {
195 unsigned const col = arch_get_irn_register(b->irn)->index;
196 max_color = MAX(max_color, col);
197 max_step = MAX(max_step, b->step);
200 dims->box.w = (max_color + 2) * opts->h_inter_gap;
201 dims->box.h = max_step * opts->v_inter_gap;
203 pmap_insert(env->block_dims, block, dims);
206 static void layout(const draw_chordal_env_t *env, ir_node *bl, int x)
208 const draw_chordal_opts_t *opts = env->opts;
209 struct block_dims *dims = pmap_get(struct block_dims, env->block_dims, bl);
210 rect_t *rect = &dims->subtree_box;
215 memset(rect, 0, sizeof(*rect));
218 dominates_for_each(bl, sub) {
219 struct block_dims *bl_dim = pmap_get(struct block_dims, env->block_dims, sub);
221 layout(env, sub, rect->x + rect->w);
223 rect->w += h_space + bl_dim->subtree_box.w;
224 rect->h = MAX(rect->h, bl_dim->subtree_box.h);
226 h_space = opts->h_gap;
227 v_space = opts->v_gap;
230 rect->w = MAX(rect->w, dims->box.w + opts->h_gap);
232 dims->box.x = x + doz(rect->w, dims->box.w) / 2;
233 dims->box.y = rect->h + v_space;
235 rect->h = dims->box.y + dims->box.h;
238 static void set_y(const draw_chordal_env_t *env, ir_node *bl, int up)
240 const draw_chordal_opts_t *opts = env->opts;
241 struct block_dims *dims = pmap_get(struct block_dims, env->block_dims, bl);
242 int max_height = dims->subtree_box.h - dims->box.h - opts->v_gap;
245 dominates_for_each(bl, sub) {
246 struct block_dims *bl_dim = pmap_get(struct block_dims, env->block_dims, sub);
247 int height_diff = max_height - bl_dim->subtree_box.h;
249 set_y(env, sub, up + height_diff);
252 dims->subtree_box.y += up;
256 static color_t *reg_to_color(const draw_chordal_env_t *env,
257 ir_node *rel_bl, ir_node *irn, color_t *color)
263 foreach_out_edge(irn, edge)
264 phi_arg |= is_Phi(edge->src);
266 color->r = is_Phi(irn) ? 0.5 : 0.0;
267 color->g = phi_arg ? 0.5 : 0.0;
272 static void draw_block(ir_node *bl, void *data)
274 static const color_t black = { 0, 0, 0 };
275 const draw_chordal_env_t *env = (const draw_chordal_env_t*)data;
276 const be_lv_t *lv = be_get_irg_liveness(env->chordal_env->irg);
277 struct list_head *head = get_block_border_head(env->chordal_env, bl);
278 ir_node *dom = get_Block_idom(bl);
279 const draw_chordal_opts_t *opts = env->opts;
280 struct block_dims *dims = pmap_get(struct block_dims, env->block_dims, bl);
283 ir_snprintf(buf, sizeof(buf), "%F", bl);
285 env->plotter->vtab->set_color(env->plotter, &black);
286 env->plotter->vtab->box(env->plotter, &dims->box);
288 env->plotter->vtab->text(env->plotter, dims->box.x, dims->box.y, buf);
290 foreach_border_head(head, b) {
292 /* Walk from def to use, so the link is set before retrieved. */
293 set_irn_link(b->irn, b);
295 ir_node *const irn = b->irn;
296 border_t const *const def = (border_t const*)get_irn_link(irn);
297 arch_register_t const *const reg = arch_get_irn_register(irn);
299 int live_out = be_is_live_out(lv, bl, irn);
300 int x = (reg->index + 1) * opts->h_inter_gap;
301 int ystart = def->step * opts->v_inter_gap;
302 int ystop = b->step * opts->v_inter_gap + (live_out ? 0 : opts->v_inter_gap / 2);
305 reg_to_color(env, bl, irn, &color);
308 ystart += dims->box.y;
309 ystop += dims->box.y;
311 env->plotter->vtab->set_color(env->plotter, &color);
312 env->plotter->vtab->line(env->plotter, x, ystart, x, ystop);
314 env->plotter->vtab->line(env->plotter, x - 2, ystart, x + 2, ystart);
315 env->plotter->vtab->line(env->plotter, x - 2, ystop, x + 2, ystop);
320 struct block_dims *dom_dims = pmap_get(struct block_dims, env->block_dims, dom);
322 be_lv_foreach_cls(lv, bl, be_lv_state_in, env->cls, irn) {
323 const arch_register_t *reg = arch_get_irn_register(irn);
324 int x = (reg->index + 1) * opts->h_inter_gap;
327 reg_to_color(env, bl, irn, &color);
329 env->plotter->vtab->set_color(env->plotter, &color);
330 env->plotter->vtab->line(env->plotter,
332 dims->box.y + dims->box.h,
339 static void draw(draw_chordal_env_t *env, const rect_t *start_box)
341 ir_graph *irg = env->chordal_env->irg;
342 plotter_t *p = env->plotter;
346 bbox.w = start_box->w + 2 * env->opts->x_margin;
347 bbox.h = start_box->h + 2 * env->opts->y_margin;
349 be_assure_live_sets(irg);
350 be_assure_live_chk(irg);
352 p->vtab->begin(p, &bbox);
353 ir_reserve_resources(irg, IR_RESOURCE_IRN_LINK);
354 irg_block_walk_graph(irg, draw_block, NULL, env);
355 ir_free_resources(irg, IR_RESOURCE_IRN_LINK);
359 void draw_interval_tree(const draw_chordal_opts_t *opts,
360 const be_chordal_env_t *chordal_env, plotter_t *plotter)
362 draw_chordal_env_t env;
363 struct block_dims *start_dims;
364 ir_node *start_block = get_irg_start_block(chordal_env->irg);
367 env.block_dims = pmap_create();
368 env.plotter = plotter;
369 env.cls = chordal_env->cls;
370 env.chordal_env = chordal_env;
371 obstack_init(&env.obst);
373 irg_block_walk_graph(chordal_env->irg, block_dims_walker, NULL, &env);
374 layout(&env, start_block, opts->x_margin);
375 set_y(&env, start_block, opts->y_margin);
376 start_dims = pmap_get(struct block_dims, env.block_dims, start_block);
377 draw(&env, &start_dims->subtree_box);
379 pmap_destroy(env.block_dims);
380 obstack_free(&env.obst, NULL);