/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 viz/graphcell.py

  • Committer: Jelmer Vernooij
  • Date: 2007-02-03 15:24:48 UTC
  • Revision ID: jelmer@samba.org-20070203152448-65ovluij1jetb71l
Fix a couple of smaller bugs.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/graphcell.py
30
30
      in_lines          (start, end, colour) tuple list to draw inward lines,
31
31
      out_lines         (start, end, colour) tuple list to draw outward lines.
32
32
    """
33
 
    
34
 
    columns_len = 0
35
33
 
36
34
    __gproperties__ = {
37
35
        "node":         ( gobject.TYPE_PYOBJECT, "node",
38
36
                          "revision node instruction",
39
37
                          gobject.PARAM_WRITABLE
40
38
                        ),
41
 
        "tags":         ( gobject.TYPE_PYOBJECT, "tags",
42
 
                          "list of tags associated with the node",
43
 
                          gobject.PARAM_WRITABLE
44
 
                        ),
45
39
        "in-lines":     ( gobject.TYPE_PYOBJECT, "in-lines",
46
40
                          "instructions to draw lines into the cell",
47
41
                          gobject.PARAM_WRITABLE
51
45
                          gobject.PARAM_WRITABLE
52
46
                        ),
53
47
        }
54
 
    
 
48
 
55
49
    def do_set_property(self, property, value):
56
50
        """Set properties from GObject properties."""
57
51
        if property.name == "node":
58
52
            self.node = value
59
 
        elif property.name == "tags":
60
 
            self.tags = value
61
53
        elif property.name == "in-lines":
62
54
            self.in_lines = value
63
55
        elif property.name == "out-lines":
92
84
        colours and the fg parameter provides the multiplier that should be
93
85
        applied to the foreground colours.
94
86
        """
95
 
        mainline_color = ( 0.0, 0.0, 0.0 )
96
87
        colours = [
97
88
            ( 1.0, 0.0, 0.0 ),
98
89
            ( 1.0, 1.0, 0.0 ),
102
93
            ( 1.0, 0.0, 1.0 ),
103
94
            ]
104
95
 
105
 
        if colour == 0:
106
 
            colour_rgb = mainline_color
107
 
        else:
108
 
            colour_rgb = colours[colour % len(colours)]
109
 
 
110
 
        red   = (colour_rgb[0] * fg) or bg
111
 
        green = (colour_rgb[1] * fg) or bg
112
 
        blue  = (colour_rgb[2] * fg) or bg
 
96
        colour %= len(colours)
 
97
        red   = (colours[colour][0] * fg) or bg
 
98
        green = (colours[colour][1] * fg) or bg
 
99
        blue  = (colours[colour][2] * fg) or bg
113
100
 
114
101
        ctx.set_source_rgb(red, green, blue)
115
102
 
122
109
        """
123
110
        box_size = self.box_size(widget) + 1
124
111
 
125
 
        width = box_size * (self.columns_len + 1)
 
112
        cols = self.node[0]
 
113
        for start, end, colour in self.in_lines + self.out_lines:
 
114
            cols = max(cols, start, end)
 
115
 
 
116
        width = box_size * (cols + 1)
126
117
        height = box_size
127
118
 
128
119
        # FIXME I have no idea how to use cell_area properly
149
140
        box_size = self.box_size(widget)
150
141
 
151
142
        ctx.set_line_width(box_size / 8)
 
143
        ctx.set_line_cap(cairo.LINE_CAP_SQUARE)
152
144
 
153
145
        # Draw lines into the cell
154
146
        for start, end, colour in self.in_lines:
155
 
            self.render_line (ctx, cell_area, box_size,
156
 
                         bg_area.y, bg_area.height,
157
 
                         start, end, colour, flags)
 
147
            ctx.move_to(cell_area.x + box_size * start + box_size / 2,
 
148
                        bg_area.y - bg_area.height / 2)
 
149
 
 
150
            if start - end > 1:
 
151
                ctx.line_to(cell_area.x + box_size * start, bg_area.y)
 
152
                ctx.line_to(cell_area.x + box_size * end + box_size, bg_area.y)
 
153
            elif start - end < -1:
 
154
                ctx.line_to(cell_area.x + box_size * start + box_size,
 
155
                            bg_area.y)
 
156
                ctx.line_to(cell_area.x + box_size * end, bg_area.y)
 
157
 
 
158
            ctx.line_to(cell_area.x + box_size * end + box_size / 2,
 
159
                        bg_area.y + bg_area.height / 2)
 
160
 
 
161
            self.set_colour(ctx, colour, 0.0, 0.65)
 
162
            ctx.stroke()
158
163
 
159
164
        # Draw lines out of the cell
160
165
        for start, end, colour in self.out_lines:
161
 
            self.render_line (ctx, cell_area, box_size,
162
 
                         bg_area.y + bg_area.height, bg_area.height,
163
 
                         start, end, colour, flags)
 
166
            ctx.move_to(cell_area.x + box_size * start + box_size / 2,
 
167
                        bg_area.y + bg_area.height / 2)
 
168
 
 
169
            if start - end > 1:
 
170
                ctx.line_to(cell_area.x + box_size * start,
 
171
                            bg_area.y + bg_area.height)
 
172
                ctx.line_to(cell_area.x + box_size * end + box_size,
 
173
                            bg_area.y + bg_area.height)
 
174
            elif start - end < -1:
 
175
                ctx.line_to(cell_area.x + box_size * start + box_size,
 
176
                            bg_area.y + bg_area.height)
 
177
                ctx.line_to(cell_area.x + box_size * end,
 
178
                            bg_area.y + bg_area.height)
 
179
 
 
180
            ctx.line_to(cell_area.x + box_size * end + box_size / 2,
 
181
                        bg_area.y + bg_area.height / 2 + bg_area.height)
 
182
 
 
183
            self.set_colour(ctx, colour, 0.0, 0.65)
 
184
            ctx.stroke()
164
185
 
165
186
        # Draw the revision node in the right column
166
187
        (column, colour) = self.node
168
189
                cell_area.y + cell_area.height / 2,
169
190
                box_size / 4, 0, 2 * math.pi)
170
191
 
171
 
        if flags & gtk.CELL_RENDERER_SELECTED:
172
 
            ctx.set_source_rgb(1.0, 1.0, 1.0)
173
 
            ctx.set_line_width(box_size / 4)
174
 
            ctx.stroke_preserve()
175
 
            ctx.set_line_width(box_size / 8)
176
 
 
177
192
        self.set_colour(ctx, colour, 0.0, 0.5)
178
193
        ctx.stroke_preserve()
179
194
 
180
195
        self.set_colour(ctx, colour, 0.5, 1.0)
181
196
        ctx.fill()
182
 
 
183
 
        self.render_tags(ctx, widget.create_pango_context(), cell_area, box_size)
184
 
    
185
 
    def render_line(self, ctx, cell_area, box_size, mid, height, start, end, colour, flags):
186
 
        if start is None:
187
 
            ctx.set_line_cap(cairo.LINE_CAP_ROUND)
188
 
            x = cell_area.x + box_size * end + box_size / 2
189
 
            ctx.move_to(x, mid + height / 3)
190
 
            ctx.line_to(x, mid + height / 3)
191
 
            ctx.move_to(x, mid + height / 6)
192
 
            ctx.line_to(x, mid + height / 6)
193
 
            
194
 
        elif end is None:
195
 
            ctx.set_line_cap(cairo.LINE_CAP_ROUND)
196
 
            x = cell_area.x + box_size * start + box_size / 2
197
 
            ctx.move_to(x, mid - height / 3)
198
 
            ctx.line_to(x, mid - height / 3)
199
 
            ctx.move_to(x, mid - height / 6)
200
 
            ctx.line_to(x, mid - height / 6)
201
 
 
202
 
        else:
203
 
            ctx.set_line_cap(cairo.LINE_CAP_BUTT)
204
 
            startx = cell_area.x + box_size * start + box_size / 2
205
 
            endx = cell_area.x + box_size * end + box_size / 2
206
 
            
207
 
            ctx.move_to(startx, mid - height / 2)
208
 
            
209
 
            if start - end == 0 :
210
 
                ctx.line_to(endx, mid + height / 2 + 1)
211
 
            else:
212
 
                ctx.curve_to(startx, mid - height / 5,
213
 
                             startx, mid - height / 5,
214
 
                             startx + (endx - startx) / 2, mid)
215
 
 
216
 
                ctx.curve_to(endx, mid + height / 5,
217
 
                             endx, mid + height / 5 ,
218
 
                             endx, mid + height / 2 + 1)
219
 
 
220
 
        if flags & gtk.CELL_RENDERER_SELECTED:
221
 
            ctx.set_source_rgb(1.0, 1.0, 1.0)
222
 
            ctx.set_line_width(box_size / 5)
223
 
            ctx.stroke_preserve()
224
 
            ctx.set_line_width(box_size / 8)
225
 
 
226
 
        self.set_colour(ctx, colour, 0.0, 0.65)
227
 
 
228
 
        ctx.stroke()
229
 
 
230
 
    def render_tags(self, ctx, pango_ctx, cell_area, box_size):
231
 
        # colour ID used in self.set_colour on the tags
232
 
        TAG_COLOUR_ID = 1
233
 
 
234
 
        (column, colour) = self.node
235
 
 
236
 
        font_desc = pango.FontDescription()
237
 
        font_desc.set_size(pango.SCALE * 7)
238
 
 
239
 
        tag_layout = pango.Layout(pango_ctx)
240
 
        tag_layout.set_font_description(font_desc)
241
 
 
242
 
        # The width of the tag label stack
243
 
        width = 0
244
 
 
245
 
        for tag_idx, tag in enumerate(self.tags):
246
 
            tag_layout.set_text(" " + tag + " ")
247
 
            text_width, text_height = tag_layout.get_pixel_size()
248
 
 
249
 
            x0 = cell_area.x + \
250
 
                 box_size * (column + 1.3) + width
251
 
 
252
 
            y0 = cell_area.y + \
253
 
                 cell_area.height / 2 - \
254
 
                 text_height / 2
255
 
 
256
 
            width += text_width + 5
257
 
 
258
 
            # Draw the tag border
259
 
            ctx.move_to(x0 - box_size / 3, y0 + text_height / 2)
260
 
            ctx.line_to(x0, y0)
261
 
            ctx.line_to(x0 + text_width, y0)
262
 
            ctx.line_to(x0 + text_width, y0 + text_height)
263
 
            ctx.line_to(x0, y0 + text_height)
264
 
            ctx.line_to(x0 - box_size / 3, y0 + text_height / 2)
265
 
 
266
 
            ctx.new_sub_path()
267
 
            ctx.arc(x0 - box_size / 12,
268
 
                        y0 + text_height / 2,
269
 
                        box_size / 7,
270
 
                        0, 2 * math.pi);
271
 
 
272
 
            self.set_colour(ctx, TAG_COLOUR_ID, 0.0, 0.5)
273
 
            ctx.stroke_preserve()
274
 
 
275
 
            ctx.set_fill_rule (cairo.FILL_RULE_EVEN_ODD)
276
 
            self.set_colour(ctx, TAG_COLOUR_ID, 0.5, 1.0)
277
 
            ctx.fill()
278
 
 
279
 
            # Draw the tag text
280
 
            self.set_colour(ctx, 0, 0.0, 0.0)
281
 
            ctx.move_to(x0, y0)
282
 
            ctx.show_layout(tag_layout)
283