forked from tensorflow/minigo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcoords.py
114 lines (95 loc) · 3.92 KB
/
coords.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# Copyright 2018 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Logic for dealing with coordinates.
This introduces some helpers and terminology that are used throughout MiniGo.
MiniGo Coordinate: This is a tuple of the form (row, column) that is indexed
starting out at (0, 0) from the upper-left.
Flattened Coordinate: this is a number ranging from 0 - N^2 (so N^2+1
possible values). The extra value N^2 is used to mark a 'pass' move.
SGF Coordinate: Coordinate used for SGF serialization format. Coordinates use
two-letter pairs having the form (column, row) indexed from the upper-left
where 0,0 == 'aa'.
KGS Coordinate: Human-readable coordinate string indexed from bottom left, with
the first character a capital letter and the second a number from 1-19.
PYGTP Coordinate: Tuple coordinate indexed starting at 1,1 from bottom-left
in the format (column, row)
So, for a 19x19,
Coord Type upper_left upper_right pass
-------------------------------------------------------
minigo coord (0, 0) (0, 18) None
flat 0 18 361
SGF 'aa' 'sa' ''
KGS 'A19' 'T19' 'pass'
pygtp (1, 19) (19, 19) (0, 0)
"""
import go
import gtp
KGS_COLUMNS = 'ABCDEFGHJKLMNOPQRST'
SGF_COLUMNS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
def sgf_to_flat(sgf):
"""Transforms an SGF coordinate directly into a flattened coordinate."""
return flatten_coords(parse_sgf_coords(sgf))
def kgs_to_flat(sgf):
"""Transforms a KGS coordinate directly into a flattened coordinate."""
return flatten_coords(parse_kgs_coords(sgf))
def flatten_coords(c):
"""Flattens a coordinate tuple from a MiniGo coordinate"""
if c is None:
return go.N * go.N
return go.N * c[0] + c[1]
def unflatten_coords(f):
"""Unflattens a flattened coordinate into a Minigo coordinate"""
if f == go.N * go.N:
return None
return divmod(f, go.N)
def parse_sgf_coords(s):
"""Transform a SGF coordinate into a coordinate-tuple"""
if s is None or s == '':
return None
return SGF_COLUMNS.index(s[1]), SGF_COLUMNS.index(s[0])
def unparse_sgf_coords(c):
"""Turns a MiniGo coordinate tuple into a SGF coordinate."""
if c is None:
return ''
return SGF_COLUMNS[c[1]] + SGF_COLUMNS[c[0]]
def parse_kgs_coords(s):
"""Interprets KGS coordinates returning a minigo coordinate tuple."""
if s == 'pass':
return None
s = s.upper()
col = KGS_COLUMNS.index(s[0])
row_from_bottom = int(s[1:]) - 1
return go.N - row_from_bottom - 1, col
def to_human_coord(coord):
"""Converts from a MiniGo coord to a human readable string.
This is equivalent to a KGS coordinate.
"""
if coord == None:
return "pass"
else:
y, x = coord
return "{}{}".format("ABCDEFGHJKLMNOPQRSTYVWYZ"[x], go.N-y)
def parse_pygtp_coords(vertex):
"""Transforms a GTP coordinate into a standard MiniGo coordinate.
GTP has a notion of both a Pass and a Resign, both of which
are mapped to None so the conversion is not precisely bijective.
"""
if vertex in (gtp.PASS, gtp.RESIGN):
return None
return go.N - vertex[1], vertex[0] - 1
def unparse_pygtp_coords(c):
"""Transforms a MiniGo Coordinate back into a GTP coordinate."""
if c is None:
return gtp.PASS
return c[1] + 1, go.N - c[0]