/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to branchview/graphcell.py

  • Committer: Jelmer Vernooij
  • Date: 2012-07-09 15:23:26 UTC
  • mto: This revision was merged to the branch mainline in revision 794.
  • Revision ID: jelmer@samba.org-20120709152326-dzxb8zoz0btull7n
Remove bzr-notify.

Show diffs side-by-side

added added

removed removed

Lines of Context:
10
10
just be for the background.
11
11
"""
12
12
 
13
 
__copyright__ = "Copyright © 2005 Canonical Ltd."
 
13
__copyright__ = "Copyright © 2005-2011 Canonical Ltd."
14
14
__author__    = "Scott James Remnant <scott@ubuntu.com>"
15
15
 
16
16
 
17
17
import math
18
18
 
19
 
import gtk
20
 
import gobject
21
 
import pango
22
 
import cairo
23
 
 
24
 
 
25
 
class CellRendererGraph(gtk.GenericCellRenderer):
 
19
from gi.repository import Gtk
 
20
from gi.repository import GObject
 
21
from gi.repository import Pango
 
22
from gi.repository import PangoCairo
 
23
 
 
24
 
 
25
# Cairo constants are not exported yet. These are taken from documentation.
 
26
CAIRO_LINE_CAP_BUTT = 0
 
27
CAIRO_LINE_CAP_ROUND = 1
 
28
CAIRO_LINE_CAP_SQUARE = 2
 
29
 
 
30
 
 
31
CAIRO_FILL_RULE_WINDING = 0
 
32
CAIRO_FILL_RULE_EVEN_ODD = 1
 
33
 
 
34
 
 
35
# Macro from  Pango header.
 
36
def PANGO_PIXELS(d):
 
37
    return (d + 512) / 1000
 
38
 
 
39
 
 
40
class CellRendererGraph(Gtk.CellRendererPixbuf):
26
41
    """Cell renderer for directed graph.
27
42
 
28
43
    Properties:
30
45
      in_lines          (start, end, colour) tuple list to draw inward lines,
31
46
      out_lines         (start, end, colour) tuple list to draw outward lines.
32
47
    """
33
 
    
 
48
 
34
49
    columns_len = 0
35
50
 
36
51
    __gproperties__ = {
37
 
        "node":         ( gobject.TYPE_PYOBJECT, "node",
 
52
        "node":         ( GObject.TYPE_PYOBJECT, "node",
38
53
                          "revision node instruction",
39
 
                          gobject.PARAM_WRITABLE
 
54
                          GObject.PARAM_WRITABLE
40
55
                        ),
41
 
        "tags":         ( gobject.TYPE_PYOBJECT, "tags",
 
56
        "tags":         ( GObject.TYPE_PYOBJECT, "tags",
42
57
                          "list of tags associated with the node",
43
 
                          gobject.PARAM_WRITABLE
 
58
                          GObject.PARAM_WRITABLE
44
59
                        ),
45
 
        "in-lines":     ( gobject.TYPE_PYOBJECT, "in-lines",
 
60
        "in-lines":     ( GObject.TYPE_PYOBJECT, "in-lines",
46
61
                          "instructions to draw lines into the cell",
47
 
                          gobject.PARAM_WRITABLE
 
62
                          GObject.PARAM_WRITABLE
48
63
                        ),
49
 
        "out-lines":    ( gobject.TYPE_PYOBJECT, "out-lines",
 
64
        "out-lines":    ( GObject.TYPE_PYOBJECT, "out-lines",
50
65
                          "instructions to draw lines out of the cell",
51
 
                          gobject.PARAM_WRITABLE
 
66
                          GObject.PARAM_WRITABLE
52
67
                        ),
53
68
        }
54
 
    
 
69
 
55
70
    def do_set_property(self, property, value):
56
71
        """Set properties from GObject properties."""
57
72
        if property.name == "node":
76
91
        except AttributeError:
77
92
            pango_ctx = widget.get_pango_context()
78
93
            font_desc = widget.get_style().font_desc
79
 
            metrics = pango_ctx.get_metrics(font_desc)
80
 
 
81
 
            ascent = pango.PIXELS(metrics.get_ascent())
82
 
            descent = pango.PIXELS(metrics.get_descent())
83
 
 
 
94
            metrics = pango_ctx.get_metrics(font_desc, None)
 
95
            ascent = PANGO_PIXELS(metrics.get_ascent())
 
96
            descent = PANGO_PIXELS(metrics.get_descent())
84
97
            self._box_size = ascent + descent + 6
85
98
            return self._box_size
86
99
 
113
126
 
114
127
        ctx.set_source_rgb(red, green, blue)
115
128
 
116
 
    def on_get_size(self, widget, cell_area):
 
129
    def do_activate(event, widget, path, bg_area, cell_area, flags):
 
130
        """Renderers cannot be activated; always return True."""
 
131
        return True
 
132
 
 
133
    def do_editing_started(event, widget, path, fb_area, cell_area, flags):
 
134
        """Renderers cannot be edited; always return None."""
 
135
        return None
 
136
 
 
137
    def do_get_size(self, widget, cell_area):
117
138
        """Return the size we need for this cell.
118
139
 
119
140
        Each cell is drawn individually and is only as wide as it needs
128
149
        # FIXME I have no idea how to use cell_area properly
129
150
        return (0, 0, width, height)
130
151
 
131
 
    def on_render(self, window, widget, bg_area, cell_area, exp_area, flags):
 
152
    def do_render(self, ctx, widget, bg_area, cell_area, flags):
132
153
        """Render an individual cell.
133
154
 
134
155
        Draws the cell contents using cairo, taking care to clip what we
142
163
        instead of a pure diagonal ... this reduces confusion by an
143
164
        incredible amount.
144
165
        """
145
 
        ctx = window.cairo_create()
146
166
        ctx.rectangle(bg_area.x, bg_area.y, bg_area.width, bg_area.height)
147
167
        ctx.clip()
148
168
 
149
169
        box_size = self.box_size(widget)
150
170
 
151
171
        ctx.set_line_width(box_size / 8)
152
 
        ctx.set_line_cap(cairo.LINE_CAP_ROUND)
153
172
 
154
173
        # Draw lines into the cell
155
174
        for start, end, colour in self.in_lines:
156
175
            self.render_line (ctx, cell_area, box_size,
157
176
                         bg_area.y, bg_area.height,
158
 
                         start, end, colour)
 
177
                         start, end, colour, flags)
159
178
 
160
179
        # Draw lines out of the cell
161
180
        for start, end, colour in self.out_lines:
162
181
            self.render_line (ctx, cell_area, box_size,
163
182
                         bg_area.y + bg_area.height, bg_area.height,
164
 
                         start, end, colour)
 
183
                         start, end, colour, flags)
165
184
 
166
185
        # Draw the revision node in the right column
167
186
        (column, colour) = self.node
169
188
                cell_area.y + cell_area.height / 2,
170
189
                box_size / 4, 0, 2 * math.pi)
171
190
 
 
191
        if flags & Gtk.CellRendererState.SELECTED:
 
192
            ctx.set_source_rgb(1.0, 1.0, 1.0)
 
193
            ctx.set_line_width(box_size / 4)
 
194
            ctx.stroke_preserve()
 
195
            ctx.set_line_width(box_size / 8)
 
196
 
172
197
        self.set_colour(ctx, colour, 0.0, 0.5)
173
198
        ctx.stroke_preserve()
174
199
 
176
201
        ctx.fill()
177
202
 
178
203
        self.render_tags(ctx, widget.create_pango_context(), cell_area, box_size)
179
 
    
180
 
    def render_line(self, ctx, cell_area, box_size, mid, height, start, end, colour):
 
204
 
 
205
    def render_line(self, ctx, cell_area, box_size,
 
206
                    mid, height, start, end, colour, flags):
181
207
        if start is None:
 
208
            ctx.set_line_cap(CAIRO_LINE_CAP_ROUND)
182
209
            x = cell_area.x + box_size * end + box_size / 2
183
210
            ctx.move_to(x, mid + height / 3)
184
211
            ctx.line_to(x, mid + height / 3)
185
212
            ctx.move_to(x, mid + height / 6)
186
213
            ctx.line_to(x, mid + height / 6)
187
 
            
 
214
 
188
215
        elif end is None:
 
216
            ctx.set_line_cap(CAIRO_LINE_CAP_ROUND)
189
217
            x = cell_area.x + box_size * start + box_size / 2
190
218
            ctx.move_to(x, mid - height / 3)
191
219
            ctx.line_to(x, mid - height / 3)
192
220
            ctx.move_to(x, mid - height / 6)
193
221
            ctx.line_to(x, mid - height / 6)
 
222
 
194
223
        else:
 
224
            ctx.set_line_cap(CAIRO_LINE_CAP_BUTT)
195
225
            startx = cell_area.x + box_size * start + box_size / 2
196
226
            endx = cell_area.x + box_size * end + box_size / 2
197
 
            
 
227
 
198
228
            ctx.move_to(startx, mid - height / 2)
199
 
            
 
229
 
200
230
            if start - end == 0 :
201
 
                ctx.line_to(endx, mid + height / 2)
 
231
                ctx.line_to(endx, mid + height / 2 + 1)
202
232
            else:
203
233
                ctx.curve_to(startx, mid - height / 5,
204
234
                             startx, mid - height / 5,
205
235
                             startx + (endx - startx) / 2, mid)
206
 
                
 
236
 
207
237
                ctx.curve_to(endx, mid + height / 5,
208
238
                             endx, mid + height / 5 ,
209
 
                             endx, mid + height / 2)
210
 
                
 
239
                             endx, mid + height / 2 + 1)
 
240
 
 
241
        if flags & Gtk.CellRendererState.SELECTED:
 
242
            ctx.set_source_rgb(1.0, 1.0, 1.0)
 
243
            ctx.set_line_width(box_size / 5)
 
244
            ctx.stroke_preserve()
 
245
            ctx.set_line_width(box_size / 8)
 
246
 
211
247
        self.set_colour(ctx, colour, 0.0, 0.65)
 
248
 
212
249
        ctx.stroke()
213
250
 
214
251
    def render_tags(self, ctx, pango_ctx, cell_area, box_size):
217
254
 
218
255
        (column, colour) = self.node
219
256
 
220
 
        font_desc = pango.FontDescription()
221
 
        font_desc.set_size(pango.SCALE * 7)
 
257
        font_desc = Pango.FontDescription()
 
258
        font_desc.set_size(Pango.SCALE * 7)
222
259
 
223
 
        tag_layout = pango.Layout(pango_ctx)
 
260
        tag_layout = Pango.Layout(pango_ctx)
224
261
        tag_layout.set_font_description(font_desc)
225
262
 
226
263
        # The width of the tag label stack
227
264
        width = 0
228
265
 
229
266
        for tag_idx, tag in enumerate(self.tags):
230
 
            tag_layout.set_text(" " + tag + " ")
 
267
            tag_layout.set_text(" " + tag + " ", -1)
231
268
            text_width, text_height = tag_layout.get_pixel_size()
232
269
 
233
270
            x0 = cell_area.x + \
256
293
            self.set_colour(ctx, TAG_COLOUR_ID, 0.0, 0.5)
257
294
            ctx.stroke_preserve()
258
295
 
259
 
            ctx.set_fill_rule (cairo.FILL_RULE_EVEN_ODD)
 
296
            ctx.set_fill_rule (CAIRO_FILL_RULE_EVEN_ODD)
260
297
            self.set_colour(ctx, TAG_COLOUR_ID, 0.5, 1.0)
261
298
            ctx.fill()
262
299
 
263
300
            # Draw the tag text
264
301
            self.set_colour(ctx, 0, 0.0, 0.0)
265
302
            ctx.move_to(x0, y0)
266
 
            ctx.show_layout(tag_layout)
267
 
 
 
303
            PangoCairo.show_layout(ctx, tag_layout)