svgwrite_diagram: implement draw_circle, draw_line and draw_rect

[flexagon-toolkit.git] / src / diagram / diagram.py

diff --git a/src/diagram/diagram.py b/src/diagram/diagram.py
index eee9fc6..a949577 100755 (executable)
--- a/src/diagram/diagram.py
+++ b/src/diagram/diagram.py
@@ -86,3 +86,22 @@ class Diagram(object):
         apothem_angle = theta + pi / sides
 
         return self.draw_star(cx, cy, sides, apothem, apothem_angle, stroke_color)
+
+    def draw_rect(self, x, y, width, height, theta=0,
+                  stroke_color=None,
+                  fill_color=(1, 1, 1, 0.8)):
+        raise NotImplementedError
+
+    def draw_rect_from_center(self, cx, cy, width, height, theta=0.0,
+                              stroke_color=None,
+                              fill_color=(1, 1, 1, 0.8)):
+        # the position of the center of a rectangle at (0,0)
+        mx = width / 2.0
+        my = height / 2.0
+
+        # calculate the position of the bottom-left corner after rotating the
+        # rectangle around the center
+        rx = cx - (mx * cos(theta) - my * sin(theta))
+        ry = cy - (mx * sin(theta) + my * cos(theta))
+
+        self.draw_rect(rx, ry, width, height, theta, stroke_color, fill_color)