Source code for timelinelib.canvas.eventboxdrawers.defaultmilestonedrawer

# Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018  Rickard Lindberg, Roger Lindberg
#
# This file is part of Timeline.
#
# Timeline is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Timeline is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Timeline.  If not, see <http://www.gnu.org/licenses/>.


import wx

from timelinelib.canvas.drawing.utils import black_solid_pen
from timelinelib.canvas.eventboxdrawers.handlerect import HandleRect, MIDDLE_HANDLE, LEFT_HANDLE, RIGHT_HANDLE



[docs]class DefaultMilestoneDrawer:

    HANDLE_SIZE = 6
    HALF_HANDLE_SIZE = 3


[docs]    def __init__(self, rect, event, selected):
        self._rect = rect
        self._event = event
        self._selected = selected



[docs]    def draw(self, dc):
        dc.DestroyClippingRegion()
        self._draw_rectangle(dc)
        self.draw_label(dc)
        if self._selected:
            self.draw_move_handle(dc)


    def _draw_rectangle(self, dc):
        dc.SetPen(black_solid_pen(1))
        dc.SetBrush(wx.Brush(wx.Colour(*self._event.get_color()), wx.BRUSHSTYLE_SOLID))
        dc.DrawRectangle(self._rect)


[docs]    def draw_label(self, dc):
        label = self._event.text[0] if self._event.text else " "
        draw_centered_text(dc, self._rect, label)



[docs]    def draw_move_handle(self, dc):
        HandleRect(self._rect, MIDDLE_HANDLE).draw(dc)




[docs]def draw_centered_text(dc, rect, label):
    size = dc.GetTextExtent(label)
    point = center_point_with_offset(rect, size.width // 2, size.height // 2)
    dc.DrawText(label, point)




[docs]def center_point_with_offset(rect, dx=0, dy=0):
    y = rect.Y + rect.Height // 2 - dy
    x = rect.X + rect.Width // 2 - dx
    return wx.Point(x, y)