def test_is_valid_triangles_incremental(self):
b = Tritower('''
X.X
...
''')
self.assertFalse(b.valid_white_triangles())
self.assertFalse(b.valid_white_triangles((1, 1)))
def solve_medium_tritower():
puzzle = Tritower(
"""
---------
-----------
0------------
---------------
----1----------
-------1-2--1
-2--2------
-1-1--2--
"""
)
puzzle.solve()
def __init__(self, board=None):
self.valid = True
if board is None:
# debug board
#self.board = Board(1)
self.board = Tritower('''
1.X..
.X....X
.X.2X..
..X..
''')
self.selected_pos = (0,0)
#import os; os.environ['SDL_VIDEO_WINDOW_POS'] = '900,600'
pygame.init()
self.screen = pygame.display.set_mode(screen_size, HWSURFACE | DOUBLEBUF)
# set up numbers
self.number_font = pygame.font.Font(None, font_size)
self.number_surfaces = {}
for n in GIVENS:
self.number_surfaces[n] = self.number_font.render(str(n), True, (0,0,0))
self.graphical_main()
class GUI(object):
def __init__(self, board=None):
self.valid = True
if board is None:
# debug board
#self.board = Board(1)
self.board = Tritower('''
1.X..
.X....X
.X.2X..
..X..
''')
self.selected_pos = (0,0)
#import os; os.environ['SDL_VIDEO_WINDOW_POS'] = '900,600'
pygame.init()
self.screen = pygame.display.set_mode(screen_size, HWSURFACE | DOUBLEBUF)
# set up numbers
self.number_font = pygame.font.Font(None, font_size)
self.number_surfaces = {}
for n in GIVENS:
self.number_surfaces[n] = self.number_font.render(str(n), True, (0,0,0))
self.graphical_main()
def to_board(self, screen_pos):
"""Convert a screen position to board coordinates."""
x, y = screen_pos
cell_row = y // cell_height
cell_column_right = (cell_height * x + cell_half_base * y) // (2 * cell_height * cell_half_base)
cell_column_left = (cell_height * x - cell_half_base * y) // (2 * cell_height * cell_half_base)
return (cell_column_left + cell_column_right, cell_row)
def graphical_main(self):
# program loop #
done = False
while not done:
# events #
for e in event.get():
if e.type == QUIT: done = True
elif e.type == KEYDOWN:
if e.key == K_ESCAPE: event.post(event.Event(QUIT))
else:
character = str(e.unicode).lower()
if character in self.board.RCHARS:
self.set_color(self.board.RCHARS[character])
elif e.type == MOUSEMOTION:
old_pos = self.selected_pos
self.selected_pos = self.to_board(e.pos)
if self.selected_pos != old_pos: # dragged to a new spot
if e.buttons[0]: # left button held
self.set_color(BLACK)
elif e.buttons[1]: # middle button held
self.set_color(UNKNOWN)
elif e.buttons[2]: # right button held
self.set_color(WHITE)
elif e.type == MOUSEBUTTONDOWN:
self.selected_pos = self.to_board(e.pos)
if e.button == 1: # left click
self.set_color(BLACK)
elif e.button == 2: # middle click
self.set_color(UNKNOWN)
elif e.button == 3: # right click
self.set_color(WHITE)
# draw #
self.screen.fill(bg_color)
self.draw()
pygame.display.flip()
pygame.quit()
def set_color(self, color):
self.board.set_value(self.selected_pos, color)
self.valid = self.board.is_valid()
def draw(self):
self.draw_board((0,0))
def draw_board(self, init_pos):
x0, y0 = init_pos
grid_color = grid_color_valid if self.valid else grid_color_invalid
for pos in self.board.positions:
x, y = pos
cell_number = None
if self.board[pos] in GIVENS:
cell_color = colors[WHITE]
cell_number = self.board[pos]
else:
cell_color = colors[self.board[pos]]
self.draw_cell(pos, cell_color, grid_color)
if cell_number is not None:
self.draw_number(pos, cell_number)
if self.selected_pos is not None:
self.draw_cell(self.selected_pos, None, highlight_color)
def draw_cell(self, pos, cell_color, line_color):
x, y = pos
screen_x, screen_y = self.to_screen(pos)
if ((x + y) % 2 == 0): # triangle points down
point = (screen_x + cell_half_base, screen_y + cell_height)
left = (screen_x, screen_y)
right = (screen_x + 2 * cell_half_base, screen_y)
else: # triangle points up
point = (screen_x + cell_half_base, screen_y)
left = (screen_x, screen_y + cell_height)
right = (screen_x + 2 * cell_half_base, screen_y + cell_height)
if cell_color is not None:
pygame.draw.polygon(self.screen, cell_color, (point, left, right))
draw_sym_line(self.screen, line_color, left, point)
draw_sym_line(self.screen, line_color, right, point)
draw_sym_line(self.screen, line_color, left, right)
def draw_number(self, pos, number):
number_surface = self.number_surfaces[number]
screen_x, screen_y = self.to_screen(pos)
screen_x = screen_x + cell_half_base - number_surface.get_width() // 2
screen_y = screen_y - number_surface.get_height() // 2
x, y = pos
if ((x + y) % 2 == 0): # triangle points down
screen_y += cell_height // 3
else: # triangle points up
screen_y += 2 * cell_height // 3
self.screen.blit(number_surface, (screen_x, screen_y))
def to_screen(self, pos):
x, y = pos
x0, y0 = (0,0) #STUB
return (x0 + x * cell_half_base, y0 + y * cell_height)
