5D_Heredero_Louis_TM2022/python/lycacode_gen_mini.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
This module can be used to create and display Mini-Lycacodes

(C) 2022 Louis Heredero  louis.heredero@edu.vs.ch
"""

import pygame
import numpy as np
import hamming

S = 600

class LycacodeError(Exception):
    pass

class LycacodeMini:
    BLACK = (158,17,26)
    #BLACK = (0,0,0)
    WHITE = (255,255,255)
    
    OFFSETS = [(0,-1), (1,0), (0,1), (-1,0)]
    
    FRAME = True
    CIRCLES = True
    DOTS = True
    DB_SQUARES = False

    def __init__(self, id_):
        self.id = id_
        self.encode()
        self.create_matrix()

    def encode(self):
        self.bits = f"{self.id:020b}"

        self.bits = list(map(int, self.bits))
        parity = [sum(self.bits[i*5:i*5+5])%2 for i in range(4)]
        for i in range(4):
            self.bits.insert((4-i)*5, parity[3-i])


    def create_matrix(self):
        self.matrix = np.zeros([9, 9])-1
        self.matrix[4:5, :] = 0
        self.matrix[:, 4:5] = 0
        self.matrix[1:2, 3:6] = 0
        self.matrix[3:6, 1:2] = 0
        self.matrix[-2:-1, -6:-3] = 0
        self.matrix[-6:-3, -2:-1] = 0
        self.matrix[4,4] = -1

        for y in range(9):
            for x in range(9):
                if self.matrix[y,x] == 0:
                    self.matrix[y,x] = self.bits.pop(0)

                if len(self.bits) == 0:
                    break

            if len(self.bits) == 0:
                break

    def display(self, surf):
        S = min(surf.get_size())
        s = int(S/12/3)*3
        O = (S-s*9)/2

        surf.fill(self.WHITE)

        # Frame
        if self.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
            if self.CIRCLES:
                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])

        # Dots
        if self.DOTS:
            for y in range(9):
                for x in range(9):
                    if self.matrix[y, x] == 1:
                        pygame.draw.circle(surf, self.WHITE, [O+(x+0.5)*s, O+(y+0.5)*s], s/3)

        # Center
        pygame.draw.circle(surf, self.WHITE, [O+4.5*s, O+4.25*s], s/6)
        pygame.draw.circle(surf, self.WHITE, [O+4.25*s, O+4.75*s], s/6)
    
    def save(self, path):
        S = 600
        s = int(S/12)
        O = (S-s*9)/2
        
        BLACK = "#9E111A"
        WHITE = "#FFFFFF"
        
        with open(path, "w") as f:
            f.write(f"<svg width='{S}px' height='{S}px' viewbox='0 0 {S} {S}'>\n")
            # Background
            f.write(f"<rect x='0' y='0' width='{S}' height='{S}' fill='{WHITE}' />\n")
            
            # Frame
            f.write(f"<rect x='{O-s*0.75}' y='{O-s*0.75}' width='{s*10.5}' height='{s*10.5}' style='fill:none;stroke:{BLACK};stroke-width:{0.5*s};' />\n")
            
            # Cross
            for i in range(4):
                dx, dy = self.OFFSETS[i]
                X, Y = S/2 + dx*s*3, S/2 + dy*s*3
                if self.CIRCLES:
                    for j in range(3):
                        dx2, dy2 = self.OFFSETS[(i+j-1)%4]
                        f.write(f"<circle cx='{X+dx2*s}' cy='{Y+dy2*s}' r='{0.75*s}' style='fill:{BLACK};stroke:none;stroke-width:0;' />\n")
                
                f.write(f"<rect x='{X-(1.5-abs(dx))*s}' y='{Y-(1.5-abs(dy))*s}' width='{s*(3-abs(dx)*2)}' height='{s*(3-abs(dy)*2)}' style='fill:{BLACK};stroke:none;stroke-width:0;' />\n")
            
            # Cross
            f.write(f"<rect x='{O}' y='{S/2-s/2}' width='{9*s}' height='{s}' style='fill:{BLACK};stroke:none;stroke-width:0;' />\n")
            f.write(f"<rect x='{S/2-s/2}' y='{O}' width='{s}' height='{9*s}' style='fill:{BLACK};stroke:none;stroke-width:0;' />\n")
            
            # Dots
            if self.DOTS:
                for y in range(9):
                    for x in range(9):
                        if self.matrix[y, x] == 1:
                            f.write(f"<circle cx='{O+(x+0.5)*s}' cy='{O+(y+0.5)*s}' r='{s/3}' style='fill:{WHITE};stroke:none;stroke-width:0;' />\n")
            
            # Center
            f.write(f"<circle cx='{O+4.5*s}' cy='{O+4.25*s}' r='{s/6}' style='fill:{WHITE};stroke:none;stroke-width:0;' />\n")
            f.write(f"<circle cx='{O+4.25*s}' cy='{O+4.75*s}' r='{s/6}' style='fill:{WHITE};stroke:none;stroke-width:0;' />\n")
            
            f.write("</svg>")

def save(self):
    path = input("Save as (.png or .svg): ")
    
    if path.endswith(".svg"):
        code.save(path)
    
    else:
        pygame.image.save(w, path)

if __name__ == "__main__":
    import base
    
    b = base.Base(S, S, "Mini-Lycacode generator")
    
    code = LycacodeMini(16048)
    code.display(b.w)
    
    b.save = save
    b.main()