src/diagram/diagram.py

   1 #!/usr/bin/env python
   2 #
   3 # A Diagram base class
   4 #
   5 # Copyright (C) 2018  Antonio Ospite <ao2@ao2.it>
   6 #
   7 # This program is free software: you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 from math import cos, sin, pi, fmod
  21
  22
  23 class Diagram(object):
  24     def __init__(self, width, height, background=(1, 1, 1), font_size=20, stroke_width=2):
  25         self.width = width
  26         self.height = height
  27         self.background = background
  28         self.font_size = font_size
  29         self.stroke_width = stroke_width
  30
  31     def clear(self):
  32         raise NotImplementedError
  33
  34     @staticmethod
  35     def test(diagram):
  36         diagram.clear()
  37
  38         x = 40
  39         y = 200
  40
  41         x_offset = x
  42
  43         theta = 0
  44
  45         diagram.draw_line(0, y, 400, y, (1, 0, 0, 1))
  46
  47         advance = diagram.draw_centered_text(x_offset, y, "Ciao", theta,
  48                                              align_baseline=True,
  49                                              bb_stroke_color=(0, 0, 0, 0.5),
  50                                              bb_fill_color=(1, 1, 1, 0.8))
  51         x_offset += advance
  52
  53         advance = diagram.draw_centered_text(x_offset, y, "____", theta + pi / 4,
  54                                              align_baseline=True,
  55                                              bb_stroke_color=(0, 0, 0, 0.5),
  56                                              bb_fill_color=(1, 1, 1, 0.8))
  57         x_offset += advance
  58
  59         advance = diagram.draw_centered_text(x_offset, y, "jxpqdlf", theta + pi / 2,
  60                                              align_baseline=True,
  61                                              bb_stroke_color=(0, 0, 0, 0.5),
  62                                              bb_fill_color=(1, 1, 1, 0.8))
  63         x_offset += advance
  64
  65         advance = diagram.draw_centered_text(x_offset, y, "pppp", theta + 3 * pi / 4,
  66                                              align_baseline=True,
  67                                              bb_stroke_color=(0, 0, 0, 0.5),
  68                                              bb_fill_color=(1, 1, 1, 0.8))
  69         x_offset += advance
  70
  71         advance = diagram.draw_centered_text(x_offset, y, "dddd", theta + pi,
  72                                              align_baseline=True,
  73                                              bb_stroke_color=(0, 0, 0, 0.5),
  74                                              bb_fill_color=(1, 1, 1, 0.8))
  75         x_offset += advance
  76
  77         advance = diagram.draw_centered_text(x_offset, y, "Jjjj", theta + 5 * pi / 4,
  78                                              align_baseline=True,
  79                                              bb_stroke_color=(0, 0, 0, 0.5),
  80                                              bb_fill_color=(1, 1, 1, 0.8))
  81         x_offset += advance
  82
  83         advance = diagram.draw_centered_text(x_offset, y, "1369", theta + 3 * pi / 2,
  84                                              color=(0, 1, 0),
  85                                              align_baseline=True,
  86                                              bb_stroke_color=(0, 0, 0, 0.5),
  87                                              bb_fill_color=(1, 1, 1, 0.8))
  88         x_offset += advance
  89
  90         advance = diagram.draw_centered_text(x_offset, y, "qqqq", theta + 7 * pi / 4,
  91                                              align_baseline=True,
  92                                              bb_stroke_color=(0, 0, 0, 0.5),
  93                                              bb_fill_color=(1, 1, 1, 0.8))
  94         x_offset += advance
  95
  96         diagram.draw_rect(40, 40, 300, 100, stroke_color=(0, 0, 0, 0.8))
  97         diagram.draw_rect(40, 40, 300, 100, pi / 30, stroke_color=(0, 0, 0, 0.8))
  98
  99         verts = diagram.draw_regular_polygon(190, 90, 3, 20)
 100
 101         diagram.draw_rect(40, 250, 300, 100, stroke_color=(0, 0, 0, 0.8))
 102         diagram.draw_rect_from_center(40 + 150, 250 + 50, 300, 100, theta=(pi / 40),
 103                                       stroke_color=(1, 0, 0),
 104                                       fill_color=None)
 105
 106         verts = diagram.draw_regular_polygon(190, 300, 6, 20, pi / 3., (0, 0, 1, 0.5), (0, 1, 0.5))
 107         diagram.draw_apothem_star(190, 300, 6, 20, 0, (1, 0, 1))
 108
 109         diagram.draw_star_by_verts(190, 300, verts, (1, 0, 0, 0.5))
 110         diagram.draw_star(190, 300, 6, 25, 0, (1, 0, 1, 0.2))
 111
 112         diagram.draw_circle(190, 300, 30, (0, 1, 0, 0.5), None)
 113         diagram.draw_circle(100, 300, 30, (1, 0, 0, 0.5), (0, 1, 1, 0.5))
 114
 115     @staticmethod
 116     def color_to_rgba(color):
 117         assert len(color) >= 3
 118
 119         color = tuple(float(c) for c in color)
 120         if len(color) == 3:
 121             color += (1.0,)
 122
 123         return color
 124
 125     @staticmethod
 126     def normalized_angle_01(theta):
 127         return fmod(theta, 2 * pi) / (2 * pi)
 128
 129     @staticmethod
 130     def get_regular_polygon(x, y, sides, r, theta0=0.0):
 131         """Calc the coordinates of the regular polygon.
 132
 133         NOTE: the first point will be in axis with y."""
 134         theta = 2 * pi / sides
 135
 136         verts = []
 137         for i in range(sides):
 138             px = x + r * sin(theta0 + i * theta)
 139             py = y + r * cos(theta0 + i * theta)
 140             verts.append((px, py))
 141
 142         return verts
 143
 144     def draw_polygon_by_verts(self, verts,
 145                               stroke_color=(0, 0, 0),
 146                               fill_color=None):
 147         raise NotImplementedError
 148
 149     def draw_regular_polygon(self, cx, cy, sides, r, theta=0.0,
 150                              stroke_color=(0, 0, 0),
 151                              fill_color=None):
 152         verts = self.get_regular_polygon(cx, cy, sides, r, theta)
 153         self.draw_polygon_by_verts(verts, stroke_color, fill_color)
 154         return verts
 155
 156     def draw_star_by_verts(self, cx, cy, verts, stroke_color=(0, 0, 0)):
 157         raise NotImplementedError
 158
 159     def draw_star(self, cx, cy, sides, r, theta=0.0, stroke_color=(0, 0, 0)):
 160         verts = self.get_regular_polygon(cx, cy, sides, r, theta)
 161         self.draw_star_by_verts(cx, cy, verts, stroke_color)
 162         return verts
 163
 164     def draw_apothem_star(self, cx, cy, sides, r, theta=0.0, stroke_color=(0, 0, 0)):
 165         """Draw a star but calculate the regular polygon apothem from the passed radius."""
 166         apothem = r * cos(pi / sides)
 167         apothem_angle = theta + pi / sides
 168
 169         return self.draw_star(cx, cy, sides, apothem, apothem_angle, stroke_color)
 170
 171     def draw_rect(self, x, y, width, height, theta=0,
 172                   stroke_color=None,
 173                   fill_color=(1, 1, 1, 0.8)):
 174         raise NotImplementedError
 175
 176     def draw_rect_from_center(self, cx, cy, width, height, theta=0.0,
 177                               stroke_color=None,
 178                               fill_color=(1, 1, 1, 0.8)):
 179         # the position of the center of a rectangle at (0,0)
 180         mx = width / 2.0
 181         my = height / 2.0
 182
 183         # calculate the position of the bottom-left corner after rotating the
 184         # rectangle around the center
 185         rx = cx - (mx * cos(theta) - my * sin(theta))
 186         ry = cy - (mx * sin(theta) + my * cos(theta))
 187
 188         self.draw_rect(rx, ry, width, height, theta, stroke_color, fill_color)