华容道高性能计算引擎
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.

107 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>')