def init(self):
BasePlayer.init(self)
self.pos = list(multiply(self.player.get_center(), common.block_size))
self.move = [0, 0]
self.crosshair_pic = common.colorize(BattlePlayer.crosshair_pic,
self.player.color)
self.empty_crosshair_pic = common.colorize(
BattlePlayer.empty_crosshair_pic, self.player.color)
def timed(msg):
if rank == 0:
print(colorize(msg, color='magenta'))
tstart = time.time()
yield
print(
colorize("done in %.3f seconds" % (time.time() - tstart),
color='magenta'))
else:
yield
# coding="utf-8"
import codecs
from PIL import Image
import numpy as np
import h5py
import matplotlib.pyplot as plt
import cv2
from common import colorize, Color
from math import *
import math
db_fname = 'results/SynthText.h5'
db = h5py.File(db_fname, 'r')
dsets = sorted(db['data'].keys())
count = 0
print "total number of images : ", colorize(Color.RED,
len(dsets),
highlight=True)
def rotate(img, pt1, pt2, pt3, pt4):
# print (pt1,pt2,pt3,pt4)
withRect = math.sqrt((pt4[0] - pt1[0])**2 + (pt4[1] - pt1[1])**2)
heightRect = math.sqrt((pt1[0] - pt2[0])**2 + (pt1[1] - pt2[1])**2)
#print (withRect,heightRect)
if (withRect):
angle = acos(abs(pt4[0] - pt1[0]) / withRect) * (180 / math.pi)
#print (angle)
if pt4[1] <= pt1[1]:
class SpinBox(Widget):
up = [
common.colorize(common.load_image("arrow.png", alpha=True), col)
for col in ((0, 0, 0), (150, 0, 0))
]
down = [pygame.transform.flip(surf, False, True) for surf in up]
_hidden = False
def __init__(self, label, pos, choices, default=None, on_change=None):
self._choices = list(choices)
self.index = 0
self.label = label
if default != None:
self.value = default
if on_change:
self.on_change = on_change
Widget.__init__(self, pos, text="", on_click=None)
# add arrows
up_pos = (self.pos[0] - SpinBox.up[0].get_width(), self.pos[1])
down_pos = (
self.pos[0] - SpinBox.up[0].get_width(),
self.pos[1] + self.normal_surface.get_height() -
SpinBox.down[0].get_height(),
)
self.up_widget = Widget(up_pos,
surfaces=SpinBox.up,
on_click=lambda: self.change(-1))
self.down_widget = Widget(down_pos,
surfaces=SpinBox.down,
on_click=lambda: self.change(1))
self.update()
def choices():
def fget(self):
return self._choices
def fset(self, choices):
selected_element = self._choices[self.index]
self._choices = choices
try:
self.index = choices.index(selected_element)
except ValueError:
self.index = 0
self.update()
return locals()
choices = property(**choices())
def change(self, diff):
self.index += diff
if hasattr(self, "on_change"):
self.on_change()
self.update()
def update(self):
self.surfaces_from_text(self.label + ": " +
str(self.choices[self.index]))
self.up_widget.inactive = self.index == 0
self.down_widget.inactive = self.index == len(self.choices) - 1
def value():
def fset(self, value):
try:
self.index = self.choices.index(value)
except NameError:
raise NameError("Could not find value in list of choices!")
def fget(self):
return self.choices[self.index]
return locals()
value = property(**value())
def hidden():
def fset(self, value):
self._hidden = value
self.up_widget.hidden = value
self.down_widget.hidden = value
def fget(self):
return self._hidden
return locals()
hidden = property(**hidden())
def __init__(self, player):
BasePlayer.__init__(self)
self.player = player
self.select_pic = common.colorize(SelectPlayer.select_pic,
self.player.color)
def set_player_id(self, id):
self._player_id = id
self.name = self.names[id]
self.color = self.colors[id]
self.wall_pic = common.colorize(self.wall_pic, self.color)
self.ground_pic = common.colorize(self.ground_pic, self.color)