mirror of https://github.com/dnomd343/klotski.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.3 KiB
106 lines
3.3 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import yaml
|
|
|
|
BLOCK_LENGTH = 12
|
|
BLOCK_DISTANCE = 4
|
|
|
|
CASE_WIDTH = BLOCK_LENGTH * 4 + BLOCK_DISTANCE * 5
|
|
CASE_HEIGHT = BLOCK_LENGTH * 5 + BLOCK_DISTANCE * 6
|
|
|
|
CASE_DISTANCE_X = 80
|
|
CASE_DISTANCE_Y = 40
|
|
|
|
MAIN_MARGIN_X = 10
|
|
MAIN_MARGIN_Y = 8
|
|
|
|
LINK_OFFSET = 1
|
|
|
|
raw_data = yaml.load(open('demo.yml').read(), Loader = yaml.FullLoader)
|
|
layer = raw_data['layer']
|
|
link = raw_data['next']
|
|
|
|
# print(layer)
|
|
# print(link)
|
|
|
|
MAX_CASE_X = max([len(x)for x in layer])
|
|
MAX_CASE_Y = len(layer)
|
|
|
|
# def cal_length(num: int, single_length: int, distance: int):
|
|
# return num * single_length + (num - 1) * distance
|
|
|
|
|
|
MAX_WIDTH = MAX_CASE_X * CASE_WIDTH + (MAX_CASE_X - 1) * CASE_DISTANCE_X
|
|
|
|
|
|
def case_address(layer_num: int, layer_index: int) -> (int, int):
|
|
|
|
case_number_x = len(layer[layer_num])
|
|
|
|
MY_WIDTH = case_number_x * CASE_WIDTH + (case_number_x - 1) * CASE_DISTANCE_X
|
|
|
|
MY_WIDTH_OFFSET = int((MAX_WIDTH - MY_WIDTH) / 2)
|
|
|
|
return (
|
|
layer_index * (CASE_WIDTH + CASE_DISTANCE_X) + MY_WIDTH_OFFSET + MAIN_MARGIN_X,
|
|
layer_num * (CASE_HEIGHT + CASE_DISTANCE_Y) + MAIN_MARGIN_Y,
|
|
)
|
|
|
|
|
|
def link_address(start_layer_num: int, start_layer_index: int, end_layer_num: int, end_layer_index: int) -> (int, int, int, int):
|
|
|
|
x1, y1 = case_address(start_layer_num, start_layer_index)
|
|
x2, y2 = case_address(end_layer_num, end_layer_index)
|
|
|
|
return (
|
|
x1 + int(CASE_WIDTH / 2),
|
|
y1 + CASE_HEIGHT + LINK_OFFSET,
|
|
x2 + int(CASE_WIDTH / 2),
|
|
y2 - LINK_OFFSET,
|
|
)
|
|
|
|
|
|
def svg_rect(left: int, top: int, width: int, height: int, line_width: int):
|
|
return '<rect x="%d" y="%d" width="%d" height="%d" style="fill:pink;stroke-width:%d;stroke:rgb(0,0,0)"/>' % (
|
|
left, top, width, height, line_width
|
|
)
|
|
|
|
|
|
def svg_line(start_x: int, start_y: int, end_x: int, end_y: int):
|
|
return '<line x1="%d" y1="%d" x2="%d" y2="%d" style="stroke:black;stroke-width:1"/>' % (
|
|
start_x, start_y, end_x, end_y
|
|
)
|
|
|
|
|
|
def case_output(info: list, x_offset: int, y_offset: int):
|
|
print(' ', svg_rect(x_offset, y_offset, CASE_WIDTH, CASE_HEIGHT, 2))
|
|
for (x, y, h, w) in info:
|
|
|
|
left = BLOCK_LENGTH * x + BLOCK_DISTANCE * (x + 1) + x_offset
|
|
top = BLOCK_LENGTH * y + BLOCK_DISTANCE * (y + 1) + y_offset
|
|
width = BLOCK_LENGTH * w + BLOCK_DISTANCE * (w - 1)
|
|
height = BLOCK_LENGTH * h + BLOCK_DISTANCE * (h - 1)
|
|
|
|
print(' ', svg_rect(left, top, width, height, 1))
|
|
|
|
|
|
MAIN_WIDTH = MAX_CASE_X * CASE_WIDTH + (MAX_CASE_X - 1) * CASE_DISTANCE_X + MAIN_MARGIN_X * 2
|
|
MAIN_HEIGHT = MAX_CASE_Y * CASE_HEIGHT + (MAX_CASE_Y - 1) * CASE_DISTANCE_Y + MAIN_MARGIN_Y * 2
|
|
|
|
print('<svg xmlns="http://www.w3.org/2000/svg" version="1.1" width="%d" height="%d">' % (MAIN_WIDTH, MAIN_HEIGHT))
|
|
|
|
for layer_num in range(0, len(layer)):
|
|
for layer_index in range(0, len(layer[layer_num])):
|
|
x, y = case_address(layer_num, layer_index)
|
|
case_output(layer[layer_num][layer_index], x, y)
|
|
|
|
for layer_num in range(0, len(link)):
|
|
for layer_index in range(0, len(link[layer_num])):
|
|
for next_index in link[layer_num][layer_index]:
|
|
x1, y1, x2, y2 = link_address(layer_num, layer_index, layer_num + 1, next_index)
|
|
# print('(%d, %d) -> (%d, %d)' % (layer_num, layer_index, layer_num + 1, next_index))
|
|
print(' ', svg_line(x1, y1, x2, y2))
|
|
|
|
|
|
print('</svg>')
|
|
|