root/gaphor/tags/gaphor-0.3.0/gaphor/diagram/usecase.py

'''
eseCaseItempango diagram item
'''
# vim:sw=4

from __future__ import generators

import gobject
import pango
import diacanvas
import gaphor.UML as UML
from gaphor.diagram import initialize_item
from classifier import ClassifierItem

class UseCaseItem(ClassifierItem):
    MARGIN_X=60
    MARGIN_Y=30

    def __init__(self, id=None):
        ClassifierItem.__init__(self, id)
        self.set(height=50, width=100)
        self._border = diacanvas.shape.Ellipse()
        self._border.set_line_width(2.0)

    # DiaCanvasItem callbacks:

    def on_update(self, affine):
        # Center the text
        w, h = self.get_name_size()
        self.set(min_width=w + UseCaseItem.MARGIN_X,
                 min_height=h + UseCaseItem.MARGIN_Y)
        self.update_name(x=0, y=(self.height - h) / 2,
                         width=self.width, height=h)

        ClassifierItem.on_update(self, affine)

        self._border.ellipse(center=(self.width / 2, self.height / 2),
                              width=self.width, height=self.height)
        self.expand_bounds(1.0)

    def on_shape_iter(self):
        yield self._border
        for s in ClassifierItem.on_shape_iter(self):
            yield s


initialize_item(UseCaseItem, UML.UseCase)