def __init__(self):
super(MainWindow, self).__init__()
self.colors = np.zeros([num_players, 3], dtype=np.uint8)
self.dark_colors = np.zeros([num_players, 3], dtype=np.uint8)
ci = 0
for hue in xrange(0, 360, 360/num_players):
light_color = np.array(
[int(color) for color in hsv2rgb(hue/360.0, 0.75, 1)]
)
dark_color = np.array(
[int(color) for color in hsv2rgb(hue/360.0, 0.75, 0.75)]
)
self.colors[ci] = light_color
self.dark_colors[ci] = dark_color
ci += 1
genVoronoi.init_cache()
self.points = np.zeros([num_players, num_moves, 2], dtype=np.int)
self.points.fill(-1)
self.image = QtGui.QImage(
board_size/scale,
board_size/scale,
QtGui.QImage.Format_RGB888
)
ptr = self.image.bits()
ptr.setsize(self.image.byteCount())
self.imagenp = np.asarray(ptr).reshape(board_size/scale, board_size/scale, 3)
self.current_player = 0
self.move = 0
self.initUI()
def __init__(self, GUIOn, delay):
self.GUIOn = GUIOn
self.delay = delay
super(MainWindow, self).__init__()
self.colors = np.zeros([num_players, 3], dtype=np.uint8)
self.dark_colors = np.zeros([num_players, 3], dtype=np.uint8)
ci = 0
for hue in xrange(0, 360, 360/num_players):
light_color = np.array(
[int(color) for color in hsv2rgb(hue/360.0, 0.75, 1)]
)
dark_color = np.array(
[int(color) for color in hsv2rgb(hue/360.0, 0.75, 0.75)]
)
self.colors[ci] = light_color
self.dark_colors[ci] = dark_color
ci += 1
genVoronoi.init_cache()
self.points = np.zeros([num_players, num_moves, 2], dtype=np.int)
self.points.fill(-1)
self.player_names = []
if self.GUIOn:
self.image = QtGui.QImage(
board_size/scale,
board_size/scale,
QtGui.QImage.Format_RGB888
)
ptr = self.image.bits()
ptr.setsize(self.image.byteCount())
self.imagenp = np.asarray(ptr).reshape(board_size/scale, board_size/scale, 3)
self.initUI()
else:
# generate_voronoi_diagram needs this
self.imagenp = np.zeros([board_size/scale, board_size/scale, 3], dtype=np.uint8)
def __init__(self, name, idx, numMoves):
Client.__init__(self, name)
self.numMoves = numMoves
genVoronoi.init_cache()
self.points = np.zeros([2, numMoves, 2], dtype=np.int)
self.points.fill(-1)
self.colors = np.zeros([2, 3], dtype=np.uint8) # Dummy data to run score script
self.playerIdx = idx
self.oppIdx = idx ^ 1
self.board = np.zeros([1000, 1000], dtype=np.uint8)
self.board.fill(-1)
self.myMoves = 0
self.oppMoves = 0
"""
def __init__(self, name, numMoves): Client.__init__(self, name) self.numMoves = numMoves genVoronoi.init_cache() self.points = np.zeros([2, numMoves, 2], dtype=np.int) self.points.fill(-1) self.colors = np.zeros([2, 3], dtype=np.uint8) #Dummy data to run score script self.myMoves = set([]) self.oppMoves = set([]) self.myIdx = 1 self.oppIdx = 0 self.myCnt = 0 self.oppCnt = 0 self.time = time.time() self.timeUsed = 0 self.rng = np.random.RandomState()
def __init__(self, name, idx, numMoves):
Client.__init__(self, name)
self.numMoves = numMoves
genVoronoi.init_cache()
self.points = np.zeros([2, numMoves+1, 2], dtype=np.int)
self.points.fill(-1)
self.colors = np.zeros([2, 3], dtype=np.uint8) #Dummy data to run score script
self.playerIdx = idx
self.oppIdx = idx ^ 1
# self.board = np.zeros([1000, 1000], dtype=np.uint8)
self.board = np.zeros([1000, 1000])
self.board.fill(-1)
self.myMoves = 0
self.oppMoves = 0
self.max_time = datetime.timedelta(seconds=1)
self.C = 1.4
self.wins = {0: {}, 1: {}}
self.plays = {0: {}, 1: {}}
self.states = []
'''
if __name__ == '__main__':
num_players, num_moves = tuple(int(n) for n in raw_input().split())
points = np.zeros([num_players, num_moves, 2], dtype=np.int)
points.fill(-1)
# Can put any garbage here. Not using this in the end.
image = np.zeros([10, 10, 3], dtype=np.uint8)
colors = np.zeros([num_players, 3], dtype=np.uint8)
for p in xrange(num_players):
player, actual_move_count = tuple(int(n) for n in raw_input().split())
for m in xrange(actual_move_count):
x, y = tuple(int(n) for n in raw_input().split())
points[p, m, 0] = x
points[p, m, 1] = y
genVoronoi.init_cache()
# Important to put GUIOn as false
genVoronoi.generate_voronoi_diagram(num_players, num_moves, points, colors, image, 0, 1)
scores, colors_index = genVoronoi.get_color_indices_and_scores(num_players, num_moves, points)
for i in range(num_players):
print scores[i]
for i in range(1000):
for j in range(1000):
n = colors_index[i,j,0]
print n,
for k in range(1, n+1):
print colors_index[i,j,k],
print
