93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Generates lycacode frame dimensions figure
|
||
|
|
|
||
|
|
(C) 2022 Louis Heredero louis.heredero@edu.vs.ch
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pygame
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
S = 600
|
||
|
|
|
||
|
|
class Lycacode:
|
||
|
|
RES = 3
|
||
|
|
|
||
|
|
BLACK = (158,17,26)
|
||
|
|
#BLACK = (0,0,0)
|
||
|
|
WHITE = (255,255,255)
|
||
|
|
|
||
|
|
OFFSETS = [(0,-1), (1,0), (0,1), (-1,0)]
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
R = self.RES
|
||
|
|
self.matrix = np.zeros([R*9, R*9])-1
|
||
|
|
self.matrix[R*4:R*5, :] = 0
|
||
|
|
self.matrix[:, R*4:R*5] = 0
|
||
|
|
self.matrix[R:R*2, R*3:R*6] = 0
|
||
|
|
self.matrix[R*3:R*6, R:R*2] = 0
|
||
|
|
self.matrix[-R*2:-R, -R*6:-R*3] = 0
|
||
|
|
self.matrix[-R*6:-R*3, -R*2:-R] = 0
|
||
|
|
|
||
|
|
self.font = pygame.font.SysFont("arial", 30, bold=True)
|
||
|
|
|
||
|
|
def display(self, surf):
|
||
|
|
R = self.RES
|
||
|
|
S = min(surf.get_size())*2
|
||
|
|
s = int(S/12/R)*R
|
||
|
|
O = (S-s*9)/2
|
||
|
|
|
||
|
|
surf.fill(self.WHITE)
|
||
|
|
|
||
|
|
# Frame
|
||
|
|
pygame.draw.rect(surf, self.BLACK, [O-s, O-s, s*11, s*11])
|
||
|
|
pygame.draw.rect(surf, self.WHITE, [O-s*0.5, O-s*0.5, s*10, s*10])
|
||
|
|
|
||
|
|
# Cross
|
||
|
|
for i in range(4):
|
||
|
|
dx, dy = self.OFFSETS[i]
|
||
|
|
X, Y = S/2 + dx*s*3, S/2 + dy*s*3
|
||
|
|
for j in range(3):
|
||
|
|
dx2, dy2 = self.OFFSETS[(i+j-1)%4]
|
||
|
|
pygame.draw.circle(surf, self.BLACK, [X+dx2*s, Y+dy2*s], 0.75*s)
|
||
|
|
|
||
|
|
pygame.draw.rect(surf, self.BLACK, [X-(1.5-abs(dx))*s, Y-(1.5-abs(dy))*s, s*(3-abs(dx)*2), s*(3-abs(dy)*2)])
|
||
|
|
|
||
|
|
pygame.draw.rect(surf, self.BLACK, [O, S/2-s/2, 9*s, s])
|
||
|
|
pygame.draw.rect(surf, self.BLACK, [S/2-s/2, O, s, 9*s])
|
||
|
|
|
||
|
|
for y in range(9):
|
||
|
|
for x in range(9):
|
||
|
|
if self.matrix[y*R, x*R] != -1:
|
||
|
|
pygame.draw.rect(surf, (0,0,0), [O+x*s, O+y*s, s+1, s+1], 2)
|
||
|
|
|
||
|
|
col = (17,158,147)
|
||
|
|
col = (0,0,0)
|
||
|
|
pygame.draw.line(surf, col, [O-s, O+s*3], [O, O+s*3], 3)
|
||
|
|
pygame.draw.line(surf, col, [O-s, O+s*3-s/6], [O-s, O+s*3+s/6], 3)
|
||
|
|
pygame.draw.line(surf, col, [O, O+s*3-s/6], [O, O+s*3+s/6], 3)
|
||
|
|
txt = self.font.render("S", True, col)
|
||
|
|
surf.blit(txt, [O-s/2-txt.get_width()/2, O+s*3-s/6-txt.get_height()])
|
||
|
|
|
||
|
|
pygame.draw.line(surf, col, [O+s, O+s*3-s/6], [O+s*2, O+s*3-s/6], 3)
|
||
|
|
pygame.draw.line(surf, col, [O+s, O+s*3-s/3], [O+s, O+s*3], 3)
|
||
|
|
pygame.draw.line(surf, col, [O+s*2, O+s*3-s/3], [O+s*2, O+s*3], 3)
|
||
|
|
txt = self.font.render("S", True, col)
|
||
|
|
surf.blit(txt, [O+3*s/2-txt.get_width()/2, O+s*3-s/3-txt.get_height()])
|
||
|
|
|
||
|
|
pygame.draw.line(surf, col, [O+s, O-s], [O+s, O-s/2], 3)
|
||
|
|
pygame.draw.line(surf, col, [O+s-s/6, O-s], [O+s+s/6, O-s], 3)
|
||
|
|
pygame.draw.line(surf, col, [O+s-s/6, O-s/2], [O+s+s/6, O-s/2], 3)
|
||
|
|
txt = self.font.render("S/2", True, col)
|
||
|
|
surf.blit(txt, [O+s-s/6-txt.get_width(), O-s*0.75-txt.get_height()/2])
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
pygame.init()
|
||
|
|
|
||
|
|
surf = pygame.Surface([S, S])
|
||
|
|
|
||
|
|
code = Lycacode()
|
||
|
|
code.display(surf)
|
||
|
|
pygame.image.save(surf, "lycacode_frame.png")
|