79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
import gdb.printing
|
|
|
|
class Square(object):
|
|
"""
|
|
Wrapper around GDB's representation of a 'seer::board::square::Square' in
|
|
memory.
|
|
"""
|
|
|
|
FILES = list(map(lambda n: chr(ord('A') + n), range(8)))
|
|
RANKS = list(map(lambda n: str(n + 1), range(8)))
|
|
|
|
def __init__(self, val):
|
|
self._val = val
|
|
|
|
def __str__(self):
|
|
return self.FILES[self.file] + self.RANKS[self.rank]
|
|
|
|
@property
|
|
def rank(self):
|
|
return int(self._val) % 8
|
|
|
|
@property
|
|
def file(self):
|
|
return int(self._val) // 8
|
|
|
|
class Bitboard(object):
|
|
"""
|
|
Wrapper around GDB's representation of a 'seer::board::bitboard::Bitboard'
|
|
in memory.
|
|
"""
|
|
|
|
def __init__(self, val):
|
|
self._val = val
|
|
|
|
def __str__(self):
|
|
return "[" + ", ".join(map(str, self.squares)) + "]"
|
|
|
|
@property
|
|
def squares(self):
|
|
n = int(self._val["__0"])
|
|
while n:
|
|
b = n & (~n+1)
|
|
yield Square(b.bit_length() - 1)
|
|
n ^= b
|
|
|
|
class SquarePrinter(object):
|
|
"Print a seer::board::square::Square"
|
|
|
|
def __init__(self, val):
|
|
self._val = Square(val)
|
|
|
|
def to_string(self):
|
|
return str(self._val)
|
|
|
|
def display_hint(self):
|
|
return 'string'
|
|
|
|
class BitboardPrinter(object):
|
|
"Print a seer::board::bitboard::Bitboard"
|
|
|
|
def __init__(self, val):
|
|
self._val = Bitboard(val)
|
|
|
|
def to_string(self):
|
|
return "Bitboard{" + str(self._val)[1:-1] + "}"
|
|
|
|
def display_hint(self):
|
|
return 'string'
|
|
|
|
def build_pretty_printer():
|
|
pp = gdb.printing.RegexpCollectionPrettyPrinter('seer')
|
|
|
|
pp.add_printer('BigNum', '^seer::board::square::Square$', SquarePrinter)
|
|
pp.add_printer('BigNum', '^seer::board::bitboard::Bitboard$', BitboardPrinter)
|
|
|
|
return pp
|
|
|
|
gdb.printing.register_pretty_printer(gdb.current_objfile(), build_pretty_printer(), True)
|