def __init__(self, line, pt):
self.cycle = 0
self.players = []
self.ball = ball.Ball()
self.kicker = player.Player()
self.side_playing = "n"
self.make(line, pt)
def run_logzip(log, player_types):
os.chdir("../samples")
new_log = open(str(counter()) + ".rca.gz", "wb")
play_on = False
# write types in file
for pt in player_types:
new_line = "pt " + str(pt.id) + " " + str(pt.kickAble_area) + "\n"
new_log.write(gzip.compress(new_line.encode()))
for line in log.split("\n"):
if line.find("playmode") != -1:
if line.find("play_on") != -1:
play_on = True
new_log.write(gzip.compress(b"*\n"))
else:
play_on = False
# if not play_on:
# continue
if line.find("show") == -1:
continue
#Seprate
show_str = line[:line.find("(b)") - 2].strip("()")
line = line[line.find("(b)") - 1:]
ball_str = line[:line.find("(l ") - 2].strip("()")
line = line[line.find("(l ") - 0:].strip("()")
players_str = line.split(")) ((")
#Cycle
cycle = int(show_str.split(" ")[1].strip(")("))
#Ball
ball_str = ball_str.split(" ")
ballobj = ball.Ball()
ballobj.pos.set(ball_str[1], ball_str[2])
ballobj.vel.set(ball_str[3], ball_str[4])
#setPlayers
players = []
for player_str in players_str:
player_str = player_str.split(" (")
player_details = []
for player_d in player_str:
x = player_d.split(" ")
xx = []
for i in range(len(x)):
x[i] = x[i].strip(")(")
if (x[i][0] != 'f'):
xx.append(x[i])
player_details.append(xx)
plyr = player.Player()
plyr.set(player_details, player_types)
players.append(plyr)
# Write in file
# new_line = "R|"
new_line = ""
new_line += str(cycle) + "|"
new_line += str(ballobj.pos.x) + " " + str(
ballobj.pos.y
) + "|" #+ " " + str(ballobj.vel.x) + " " + str(ballobj.vel.y) + "|"
for plyr in players:
new_line += plyr.side + " " + str(plyr.unum) + " " + str(
plyr.playerType.id) + " "
new_line += str(plyr.pos.x) + " " + str(
plyr.pos.y
) + "|" #" " + str(plyr.vel.x) + " " + str(plyr.vel.y) + " "
new_line += str(plyr.body) + " " + str(plyr.head) + "|"
new_line += "\n"
new_log.write(gzip.compress(new_line.encode()))
# print("samples:",cycle)
os.chdir("../orginalLogs")
def extractFeatur(log, player_types):
os.chdir("../features")
file_name = str(counter())
new_log = open(file_name + ".rcb", "w")
# write types in file
for pt in player_types:
new_log.write("pt " + str(pt.id) + " " + str(pt.kickAble_area) + "\n")
lines = log.split("\n")
last_kickAble_cycle = -10
last_kickAble_unum = -1
for line in lines:
#Reading a cycle data
if line.find("*") != -1:
last_kickAble_unum = -1
last_kickAble_cycle = -10
if (line.find("R") == -1):
continue
details = line.split("|")
cycle = int(details[1])
ballobj = ball.Ball()
ballobj.set_data(details[2])
players = []
for ply_str in details:
if ply_str == '':
continue
if ply_str[0] != "r" and ply_str[0] != "l":
continue
playerobj = player.Player()
playerobj.set_data(ply_str, player_types)
players.append(playerobj)
if not side(players, ballobj, "l"):
if side(players, ballobj, "r"):
last_kickAble_unum = -1
last_kickAble_cycle = -10
continue
n = 0
for i in range(len(players)):
if players[i].is_kickAble(ballobj.pos) and players[i].side == "l":
n = i
break
polars = []
polars_n = []
kicker = players[n]
for plyr in players:
teta = plyr.pos.teta(kicker.pos)
r = plyr.pos.dist(kicker.pos)
polar = pos.Polar(r, teta)
polars.append(polar)
next_pos = pos.plus(plyr.pos, plyr.vel)
teta_n = next_pos.teta(kicker.pos)
r_n = next_pos.dist(kicker.pos)
polar_n = pos.Polar(r_n, teta_n)
polars_n.append(polar_n)
new_line = ""
new_line += "R|" + str(cycle) + "|"
#find polar pos and vel of ball
teta_b = ballobj.pos.teta(kicker.pos)
r_b = ballobj.pos.dist(kicker.pos)
polar_b = pos.Polar(r_b, teta_b)
next_ballpos = pos.Pos(ballobj.pos.x + ballobj.vel.x,
ballobj.pos.y + ballobj.vel.y)
teta_vb = next_ballpos.teta(ballobj.pos)
r_vb = ballobj.vel.r()
polar_vb = pos.Polar(r_vb, teta_vb)
new_line += str(polar_b.r) + " " + str(polar_b.teta) + " " + str(
polar_vb.r) + " " + str(polar_vb.teta) + "|"
for i_p in range(len(polars)):
new_line += str(polars[i_p].r) + " " + str(polars[i_p].teta) + "|"
# new_line += str(polars_n[i_p].r) + " " + str(polars_n[i_p].teta) + "|"
# for plyr in players:
# new_line += str(plyr.pos.x/52.5) + " " + str(plyr.pos.y/34) + "|"
new_line += str(last_kickAble_unum) + " " + str(kicker.unum) + "|"
if last_kickAble_unum != -1:
new_log.write(new_line + "\n")
last_kickAble_unum = kicker.unum
# print("feature:",cycle)
new_log.close()
f_log = open(file_name + ".rcb", "r").read()
new_log = open(file_name + ".rcb", "w")
lines = f_log.split("\n")
for i in range(len(lines) - 2):
if lines[i].find("pt") != -1:
continue
line = lines[i].split("|")[:-1]
next_line = lines[i + 1].split("|")[:-1]
line[-1] = next_line[-1]
newline = ""
for w in line:
newline += w + "|"
new_log.write(newline + "\n")
os.chdir("../samples")