1
# -*- coding: UTF-8 -*-
2
"""Cell renderer for directed graph.
4
This module contains the implementation of a custom GtkCellRenderer that
5
draws part of the directed graph based on the lines suggested by the code
8
Because we're shiny, we use Cairo to do this, and because we're naughty
9
we cheat and draw over the bits of the TreeViewColumn that are supposed to
10
just be for the background.
13
__copyright__ = "Copyright © 2005 Canonical Ltd."
14
__author__ = "Scott James Remnant <scott@ubuntu.com>"
25
class CellRendererGraph(gtk.GenericCellRenderer):
26
"""Cell renderer for directed graph.
29
node (column, colour) tuple to draw revision node,
30
in_lines (start, end, colour) tuple list to draw inward lines,
31
out_lines (start, end, colour) tuple list to draw outward lines.
37
"node": ( gobject.TYPE_PYOBJECT, "node",
38
"revision node instruction",
39
gobject.PARAM_WRITABLE
41
"in-lines": ( gobject.TYPE_PYOBJECT, "in-lines",
42
"instructions to draw lines into the cell",
43
gobject.PARAM_WRITABLE
45
"out-lines": ( gobject.TYPE_PYOBJECT, "out-lines",
46
"instructions to draw lines out of the cell",
47
gobject.PARAM_WRITABLE
51
def do_set_property(self, property, value):
52
"""Set properties from GObject properties."""
53
if property.name == "node":
55
elif property.name == "in-lines":
57
elif property.name == "out-lines":
58
self.out_lines = value
60
raise AttributeError, "no such property: '%s'" % property.name
62
def box_size(self, widget):
63
"""Calculate box size based on widget's font.
65
Cache this as it's probably expensive to get. It ensures that we
66
draw the graph at least as large as the text.
70
except AttributeError:
71
pango_ctx = widget.get_pango_context()
72
font_desc = widget.get_style().font_desc
73
metrics = pango_ctx.get_metrics(font_desc)
75
ascent = pango.PIXELS(metrics.get_ascent())
76
descent = pango.PIXELS(metrics.get_descent())
78
self._box_size = ascent + descent + 6
81
def set_colour(self, ctx, colour, bg, fg):
82
"""Set the context source colour.
84
Picks a distinct colour based on an internal wheel; the bg
85
parameter provides the value that should be assigned to the 'zero'
86
colours and the fg parameter provides the multiplier that should be
87
applied to the foreground colours.
89
mainline_color = ( 0.0, 0.0, 0.0 )
100
colour_rgb = mainline_color
102
colour_rgb = colours[colour % len(colours)]
104
red = (colour_rgb[0] * fg) or bg
105
green = (colour_rgb[1] * fg) or bg
106
blue = (colour_rgb[2] * fg) or bg
108
ctx.set_source_rgb(red, green, blue)
110
def on_get_size(self, widget, cell_area):
111
"""Return the size we need for this cell.
113
Each cell is drawn individually and is only as wide as it needs
114
to be, we let the TreeViewColumn take care of making them all
117
box_size = self.box_size(widget) + 1
119
width = box_size * (self.columns_len + 1)
122
# FIXME I have no idea how to use cell_area properly
123
return (0, 0, width, height)
125
def on_render(self, window, widget, bg_area, cell_area, exp_area, flags):
126
"""Render an individual cell.
128
Draws the cell contents using cairo, taking care to clip what we
129
do to within the background area so we don't draw over other cells.
130
Note that we're a bit naughty there and should really be drawing
131
in the cell_area (or even the exposed area), but we explicitly don't
134
We try and be a little clever, if the line we need to draw is going
135
to cross other columns we actually draw it as in the .---' style
136
instead of a pure diagonal ... this reduces confusion by an
139
ctx = window.cairo_create()
140
ctx.rectangle(bg_area.x, bg_area.y, bg_area.width, bg_area.height)
143
box_size = self.box_size(widget)
145
ctx.set_line_width(box_size / 8)
146
ctx.set_line_cap(cairo.LINE_CAP_ROUND)
148
# Draw lines into the cell
149
for start, end, colour in self.in_lines:
150
self.render_line (ctx, cell_area, box_size,
151
bg_area.y, bg_area.height,
154
# Draw lines out of the cell
155
for start, end, colour in self.out_lines:
156
self.render_line (ctx, cell_area, box_size,
157
bg_area.y + bg_area.height, bg_area.height,
160
# Draw the revision node in the right column
161
(column, colour) = self.node
162
ctx.arc(cell_area.x + box_size * column + box_size / 2,
163
cell_area.y + cell_area.height / 2,
164
box_size / 4, 0, 2 * math.pi)
166
self.set_colour(ctx, colour, 0.0, 0.5)
167
ctx.stroke_preserve()
169
self.set_colour(ctx, colour, 0.5, 1.0)
172
def render_line (self, ctx, cell_area, box_size, mid, height, start, end, colour):
174
x = cell_area.x + box_size * end + box_size / 2
175
ctx.move_to(x, mid + height / 3)
176
ctx.line_to(x, mid + height / 3)
177
ctx.move_to(x, mid + height / 6)
178
ctx.line_to(x, mid + height / 6)
181
x = cell_area.x + box_size * start + box_size / 2
182
ctx.move_to(x, mid - height / 3)
183
ctx.line_to(x, mid - height / 3)
184
ctx.move_to(x, mid - height / 6)
185
ctx.line_to(x, mid - height / 6)
187
startx = cell_area.x + box_size * start + box_size / 2
188
endx = cell_area.x + box_size * end + box_size / 2
190
ctx.move_to(startx, mid - height / 2)
192
if start - end == 0 :
193
ctx.line_to(endx, mid + height / 2)
195
ctx.curve_to(startx, mid - height / 5,
196
startx, mid - height / 5,
197
startx + (endx - startx) / 2, mid)
199
ctx.curve_to(endx, mid + height / 5,
200
endx, mid + height / 5 ,
201
endx, mid + height / 2)
203
self.set_colour(ctx, colour, 0.0, 0.65)
b'\\ No newline at end of file'
added
removed
viz/graphcell.py
