src/flexagon/tetraflexagon_diagram.py

   1 #!/usr/bin/env python
   2 #
   3 # A class to draw tetraflexagons
   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 .tritetraflexagon import TriTetraflexagon
  21
  22
  23 class TetraflexagonDiagram(object):
  24     def __init__(self, x_border, backend=None):
  25         self.x_border = x_border
  26         self.backend = backend
  27
  28         self.tetraflexagon = TriTetraflexagon()
  29
  30         num_squares = len(self.tetraflexagon.squares)
  31         self.square_side = (self.backend.height - (x_border * 3)) / (num_squares)
  32         self.square_radius = self.square_side / 2
  33         self.tile_side = self.square_radius
  34         self.tile_radius = self.tile_side / 2
  35
  36         self._init_centers()
  37
  38         # draw the plan centered wrt. the squares
  39         self.plan_origin = ((self.backend.width - self.tile_side * 5) / 2,
  40                             self.x_border)
  41
  42         self.squares_color_map = [(1, 0, 0), (0, 1, 0), (0, 0, 1)]
  43
  44     def _init_centers(self):
  45         # Preallocate the lists to be able to access them by indices in the
  46         # loops below.
  47         self.squares_centers = [None for h in self.tetraflexagon.squares]
  48         self.tiles_centers = [[None for t in h.tiles] for h in self.tetraflexagon.squares]
  49
  50         cy = self.backend.height - (self.square_radius + self.x_border)
  51         for square in self.tetraflexagon.squares:
  52             cx = self.x_border + (2 * self.square_radius + self.x_border) * (square.index + 1)
  53             self.squares_centers[square.index] = (cx, cy)
  54
  55             for tile in square.tiles:
  56                 # offset by 1 or -1 times the tile radius
  57                 tile_xoffset, tile_yoffset = tile.calc_offset_in_square(self.tile_side)
  58                 self.tiles_centers[square.index][tile.index] = (cx + tile_xoffset, cy + tile_yoffset)
  59
  60     def get_square_center(self, square):
  61         return self.squares_centers[square.index]
  62
  63     def get_tile_center(self, tile):
  64         return self.tiles_centers[tile.square.index][tile.index]
  65
  66     def get_tile_center_in_plan(self, tile):
  67         x0, y0 = self.plan_origin
  68         i, j = self.tetraflexagon.get_tile_plan_position(tile)
  69         x, y = tile.calc_plan_coordinates(self.tile_side, i, j)
  70         return x0 + x, y0 + y
  71
  72     def get_tile_transform(self, tile):
  73         """Calculate the transformation matrix from a tile in an square to
  74         the correspondent tile in the plan.
  75
  76         Return the matrix as a list of values sorted in row-major order."""
  77
  78         src_x, src_y = self.get_tile_center(tile)
  79         dest_x, dest_y = self.get_tile_center_in_plan(tile)
  80
  81         i, j = self.tetraflexagon.get_tile_plan_position(tile)
  82         theta = tile.calc_angle_in_plan(i, j)
  83
  84         return self.backend.calc_rotate_translate_transform(src_x, src_y,
  85                                                             dest_x, dest_y, theta)
  86
  87     def draw_square_template(self, square):
  88         for tile in square.tiles:
  89             cx, cy = self.get_tile_center(tile)
  90             self.draw_tile_template(tile, cx, cy, 0)
  91
  92     def draw_tile_template(self, tile, cx, cy, theta):
  93         side = self.tile_side
  94         color = self.squares_color_map[tile.square.index]
  95
  96         self.backend.draw_rect_from_center(cx, cy, side, side, theta,
  97                                            stroke_color=color,
  98                                            fill_color=None)
  99
 100         corners_labels = "ABC"
 101         corner_text = corners_labels[tile.square.index] + str(tile.index + 1)
 102         self.backend.draw_centered_text(cx, cy, corner_text, 0, color)
 103
 104     def draw_plan_template(self):
 105         x0, y0 = self.plan_origin
 106         for square in self.tetraflexagon.squares:
 107             for tile in square.tiles:
 108                 i, j = self.tetraflexagon.get_tile_plan_position(tile)
 109                 x, y = tile.calc_plan_coordinates(self.tile_radius, i, j)
 110                 theta = tile.get_angle_in_plan(i, j)
 111                 self.draw_tile_template(tile, x0 + x, y0 + y, theta)
 112
 113     def draw_template(self):
 114         for square in self.tetraflexagon.squares:
 115             self.draw_square_template(square)
 116
 117         self.draw_plan_template()