Few changes to board

This commit is contained in:
Alexander Munch-Hansen 2018-04-14 14:13:27 +02:00
parent dec12d989e
commit c08e7fe540

View File

@ -1,4 +1,9 @@
# TODO: The bar is just for show at the moment. Home doesn't work either.
# TODO: An issue with the bouncing back things. It appears to do the move and then
# it doesn't properly restore the buckets to where they should be.
import pygame
from board import Board
import numpy as np
# --- constants --- (UPPER_CASE names)
@ -8,8 +13,6 @@ SCREEN_HEIGHT = 400
SPACING = 83.333
BUCKETS = [[]]*24
#BLACK = ( 0, 0, 0)
#242 209 107
@ -88,10 +91,50 @@ is_true = False
clock = pygame.time.Clock()
buckets = [5,0,0,0,3,0,5,0,0,0,0,2,5,0,0,0,3,0,5,0,0,0,0,2]
buckets = [[5,-1],[0,0],[0,0],[0,0],[3,1],[0,0],[5,1],[0,0],[0,0],[0,0],[0,0],[2,-1],[5,1],[0,0],[0,0],[0,0],[3,-1],[0,0],[5,-1],[0,0],[0,0],[0,0],[0,0],[2,1]]
running = True
player = 1
roll = []
def switch_player(roll):
roll = roll
player *= -1
def gen_buckets_from_board(board):
meh = []
for i in range(13,25):
pin = board[i]
print(pin)
meh.append([abs(pin), np.sign(pin)])
for i in range(1,13):
pin = board[i]
meh.append([abs(pin), np.sign(pin)])
return meh
def gen_board_from_buckets(buckets):
board = []
board.append([0,0])
for i in range(-1,-13,-1):
board.append(buckets[i])
for i in range(0,12):
board.append(buckets[i])
board.append([0,0])
board = [x*y for x,y in board]
return board
def move_legal(from_board, buckets, roll):
board = gen_board_from_buckets(buckets)
legal_states = Board.calculate_legal_states(from_board, 1, roll)
# print(legal_states)
if board in [list(state) for state in list(legal_states)]:
return True
return False
def find_pin(pos):
x,y = pos
x -= 50 if x > 550 else 0
@ -105,13 +148,12 @@ def find_pin(pos):
# Find the y position based on the chosen pin
def calc_pos(buckets, chosen):
amount = buckets[chosen]
amount = buckets[chosen][0]
# My math is incredibly off
if chosen > 11:
print(amount)
print("Amount at pin:", amount)
y = 378 - (30 * amount)
print(y)
chosen -= 11
x = (SPACING*(chosen-1))+(SPACING/2)
x += 50 if x > 500 else 0
@ -124,7 +166,9 @@ def calc_pos(buckets, chosen):
return x,y
from_board = None
from_buckets = [x for x in buckets]
from_locat = None
while running:
# - events -
@ -145,27 +189,63 @@ while running:
rectangles_drag[is_true] = True
mouse_x, mouse_y = event.pos
# Need this to be a deepcopy :<
from_buckets = []
for x in buckets:
tmp = []
for y in x:
tmp.append(y)
from_buckets.append(tmp)
from_board = [x for x in gen_board_from_buckets(buckets)]
# print("From board in mousedown:", from_board)
pin, idx = find_pin(event.pos)
buckets[idx] -= 1
from_pin = pin
buckets[idx][0] -= 1
if buckets[idx][0] == 0:
buckets[idx][1] = 0
offsets[is_true][0] = rectangles[is_true].x - mouse_x
offsets[is_true][1] = rectangles[is_true].y - mouse_y
from_locat = [rectangles[is_true].x, rectangles[is_true].y]
elif event.type == pygame.MOUSEBUTTONUP:
if event.button == 1:
if is_true is not None:
meh = [rect.collidepoint(event.pos) for rect in rectangles]
if any(meh):
is_true = np.where(meh)[0][0]
pin, idx = find_pin(event.pos)
print(pin,idx)
print(calc_pos(buckets, idx))
x, y = calc_pos(buckets,idx)
buckets[idx] += 1
buckets[idx][0] += 1
buckets[idx][1] = player
rectangles_drag = all_drag[player]
rectangles_drag[is_true] = False
rectangles[is_true].x = x
rectangles[is_true].y = y
print(from_board)
print("To :",gen_board_from_buckets(buckets))
# print(move_legal(from_board, buckets, [1,2]))
if move_legal(from_board, buckets, [1,0]):
print("From:",gen_board_from_buckets(from_buckets))
print("WOHO!"*10)
rectangles_drag = all_drag[player]
rectangles_drag[is_true] = False
rectangles[is_true].x = x
rectangles[is_true].y = y
else:
print("From:",gen_board_from_buckets(from_buckets))
buckets = [x for x in from_buckets]
rectangles_drag = all_drag[player]
rectangles_drag[is_true] = False
rectangles[is_true].x = from_locat[0]
rectangles[is_true].y = from_locat[1]
print("End :",gen_board_from_buckets(buckets))
elif event.type == pygame.MOUSEMOTION:
rectangles_drag = all_drag[player]
@ -173,59 +253,54 @@ while running:
offsets = all_off[player]
if any(rectangles_drag):
is_true = np.where(rectangles_drag)[0][0]
if any(rectangles_drag):
mouse_x, mouse_y = event.pos
rectangles[is_true].x = mouse_x + offsets[is_true][0]
rectangles[is_true].y = mouse_y + offsets[is_true][1]
# - updates (without draws) -
mouse_x, mouse_y = event.pos
rectangles[is_true].x = mouse_x + offsets[is_true][0]
rectangles[is_true].y = mouse_y + offsets[is_true][1]
# empty
# - draws (without updates) -
screen.fill(GREEN_FILT)
# pygame.draw.polygon(screen, (RED), [[0, 0], [50,0],[25,100]], 2)
screen.fill(GREEN_FILT)
# pygame.draw.polygon(screen, (RED), [[0, 0], [50,0],[25,100]], 2)
color = LIGHT_GREY
x = 0
y = 150
# for _ in range(2):
for i in range(12):
if x < 500 and x+SPACING > 500:
x = 550
color = SALMON if color == LIGHT_GREY else LIGHT_GREY
pygame.draw.polygon(screen, color, [[x, 0], [x+SPACING, 0], [(2*x+SPACING)/2, y]])
x += SPACING
color = LIGHT_GREY
x = 0
y = 150
# for _ in range(2):
for i in range(12):
if x < 500 and x+SPACING > 500:
x = 550
color = SALMON if color == LIGHT_GREY else LIGHT_GREY
pygame.draw.polygon(screen, color, [[x, 0], [x+SPACING, 0], [(2*x+SPACING)/2, y]])
x += SPACING
# y += 50
x = 0
y = 250
# for _ in range(2):
color = SALMON if color == LIGHT_GREY else LIGHT_GREY
for i in range(12):
if x < 500 and x+SPACING > 500:
x = 550
x = 0
y = 250
# for _ in range(2):
color = SALMON if color == LIGHT_GREY else LIGHT_GREY
pygame.draw.polygon(screen, color, [[x, 400], [x+SPACING, 400], [(2*x+SPACING)/2, y]])
x += SPACING
for i in range(12):
if x < 500 and x+SPACING > 500:
x = 550
color = SALMON if color == LIGHT_GREY else LIGHT_GREY
pygame.draw.polygon(screen, color, [[x, 400], [x+SPACING, 400], [(2*x+SPACING)/2, y]])
x += SPACING
pygame.draw.rect(screen, BROWN, pygame.rect.Rect((500, 0, 50, 400)))
for p in [-1,1]:
for rect in all_rects[p]:
# print(gen_board_from_buckets(buckets))
pygame.draw.rect(screen, BROWN, pygame.rect.Rect((500, 0, 50, 400)))
for p in [-1,1]:
for rect in all_rects[p]:
pygame.draw.rect(screen, RED if p == -1 else BLACK, rect)
pygame.draw.rect(screen, RED if p == -1 else BLACK, rect)
pygame.display.flip()
pygame.display.flip()
# - constant game speed / FPS -
# - constant game speed / FPS -
clock.tick(FPS)
clock.tick(FPS)
# - end -