Пример #1
0
def sgf_part_from_node(board, node_id):
    part = ';'
    node = board.tree[node_id]

    if node.action == NODE_BLACK:
        part += 'B[%s]' % coord_to_sgf(node.move, board.size)
    elif node.action == NODE_WHITE:
        part += 'W[%s]' % coord_to_sgf(node.move, board.size)
    elif node.action == NODE_EDIT:
        edits = defaultdict(list)
        for coord, color in node.edits.items():
            edits[color].append(coord_to_sgf(int(coord), board.size))

        ab = ''.join('[%s]' % c for c in edits[BLACK])
        aw = ''.join('[%s]' % c for c in edits[WHITE])
        ae = ''.join('[%s]' % c for c in edits[EMPTY])

        if ab:
            part += 'AB%s' % ab
        if aw:
            part += 'AW%s' % aw
        if ae:
            part += 'AE%s' % ae

    part += _part_symbols(board, node, SYMBOL_TRIANGLE)
    part += _part_symbols(board, node, SYMBOL_SQUARE)
    part += _part_symbols(board, node, SYMBOL_CIRCLE)
    part += _part_labels(board, node)

    for child in node.children:
        p = sgf_part_from_node(board, child)

        if len(node.children) == 1:
            part += p
        else:
            part += '(' + p + ')'

    return part
Пример #2
0
def _part_symbols(board, node, symbol):
    symbols = ['[%s]' % coord_to_sgf(int(coord), board.size)
               for coord, sym in node.symbols.items() if sym == symbol]
    return (symbol + ''.join(symbols)) if symbols else ''
Пример #3
0
def _part_labels(board, node):
    labels = ['[%s:%s]' % (coord_to_sgf(int(coord), board.size), lbl)
              for coord, lbl in node.labels.items()]
    return ('LB' + ''.join(labels)) if labels else ''
Пример #4
0
def test_coord_to_sgf():
    assert coord_to_sgf(coord2d(1, 1, 19), 19) == 'aa'
    assert coord_to_sgf(coord2d(3, 1, 19), 19) == 'ca'
    assert coord_to_sgf(coord2d(4, 4, 19), 19) == 'dd'
    assert coord_to_sgf(coord2d(4, 16, 19), 19) == 'dp'
    assert coord_to_sgf(coord2d(19, 19, 19), 19) == 'ss'